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

2025-09-05 17:21:02 +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/completions/completions_router.py
+++ b/private_gpt/server/completions/completions_router.py
@@ -42,6 +42,13 @@ class CompletionsBody(BaseModel):
    summary="Completion",
    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 prompt_completion(
    request: Request, body: CompletionsBody