diff --git a/router/src/server.rs b/router/src/server.rs index 8088beb2..35292410 100644 --- a/router/src/server.rs +++ b/router/src/server.rs @@ -121,7 +121,7 @@ async fn get_model_info(info: Extension) -> Json { tag = "Text Generation Inference", path = "/chat_tokenize", request_body = ChatRequest, - responses((status = 200, description = "Templated Chat Request", body = Value)) + responses((status = 200, description = "Templated and tokenized ChatRequest", body = ChatTokenizeResponse)) )] async fn get_chat_tokenize( Extension(infer): Extension,