hyperforge-nucliadb-agentic 1.0.0.post64__tar.gz → 1.0.0.post78__tar.gz

{hyperforge_nucliadb_agentic-1.0.0.post64 → hyperforge_nucliadb_agentic-1.0.0.post78}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperforge_nucliadb_agentic
-Version: 1.0.0.post64
+Version: 1.0.0.post78
 Summary: NucliaDB Hyperforge agent
 Author: Nuclia
 Author-email: Nuclia <nucliadb@nuclia.com>

{hyperforge_nucliadb_agentic-1.0.0.post64 → hyperforge_nucliadb_agentic-1.0.0.post78}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "hyperforge_nucliadb_agentic"
-version = "1.0.0.post64"
+version = "1.0.0.post78"
 license = "Apache-2.0"
 description = "NucliaDB Hyperforge agent"
 authors = [{ name = "Nuclia", email = "nucliadb@nuclia.com" }]

{hyperforge_nucliadb_agentic-1.0.0.post64 → hyperforge_nucliadb_agentic-1.0.0.post78}/src/hyperforge_nucliadb_agentic/agent.py

@@ -1056,7 +1056,7 @@ class NucliaDBAgent(ContextAgent, Agent[NucliaDBAgentConfig]):
             )
         )
 
-        paragraphs = await paragraphs_result.model_dump()
+        paragraphs = await paragraphs_result.to_sync_response()
         answer = None
         input_tokens = (
             paragraphs.consumption.normalized_tokens.input

{hyperforge_nucliadb_agentic-1.0.0.post64 → hyperforge_nucliadb_agentic-1.0.0.post78}/src/hyperforge_nucliadb_agentic/ask/predict.py

@@ -296,13 +296,18 @@ class PredictEngine:
         headers = self.get_predict_headers(kbid)
         headers["Accept"] = "application/x-ndjson"
 
-        resp = await self.make_stream_request(
-            "POST",
-            url=self.get_predict_url(CHAT, kbid),
-            json=item.model_dump(),
-            headers={**headers, **(extra_headers or {})},
-            timeout=httpx.Timeout(180.0, read=120.0),
-        )
+        try:
+            resp = await self.make_stream_request(
+                "POST",
+                url=self.get_predict_url(CHAT, kbid),
+                json=item.model_dump(),
+                headers={**headers, **(extra_headers or {})},
+                timeout=httpx.Timeout(30.0, read=None),
+            )
+        except httpx.TimeoutException as exc:
+            raise ProxiedPredictAPIError(
+                status=504, detail="Timeout while connecting to predict"
+            ) from exc
         await self.check_response(kbid, resp, expected_status=200)
         ident = resp.headers.get(NUCLIA_LEARNING_ID_HEADER) or "unknown"
         model = resp.headers.get(NUCLIA_LEARNING_MODEL_HEADER) or "unknown"

{hyperforge_nucliadb_agentic-1.0.0.post64 → hyperforge_nucliadb_agentic-1.0.0.post78}/src/hyperforge_nucliadb_agentic/ask/search/ask.py

@@ -378,7 +378,7 @@ class AskResult:
                 metrics=self.metrics.dump(),
             )
 
-    async def model_dump(self) -> SyncAskResponse:
+    async def to_sync_response(self) -> SyncAskResponse:
         # Run the stream to completion to ensure all data is available in memory
         # First, run the stream in memory to get all the data in memory
         async for _ in self._stream():
@@ -458,7 +458,7 @@ class AskResult:
         return response
 
     async def json(self) -> str:
-        response = await self.model_dump()
+        response = await self.to_sync_response()
         return response.model_dump_json(exclude_none=True, by_alias=True)
 
     async def get_relations_results(self) -> Relations:
@@ -551,7 +551,7 @@ class NotEnoughContextAskResult(AskResult):
                 )
             )
 
-    async def json(self) -> str:
+    async def to_sync_response(self) -> SyncAskResponse:
         prequeries = (
             {
                 prequery.id or f"prequery_{index}": prequery_result
@@ -567,7 +567,10 @@ class NotEnoughContextAskResult(AskResult):
             retrieval_results=self.main_results,
             prequeries=prequeries,
             status=AnswerStatusCode.NO_RETRIEVAL_DATA.prettify(),
-        ).model_dump_json()
+        )
+
+    async def json(self) -> str:
+        return (await self.to_sync_response()).model_dump_json()
 
 
 async def rephrase_query(