huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add router name to /info endpoint

Browse Source
This commit is contained in:
Wauplin 2024-05-03 12:11:57 +02:00
parent a25737139d
commit afb02914e6
No known key found for this signature in database
GPG Key ID: 9838FE02BECE1A02
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
router/src/lib.rs
View File

@ -159,6 +159,8 @@ pub struct Info {
#[schema(example = "32")] #[schema(example = "32")]
pub max_client_batch_size: usize, pub max_client_batch_size: usize,
/// Router Info /// Router Info
#[schema(example = "text-generation-router")]
pub router: &'static str,
#[schema(example = "0.5.0")] #[schema(example = "0.5.0")]
pub version: &'static str, pub version: &'static str,
#[schema(nullable = true, example = "null")] #[schema(nullable = true, example = "null")]

1
router/src/server.rs
View File

@ -1564,6 +1564,7 @@ pub async fn run(
max_batch_size, max_batch_size,
validation_workers, validation_workers,
max_client_batch_size, max_client_batch_size,
router: env!("CARGO_PKG_NAME"),
version: env!("CARGO_PKG_VERSION"), version: env!("CARGO_PKG_VERSION"),
sha: option_env!("VERGEN_GIT_SHA"), sha: option_env!("VERGEN_GIT_SHA"),
docker_label: option_env!("DOCKER_LABEL"), docker_label: option_env!("DOCKER_LABEL"),