nuclia 4.8.7__tar.gz → 4.8.8__tar.gz

{nuclia-4.8.7 → nuclia-4.8.8}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+
+## 4.8.8 (2025-05-14)
+
+
+- Fix `nuclia version`
+- Fix for Tool validation in ChatModel
+- Fix streaming trying to validate error payload as GenerativeChunk
+
 ## 4.8.7 (2025-05-13)
 
 

{nuclia-4.8.7 → nuclia-4.8.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nuclia
-Version: 4.8.7
+Version: 4.8.8
 Summary: Nuclia Python SDK
 Author-email: Nuclia <info@nuclia.com>
 License-Expression: MIT

nuclia-4.8.8/VERSION

@@ -0,0 +1 @@
+4.8.8

{nuclia-4.8.7 → nuclia-4.8.8}/nuclia/cli/run.py

@@ -1,6 +1,6 @@
 import logging
 import sys
-from pathlib import Path
+from importlib.metadata import version
 
 import fire  # type: ignore
 from nucliadb_sdk import exceptions
@@ -9,12 +9,13 @@ from nuclia.data import get_auth
 from nuclia.exceptions import NeedUserToken, UserTokenExpired
 from nuclia.lib.utils import serialize
 from nuclia.sdk.accounts import NucliaAccounts
+from nuclia.sdk.backup import NucliaBackup
 from nuclia.sdk.kb import NucliaKB
 from nuclia.sdk.kbs import NucliaKBS
-from nuclia.sdk.backup import NucliaBackup
 from nuclia.sdk.logger import logger
 from nuclia.sdk.nua import NucliaNUA
 from nuclia.sdk.zones import NucliaZones
+
 from .utils import CustomFormatter
 
 
@@ -30,9 +31,7 @@ class NucliaCLI(object):
 
     def version(self):
         """Print the version of the CLI"""
-        _dir = Path(__file__).resolve().parent.parent.parent
-        VERSION = _dir.joinpath("VERSION").open().read().strip()
-        return VERSION
+        return version("nuclia")
 
 
 def run():

{nuclia-4.8.7 → nuclia-4.8.8}/nuclia/lib/nua.py

@@ -56,6 +56,8 @@ import os
 from tqdm import tqdm
 import asyncio
 from nuclia.lib.utils import build_httpx_client, build_httpx_async_client
+from pydantic import ValidationError
+
 
 if TYPE_CHECKING:
     from nucliadb_protos.writer_pb2 import BrokerMessage
@@ -147,8 +149,19 @@ class NuaClient:
             json=payload,
             timeout=timeout,
         ) as response:
-            for json_body in response.iter_lines():
-                yield GenerativeChunk.model_validate_json(json_body)  # type: ignore
+            if response.headers.get("content-type") == "application/x-ndjson":
+                for json_body in response.iter_lines():
+                    try:
+                        yield GenerativeChunk.model_validate_json(json_body)  # type: ignore
+                    except ValidationError as e:
+                        raise RuntimeError(f"Invalid stream chunk: {json_body}") from e
+
+            else:
+                # Read the full error body and raise an appropriate exception
+                error_content = response.content
+                raise RuntimeError(
+                    f"Stream request failed with status {response.status_code}: {error_content.decode('utf-8')}"
+                )
 
     def add_config_predict(self, kbid: str, config: LearningConfigurationCreation):
         endpoint = f"{self.url}{CONFIG}/{kbid}"
@@ -456,8 +469,19 @@ class AsyncNuaClient:
             json=payload,
             timeout=timeout,
         ) as response:
-            async for json_body in response.aiter_lines():
-                yield GenerativeChunk.model_validate_json(json_body)  # type: ignore
+            if response.headers.get("content-type") == "application/x-ndjson":
+                async for json_body in response.aiter_lines():
+                    try:
+                        yield GenerativeChunk.model_validate_json(json_body)  # type: ignore
+                    except ValidationError as e:
+                        raise RuntimeError(f"Invalid stream chunk: {json_body}") from e
+
+            else:
+                # Read the full error body and raise an appropriate exception
+                error_content = await response.aread()
+                raise RuntimeError(
+                    f"Stream request failed with status {response.status_code}: {error_content.decode('utf-8')}"
+                )
 
     async def add_config_predict(
         self, kbid: str, config: LearningConfigurationCreation

{nuclia-4.8.7 → nuclia-4.8.8}/nuclia/lib/nua_responses.py

@@ -120,10 +120,10 @@ class ChatModel(BaseModel):
             raise ValueError("Can not setup markdown and JSON Schema at the same time")
         if self.citations is True and self.json_schema is not None:
             raise ValueError("Can not setup citations and JSON Schema at the same time")
-        if self.citations is True and self.tools is not None:
+        if self.citations is True and len(self.tools) > 0:
             raise ValueError("Can not setup citations and Tools at the same time")
-        if self.tools is True and self.json_schema is not None:
-            raise ValueError("Can not setup tools and JSON Schema at the same time")
+        if len(self.tools) > 0 and self.json_schema is not None:
+            raise ValueError("Can not setup Tools and JSON Schema at the same time")
         return self
 
 

{nuclia-4.8.7 → nuclia-4.8.8}/nuclia/tests/test_kb/test_search.py

@@ -69,7 +69,7 @@ def test_search(testing_config):
 def test_catalog(testing_config):
     search = NucliaSearch()
     results = search.catalog()
-    assert len(results.resources.keys()) == 2
+    assert len(results.resources.keys()) >= 2
 
 
 def test_search_object(testing_config):

{nuclia-4.8.7 → nuclia-4.8.8}/nuclia.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nuclia
-Version: 4.8.7
+Version: 4.8.8
 Summary: Nuclia Python SDK
 Author-email: Nuclia <info@nuclia.com>
 License-Expression: MIT

{nuclia-4.8.7 → nuclia-4.8.8}/pyproject.toml

@@ -54,3 +54,4 @@ version = { file = "VERSION" }
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"
+

{nuclia-4.8.7 → nuclia-4.8.8}/requirements.txt

@@ -8,7 +8,7 @@ prompt_toolkit
 nucliadb_sdk>=6.4,<7
 nucliadb_models>=6.4,<7
 nucliadb_protos>=6.4,<7
-nuclia-models>=0.32.0
+nuclia-models>=0.39.0
 tqdm
 aiofiles
 backoff

nuclia-4.8.7/VERSION

@@ -1 +0,0 @@
-4.8.7