From 3d73c99ebe14cbad24408d1147871256e7417088 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 12 Sep 2024 14:47:18 +0200
Subject: [PATCH] Attempt at integration tests.

---
 flake.nix      |  2 ++
 nix/client.nix | 26 ++++++++++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 nix/client.nix

diff --git a/flake.nix b/flake.nix
index a8e53cd5..4ebbf119 100644
--- a/flake.nix
+++ b/flake.nix
@@ -67,6 +67,7 @@
             '';
           };
         server = pkgs.python3.pkgs.callPackage ./nix/server.nix { inherit nix-filter; };
+        client = pkgs.python3.pkgs.callPackage ./nix/client.nix { inherit nix-filter; };
       in
       {
         formatter = pkgs.nixfmt-rfc-style;
@@ -88,6 +89,7 @@
                 launcher
                 router
                 server
+                client
                 openssl.dev
                 pkg-config
                 cargo
diff --git a/nix/client.nix b/nix/client.nix
new file mode 100644
index 00000000..f6fe76f0
--- /dev/null
+++ b/nix/client.nix
@@ -0,0 +1,26 @@
+{
+  buildPythonPackage,
+  poetry-core,
+  huggingface-hub,
+  pydantic,
+}:
+
+buildPythonPackage {
+  name = "text-generation";
+
+  src = ../clients/python;
+
+  pyproject = true;
+
+  build-system = [ poetry-core ];
+
+  nativeBuildInputs = [ ];
+
+  pythonRemoveDeps = [ ];
+
+  dependencies = [
+    huggingface-hub
+    transformers
+    pydantic
+  ];
+}