feat(router): Adding response schema for compat_generate (#292)

926fd9a0 · Sai Vinay G · GitHub · e9b01b34 · 926fd9a0
Unverified Commit 926fd9a0 authored May 09, 2023 by Sai Vinay G Committed by GitHub May 09, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

router/src/server.rs router/src/server.rs +5 -1

No files found.
--- 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,