fere-sdk 0.1.0.dev9__tar.gz → 0.1.0.dev12__tar.gz

{fere_sdk-0.1.0.dev9 → fere_sdk-0.1.0.dev12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fere-sdk
-Version: 0.1.0.dev9
+Version: 0.1.0.dev12
 Summary: Python SDK for the FereAI Gateway API
 Author-email: Fere AI <info@fere.ai>
 License-Expression: MIT

{fere_sdk-0.1.0.dev9 → fere_sdk-0.1.0.dev12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fere-sdk"
-version = "0.1.0.dev9"
+version = "0.1.0.dev12"
 description = "Python SDK for the FereAI Gateway API"
 readme = "README.md"
 requires-python = ">=3.10"

{fere_sdk-0.1.0.dev9 → fere_sdk-0.1.0.dev12}/src/fere_sdk/client.py

@@ -98,7 +98,9 @@ class FereClient:
             elif event.event == "done":
                 result["done"] = event.data
             elif event.event == "error":
-                raise RuntimeError(event.data.get("message", str(event.data)))
+                raise RuntimeError(
+                    event.data.get("message", str(event.data))
+                )
 
         if tool_responses:
             result["tool_responses"] = tool_responses

{fere_sdk-0.1.0.dev9 → fere_sdk-0.1.0.dev12}/src/fere_sdk/low_level.py

@@ -84,7 +84,6 @@ class FereAPI:
             params=params,
         )
         if resp.status_code == 401:
-            # Token expired — refresh and retry
             self._auth.clear_token()
             headers = await self._headers()
             resp = await self._client.get(
@@ -144,9 +143,14 @@ class FereAPI:
         thread_id: str | None = None,
         agent: str = "ProAgent",
     ) -> AsyncIterator[ChatEvent]:
-        """POST /v1/chat — returns an async iterator of SSE events."""
+        """POST /v1/chat — returns an async iterator of SSE events.
+
+        Uses a 1200s read timeout because research queries (especially
+        ProDeepResearchAgent) can take up to 20 minutes with long gaps
+        between SSE events while the server plans and researches.
+        """
         headers = await self._headers()
-        body = {
+        body: dict[str, Any] = {
             "query": query,
             "stream": True,
             "agent": agent,
@@ -154,11 +158,15 @@ class FereAPI:
         if thread_id:
             body["thread_id"] = thread_id
 
+        chat_timeout = httpx.Timeout(
+            connect=30.0, read=1200.0, write=30.0, pool=30.0
+        )
         async with self._client.stream(
             "POST",
             f"{self.base_url}/v1/chat",
             json=body,
             headers=headers,
+            timeout=chat_timeout,
         ) as resp:
             resp.raise_for_status()
             event_type = ""
@@ -172,9 +180,13 @@ class FereAPI:
                     try:
                         data = json.loads(raw)
                     except (json.JSONDecodeError, ValueError):
-                        logger.debug("Unparseable SSE data: %r", raw)
+                        logger.debug(
+                            "Unparseable SSE data: %r", raw
+                        )
                         continue
-                    yield ChatEvent(event=event_type, data=data)
+                    yield ChatEvent(
+                        event=event_type, data=data
+                    )
 
     async def get_threads(self, skip: int = 0, limit: int = 10) -> list[dict]:
         """GET /v1/chat/threads"""