nuclia 4.8.9__py3-none-any.whl → 4.9.0__py3-none-any.whl

nuclia/lib/kb.py

@@ -1,5 +1,4 @@
 import base64
-import csv
 import os
 from enum import Enum
 from typing import Dict, Optional, Union
@@ -15,10 +14,10 @@ from tqdm import tqdm
 from nuclia_models.events.activity_logs import (  # type: ignore
     ActivityLogsQuery,
     ActivityLogsSearchQuery,
-    ActivityLogsChatQuery,
+    ActivityLogsAskQuery,
     DownloadActivityLogsQuery,
     DownloadActivityLogsSearchQuery,
-    DownloadActivityLogsChatQuery,
+    DownloadActivityLogsAskQuery,
     EventType,
     DownloadFormat,
 )
@@ -46,11 +45,9 @@ DOWNLOAD_EXPORT_URL = "/export/{export_id}"
 DOWNLOAD_URL = "/{uri}"
 TUS_UPLOAD_RESOURCE_URL = "/resource/{rid}/file/{field}/tusupload"
 TUS_UPLOAD_URL = "/tusupload"
-LEGACY_ACTIVITY_LOG_URL = "/activity/download?type={type}&month={month}"
 ACTIVITY_LOG_URL = "/activity/{type}/query/download"
 ACTIVITY_LOG_DOWNLOAD_REQUEST_URL = "/activity/download_request/{request_id}"
 ACTIVITY_LOG_QUERY_URL = "/activity/{type}/query"
-FEEDBACK_LOG_URL = "/feedback/{month}"
 NOTIFICATIONS = "/notifications"
 REMI_QUERY_URL = "/remi/query"
 REMI_EVENT_URL = "/remi/events/{event_id}"
@@ -73,23 +70,6 @@ class Environment(str, Enum):
     OSS = "OSS"
 
 
-class LogType(str, Enum):
-    # Nucliadb
-    VISITED = "visited"
-    MODIFIED = "modified"
-    DELETED = "deleted"
-    NEW = "new"
-    SEARCH = "search"
-    SUGGEST = "suggest"
-    INDEXED = "indexed"
-    CHAT = "chat"
-    # Tasks
-    STARTED = "started"
-    STOPPED = "stopped"
-    # Processor
-    PROCESSED = "processed"
-
-
 class BaseNucliaDBClient:
     environment: Environment
     base_url: str
@@ -352,41 +332,10 @@ class NucliaDBClient(BaseNucliaDBClient):
         handle_http_sync_errors(response)
         return response
 
-    def logs(self, type: LogType, month: str) -> list[list[str]]:
-        if self.reader_session is None:
-            raise Exception("KB not configured")
-
-        if type != "feedback":
-            url = LEGACY_ACTIVITY_LOG_URL.format(type=type.value, month=month)
-            response: httpx.Response = self.reader_session.get(url)
-            handle_http_sync_errors(response)
-            return [row for row in csv.reader(response.iter_lines())]
-        else:
-            feedback_url = f"{self.url}{FEEDBACK_LOG_URL.format(month=month)}"
-            feedback_response: httpx.Response = self.reader_session.get(feedback_url)
-            handle_http_sync_errors(feedback_response)
-            feedbacks = [row for row in csv.reader(feedback_response.iter_lines())]
-            answers = self.logs(type=LogType.CHAT, month=month)
-            # first row with the columns headers
-            results = [[*feedbacks[0], *answers[0][:-1]]]
-            for feedback in feedbacks[1:]:
-                learning_id = feedback[1]
-                # search for the corresponding question/answer
-                # (the learning id is the same for both question/answer and feedback,
-                # and is the second column in the Q/A csv)
-                matching_answers = [
-                    answer for answer in answers if answer[1] == learning_id
-                ]
-                if len(matching_answers) > 0:
-                    results.append([*feedback, *matching_answers[0][:-1]])
-                else:
-                    results.append(feedback)
-            return results
-
     def logs_query(
         self,
         type: EventType,
-        query: Union[ActivityLogsQuery, ActivityLogsSearchQuery, ActivityLogsChatQuery],
+        query: Union[ActivityLogsQuery, ActivityLogsSearchQuery, ActivityLogsAskQuery],
     ) -> requests.Response:
         if self.stream_session is None:
             raise Exception("KB not configured")
@@ -405,7 +354,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         query: Union[
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ],
         download_format: DownloadFormat,
     ):
@@ -709,43 +658,10 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         await handle_http_async_errors(response)
         return response
 
-    async def logs(self, type: LogType, month: str) -> list[list[str]]:
-        if self.reader_session is None:
-            raise Exception("KB not configured")
-
-        if type != "feedback":
-            url = LEGACY_ACTIVITY_LOG_URL.format(type=type.value, month=month)
-            response: httpx.Response = await self.reader_session.get(url)
-            await handle_http_async_errors(response)
-            return [row for row in csv.reader(response.iter_lines())]
-        else:
-            feedback_url = f"{self.url}{FEEDBACK_LOG_URL.format(month=month)}"
-            feedback_response: httpx.Response = await self.reader_session.get(
-                feedback_url
-            )
-            await handle_http_async_errors(feedback_response)
-            feedbacks = [row for row in csv.reader(feedback_response.iter_lines())]
-            answers = await self.logs(type=LogType.CHAT, month=month)
-            # first row with the columns headers
-            results = [[*feedbacks[0], *answers[0][:-1]]]
-            for feedback in feedbacks[1:]:
-                learning_id = feedback[1]
-                # search for the corresponding question/answer
-                # (the learning id is the same for both question/answer and feedback,
-                # and is the second column in the Q/A csv)
-                matching_answers = [
-                    answer for answer in answers if answer[1] == learning_id
-                ]
-                if len(matching_answers) > 0:
-                    results.append([*feedback, *matching_answers[0][:-1]])
-                else:
-                    results.append(feedback)
-            return results
-
     async def logs_query(
         self,
         type: EventType,
-        query: Union[ActivityLogsQuery, ActivityLogsSearchQuery, ActivityLogsChatQuery],
+        query: Union[ActivityLogsQuery, ActivityLogsSearchQuery, ActivityLogsAskQuery],
     ) -> httpx.Response:
         if self.reader_session is None:
             raise Exception("KB not configured")
@@ -763,7 +679,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         query: Union[
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ],
         download_format: DownloadFormat,
     ):

nuclia/lib/models.py

@@ -48,7 +48,7 @@ class GraphRelation(BaseModel):
     destination: GraphEntity
 
     def to_relation(self) -> Relation:
-        return Relation.parse_obj(
+        return Relation.model_validate(
             {
                 "from": self.source.to_dict(),
                 "to": self.destination.to_dict(),
@@ -60,6 +60,6 @@ class GraphRelation(BaseModel):
 
 def get_relation(relation: Union[GraphRelation, dict]) -> GraphRelation:
     if isinstance(relation, dict):
-        return GraphRelation.parse_obj(relation)
+        return GraphRelation.model_validate(relation)
     else:
         return relation

nuclia/sdk/logs.py

@@ -1,11 +1,11 @@
 from nuclia.decorators import kb
-from nuclia.lib.kb import LogType, NucliaDBClient, AsyncNucliaDBClient
+from nuclia.lib.kb import NucliaDBClient, AsyncNucliaDBClient
 from nuclia_models.events.activity_logs import (  # type: ignore
     ActivityLogsQuery,
-    ActivityLogsChatQuery,
+    ActivityLogsAskQuery,
     ActivityLogsSearchQuery,
     DownloadActivityLogsQuery,
-    DownloadActivityLogsChatQuery,
+    DownloadActivityLogsAskQuery,
     DownloadActivityLogsSearchQuery,
     DownloadFormat,
     EventType,
@@ -25,23 +25,6 @@ WAIT_FOR_DOWNLOAD_TIMEOUT = 120
 
 
 class NucliaLogs:
-    @kb
-    def get(
-        self, *args, type: Union[LogType, str], month: str, **kwargs
-    ) -> list[list[str]]:
-        """
-        Download activity logs.
-
-        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, STARTED, STOPPED, PROCESSED
-        :param month: YYYY-MM
-        """
-        if isinstance(type, str):
-            type = LogType[type.upper()]
-
-        ndb: NucliaDBClient = kwargs["ndb"]
-        resp = ndb.logs(type=type, month=month)
-        return resp
-
     @kb
     def query(
         self,
@@ -51,25 +34,25 @@ class NucliaLogs:
             dict,
             ActivityLogsQuery,
             ActivityLogsSearchQuery,
-            ActivityLogsChatQuery,
+            ActivityLogsAskQuery,
         ],
         **kwargs,
     ) -> ActivityLogsOutput:
         """
         Query activity logs.
 
-        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, STARTED, STOPPED, PROCESSED
+        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, ASK, STARTED, STOPPED, PROCESSED
         :param query: ActivityLogsQuery
         """
         _type = EventType[type.upper()] if isinstance(type, str) else type
         _query: Union[
             ActivityLogsQuery,
             ActivityLogsSearchQuery,
-            ActivityLogsChatQuery,
+            ActivityLogsAskQuery,
         ]
         if isinstance(query, dict):
-            if _type is EventType.CHAT:
-                _query = ActivityLogsChatQuery.model_validate(query)
+            if _type in (EventType.ASK, EventType.CHAT):  # TODO: deprecate chat event
+                _query = ActivityLogsAskQuery.model_validate(query)
             elif type is EventType.SEARCH:
                 _query = ActivityLogsSearchQuery.model_validate(query)
             else:
@@ -95,7 +78,7 @@ class NucliaLogs:
             dict,
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ],
         download_format: Union[DownloadFormat, str],
         wait: bool = False,
@@ -104,8 +87,8 @@ class NucliaLogs:
         """
         Download activity logs.
 
-        :param type: SEARCH, CHAT, VISITED, MODIFIED, DELETED, NEW, STARTED, STOPPED, PROCESSED
-        :param type: NDJSON, CSV
+        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, ASK, STARTED, STOPPED, PROCESSED
+        :param download_format: NDJSON, CSV
         :param query: DownloadActivityLogsQuery
         """
         _type = EventType[type.upper()] if isinstance(type, str) else type
@@ -119,11 +102,11 @@ class NucliaLogs:
             dict,
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ]
         if isinstance(query, dict):
-            if _type is EventType.CHAT:
-                _query = DownloadActivityLogsChatQuery.model_validate(query)
+            if _type in (EventType.ASK, EventType.CHAT):  # TODO: deprecate chat event
+                _query = DownloadActivityLogsAskQuery.model_validate(query)
             elif type is EventType.SEARCH:
                 _query = DownloadActivityLogsSearchQuery.model_validate(query)
             else:
@@ -168,23 +151,6 @@ class NucliaLogs:
 
 
 class AsyncNucliaLogs:
-    @kb
-    async def get(
-        self, *args, type: Union[LogType, str], month: str, **kwargs
-    ) -> list[list[str]]:
-        """
-        Download activity logs.
-
-        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, STARTED, STOPPED, PROCESSED
-        :param month: YYYY-MM
-        """
-        if isinstance(type, str):
-            type = LogType[type.upper()]
-
-        ndb: AsyncNucliaDBClient = kwargs["ndb"]
-        resp = await ndb.logs(type=type, month=month)
-        return resp
-
     @kb
     async def query(
         self,
@@ -194,25 +160,25 @@ class AsyncNucliaLogs:
             dict,
             ActivityLogsQuery,
             ActivityLogsSearchQuery,
-            ActivityLogsChatQuery,
+            ActivityLogsAskQuery,
         ],
         **kwargs,
     ) -> ActivityLogsOutput:
         """
         Query activity logs.
 
-        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, STARTED, STOPPED, PROCESSED
+        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, ASK, STARTED, STOPPED, PROCESSED
         :param query: ActivityLogsQuery
         """
         _type = EventType[type.upper()] if isinstance(type, str) else type
         _query: Union[
             ActivityLogsQuery,
             ActivityLogsSearchQuery,
-            ActivityLogsChatQuery,
+            ActivityLogsAskQuery,
         ]
         if isinstance(query, dict):
-            if _type is EventType.CHAT:
-                _query = ActivityLogsChatQuery.model_validate(query)
+            if _type in (EventType.ASK, EventType.CHAT):  # TODO: deprecate chat event
+                _query = ActivityLogsAskQuery.model_validate(query)
             elif type is EventType.SEARCH:
                 _query = ActivityLogsSearchQuery.model_validate(query)
             else:
@@ -238,7 +204,7 @@ class AsyncNucliaLogs:
             dict,
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ],
         download_format: Union[DownloadFormat, str],
         wait: bool = False,
@@ -247,8 +213,8 @@ class AsyncNucliaLogs:
         """
         Download activity logs.
 
-        :param type: SEARCH, CHAT, VISITED, MODIFIED, DELETED, NEW, STARTED, STOPPED, PROCESSED
-        :param type: NDJSON, CSV
+        :param type: VISITED, MODIFIED, DELETED, NEW, SEARCH, SUGGEST, INDEXED, CHAT, ASK, STARTED, STOPPED, PROCESSED
+        :param download_format: NDJSON, CSV
         :param query: DownloadActivityLogsQuery
         """
         _type = EventType[type.upper()] if isinstance(type, str) else type
@@ -262,11 +228,11 @@ class AsyncNucliaLogs:
             dict,
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ]
         if isinstance(query, dict):
-            if _type is EventType.CHAT:
-                _query = DownloadActivityLogsChatQuery.model_validate(query)
+            if _type in (EventType.ASK, EventType.CHAT):  # TODO: deprecate chat event
+                _query = DownloadActivityLogsAskQuery.model_validate(query)
             elif type is EventType.SEARCH:
                 _query = DownloadActivityLogsSearchQuery.model_validate(query)
             else:

nuclia/sdk/search.py

@@ -20,6 +20,8 @@ from nucliadb_models.search import (
     SyncAskResponse,
     ChatModel,
 )
+from nucliadb_models.graph.requests import GraphSearchRequest
+from nucliadb_models.graph.responses import GraphSearchResponse
 from pydantic import ValidationError
 
 from nuclia.data import get_async_auth, get_auth
@@ -331,6 +333,33 @@ class NucliaSearch:
                 result.tokens = ask_response.metadata.tokens.model_dump()
         return result
 
+    @kb
+    def graph(
+        self,
+        *,
+        query: Union[dict, GraphSearchRequest],
+        **kwargs,
+    ) -> GraphSearchResponse:
+        """
+        Perform a graph path query.
+
+        See https://docs.nuclia.dev/docs/api#tag/Search/operation/graph_search_knowledgebox_kb__kbid__graph_post
+        """
+        ndb: NucliaDBClient = kwargs["ndb"]
+
+        if isinstance(query, GraphSearchRequest):
+            req = query
+        elif isinstance(query, dict):
+            try:
+                req = GraphSearchRequest.model_validate(query)
+            except ValidationError:
+                logger.exception("Error validating query")
+                raise
+        else:
+            raise Exception(f"Invalid query: '{query}'")
+
+        return ndb.ndb.graph_search(req, kbid=ndb.kbid)
+
 
 class AsyncNucliaSearch:
     """
@@ -659,3 +688,30 @@ class AsyncNucliaSearch:
             else:  # pragma: no cover
                 warnings.warn(f"Unknown ask stream item type: {ask_response_item.type}")
         return result
+
+    @kb
+    async def graph(
+        self,
+        *,
+        query: Union[dict, GraphSearchRequest],
+        **kwargs,
+    ) -> GraphSearchResponse:
+        """
+        Perform a graph path query.
+
+        See https://docs.nuclia.dev/docs/api#tag/Search/operation/graph_search_knowledgebox_kb__kbid__graph_post
+        """
+        ndb: AsyncNucliaDBClient = kwargs["ndb"]
+
+        if isinstance(query, GraphSearchRequest):
+            req = query
+        elif isinstance(query, dict):
+            try:
+                req = GraphSearchRequest.model_validate(query)
+            except ValidationError:
+                logger.exception("Error validating query")
+                raise
+        else:
+            raise Exception(f"Invalid query: '{query}'")
+
+        return await ndb.ndb.graph_search(req, kbid=ndb.kbid)

nuclia/tests/fixtures.py

@@ -49,8 +49,8 @@ def testing_config(testing_kb, testing_nua, testing_user):
         nuclia_auth._config.set_default_kb(TESTING_KBID)
         nuclia_auth._config.set_default_nua(client_id)
 
-    yield
-    reset_config_file()
+        yield
+        reset_config_file()
 
 
 @pytest.fixture(autouse=True)

nuclia/tests/test_kb/test_conversation.py

@@ -2,7 +2,6 @@ import os
 
 from nucliadb_sdk.v2.exceptions import NotFoundError
 
-from nuclia.sdk.kb import NucliaKB
 from nuclia.sdk.resource import NucliaResource
 from nuclia.sdk.upload import NucliaUpload
 
@@ -10,7 +9,6 @@ path = f"{os.path.dirname(__file__)}/../assets/conversation.json"
 
 
 def test_conversation(testing_config):
-    nkb = NucliaKB()
     nresource = NucliaResource()
     try:
         res = nresource.get(slug="conversation1")
@@ -18,9 +16,6 @@ def test_conversation(testing_config):
     except NotFoundError:
         pass
 
-    res = nkb.list()
-    assert len(res.resources) == 2
-
     nu = NucliaUpload()
     nu.conversation(path=path, slug="conversation1", field="c1")
 

nuclia/tests/test_kb/test_graph.py

@@ -6,6 +6,13 @@ from time import sleep
 
 def test_graph(testing_config):
     nkb = NucliaKB()
+
+    # ensure KB is clean
+    try:
+        nkb.delete_graph(slug="graph1")
+    except NotFoundError:
+        pass
+
     nkb.add_graph(
         slug="graph1",
         graph=[
@@ -28,6 +35,25 @@ def test_graph(testing_config):
     )
     relations = nkb.get_graph(slug="graph1")
     assert len(relations) == 3
+
+    # XXX: sometimes, nucliadb needs some more time to index the graph. We retry
+    # some times and fail if that keeps hapenning
+    RETRIES = 5
+    DELAY = 0.5
+    for _ in range(RETRIES):
+        paths = nkb.search.graph(
+            query={"query": {"prop": "path", "source": {"value": "Alice"}}}
+        )
+        try:
+            assert len(paths.paths) == 2
+        except AssertionError:
+            # wait for a bit before retrying
+            print("Graph was not indexed yet, waiting a bit...")
+            sleep(DELAY)
+        else:
+            break
+    assert len(paths.paths) == 2
+
     nkb.update_graph(
         slug="graph1",
         graph=[
@@ -40,6 +66,20 @@ def test_graph(testing_config):
     )
     relations = nkb.get_graph(slug="graph1")
     assert len(relations) == 4
+
+    for _ in range(RETRIES):
+        paths = nkb.search.graph(
+            query={"query": {"prop": "path", "source": {"value": "Victor"}}}
+        )
+        try:
+            assert len(paths.paths) == 1
+        except AssertionError:
+            print("Graph was not indexed yet, waiting a bit...")
+            sleep(DELAY)
+        else:
+            break
+    assert len(paths.paths) == 1
+
     nkb.delete_graph(slug="graph1")
     try:
         sleep(0.5)

nuclia/tests/test_kb/test_logs.py

@@ -1,24 +1,15 @@
 from nuclia.sdk.kb import NucliaKB, AsyncNucliaKB
 from nuclia.tests.fixtures import IS_PROD
-from nuclia.lib.kb import LogType
 from nuclia_models.events.activity_logs import (
     ActivityLogsQuery,
     Pagination,
     DownloadActivityLogsQuery,
     DownloadFormat,
+    EventType,
 )
 import pytest
 
 
-def test_logs(testing_config):
-    if not IS_PROD:
-        assert True
-        return
-    nkb = NucliaKB()
-    logs = nkb.logs.get(type=LogType.NEW, month="2024-06")
-    assert len(logs) == 23
-
-
 def test_activity_logs_query(testing_config):
     if not IS_PROD:
         assert True
@@ -30,7 +21,7 @@ def test_activity_logs_query(testing_config):
         pagination=Pagination(limit=10),
     )
     nkb = NucliaKB()
-    output = nkb.logs.query(type=LogType.CHAT, query=query)
+    output = nkb.logs.query(type=EventType.CHAT, query=query)
     assert len(output.data) == 10
     assert output.has_more
 
@@ -46,24 +37,48 @@ def test_activity_logs_download(testing_config):
     )
     nkb = NucliaKB()
     output = nkb.logs.download(
-        type=LogType.CHAT, query=query, download_format=DownloadFormat.NDJSON
+        type=EventType.CHAT, query=query, download_format=DownloadFormat.NDJSON
     )
     assert output.request_id
     assert output.download_url is None
 
 
 @pytest.mark.asyncio
-async def test_logs_async(testing_config):
+async def test_activity_logs_query_async(testing_config):
     if not IS_PROD:
         assert True
         return
+    query = ActivityLogsQuery(
+        year_month="2024-10",
+        show=["id", "date", "client_type", "total_duration"],
+        filters={},
+        pagination=Pagination(limit=10),
+    )
     nkb = AsyncNucliaKB()
-    logs = await nkb.logs.get(type=LogType.NEW, month="2024-06")
-    assert len(logs) == 23
+    output = await nkb.logs.query(type=EventType.CHAT, query=query)
+    assert len(output.data) == 10
+    assert output.has_more
 
 
 @pytest.mark.asyncio
-async def test_activity_logs_query_async(testing_config):
+async def test_activity_logs_download_async(testing_config):
+    if not IS_PROD:
+        assert True
+        return
+    query = DownloadActivityLogsQuery(
+        year_month="2024-10",
+        show=["id", "date", "client_type", "total_duration"],
+        filters={},
+    )
+    nkb = AsyncNucliaKB()
+    output = await nkb.logs.download(
+        type=EventType.CHAT, query=query, download_format=DownloadFormat.NDJSON
+    )
+    assert output.request_id
+    assert output.download_url is None
+
+
+def test_activity_logs_ask_query(testing_config):
     if not IS_PROD:
         assert True
         return
@@ -73,14 +88,14 @@ async def test_activity_logs_query_async(testing_config):
         filters={},
         pagination=Pagination(limit=10),
     )
-    nkb = AsyncNucliaKB()
-    output = await nkb.logs.query(type=LogType.CHAT, query=query)
+    nkb = NucliaKB()
+    output = nkb.logs.query(type=EventType.ASK, query=query)
     assert len(output.data) == 10
     assert output.has_more
 
 
 @pytest.mark.asyncio
-async def test_activity_logs_download_async(testing_config):
+async def test_activity_logs_ask_download_async(testing_config):
     if not IS_PROD:
         assert True
         return
@@ -91,7 +106,7 @@ async def test_activity_logs_download_async(testing_config):
     )
     nkb = AsyncNucliaKB()
     output = await nkb.logs.download(
-        type=LogType.CHAT, query=query, download_format=DownloadFormat.NDJSON
+        type=EventType.ASK, query=query, download_format=DownloadFormat.NDJSON
     )
     assert output.request_id
     assert output.download_url is None

{nuclia-4.8.9.dist-info → nuclia-4.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nuclia
-Version: 4.8.9
+Version: 4.9.0
 Summary: Nuclia Python SDK
 Author-email: Nuclia <info@nuclia.com>
 License-Expression: MIT
@@ -24,9 +24,9 @@ Requires-Dist: requests
 Requires-Dist: httpx
 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.39.0
+Requires-Dist: nucliadb_sdk<7,>=6.5
+Requires-Dist: nucliadb_models<7,>=6.5
+Requires-Dist: nuclia-models>=0.41.1
 Requires-Dist: tqdm
 Requires-Dist: aiofiles
 Requires-Dist: backoff

{nuclia-4.8.9.dist-info → nuclia-4.9.0.dist-info}/RECORD

@@ -9,8 +9,8 @@ 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=uUBts_s_St4DBeUTn3wSUdRSqtX2GbeLC5jAUVQAy-c,32190
-nuclia/lib/models.py,sha256=lO3TZa4jTQ6C8ptGO1oX-5pxDgv8ocO13z7j5jXOC0Y,1554
+nuclia/lib/kb.py,sha256=jV-L8-a5oFVHIbFfeGO17hXf2IQk9UuW-9fq5D3LO6Y,28508
+nuclia/lib/models.py,sha256=ekEQrVIFU3aFvt60yQh-zpWkGNORBMSc7c5Hd_VzPzI,1564
 nuclia/lib/nua.py,sha256=sUVFdCjvLigTqUUhILywdHpiC0qKCtKPABn5kUXfuxQ,27064
 nuclia/lib/nua_chat.py,sha256=ApL1Y1FWvAVUt-Y9a_8TUSJIhg8-UmBSy8TlDPn6tD8,3874
 nuclia/lib/nua_responses.py,sha256=9WRGpvvhA2ZOyOv9kzX1zI7C3ypM8ySqciVYCSibUjo,12724
@@ -24,27 +24,27 @@ nuclia/sdk/export_import.py,sha256=y5cTOxhILwRPIvR2Ya12bk-ReGbeDzA3C9TPxgnOHD4,9
 nuclia/sdk/kb.py,sha256=iDJimbzw_R9VeUu7a4F5wXoLdLxCim7W3UeEaAQ5dGI,26820
 nuclia/sdk/kbs.py,sha256=nXEvg5ddZYdDS8Kie7TrN-s1meU9ecYLf9FlT5xr-ro,9131
 nuclia/sdk/logger.py,sha256=UHB81eS6IGmLrsofKxLh8cmF2AsaTj_HXP0tGqMr_HM,57
-nuclia/sdk/logs.py,sha256=P7PJwrSfJMTJqykWZZm9yhtWtIODZYvPqsaStKoPnmo,10400
+nuclia/sdk/logs.py,sha256=3jfORpo8fzZiXFFSbGY0o3Bre1ZgJaKQCXgxP1keNHw,9614
 nuclia/sdk/nua.py,sha256=6t0m0Sx-UhqNU2Hx9v6vTwy0m3a30K4T0KmP9G43MzY,293
 nuclia/sdk/nucliadb.py,sha256=bOESIppPgY7IrNqrYY7T3ESoxwttbOSTm5zj1xUS1jI,1288
 nuclia/sdk/predict.py,sha256=KF7iT2aasaB9DIEAwqktXbOl2H_Y_ne-6-SEErN7YOk,9095
 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=OpW2Tl0RQwG6e2Ytw6Iz7L0SMzaX06qTpb-f4mzw6_g,23435
+nuclia/sdk/search.py,sha256=1mLJzDO-W3ObReDL1xK8zgjIxkdpuIhyf87-L4bISKQ,25180
 nuclia/sdk/task.py,sha256=UawH-7IneRIGVOiLdDQ2vDBwe5eI51UXRbLRRVeu3C4,6095
 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
 nuclia/tests/conftest.py,sha256=-Gv9Q20tBVQNNWqWHGNmzkXJpijmgTdcvNnaRAUluGk,115
-nuclia/tests/fixtures.py,sha256=ppjT1VgEdkrqvRPoqt1LU7T6rUf9qZ1UkplQz4hT0IQ,1787
+nuclia/tests/fixtures.py,sha256=rcGGbtxQbGK9vR9efL7BsE1ugJl15QQDIxUyFZLfv-E,1795
 nuclia/tests/assets/conversation.json,sha256=jLmVngHEW8QdAzSMa-MYgbJgYxOO8tHq6Ia029-CD0E,348
 nuclia/tests/test_kb/test_backup.py,sha256=67fAw1C_NB363G-wuI9LnSH6nG_YrYs-uBcFyBx6Peg,2932
-nuclia/tests/test_kb/test_conversation.py,sha256=L__KMBucqChPGeYR_XwXl7Elt4LDlp1mcmFdD71U8cQ,840
+nuclia/tests/test_kb/test_conversation.py,sha256=plsWY_gEhK3P4CsIO9jkBmimZk3UDGeMYPu7YVFeuIQ,727
 nuclia/tests/test_kb/test_export_import.py,sha256=lQEww2jFNHZYcudFJqcHhoWAPrmtvvnPvcFqrijxLbo,1019
-nuclia/tests/test_kb/test_graph.py,sha256=PQSdqvPvOHr71pfdbUpsAK-_DxBQh7spxtJaaleF87A,1434
+nuclia/tests/test_kb/test_graph.py,sha256=ecAB-lWqm3_796HUuF-fSpGzlhsXV9NGZnAjvxXANps,2540
 nuclia/tests/test_kb/test_labels.py,sha256=IUdTq4mzv0OrOkwBWWy4UwKGKyJybtoHrgvXr676vyY,961
-nuclia/tests/test_kb/test_logs.py,sha256=Iw0v-R5QLlfHKxL8roPeOQKjTylgpYgot0wG4vsXQTE,2609
+nuclia/tests/test_kb/test_logs.py,sha256=Z9ELtiiU9NniITJzeWt92GCcERKYy9Nwc_fUVPboRU0,3121
 nuclia/tests/test_kb/test_remi.py,sha256=OX5N-MHbgcwpLg6fBjrAK_KhqkMspJo_VKQHCBCayZ8,2080
 nuclia/tests/test_kb/test_resource.py,sha256=05Xgmg5fwcPW2PZKnUSSjr6MPXp5w8XDgx8plfNcR68,1102
 nuclia/tests/test_kb/test_search.py,sha256=bsQhfB6-NYFwY3gqkrVJJnru153UgZqEhV22ho4VeWM,4660
@@ -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.9.dist-info/licenses/LICENSE,sha256=Ops2LTti_HJtpmWcanuUTdTY3vKDR1myJ0gmGBKC0FA,1063
-nuclia-4.8.9.dist-info/METADATA,sha256=1Jn5Qc4x7iXS9dfqyY__tEBakgz8xpPxbnl-rCqTWLM,2337
-nuclia-4.8.9.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-nuclia-4.8.9.dist-info/entry_points.txt,sha256=iZHOyXPNS54r3eQmdi5So20xO1gudI9K2oP4sQsCJRw,46
-nuclia-4.8.9.dist-info/top_level.txt,sha256=cqn_EitXOoXOSUvZnd4q6QGrhm04pg8tLAZtem-Zfdo,7
-nuclia-4.8.9.dist-info/RECORD,,
+nuclia-4.9.0.dist-info/licenses/LICENSE,sha256=Ops2LTti_HJtpmWcanuUTdTY3vKDR1myJ0gmGBKC0FA,1063
+nuclia-4.9.0.dist-info/METADATA,sha256=ZHT0wCR1AhZtfnWdcD-uSmfNkb8QznVDuWRt_2f6yP8,2337
+nuclia-4.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nuclia-4.9.0.dist-info/entry_points.txt,sha256=iZHOyXPNS54r3eQmdi5So20xO1gudI9K2oP4sQsCJRw,46
+nuclia-4.9.0.dist-info/top_level.txt,sha256=cqn_EitXOoXOSUvZnd4q6QGrhm04pg8tLAZtem-Zfdo,7
+nuclia-4.9.0.dist-info/RECORD,,

{nuclia-4.8.9.dist-info → nuclia-4.9.0.dist-info}/WHEEL

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