wip - chat stream

ion2088 · Feb 27, 2024 · 4cbaf32 · 4cbaf32
1 parent ef54fa4
commit 4cbaf32
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/firedust/_utils/api.py b/src/firedust/_utils/api.py
@@ -66,7 +66,9 @@ def post_stream(
         self, url: str, data: Dict[str, Any] | None = None
     ) -> Iterator[bytes]:
         url = self.base_url + url
-        with httpx.stream("post", url, json=data, headers=self.headers) as response:
+        with httpx.stream(
+            "post", url, json=data, headers=self.headers, timeout=300
+        ) as response:
             for chunk in response.iter_bytes():
                 yield chunk