feat: Add stream information to generate SDKs (#1569)

2025-09-13 05:43:44 +00:00 · 2024-02-02 16:14:22 +01:00
parent 3e67e21d38
commit 24fae660e6
3 changed files with 32 additions and 0 deletions
--- a/private_gpt/server/chat/chat_router.py
+++ b/private_gpt/server/chat/chat_router.py
@@ -54,6 +54,13 @@ class ChatBody(BaseModel):
    response_model=None,
    responses={200: {"model": OpenAICompletion}},
    tags=["Contextual Completions"],
+    openapi_extra={
+        "x-fern-streaming": {
+            "stream-condition": "stream",
+            "response": {"$ref": "#/components/schemas/OpenAICompletion"},
+            "response-stream": {"$ref": "#/components/schemas/OpenAICompletion"},
+        }
+    },
 )
 def chat_completion(
    request: Request, body: ChatBody