From 926fd9a0105a2351055c4c526851ed86d3334105 Mon Sep 17 00:00:00 2001 From: Sai Vinay G Date: Tue, 9 May 2023 16:08:09 +0530 Subject: [PATCH] feat(router): Adding response schema for compat_generate (#292) --- router/src/server.rs | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/router/src/server.rs b/router/src/server.rs index c5fc41ef..a9baf288 100644 --- a/router/src/server.rs +++ b/router/src/server.rs @@ -37,7 +37,11 @@ use utoipa_swagger_ui::SwaggerUi; path = "/", request_body = CompatGenerateRequest, responses( - (status = 200, description = "See /generate or /generate_stream"), + (status = 200, description = "See /generate or /generate_stream", + content( + ("application/json" = GenerateResponse), + ("text/event-stream" = StreamResponse), + )), (status = 424, description = "Generation Error", body = ErrorResponse, example = json ! ({"error": "Request failed during generation"})), (status = 429, description = "Model is overloaded", body = ErrorResponse,