hivetrace 1.3.8__py3-none-any.whl → 1.3.9__py3-none-any.whl

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/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/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/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.dist-info → hivetrace-1.3.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hivetrace
-Version: 1.3.8
+Version: 1.3.9
 Summary: Hivetrace SDK for monitoring LLM applications
 Home-page: http://hivetrace.ai
 Author: Raft
@@ -11,6 +11,7 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: httpx >=0.28.1
+Requires-Dist: pydantic >=2.11.7
 Requires-Dist: python-dotenv >=1.0.1
 Provides-Extra: all
 Requires-Dist: crewai >=0.95.0 ; extra == 'all'
@@ -20,9 +21,11 @@ Requires-Dist: langchain-openai ==0.2.5 ; extra == 'all'
 Requires-Dist: langchain ==0.3.19 ; extra == 'all'
 Requires-Dist: langchain-experimental ==0.3.4 ; extra == 'all'
 Requires-Dist: openai-agents >=0.1.0 ; extra == 'all'
+Requires-Dist: pydantic >=2.11.7 ; extra == 'all'
 Requires-Dist: python-dotenv >=1.0.1 ; extra == 'all'
 Provides-Extra: base
 Requires-Dist: httpx >=0.28.1 ; extra == 'base'
+Requires-Dist: pydantic >=2.11.7 ; extra == 'base'
 Requires-Dist: python-dotenv >=1.0.1 ; extra == 'base'
 Provides-Extra: crewai
 Requires-Dist: crewai >=0.95.0 ; extra == 'crewai'
@@ -78,6 +81,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)
@@ -87,6 +100,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,
+)
 ```
 
 ---
@@ -107,6 +130,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)
@@ -116,6 +149,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,
+)
 ```
 
 ---
@@ -148,10 +191,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.
@@ -159,11 +212,15 @@ 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.)
+* `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,
@@ -181,10 +238,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.
@@ -192,11 +259,17 @@ 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,

{hivetrace-1.3.8.dist-info → hivetrace-1.3.9.dist-info}/RECORD

@@ -21,9 +21,9 @@ hivetrace/adapters/openai_agents/tracing.py,sha256=aOmJV2PT77x0bKZHXy46GRsExFmGw
 hivetrace/adapters/utils/__init__.py,sha256=AkdJzecQlhT3hHFOIO5zWbAIEXvbgH_5vmzlPViedt0,142
 hivetrace/adapters/utils/logging.py,sha256=UxCMFvlpP6vJfzRwMYhhJIi7RTWdgVK2sWtCeEB67_w,1126
 hivetrace/client/__init__.py,sha256=Daz_KxOMzGSBKUpv48tTooGZrmzk0wzDq8QUTHuBZBU,313
-hivetrace/client/async_client.py,sha256=UgGIaa5QJK2BV7eVvPCe9eUf4Q9Dwlkqp09vd5XUvVM,3790
-hivetrace/client/base.py,sha256=tOqdM7aB7wPUFlI0JzAiQPEaSnt3rHWcEC-wGx-B5u0,5995
-hivetrace/client/sync_client.py,sha256=5wixTkwsQezeSupeTZUuCBaNg5FpTxMS64jS9H5NcW4,3579
+hivetrace/client/async_client.py,sha256=9uvimxP-Q7vY-bbZW5RiTEYQd8_uP3wyqR3s92kPGUs,6518
+hivetrace/client/base.py,sha256=JSVL60S7jaGNCovkTP9MPZGzSPPe7FcgMJne0QKCs1U,7960
+hivetrace/client/sync_client.py,sha256=J3nKDVfVtfYL3NjlMtj5l7EpZKXupJnI1bpHy-tWgyE,6347
 hivetrace/errors/__init__.py,sha256=3Sqr2Pz4NwE_u8CgTzVbzoNj-B52cwzdsHINnpWO3Yg,1006
 hivetrace/errors/api.py,sha256=ThIoH8akPWuSqF6bqnex5f25p8ZBnRFsz2IrcivAHgc,1029
 hivetrace/errors/base.py,sha256=_2o8TbvlPzJEzKGl4T3u1XbTAJgP5fO19T6XQSfH3pI,686
@@ -34,12 +34,12 @@ hivetrace/handlers/error_handler.py,sha256=aWLL--HKBm4h8AO4oQSpsFObq2TjHXntYXBDX
 hivetrace/handlers/response_builder.py,sha256=J3qDWxvi_jxiZ7AgmYscHfukbz4rmnSwBF2UaWkqzb4,2558
 hivetrace/models/__init__.py,sha256=qQvtDkI0Awlch6c_kedH5Jq8aWn3XhDXbxbch5nf-RI,1020
 hivetrace/models/requests.py,sha256=dMKUMOqerDmZj4CcQ3AF-N4HSl_csv9lafA2bpggNVc,2640
-hivetrace/models/responses.py,sha256=zTcbCZ8GbIjhswb_yR7O3PmRuZaxGWOVtqSRfCy2jNA,3059
+hivetrace/models/responses.py,sha256=U06Gc4ZU2WmaPxjy1Dg5BtTDad6MOBURiDPTeLyDtcE,3209
 hivetrace/utils/__init__.py,sha256=BNYbeSuUbrZL7RradjE_OFAxam3L6eexbL2IMfjImv0,747
 hivetrace/utils/error_helpers.py,sha256=egVQpENputLR8exNpV1cui2LSHqbf8pI6SHRbLdxOX8,2661
 hivetrace/utils/uuid_generator.py,sha256=W4i2tUSyClNKNgm4O-bk_Qkkmw3cWIuf29DjwXftx0c,344
-hivetrace-1.3.8.dist-info/LICENSE,sha256=8d3g3prbWPDLQ5AV0dtyWfYTj5QPl8MJ_wlr2l8pjEU,11333
-hivetrace-1.3.8.dist-info/METADATA,sha256=cr1NUfYq_-_ppIS7lDTIx7nJRdjggy_50ZoBAP5KE3s,25863
-hivetrace-1.3.8.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-hivetrace-1.3.8.dist-info/top_level.txt,sha256=F6mZCzZ5CSftMc-M0NeOYWbwyTzjybR72P4qSBMyZZM,10
-hivetrace-1.3.8.dist-info/RECORD,,
+hivetrace-1.3.9.dist-info/LICENSE,sha256=8d3g3prbWPDLQ5AV0dtyWfYTj5QPl8MJ_wlr2l8pjEU,11333
+hivetrace-1.3.9.dist-info/METADATA,sha256=WMOawn28WE9xSOozoUgigYH2UkRCMCtcreYj7r_ccuY,27727
+hivetrace-1.3.9.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+hivetrace-1.3.9.dist-info/top_level.txt,sha256=F6mZCzZ5CSftMc-M0NeOYWbwyTzjybR72P4qSBMyZZM,10
+hivetrace-1.3.9.dist-info/RECORD,,