fix response type of document for Text Generation Inference

Signed-off-by: jitokim <pigberger70@gmail.com>
This commit is contained in:
jitokim 2024-11-14 01:13:03 +09:00
parent a785000842
commit 9d0aaf6c9d
2 changed files with 5 additions and 2 deletions

View File

@ -36,7 +36,10 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/GenerateResponse" "type": "array",
"items": {
"$ref": "#/components/schemas/GenerateResponse"
}
} }
}, },
"text/event-stream": { "text/event-stream": {

View File

@ -109,7 +109,7 @@ request_body = CompatGenerateRequest,
responses( responses(
(status = 200, description = "Generated Text", (status = 200, description = "Generated Text",
content( content(
("application/json" = GenerateResponse), ("application/json" = Vec<GenerateResponse>),
("text/event-stream" = StreamResponse), ("text/event-stream" = StreamResponse),
)), )),
(status = 424, description = "Generation Error", body = ErrorResponse, (status = 424, description = "Generation Error", body = ErrorResponse,