nuclia 4.8.5__py3-none-any.whl → 4.8.7__py3-none-any.whl

nuclia/lib/nua.py

@@ -26,6 +26,7 @@ from nuclia_models.predict.generative_responses import (
     CitationsGenerativeResponse,
     MetaGenerativeResponse,
     StatusGenerativeResponse,
+    ToolsGenerativeResponse,
 )
 from nuclia.lib.nua_responses import (
     ChatModel,
@@ -233,6 +234,8 @@ class NuaClient:
                 result.citations = chunk.chunk.citations
             elif isinstance(chunk.chunk, StatusGenerativeResponse):
                 result.code = chunk.chunk.code
+            elif isinstance(chunk.chunk, ToolsGenerativeResponse):
+                result.tools = chunk.chunk.tools
         return result
 
     def generate_stream(
@@ -564,6 +567,9 @@ class AsyncNuaClient:
                 result.citations = chunk.chunk.citations
             elif isinstance(chunk.chunk, StatusGenerativeResponse):
                 result.code = chunk.chunk.code
+            elif isinstance(chunk.chunk, ToolsGenerativeResponse):
+                result.tools = chunk.chunk.tools
+
         return result
 
     async def generate_stream(

nuclia/lib/nua_responses.py

@@ -72,6 +72,14 @@ class Image(BaseModel):
     b64encoded: str
 
 
+class Tool(BaseModel):
+    name: str
+    description: str
+    parameters: Dict[str, Any] = Field(
+        default_factory=dict, description="Schema of the tool"
+    )
+
+
 class ChatModel(BaseModel):
     question: str
     retrieval: bool = True
@@ -81,7 +89,7 @@ class ChatModel(BaseModel):
     context: List[Message] = []
     query_context: Union[List[str], Dict[str, str]] = {}
     query_context_order: Dict[str, int] = {}
-    truncate: Optional[bool] = False
+    truncate: Optional[bool] = True
     user_prompt: Optional[UserPrompt] = None
     citations: Optional[bool] = False
     citation_threshold: Optional[float] = Field(
@@ -102,6 +110,9 @@ class ChatModel(BaseModel):
         default=False,
         description="Whether to reorder the query context based on a reranker. This option will also make it so the first response will contain the scores given for each context piece.",
     )
+    tools: List[Tool] = Field(
+        default_factory=list, description="List of tools to choose"
+    )
 
     @model_validator(mode="after")
     def validate_model(self) -> Self:
@@ -109,6 +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:
+            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")
         return self
 
 

{nuclia-4.8.5.dist-info → nuclia-4.8.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nuclia
-Version: 4.8.5
+Version: 4.8.7
 Summary: Nuclia Python SDK
 Author-email: Nuclia <info@nuclia.com>
 License-Expression: MIT
@@ -26,7 +26,7 @@ Requires-Dist: httpcore>=1.0.0
 Requires-Dist: prompt_toolkit
 Requires-Dist: nucliadb_sdk<7,>=6.4
 Requires-Dist: nucliadb_models<7,>=6.4
-Requires-Dist: nuclia-models>=0.32.0
+Requires-Dist: nuclia-models>=0.38.1
 Requires-Dist: tqdm
 Requires-Dist: aiofiles
 Requires-Dist: backoff

{nuclia-4.8.5.dist-info → nuclia-4.8.7.dist-info}/RECORD

@@ -11,9 +11,9 @@ nuclia/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nuclia/lib/conversations.py,sha256=M6qhL9NPEKroYF767S-Q2XWokRrjX02kpYTzRvZKwUE,149
 nuclia/lib/kb.py,sha256=98zpJ8Dp4rQYMNKm4nhpOms3PY-PFOd8acNgDmBud-o,31819
 nuclia/lib/models.py,sha256=lO3TZa4jTQ6C8ptGO1oX-5pxDgv8ocO13z7j5jXOC0Y,1554
-nuclia/lib/nua.py,sha256=TmjQuYKRjP6R2gjelU2Z354G_7Cm2L58kf8kql7MA3M,25626
+nuclia/lib/nua.py,sha256=Y8W0sJWnJqNLQ8y8fPbcJAV-6UtbE1FearHs68qwZQA,25888
 nuclia/lib/nua_chat.py,sha256=ApL1Y1FWvAVUt-Y9a_8TUSJIhg8-UmBSy8TlDPn6tD8,3874
-nuclia/lib/nua_responses.py,sha256=RZ2Bbfnb7F2s3HktB_ivQICIrPNghv6NmiF6e7Xw-go,12159
+nuclia/lib/nua_responses.py,sha256=zji-T-ipYRPo-e1U163Nn_saIiYJYNW3PfjTSL2fQXg,12726
 nuclia/lib/utils.py,sha256=XAVRRICPem-qOIEcssqDuv-vqD-9PICk6Iuj6DK5GKk,4808
 nuclia/sdk/__init__.py,sha256=-nAw8i53XBdmbfTa1FJZ0FNRMNakimDVpD6W4OdES-c,1374
 nuclia/sdk/accounts.py,sha256=7XQ3K9_jlSuk2Cez868FtazZ05xSGab6h3Mt1qMMwIE,647
@@ -61,9 +61,9 @@ nuclia/tests/test_nucliadb/test_crud.py,sha256=GuY76HRvt2DFaNgioKm5n0Aco1HnG7zzV
 nuclia/tests/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nuclia/tests/unit/test_export_import.py,sha256=xo_wVbjUnNlVV65ZGH7LtZ38qy39EkJp2hjOuTHC1nU,980
 nuclia/tests/unit/test_nua_responses.py,sha256=t_hIdVztTi27RWvpfTJUYcCL0lpKdZFegZIwLdaPNh8,319
-nuclia-4.8.5.dist-info/licenses/LICENSE,sha256=Ops2LTti_HJtpmWcanuUTdTY3vKDR1myJ0gmGBKC0FA,1063
-nuclia-4.8.5.dist-info/METADATA,sha256=OkQrkW4V6ydhryMPtBHg5Oeix3kaGuINUZP_VnOpcGs,2337
-nuclia-4.8.5.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-nuclia-4.8.5.dist-info/entry_points.txt,sha256=iZHOyXPNS54r3eQmdi5So20xO1gudI9K2oP4sQsCJRw,46
-nuclia-4.8.5.dist-info/top_level.txt,sha256=cqn_EitXOoXOSUvZnd4q6QGrhm04pg8tLAZtem-Zfdo,7
-nuclia-4.8.5.dist-info/RECORD,,
+nuclia-4.8.7.dist-info/licenses/LICENSE,sha256=Ops2LTti_HJtpmWcanuUTdTY3vKDR1myJ0gmGBKC0FA,1063
+nuclia-4.8.7.dist-info/METADATA,sha256=L_ftUZ5c8tQ5NA4gpDuNzKTkn8Id_XfSxjEk1pleolU,2337
+nuclia-4.8.7.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+nuclia-4.8.7.dist-info/entry_points.txt,sha256=iZHOyXPNS54r3eQmdi5So20xO1gudI9K2oP4sQsCJRw,46
+nuclia-4.8.7.dist-info/top_level.txt,sha256=cqn_EitXOoXOSUvZnd4q6QGrhm04pg8tLAZtem-Zfdo,7
+nuclia-4.8.7.dist-info/RECORD,,

{nuclia-4.8.5.dist-info → nuclia-4.8.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.4.0)
 Root-Is-Purelib: true
 Tag: py3-none-any