nix: add `text-generation-benchmark` to pure devshell (#2431)

nix: add text-generation-benchmark to pure devshell

flake.nix

@@ -43,6 +43,9 @@
           ];
         };
         crateOverrides = import ./nix/crate-overrides.nix { inherit pkgs nix-filter; };
+        benchmark = cargoNix.workspaceMembers.text-generation-benchmark.build.override {
+          inherit crateOverrides;
+        };
         launcher = cargoNix.workspaceMembers.text-generation-launcher.build.override {
           inherit crateOverrides;
         };
@@ -57,6 +60,7 @@
 
           pure = mkShell {
             buildInputs = [
+              benchmark
               launcher
               router
               server

nix/crate-overrides.nix

@@ -20,8 +20,7 @@ defaultCrateOverrides
   rav1e = attrs: { env.CARGO_ENCODED_RUSTFLAGS = "-C target-feature=-crt-static"; };
 
   grpc-metadata = attrs: {
-    src =
+    src = filter {
-      filter {
       root = ../backends/grpc-metadata;
       include = with filter; [
         isDirectory
@@ -29,9 +28,29 @@ defaultCrateOverrides
       ];
     };
   };
-  text-generation-launcer = attrs: {
+  text-generation-benchmark = attrs: {
-    src =
+    src = filter {
-      filter {
+      root = ../benchmark;
+      include = with filter; [
+        isDirectory
+        (matchExt "rs")
+      ];
+    };
+  };
+  text-generation-client = attrs: {
+    src = filter {
+      root = ../.;
+      include = with filter; [
+        isDirectory
+        (and (inDirectory "backends/client") (matchExt "rs"))
+        (and (inDirectory "proto") (matchExt "proto"))
+      ];
+    };
+    postPatch = "cd backends/client";
+    buildInputs = [ protobuf ];
+  };
+  text-generation-launcher = attrs: {
+    src = filter {
       root = ../launcher;
       include = with filter; [
         isDirectory
@@ -40,8 +59,7 @@ defaultCrateOverrides
     };
   };
   text-generation-router = attrs: {
-    src =
+    src = filter {
-      filter {
       root = ../router;
       include = with filter; [
         isDirectory