From 9d0aaf6c9db2dfe04cfad0208b6eb8810bc43948 Mon Sep 17 00:00:00 2001 From: jitokim Date: Thu, 14 Nov 2024 01:13:03 +0900 Subject: [PATCH] fix response type of document for Text Generation Inference Signed-off-by: jitokim --- docs/openapi.json | 5 ++++- router/src/server.rs | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/docs/openapi.json b/docs/openapi.json index 22b06720..e4c8ffdb 100644 --- a/docs/openapi.json +++ b/docs/openapi.json @@ -36,7 +36,10 @@ "content": { "application/json": { "schema": { - "$ref": "#/components/schemas/GenerateResponse" + "type": "array", + "items": { + "$ref": "#/components/schemas/GenerateResponse" + } } }, "text/event-stream": { diff --git a/router/src/server.rs b/router/src/server.rs index 2058bce3..a0bc1768 100644 --- a/router/src/server.rs +++ b/router/src/server.rs @@ -109,7 +109,7 @@ request_body = CompatGenerateRequest, responses( (status = 200, description = "Generated Text", content( -("application/json" = GenerateResponse), +("application/json" = Vec), ("text/event-stream" = StreamResponse), )), (status = 424, description = "Generation Error", body = ErrorResponse,