From afb02914e6f6d6a4aa3525504070e5ec288f4313 Mon Sep 17 00:00:00 2001
From: Wauplin <lucainp@gmail.com>
Date: Fri, 3 May 2024 12:11:57 +0200
Subject: [PATCH] Add router name to /info endpoint

---
 router/src/lib.rs    | 2 ++
 router/src/server.rs | 1 +
 2 files changed, 3 insertions(+)

diff --git a/router/src/lib.rs b/router/src/lib.rs
index fac4c14e..96a9fdf6 100644
--- a/router/src/lib.rs
+++ b/router/src/lib.rs
@@ -159,6 +159,8 @@ pub struct Info {
     #[schema(example = "32")]
     pub max_client_batch_size: usize,
     /// Router Info
+    #[schema(example = "text-generation-router")]
+    pub router: &'static str,
     #[schema(example = "0.5.0")]
     pub version: &'static str,
     #[schema(nullable = true, example = "null")]
diff --git a/router/src/server.rs b/router/src/server.rs
index 8657b779..5dbde21e 100644
--- a/router/src/server.rs
+++ b/router/src/server.rs
@@ -1564,6 +1564,7 @@ pub async fn run(
         max_batch_size,
         validation_workers,
         max_client_batch_size,
+        router: env!("CARGO_PKG_NAME"),
         version: env!("CARGO_PKG_VERSION"),
         sha: option_env!("VERGEN_GIT_SHA"),
         docker_label: option_env!("DOCKER_LABEL"),