nuclia 4.8.9__tar.gz → 4.9.0__tar.gz

{nuclia-4.8.9 → nuclia-4.9.0}/CHANGELOG.md

@@ -1,6 +1,18 @@
 # Changelog
 
 
+## 4.9.0 (2025-06-05)
+
+
+- Add search graph method
+
+
+## 4.8.10 (2025-05-28)
+
+
+- Activity logs: Add event type `ASK` and deprecate `CHAT`
+
+
 ## 4.8.9 (2025-05-15)
 
 

{nuclia-4.8.9 → nuclia-4.9.0}/PKG-INFO

@@ -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.9.0/VERSION

@@ -0,0 +1 @@
+4.9.0

{nuclia-4.8.9 → nuclia-4.9.0}/docs/05-search.md

@@ -139,3 +139,33 @@ In these cases, you can just pass your query as a dictionnary in the `query` par
   search.find(query={"query": "My search", "filters": ["/icon/application/pdf", "/classification.labels/region/Asia"]})
   search.ask(query={"query": "My search","top_k": 5})
   ```
+
+## Graph queries
+
+The Python SDK allows graph queries supported by the `/graph` endpoint. Although
+a bit cumbersome, the knowledge graph can be queried as in this example:
+
+- CLI:
+
+  ```bash
+  nuclia kb search graph --query='{"query": {"prop": "path", "source": {"value": "Rust"}, "destination": {"value": "Python"}}}'
+  ```
+
+- SDK:
+
+  ```python
+  from nuclia import sdk
+  search = sdk.NucliaSearch()
+  search.graph(
+      query={
+          "query": {
+              "prop": "path",
+              "source": {"value": "Rust"},
+              "destination": {"value": "Python"}
+          }
+      }
+  )
+  ```
+
+For more information about graph querying, please refer to [Nuclia's graph
+doc](https://docs.nuclia.dev/docs/rag/advanced/graph) or the [API reference](https://docs.nuclia.dev/docs/api#tag/Search/operation/graph_search_knowledgebox_kb__kbid__graph_post)

{nuclia-4.8.9 → nuclia-4.9.0}/docs/11-activity-log.md

@@ -35,7 +35,7 @@ See the examples for more information
 
 #### Event-Specific Fields
 - `SEARCH` events: Common fields + `question`, `resources_count`, `filter`, `learning_id`
-- `CHAT` events: Common fields + Search fields + `rephrased_question`, `answer`, `retrieved_context`, `chat_history`, `feedback_good`, `feedback_comment`, `model`, `rag_strategies_names`, `rag_strategies`, `status`, `time_to_first_char`
+- `ASK` events: Common fields + Search fields + `rephrased_question`, `answer`, `retrieved_context`, `chat_history`, `feedback_good`, `feedback_comment`, `model`, `rag_strategies_names`, `rag_strategies`, `status`, `time_to_first_char`
 
 
 ### Query Examples
@@ -43,7 +43,7 @@ See the examples for more information
 #### CLI Example
 
 ```bash
-nuclia kb logs query --type=CHAT --query='{
+nuclia kb logs query --type=ASK --query='{
   "year_month": "2024-10",
   "show": ["id", "date", "question", "answer", "feedback_good"],
   "filters": {
@@ -58,11 +58,10 @@ nuclia kb logs query --type=CHAT --query='{
 
 ```python
 from nuclia import sdk
-from nuclia.lib.kb import LogType
-from nuclia_models.events.activity_logs import ActivityLogsChatQuery, Pagination
+from nuclia_models.events.activity_logs import ActivityLogsAskQuery, EventType, Pagination
 
 kb = sdk.NucliaKB()
-query = ActivityLogsChatQuery(
+query = ActivityLogsAskQuery(
     year_month="2024-10",
     show=["id", "date", "question", "answer"],
     filters={
@@ -71,7 +70,7 @@ query = ActivityLogsChatQuery(
     },
     pagination=Pagination(limit=10)
 )
-kb.logs.query(type=LogType.CHAT, query=query)
+kb.logs.query(type=EventType.ASK, query=query)
 ```
 ### Special Field: `audit_metadata`
 The `audit_metadata` field is a customizable dictionary. Use the `key` operator to target specific keys within the dictionary.
@@ -103,7 +102,7 @@ The `audit_metadata` field is a customizable dictionary. Use the `key` operator
 Request download and wait until the download url is generated
 
 ```bash
->>> nuclia kb logs download --wait --type=CHAT --format=NDJSON --query='{
+>>> nuclia kb logs download --wait --type=ASK --format=NDJSON --query='{
   "year_month": "2024-10",
   "show": ["id", "date", "question", "answer", "feedback_good"],
   "filters": {
@@ -119,7 +118,7 @@ download_url=https://your-download-url
 
 Request download and ask to be notified
 ```bash
->>> nuclia kb logs download --type=CHAT --format=NDJSON --query='{
+>>> nuclia kb logs download --type=ASK --format=NDJSON --query='{
   "year_month": "2024-10",
   "show": ["id", "date", "question", "answer", "feedback_good"],
   "filters": {
@@ -136,7 +135,7 @@ download_url=null
 ```
 Request download and poll for the status
 ```bash
->>> nuclia kb logs download --type=CHAT --format=NDJSON --query='{
+>>> nuclia kb logs download --type=ASK --format=NDJSON --query='{
   "year_month": "2024-10",
   "show": ["id", "date", "question", "answer", "feedback_good"],
   "filters": {
@@ -159,11 +158,10 @@ download_url=https://your-download-url
 
 ```python
 from nuclia import sdk
-from nuclia.lib.kb import LogType
-from nuclia_models.events.activity_logs import DownloadActivityLogsChatQuery
+from nuclia_models.events.activity_logs import DownloadActivityLogsAskQuery, EventType
 
 kb = sdk.NucliaKB()
-query = DownloadActivityLogsChatQuery(
+query = DownloadActivityLogsAskQuery(
     year_month="2024-10",
     show=["id", "date", "question", "answer"],
     filters={
@@ -171,7 +169,7 @@ query = DownloadActivityLogsChatQuery(
         "feedback_good": {"eq": True}
     },
 )
-request = kb.logs.download(type=LogType.CHAT, query=query, wait=True)
+request = kb.logs.download(type=EventType.ASK, query=query, wait=True)
 return request.download_url
 ```
 
@@ -182,7 +180,7 @@ The REMi module provides tools to monitor the quality of your RAG pipeline to ge
 
 ## REMi Query
 
-Use `remi query` to retrieve a list of chat activity logs that match specified criteria for REMi scores.
+Use `remi query` to retrieve a list of ask activity logs that match specified criteria for REMi scores.
 
 ### Basic Query
 
@@ -284,7 +282,7 @@ kb.remi.query(
 
 ### REMi Get
 
-Use `remi get_event` to fetch detailed information for a specific chat activity log. This command is useful for retrieving full context and score details of an entry obtained from a previous REMi query.
+Use `remi get_event` to fetch detailed information for a specific ask activity log. This command is useful for retrieving full context and score details of an entry obtained from a previous REMi query.
 
 #### CLI Example
 ```bash

{nuclia-4.8.9 → nuclia-4.9.0}/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-4.8.9 → nuclia-4.9.0}/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-4.8.9 → nuclia-4.9.0}/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-4.8.9 → nuclia-4.9.0}/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-4.8.9 → nuclia-4.9.0}/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-4.8.9 → nuclia-4.9.0}/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")