nuclia 4.8.8__tar.gz → 4.8.10__tar.gz

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

@@ -1,6 +1,18 @@
 # Changelog
 
 
+## 4.8.10 (2025-05-28)
+
+
+- Activity logs: Add event type `ASK` and deprecate `CHAT`
+
+
+## 4.8.9 (2025-05-15)
+
+
+- Better http error handling
+
+
 ## 4.8.8 (2025-05-14)
 
 

nuclia-4.8.10/Makefile

@@ -0,0 +1,12 @@
+lint:
+	ruff check nuclia
+	ruff format --check nuclia
+	mypy nuclia
+
+fmt:
+	ruff check --fix nuclia
+	ruff format nuclia
+
+test:
+	mypy nuclia
+	pytest --asyncio-mode=auto nuclia

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nuclia
-Version: 4.8.8
+Version: 4.8.10
 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.38.1
+Requires-Dist: nuclia-models>=0.41.1
 Requires-Dist: tqdm
 Requires-Dist: aiofiles
 Requires-Dist: backoff

nuclia-4.8.10/VERSION

@@ -0,0 +1 @@
+4.8.10

{nuclia-4.8.8 → nuclia-4.8.10}/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.8 → nuclia-4.8.10}/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,17 +14,17 @@ from tqdm import tqdm
 from nuclia_models.events.activity_logs import (  # type: ignore
     ActivityLogsQuery,
     ActivityLogsSearchQuery,
-    ActivityLogsChatQuery,
+    ActivityLogsAskQuery,
     DownloadActivityLogsQuery,
     DownloadActivityLogsSearchQuery,
-    DownloadActivityLogsChatQuery,
+    DownloadActivityLogsAskQuery,
     EventType,
     DownloadFormat,
 )
 from nuclia_models.events.remi import RemiQuery
 from nuclia_models.worker.tasks import TaskStartKB
 from nuclia.exceptions import RateLimitError
-from nuclia.lib.utils import handle_http_errors
+from nuclia.lib.utils import handle_http_sync_errors, handle_http_async_errors
 from datetime import datetime
 from nuclia.lib.utils import build_httpx_client, build_httpx_async_client, USER_AGENT
 
@@ -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
@@ -220,7 +200,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             raise Exception("KB not configured")
         url = f"{self.url}{NOTIFICATIONS}"
         response = self.stream_session.get(url, stream=True, timeout=3660)
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def ask(self, request: AskRequest, timeout: int = 1000):
@@ -233,7 +213,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             stream=True,
             timeout=timeout,
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def download_export(self, export_id: str, path: str, chunk_size: int):
@@ -276,7 +256,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         new_uri = "/".join(uri_parts[3:])
         url = DOWNLOAD_URL.format(uri=new_uri)
         response: httpx.Response = self.reader_session.get(url)
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response.content
 
     @backoff.on_exception(
@@ -320,7 +300,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             headers["x-extract-strategy"] = extract_strategy
 
         response: httpx.Response = self.writer_session.post(url, headers=headers)
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response.headers.get("Location")
 
     def patch_tus_upload(self, upload_url: str, data: bytes, offset: int) -> int:
@@ -338,7 +318,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.writer_session.patch(
             url, headers=headers, content=data
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return int(response.headers.get("Upload-Offset"))
 
     def summarize(self, request: SummarizeRequest, timeout: int = 1000):
@@ -349,44 +329,13 @@ class NucliaDBClient(BaseNucliaDBClient):
         response = self.reader_session.post(
             url, json=request.model_dump(), timeout=timeout
         )
-        handle_http_errors(response)
+        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_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_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")
@@ -396,7 +345,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             json=query.model_dump(mode="json", exclude_unset=True),
             stream=True,
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def logs_download(
@@ -405,7 +354,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         query: Union[
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ],
         download_format: DownloadFormat,
     ):
@@ -420,7 +369,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             json=query.model_dump(mode="json", exclude_unset=True),
             headers={"accept": format_header_value},
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def get_download_request(
@@ -431,7 +380,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             raise Exception("KB not configured")
         download_request_url = f"{self.url}{ACTIVITY_LOG_DOWNLOAD_REQUEST_URL.format(request_id=request_id)}"
         response: httpx.Response = self.reader_session.get(download_request_url)
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def remi_query(
@@ -446,7 +395,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             json=query.model_dump(mode="json", exclude_unset=True),
             timeout=10,
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def get_remi_event(
@@ -459,7 +408,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.reader_session.get(
             f"{self.url}{REMI_EVENT_URL.format(event_id=event_id)}"
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def get_remi_scores(
@@ -476,7 +425,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.reader_session.get(
             f"{self.url}{REMI_SCORES_URL}", params=params, timeout=10
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def list_tasks(self) -> httpx.Response:
@@ -484,7 +433,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             raise Exception("KB not configured")
 
         response: httpx.Response = self.reader_session.get(f"{self.url}{LIST_TASKS}")
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def start_task(self, body: TaskStartKB) -> httpx.Response:
@@ -495,7 +444,7 @@ class NucliaDBClient(BaseNucliaDBClient):
             f"{self.url}{START_TASK}",
             json=body.model_dump(mode="json", exclude_unset=True),
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def delete_task(self, task_id: str) -> httpx.Response:
@@ -505,7 +454,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.writer_session.delete(
             f"{self.url}{DELETE_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def stop_task(self, task_id: str) -> httpx.Response:
@@ -515,7 +464,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.writer_session.post(
             f"{self.url}{STOP_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def get_task(self, task_id: str) -> httpx.Response:
@@ -525,7 +474,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.reader_session.get(
             f"{self.url}{GET_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
     def restart_task(self, task_id: str) -> httpx.Response:
@@ -535,7 +484,7 @@ class NucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = self.writer_session.post(
             f"{self.url}{RESTART_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        handle_http_sync_errors(response)
         return response
 
 
@@ -583,7 +532,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         url = f"{self.url}{NOTIFICATIONS}"
         req = self.reader_session.build_request("GET", url, timeout=3660)
         response = await self.reader_session.send(req, stream=True)
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def ask(self, request: AskRequest, timeout: int = 1000):
@@ -594,7 +543,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             "POST", url, json=request.model_dump(), timeout=timeout
         )
         response = await self.reader_session.send(req, stream=True)
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def download_export(self, export_id: str, path: str, chunk_size: int):
@@ -635,7 +584,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         new_uri = "/".join(uri_parts[3:])
         url = DOWNLOAD_URL.format(uri=new_uri)
         response = await self.reader_session.get(url)
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response.content
 
     @backoff.on_exception(
@@ -679,7 +628,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             headers["x-extract-strategy"] = extract_strategy
 
         response = await self.writer_session.post(url, headers=headers)
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response.headers.get("Location")
 
     async def patch_tus_upload(self, upload_url: str, data: bytes, offset: int) -> int:
@@ -695,7 +644,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             url = httpx.URL("/".join(url.path.split("/")[3:]))
 
         response = await self.writer_session.patch(url, headers=headers, content=data)
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return int(response.headers.get("Upload-Offset"))
 
     async def summarize(self, request: SummarizeRequest, timeout: int = 1000):
@@ -706,46 +655,13 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response = await self.reader_session.post(
             url, json=request.model_dump(), timeout=timeout
         )
-        handle_http_errors(response)
+        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)
-            handle_http_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
-            )
-            handle_http_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")
@@ -754,7 +670,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             f"{self.url}{ACTIVITY_LOG_QUERY_URL.format(type=type.value)}",
             json=query.model_dump(mode="json", exclude_unset=True),
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def logs_download(
@@ -763,7 +679,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         query: Union[
             DownloadActivityLogsQuery,
             DownloadActivityLogsSearchQuery,
-            DownloadActivityLogsChatQuery,
+            DownloadActivityLogsAskQuery,
         ],
         download_format: DownloadFormat,
     ):
@@ -778,7 +694,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             json=query.model_dump(mode="json", exclude_unset=True),
             headers={"accept": format_header_value},
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def get_download_request(
@@ -789,7 +705,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             raise Exception("KB not configured")
         download_request_url = f"{self.url}{ACTIVITY_LOG_DOWNLOAD_REQUEST_URL.format(request_id=request_id)}"
         response: httpx.Response = await self.reader_session.get(download_request_url)
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def remi_query(
@@ -804,7 +720,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             json=query.model_dump(mode="json", exclude_unset=True),
             timeout=10,
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def get_remi_event(
@@ -817,7 +733,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.reader_session.get(
             f"{self.url}{REMI_EVENT_URL.format(event_id=event_id)}"
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def get_remi_scores(
@@ -834,7 +750,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.reader_session.get(
             f"{self.url}{REMI_SCORES_URL}", params=params, timeout=10
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def list_tasks(self) -> httpx.Response:
@@ -844,7 +760,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.reader_session.get(
             f"{self.url}{LIST_TASKS}"
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def start_task(self, body: TaskStartKB) -> httpx.Response:
@@ -855,7 +771,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
             f"{self.url}{START_TASK}",
             json=body.model_dump(mode="json", exclude_unset=True),
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def delete_task(self, task_id: str) -> httpx.Response:
@@ -865,7 +781,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.writer_session.delete(
             f"{self.url}{DELETE_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def stop_task(self, task_id: str) -> httpx.Response:
@@ -875,7 +791,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.writer_session.post(
             f"{self.url}{STOP_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def get_task(self, task_id: str) -> httpx.Response:
@@ -885,7 +801,7 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.reader_session.get(
             f"{self.url}{GET_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response
 
     async def restart_task(self, task_id: str) -> httpx.Response:
@@ -895,5 +811,5 @@ class AsyncNucliaDBClient(BaseNucliaDBClient):
         response: httpx.Response = await self.writer_session.post(
             f"{self.url}{RESTART_TASK.format(task_id=task_id)}",
         )
-        handle_http_errors(response)
+        await handle_http_async_errors(response)
         return response