nuclia 4.9.4__py3-none-any.whl → 4.9.5__py3-none-any.whl

nuclia/lib/kb.py

@@ -455,12 +455,16 @@ class NucliaDBClient(BaseNucliaDBClient):
         handle_http_sync_errors(response)
         return response
 
-    def delete_task(self, task_id: str) -> httpx.Response:
+    def delete_task(self, task_id: str, cleanup: bool = False) -> httpx.Response:
         if self.writer_session is None:
             raise Exception("KB not configured")
 
+        params = None
+        if cleanup:
+            params = {"cleanup": "true"}
+
         response: httpx.Response = self.writer_session.delete(
-            f"{self.url}{DELETE_TASK.format(task_id=task_id)}",
+            f"{self.url}{DELETE_TASK.format(task_id=task_id)}", params=params
         )
         handle_http_sync_errors(response)
         return response
@@ -827,12 +831,16 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         await handle_http_async_errors(response)
         return response
 
-    async def delete_task(self, task_id: str) -> httpx.Response:
+    async def delete_task(self, task_id: str, cleanup: bool = False) -> httpx.Response:
         if self.writer_session is None:
             raise Exception("KB not configured")
 
+        params = None
+        if cleanup:
+            params = {"cleanup": "true"}
+
         response: httpx.Response = await self.writer_session.delete(
-            f"{self.url}{DELETE_TASK.format(task_id=task_id)}",
+            f"{self.url}{DELETE_TASK.format(task_id=task_id)}", params=params
         )
         await handle_http_async_errors(response)
         return response

nuclia/sdk/task.py

@@ -6,9 +6,9 @@ from nuclia.sdk.auth import NucliaAuth, AsyncNucliaAuth
 from nuclia_models.worker.tasks import (
     ApplyOptions,
     TaskStartKB,
-    TaskResponse,
     TaskList,
     TaskName,
+    TaskResponse,
     PARAMETERS_TYPING,
     PublicTaskSet,
     TASKS,
@@ -64,7 +64,7 @@ class NucliaTask:
         return TaskResponse.model_validate(response.json())
 
     @kb
-    def delete(self, *args, task_id: str, **kwargs):
+    def delete(self, *args, task_id: str, cleanup: bool = False, **kwargs):
         """
         Delete task
 
@@ -72,7 +72,7 @@ class NucliaTask:
         """
         ndb: NucliaDBClient = kwargs["ndb"]
         try:
-            _ = ndb.delete_task(task_id=task_id)
+            _ = ndb.delete_task(task_id=task_id, cleanup=cleanup)
         except InvalidPayload:
             pass
 
@@ -158,7 +158,7 @@ class AsyncNucliaTask:
         return TaskResponse.model_validate(response.json())
 
     @kb
-    async def delete(self, *args, task_id: str, **kwargs):
+    async def delete(self, *args, task_id: str, cleanup: bool = False, **kwargs):
         """
         Delete task
 
@@ -166,7 +166,7 @@ class AsyncNucliaTask:
         """
         ndb: AsyncNucliaDBClient = kwargs["ndb"]
         try:
-            _ = await ndb.delete_task(task_id=task_id)
+            _ = await ndb.delete_task(task_id=task_id, cleanup=cleanup)
         except InvalidPayload:
             pass
 

{nuclia-4.9.4.dist-info → nuclia-4.9.5.dist-info}/METADATA

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

{nuclia-4.9.4.dist-info → nuclia-4.9.5.dist-info}/RECORD

@@ -9,7 +9,7 @@ nuclia/cli/run.py,sha256=B1hP0upSbSCqqT89WAwsd93ZxkAoF6ajVyLOdYmo8fU,1560
 nuclia/cli/utils.py,sha256=iZ3P8juBdAGvaRUd2BGz7bpUXNDHdPrC5p876yyZ2Cs,1223
 nuclia/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nuclia/lib/conversations.py,sha256=M6qhL9NPEKroYF767S-Q2XWokRrjX02kpYTzRvZKwUE,149
-nuclia/lib/kb.py,sha256=hbWg-0pX1OCt9ezUXYcOfyWHahgmZLUQe2nlSycHetU,31364
+nuclia/lib/kb.py,sha256=V1hX3aPdrrPX5slN11K7IwSA22E-M-k7rxfwwh1_Dts,31606
 nuclia/lib/models.py,sha256=ekEQrVIFU3aFvt60yQh-zpWkGNORBMSc7c5Hd_VzPzI,1564
 nuclia/lib/nua.py,sha256=qP6Gv9ck117kdqNuKVl_qHodrDE8CS8xdh_bdzeJMiQ,30631
 nuclia/lib/nua_chat.py,sha256=ApL1Y1FWvAVUt-Y9a_8TUSJIhg8-UmBSy8TlDPn6tD8,3874
@@ -33,7 +33,7 @@ nuclia/sdk/process.py,sha256=WuNnqaWprp-EABWDC_z7O2woesGIlYWnDUKozh7Ibr4,2241
 nuclia/sdk/remi.py,sha256=BEb3O9R2jOFlOda4vjFucKKGO1c2eTkqYZdFlIy3Zmo,4357
 nuclia/sdk/resource.py,sha256=0lSvD4e1FpN5iM9W295dOKLJ8hsXfIe8HKdo0HsVg20,13976
 nuclia/sdk/search.py,sha256=tHpx0gwVwSepa3dn7AX2iWs7JO0pQkjJk7v8TEoL7Gg,27742
-nuclia/sdk/task.py,sha256=UawH-7IneRIGVOiLdDQ2vDBwe5eI51UXRbLRRVeu3C4,6095
+nuclia/sdk/task.py,sha256=V2M_44tkNB-edflf0mOjmgTg-w6pTjEYzlcCPkdJZYc,6175
 nuclia/sdk/upload.py,sha256=ZBzYROF3yP-77HcaR06OBsFjJAbTOCvF-nlxaqQZsT4,22720
 nuclia/sdk/zones.py,sha256=1ARWrTsTuzj8zguanpX3OaIw-3Qq_ULS_g4GG2mHxOA,342
 nuclia/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -63,9 +63,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.9.4.dist-info/licenses/LICENSE,sha256=Ops2LTti_HJtpmWcanuUTdTY3vKDR1myJ0gmGBKC0FA,1063
-nuclia-4.9.4.dist-info/METADATA,sha256=2fk95GmG2Qa-GGAOPxWXWgA-gAGv-vEIHMRftKOsgA0,2341
-nuclia-4.9.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nuclia-4.9.4.dist-info/entry_points.txt,sha256=iZHOyXPNS54r3eQmdi5So20xO1gudI9K2oP4sQsCJRw,46
-nuclia-4.9.4.dist-info/top_level.txt,sha256=cqn_EitXOoXOSUvZnd4q6QGrhm04pg8tLAZtem-Zfdo,7
-nuclia-4.9.4.dist-info/RECORD,,
+nuclia-4.9.5.dist-info/licenses/LICENSE,sha256=Ops2LTti_HJtpmWcanuUTdTY3vKDR1myJ0gmGBKC0FA,1063
+nuclia-4.9.5.dist-info/METADATA,sha256=SkwJpFksjfU0CEWqpLPdHCFeH8yQ3J9XJlp0LzGjRDE,2341
+nuclia-4.9.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nuclia-4.9.5.dist-info/entry_points.txt,sha256=iZHOyXPNS54r3eQmdi5So20xO1gudI9K2oP4sQsCJRw,46
+nuclia-4.9.5.dist-info/top_level.txt,sha256=cqn_EitXOoXOSUvZnd4q6QGrhm04pg8tLAZtem-Zfdo,7
+nuclia-4.9.5.dist-info/RECORD,,