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

zylon-ai · Feb 2, 2024 · 24fae66 · 24fae66
1 parent 3e67e21
commit 24fae66
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/fern/openapi/openapi.json b/fern/openapi/openapi.json
@@ -44,6 +44,15 @@
               }
             }
           }
+        },
+        "x-fern-streaming": {
+          "stream-condition": "stream",
+          "response": {
+            "$ref": "#/components/schemas/OpenAICompletion"
+          },
+          "response-stream": {
+            "$ref": "#/components/schemas/OpenAICompletion"
+          }
         }
       }
     },
@@ -86,6 +95,15 @@
               }
             }
           }
+        },
+        "x-fern-streaming": {
+          "stream-condition": "stream",
+          "response": {
+            "$ref": "#/components/schemas/OpenAICompletion"
+          },
+          "response-stream": {
+            "$ref": "#/components/schemas/OpenAICompletion"
+          }
         }
       }
     },

diff --git 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

diff --git 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