hivetrace 1.3.8__tar.gz → 1.3.10__tar.gz

{hivetrace-1.3.8 → hivetrace-1.3.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hivetrace
-Version: 1.3.8
+Version: 1.3.10
 Summary: Hivetrace SDK for monitoring LLM applications
 Home-page: http://hivetrace.ai
 Author: Raft
@@ -60,6 +60,16 @@ response = client.input(
     application_id="your-application-id",  # Obtained after registering the application in the UI
     message="User prompt here",
 )
+
+# Optionally attach files (filename, bytes, mime_type)
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = client.input(
+    application_id="your-application-id",
+    message="User prompt with files",
+    files=files,
+)
 ```
 
 ### Send an LLM response (output)
@@ -69,6 +79,16 @@ response = client.output(
     application_id="your-application-id",
     message="LLM response here",
 )
+
+# With files
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = client.output(
+    application_id="your-application-id",
+    message="LLM response with files",
+    files=files,
+)
 ```
 
 ---
@@ -89,6 +109,16 @@ response = await client.input(
     application_id="your-application-id",
     message="User prompt here",
 )
+
+# With files (filename, bytes, mime_type)
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = await client.input(
+    application_id="your-application-id",
+    message="User prompt with files",
+    files=files,
+)
 ```
 
 ### Send an LLM response (output)
@@ -98,6 +128,16 @@ response = await client.output(
     application_id="your-application-id",
     message="LLM response here",
 )
+
+# With files
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = await client.output(
+    application_id="your-application-id",
+    message="LLM response with files",
+    files=files,
+)
 ```
 
 ---
@@ -115,13 +155,13 @@ response = client.input(
             "agent-1-id": {"name": "Agent 1", "description": "Agent description"},
             "agent-2-id": {"name": "Agent 2"},
             "agent-3-id": {}
-        }
+        },
+        # If you want to send only to censor and avoid DB persistence on backend
+        "censor_only": True,
     }
 )
 ```
 
-> **Note:** `session_id`, `user_id`, and all agent IDs must be valid UUIDs.
-
 ---
 
 ## API
@@ -130,10 +170,20 @@ response = client.input(
 
 ```python
 # Sync
-def input(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+def input(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 
 # Async
-async def input(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+async def input(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 ```
 
 Sends a **user prompt** to Hivetrace.
@@ -141,18 +191,22 @@ Sends a **user prompt** to Hivetrace.
 * `application_id` — Application identifier (must be a valid UUID, created in the UI)
 * `message` — The user prompt
 * `additional_parameters` — Optional dictionary with extra context (session, user, agents, etc.)
+  - Supported special flags: `censor_only: bool` — when `True`, backend should not persist the message in DB and only pass it to the censor
+* `files` — Optional list of tuples `(filename: str, content: bytes, mime_type: str)`; files are attached to the created analysis record
+
+Response contains a `blocked` flag that indicates role restrictions.
 
 **Response example:**
 
 ```json
 {
+  "blocked": false,
   "status": "processed",
   "monitoring_result": {
     "is_toxic": false,
     "type_of_violation": "benign",
     "token_count": 9,
-    "token_usage_warning": false,
-    "token_usage_unbounded": false
+    "token_usage_severity": None
   }
 }
 ```
@@ -163,10 +217,20 @@ Sends a **user prompt** to Hivetrace.
 
 ```python
 # Sync
-def output(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+def output(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 
 # Async
-async def output(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+async def output(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 ```
 
 Sends an **LLM response** to Hivetrace.
@@ -174,18 +238,23 @@ Sends an **LLM response** to Hivetrace.
 * `application_id` — Application identifier (must be a valid UUID, created in the UI)
 * `message` — The LLM response
 * `additional_parameters` — Optional dictionary with extra context (session, user, agents, etc.)
+* `files` — Optional list of tuples `(filename: str, content: bytes, mime_type: str)`
+
+> Files are uploaded after the main request completes and an analysis ID is available.
+
+Response contains a `blocked` flag that indicates role restrictions.
 
 **Response example:**
 
 ```json
 {
+  "blocked": false,
   "status": "processed",
   "monitoring_result": {
     "is_toxic": false,
     "type_of_violation": "safe",
     "token_count": 21,
-    "token_usage_warning": false,
-    "token_usage_unbounded": false
+    "token_usage_severity": None
   }
 }
 ```

{hivetrace-1.3.8 → hivetrace-1.3.10}/README.md

@@ -42,6 +42,16 @@ response = client.input(
     application_id="your-application-id",  # Obtained after registering the application in the UI
     message="User prompt here",
 )
+
+# Optionally attach files (filename, bytes, mime_type)
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = client.input(
+    application_id="your-application-id",
+    message="User prompt with files",
+    files=files,
+)
 ```
 
 ### Send an LLM response (output)
@@ -51,6 +61,16 @@ response = client.output(
     application_id="your-application-id",
     message="LLM response here",
 )
+
+# With files
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = client.output(
+    application_id="your-application-id",
+    message="LLM response with files",
+    files=files,
+)
 ```
 
 ---
@@ -71,6 +91,16 @@ response = await client.input(
     application_id="your-application-id",
     message="User prompt here",
 )
+
+# With files (filename, bytes, mime_type)
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = await client.input(
+    application_id="your-application-id",
+    message="User prompt with files",
+    files=files,
+)
 ```
 
 ### Send an LLM response (output)
@@ -80,6 +110,16 @@ response = await client.output(
     application_id="your-application-id",
     message="LLM response here",
 )
+
+# With files
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = await client.output(
+    application_id="your-application-id",
+    message="LLM response with files",
+    files=files,
+)
 ```
 
 ---
@@ -97,13 +137,13 @@ response = client.input(
             "agent-1-id": {"name": "Agent 1", "description": "Agent description"},
             "agent-2-id": {"name": "Agent 2"},
             "agent-3-id": {}
-        }
+        },
+        # If you want to send only to censor and avoid DB persistence on backend
+        "censor_only": True,
     }
 )
 ```
 
-> **Note:** `session_id`, `user_id`, and all agent IDs must be valid UUIDs.
-
 ---
 
 ## API
@@ -112,10 +152,20 @@ response = client.input(
 
 ```python
 # Sync
-def input(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+def input(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 
 # Async
-async def input(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+async def input(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 ```
 
 Sends a **user prompt** to Hivetrace.
@@ -123,18 +173,22 @@ Sends a **user prompt** to Hivetrace.
 * `application_id` — Application identifier (must be a valid UUID, created in the UI)
 * `message` — The user prompt
 * `additional_parameters` — Optional dictionary with extra context (session, user, agents, etc.)
+  - Supported special flags: `censor_only: bool` — when `True`, backend should not persist the message in DB and only pass it to the censor
+* `files` — Optional list of tuples `(filename: str, content: bytes, mime_type: str)`; files are attached to the created analysis record
+
+Response contains a `blocked` flag that indicates role restrictions.
 
 **Response example:**
 
 ```json
 {
+  "blocked": false,
   "status": "processed",
   "monitoring_result": {
     "is_toxic": false,
     "type_of_violation": "benign",
     "token_count": 9,
-    "token_usage_warning": false,
-    "token_usage_unbounded": false
+    "token_usage_severity": None
   }
 }
 ```
@@ -145,10 +199,20 @@ Sends a **user prompt** to Hivetrace.
 
 ```python
 # Sync
-def output(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+def output(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 
 # Async
-async def output(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+async def output(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 ```
 
 Sends an **LLM response** to Hivetrace.
@@ -156,18 +220,23 @@ Sends an **LLM response** to Hivetrace.
 * `application_id` — Application identifier (must be a valid UUID, created in the UI)
 * `message` — The LLM response
 * `additional_parameters` — Optional dictionary with extra context (session, user, agents, etc.)
+* `files` — Optional list of tuples `(filename: str, content: bytes, mime_type: str)`
+
+> Files are uploaded after the main request completes and an analysis ID is available.
+
+Response contains a `blocked` flag that indicates role restrictions.
 
 **Response example:**
 
 ```json
 {
+  "blocked": false,
   "status": "processed",
   "monitoring_result": {
     "is_toxic": false,
     "type_of_violation": "safe",
     "token_count": 21,
-    "token_usage_warning": false,
-    "token_usage_unbounded": false
+    "token_usage_severity": None
   }
 }
 ```

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace/client/async_client.py

@@ -1,5 +1,5 @@
 import warnings
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 import httpx
 
@@ -22,7 +22,6 @@ class AsyncHivetraceSDK(BaseHivetraceSDK):
     def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
         super().__init__(config)
         self.session = httpx.AsyncClient()
-        # SDK асинхронный, задаем флаг для адаптеров
         self.async_mode = True
 
     async def __aenter__(self):
@@ -55,27 +54,90 @@ class AsyncHivetraceSDK(BaseHivetraceSDK):
         except Exception as e:
             return ErrorHandler.handle_unexpected_error(e)
 
+    async def _send_files(
+        self, endpoint: str, files: List[Tuple[str, bytes, str]]
+    ) -> HivetraceResponse:
+        request_args = self._build_files_request_args(endpoint, files)
+        try:
+            response = await self.session.post(**request_args)
+            response.raise_for_status()
+            api_data = response.json()
+            return ResponseBuilder.build_response_from_api(api_data)
+        except httpx.HTTPStatusError as e:
+            return ErrorHandler.handle_http_error(e)
+        except httpx.ConnectError as e:
+            return ErrorHandler.handle_connection_error(e)
+        except httpx.TimeoutException as e:
+            return ErrorHandler.handle_timeout_error(e)
+        except httpx.RequestError as e:
+            return ErrorHandler.handle_request_error(e)
+        except ValueError as e:
+            return ErrorHandler.handle_json_decode_error(e)
+        except Exception as e:
+            return ErrorHandler.handle_unexpected_error(e)
+
+    async def _get_blocking_status(self, endpoint: str) -> Optional[bool]:
+        url = f"{self.hivetrace_url}/{endpoint.lstrip('/')}"
+        headers = {"Authorization": f"Bearer {self.hivetrace_access_token}"}
+        try:
+            response = await self.session.get(
+                url, headers=headers, timeout=self._DEFAULT_TIMEOUT
+            )
+            response.raise_for_status()
+            data = response.json()
+            return data.get("blocked")
+        except Exception:
+            return None
+
     async def input(
         self,
         application_id: str,
         message: str,
         additional_parameters: Optional[Dict[str, Any]] = None,
+        files: Optional[List[Tuple[str, bytes, str]]] = None,
     ) -> HivetraceResponse:
         payload = self._build_message_payload(
             application_id, message, additional_parameters
         )
-        return await self._send_request("/process_request/", payload)
+        process_response = await self._send_request("/process_request/", payload)
+        if files:
+            analysis_id = self._extract_analysis_id(process_response)
+            if analysis_id:
+                await self._send_files(
+                    f"/user_prompt_analysis/{analysis_id}/attach_files", files
+                )
+        analysis_id = self._extract_analysis_id(process_response)
+        if analysis_id:
+            blocked = await self._get_blocking_status(
+                f"/user_prompt_analysis/{analysis_id}/check_blocking"
+            )
+            self._set_blocked(process_response, blocked)
+        return process_response
 
     async def output(
         self,
         application_id: str,
         message: str,
         additional_parameters: Optional[Dict[str, Any]] = None,
+        files: Optional[List[Tuple[str, bytes, str]]] = None,
     ) -> HivetraceResponse:
         payload = self._build_message_payload(
             application_id, message, additional_parameters
         )
-        return await self._send_request("/process_response/", payload)
+        process_response = await self._send_request("/process_response/", payload)
+        if files:
+            analysis_id = self._extract_analysis_id(process_response)
+            if analysis_id:
+                await self._send_files(
+                    f"/llm_response_analysis/{analysis_id}/attach_files", files
+                )
+        analysis_id = self._extract_analysis_id(process_response)
+        if analysis_id:
+            blocked = await self._get_blocking_status(
+                f"/llm_response_analysis/{analysis_id}/check_blocking"
+            )
+            self._set_blocked(process_response, blocked)
+        return process_response
 
     async def function_call(
         self,

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace/client/base.py

@@ -1,7 +1,7 @@
 import os
 import uuid
 from abc import ABC, abstractmethod
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 import httpx
 from pydantic import ValidationError
@@ -94,6 +94,58 @@ class BaseHivetraceSDK(ABC):
             "timeout": self._DEFAULT_TIMEOUT,
         }
 
+    def _build_files_request_args(
+        self,
+        endpoint: str,
+        files: List[Tuple[str, bytes, str]],
+        files_field_name: str = "attached_files",
+    ) -> Dict[str, Any]:
+        """Builds request args for multipart file upload."""
+        url = f"{self.hivetrace_url}/{endpoint.lstrip('/')}"
+        headers = {"Authorization": f"Bearer {self.hivetrace_access_token}"}
+        return {
+            "url": url,
+            "files": self._prepare_files_param(files, files_field_name),
+            "headers": headers,
+            "timeout": self._DEFAULT_TIMEOUT,
+        }
+
+    @staticmethod
+    def _prepare_files_param(
+        files: List[Tuple[str, bytes, str]],
+        files_field_name: str = "attached_files",
+    ) -> List[Tuple[str, Tuple[str, bytes, str]]]:
+        files_param: List[Tuple[str, Tuple[str, bytes, str]]] = []
+        for file_tuple in files:
+            files_param.append((files_field_name, file_tuple))
+        return files_param
+
+    @staticmethod
+    def _extract_analysis_id(response: Any) -> Optional[str]:
+        """Extracts analysis id from API response if present."""
+        try:
+            if isinstance(response, dict):
+                monitoring_result = response.get("monitoring_result", {})
+                analysis_id = monitoring_result.get("id")
+                return str(analysis_id) if analysis_id is not None else None
+        except Exception:
+            return None
+        return None
+
+    @staticmethod
+    def _set_blocked(response: Any, blocked: Optional[bool]) -> Any:
+        """Sets 'blocked' flag on response when possible."""
+        try:
+            if isinstance(response, dict):
+                response["blocked"] = blocked
+                return response
+            if hasattr(response, "blocked"):
+                setattr(response, "blocked", blocked)
+                return response
+        except Exception:
+            return response
+        return response
+
     def _handle_http_error(self, error: httpx.HTTPStatusError) -> HivetraceResponse:
         return ErrorHandler.handle_http_error(error)
 

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace/client/sync_client.py

@@ -1,5 +1,5 @@
 import weakref
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 import httpx
 
@@ -67,27 +67,92 @@ class SyncHivetraceSDK(BaseHivetraceSDK):
         except Exception as e:
             return ErrorHandler.handle_unexpected_error(e)
 
+    def _send_files(
+        self, endpoint: str, files: List[Tuple[str, bytes, str]]
+    ) -> HivetraceResponse:
+        request_args = self._build_files_request_args(endpoint, files)
+        try:
+            response = self.session.post(**request_args)
+            response.raise_for_status()
+
+            api_data = response.json()
+            return ResponseBuilder.build_response_from_api(api_data)
+
+        except httpx.HTTPStatusError as e:
+            return ErrorHandler.handle_http_error(e)
+        except httpx.ConnectError as e:
+            return ErrorHandler.handle_connection_error(e)
+        except httpx.TimeoutException as e:
+            return ErrorHandler.handle_timeout_error(e)
+        except httpx.RequestError as e:
+            return ErrorHandler.handle_request_error(e)
+        except ValueError as e:
+            return ErrorHandler.handle_json_decode_error(e)
+        except Exception as e:
+            return ErrorHandler.handle_unexpected_error(e)
+
+    def _get_blocking_status(self, endpoint: str) -> Optional[bool]:
+        url = f"{self.hivetrace_url}/{endpoint.lstrip('/')}"
+        headers = {"Authorization": f"Bearer {self.hivetrace_access_token}"}
+        try:
+            response = self.session.get(
+                url, headers=headers, timeout=self._DEFAULT_TIMEOUT
+            )
+            response.raise_for_status()
+            data = response.json()
+            return data.get("blocked")
+        except Exception:
+            return None
+
     def input(
         self,
         application_id: str,
         message: str,
         additional_parameters: Optional[Dict[str, Any]] = None,
+        files: Optional[List[Tuple[str, bytes, str]]] = None,
     ) -> HivetraceResponse:
         payload = self._build_message_payload(
             application_id, message, additional_parameters
         )
-        return self._send_request("/process_request/", payload)
+        process_response = self._send_request("/process_request/", payload)
+        if files:
+            analysis_id = self._extract_analysis_id(process_response)
+            if analysis_id:
+                self._send_files(
+                    f"/user_prompt_analysis/{analysis_id}/attach_files", files
+                )
+        analysis_id = self._extract_analysis_id(process_response)
+        if analysis_id:
+            blocked = self._get_blocking_status(
+                f"/user_prompt_analysis/{analysis_id}/check_blocking"
+            )
+            self._set_blocked(process_response, blocked)
+        return process_response
 
     def output(
         self,
         application_id: str,
         message: str,
         additional_parameters: Optional[Dict[str, Any]] = None,
+        files: Optional[List[Tuple[str, bytes, str]]] = None,
     ) -> HivetraceResponse:
         payload = self._build_message_payload(
             application_id, message, additional_parameters
         )
-        return self._send_request("/process_response/", payload)
+        process_response = self._send_request("/process_response/", payload)
+        if files:
+            analysis_id = self._extract_analysis_id(process_response)
+            if analysis_id:
+                self._send_files(
+                    f"/llm_response_analysis/{analysis_id}/attach_files", files
+                )
+        analysis_id = self._extract_analysis_id(process_response)
+        if analysis_id:
+            blocked = self._get_blocking_status(
+                f"/llm_response_analysis/{analysis_id}/check_blocking"
+            )
+            self._set_blocked(process_response, blocked)
+        return process_response
 
     def function_call(
         self,

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace/models/requests.py

@@ -15,10 +15,10 @@ class BaseRequest(BaseModel):
 class MessageRequest(BaseRequest):
     """Base model for requests with message and additional parameters."""
 
-    application_id: str = Field(..., description="ID приложения (UUID)")
-    message: str = Field(..., min_length=1, description="Текст сообщения")
+    application_id: str = Field(..., description="Application ID (UUID)")
+    message: str = Field(..., min_length=1, description="Message text")
     additional_parameters: Optional[Dict[str, Any]] = Field(
-        None, description="Дополнительные параметры"
+        None, description="Additional parameters"
     )
 
     @validator("application_id")

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace/models/responses.py

@@ -24,6 +24,10 @@ class ProcessResponse(SuccessResponse):
 
     message_id: Optional[str] = Field(None, description="ID of processed message")
     trace_id: Optional[str] = Field(None, description="Trace ID")
+    blocked: Optional[bool] = Field(
+        None,
+        description="Role restriction flag. True if message/response is blocked by policy.",
+    )
 
 
 class ErrorResponse(BaseResponse):

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hivetrace
-Version: 1.3.8
+Version: 1.3.10
 Summary: Hivetrace SDK for monitoring LLM applications
 Home-page: http://hivetrace.ai
 Author: Raft
@@ -60,6 +60,16 @@ response = client.input(
     application_id="your-application-id",  # Obtained after registering the application in the UI
     message="User prompt here",
 )
+
+# Optionally attach files (filename, bytes, mime_type)
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = client.input(
+    application_id="your-application-id",
+    message="User prompt with files",
+    files=files,
+)
 ```
 
 ### Send an LLM response (output)
@@ -69,6 +79,16 @@ response = client.output(
     application_id="your-application-id",
     message="LLM response here",
 )
+
+# With files
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = client.output(
+    application_id="your-application-id",
+    message="LLM response with files",
+    files=files,
+)
 ```
 
 ---
@@ -89,6 +109,16 @@ response = await client.input(
     application_id="your-application-id",
     message="User prompt here",
 )
+
+# With files (filename, bytes, mime_type)
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = await client.input(
+    application_id="your-application-id",
+    message="User prompt with files",
+    files=files,
+)
 ```
 
 ### Send an LLM response (output)
@@ -98,6 +128,16 @@ response = await client.output(
     application_id="your-application-id",
     message="LLM response here",
 )
+
+# With files
+files = [
+    ("doc1.txt", open("doc1.txt", "rb"), "text/plain"),
+]
+response_with_files = await client.output(
+    application_id="your-application-id",
+    message="LLM response with files",
+    files=files,
+)
 ```
 
 ---
@@ -115,13 +155,13 @@ response = client.input(
             "agent-1-id": {"name": "Agent 1", "description": "Agent description"},
             "agent-2-id": {"name": "Agent 2"},
             "agent-3-id": {}
-        }
+        },
+        # If you want to send only to censor and avoid DB persistence on backend
+        "censor_only": True,
     }
 )
 ```
 
-> **Note:** `session_id`, `user_id`, and all agent IDs must be valid UUIDs.
-
 ---
 
 ## API
@@ -130,10 +170,20 @@ response = client.input(
 
 ```python
 # Sync
-def input(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+def input(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 
 # Async
-async def input(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+async def input(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 ```
 
 Sends a **user prompt** to Hivetrace.
@@ -141,18 +191,22 @@ Sends a **user prompt** to Hivetrace.
 * `application_id` — Application identifier (must be a valid UUID, created in the UI)
 * `message` — The user prompt
 * `additional_parameters` — Optional dictionary with extra context (session, user, agents, etc.)
+  - Supported special flags: `censor_only: bool` — when `True`, backend should not persist the message in DB and only pass it to the censor
+* `files` — Optional list of tuples `(filename: str, content: bytes, mime_type: str)`; files are attached to the created analysis record
+
+Response contains a `blocked` flag that indicates role restrictions.
 
 **Response example:**
 
 ```json
 {
+  "blocked": false,
   "status": "processed",
   "monitoring_result": {
     "is_toxic": false,
     "type_of_violation": "benign",
     "token_count": 9,
-    "token_usage_warning": false,
-    "token_usage_unbounded": false
+    "token_usage_severity": None
   }
 }
 ```
@@ -163,10 +217,20 @@ Sends a **user prompt** to Hivetrace.
 
 ```python
 # Sync
-def output(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+def output(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 
 # Async
-async def output(application_id: str, message: str, additional_parameters: dict | None = None) -> dict: ...
+async def output(
+    application_id: str,
+    message: str,
+    additional_parameters: dict | None = None,
+    files: list[tuple[str, bytes, str]] | None = None,
+) -> dict: ...
 ```
 
 Sends an **LLM response** to Hivetrace.
@@ -174,18 +238,23 @@ Sends an **LLM response** to Hivetrace.
 * `application_id` — Application identifier (must be a valid UUID, created in the UI)
 * `message` — The LLM response
 * `additional_parameters` — Optional dictionary with extra context (session, user, agents, etc.)
+* `files` — Optional list of tuples `(filename: str, content: bytes, mime_type: str)`
+
+> Files are uploaded after the main request completes and an analysis ID is available.
+
+Response contains a `blocked` flag that indicates role restrictions.
 
 **Response example:**
 
 ```json
 {
+  "blocked": false,
   "status": "processed",
   "monitoring_result": {
     "is_toxic": false,
     "type_of_violation": "safe",
     "token_count": 21,
-    "token_usage_warning": false,
-    "token_usage_unbounded": false
+    "token_usage_severity": None
   }
 }
 ```

{hivetrace-1.3.8 → hivetrace-1.3.10}/hivetrace.egg-info/requires.txt

@@ -1,4 +1,5 @@
 httpx>=0.28.1
+pydantic>=2.11.7
 python-dotenv>=1.0.1
 
 [all]
@@ -9,10 +10,12 @@ langchain-openai==0.2.5
 langchain==0.3.19
 langchain_experimental==0.3.4
 openai-agents>=0.1.0
+pydantic>=2.11.7
 python-dotenv>=1.0.1
 
 [base]
 httpx>=0.28.1
+pydantic>=2.11.7
 python-dotenv>=1.0.1
 
 [crewai]

{hivetrace-1.3.8 → hivetrace-1.3.10}/setup.py

@@ -9,6 +9,7 @@ def readme():
 base_requires = [
     "httpx>=0.28.1",
     "python-dotenv>=1.0.1",
+    "pydantic>=2.11.7",
 ]
 
 langchain_requires = [
@@ -28,7 +29,7 @@ openai_agents_requires = [
 
 setup(
     name="hivetrace",
-    version="1.3.8",
+    version="1.3.10",
     author="Raft",
     author_email="sales@raftds.com",
     description="Hivetrace SDK for monitoring LLM applications",