veris-ai 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

veris_ai/tool_mock.py

@@ -3,62 +3,81 @@ import json
 import logging
 import os
 from collections.abc import Callable
+from contextlib import suppress
 from functools import wraps
-from typing import Any, Union, get_type_hints
+from typing import Any, TypeVar, Union, get_args, get_origin, get_type_hints
 
 import httpx
 
 logger = logging.getLogger(__name__)
 
+T = TypeVar("T")  # Generic type for return value
+
+
 class ToolMock:
     """Class for mocking tool calls."""
-    def __init__(self):
-        self.endpoint = os.getenv("VERIS_MOCK_ENDPOINT_URL")
-        if not self.endpoint:
-            raise ValueError("VERIS_MOCK_ENDPOINT_URL environment variable is not set")
-        # Default timeout of 30 seconds
-        self.timeout = float(os.getenv("VERIS_MOCK_TIMEOUT", "30.0"))
 
-    def _convert_to_type(self, value: Any, target_type: type) -> Any: # type: ignore
+    def __init__(self) -> None:
+        """Initialize the ToolMock class."""
+
+    def _convert_to_type(self, value: object, target_type: type) -> object:
         """Convert a value to the specified type."""
-        if target_type == Any:
+        if target_type is Any:
             return value
-            
+
         # Handle basic types
         if target_type in (str, int, float, bool):
             return target_type(value)
-            
+
         # Handle List types
-        if hasattr(target_type, "__origin__") and target_type.__origin__ == list:
+        origin = get_origin(target_type)
+        if origin is list:
             if not isinstance(value, list):
-                raise ValueError(f"Expected list but got {type(value)}")
-            item_type = target_type.__args__[0]
+                error_msg = f"Expected list but got {type(value)}"
+                raise ValueError(error_msg)
+            item_type = get_args(target_type)[0]
             return [self._convert_to_type(item, item_type) for item in value]
-            
+
         # Handle Dict types
-        if hasattr(target_type, "__origin__") and target_type.__origin__ == dict:
+        if origin is dict:
             if not isinstance(value, dict):
-                raise ValueError(f"Expected dict but got {type(value)}")
-            key_type, value_type = target_type.__args__
-            return {self._convert_to_type(k, key_type): self._convert_to_type(v, value_type) 
-                   for k, v in value.items()}
-            
+                error_msg = f"Expected dict but got {type(value)}"
+                raise ValueError(error_msg)
+            key_type, value_type = get_args(target_type)
+            return {
+                self._convert_to_type(k, key_type): self._convert_to_type(v, value_type)
+                for k, v in value.items()
+            }
+
         # Handle Union types
-        if hasattr(target_type, "__origin__") and target_type.__origin__ == Union:
-            for possible_type in target_type.__args__:
+        if origin is Union:
+            error_msg = (
+                f"Could not convert {value} to any of the union types {get_args(target_type)}"
+            )
+            for possible_type in get_args(target_type):
                 try:
                     return self._convert_to_type(value, possible_type)
                 except (ValueError, TypeError):
                     continue
-            raise ValueError(f"Could not convert {value} to any of the union types {target_type.__args__}") # noqa
-            
+            raise ValueError(error_msg)
+
         # For other types, try direct conversion
         return target_type(value)
 
     def mock(self, func: Callable) -> Callable:
         """Decorator for mocking tool calls."""
+        endpoint = os.getenv("VERIS_MOCK_ENDPOINT_URL")
+        if not endpoint:
+            error_msg = "VERIS_MOCK_ENDPOINT_URL environment variable is not set"
+            raise ValueError(error_msg)
+        # Default timeout of 30 seconds
+        timeout = float(os.getenv("VERIS_MOCK_TIMEOUT", "30.0"))
+
         @wraps(func)
-        async def wrapper(*args, **kwargs):
+        async def wrapper(
+            *args: tuple[object, ...],
+            **kwargs: dict[str, object],
+        ) -> object:
             # Check if we're in simulation mode
             env_mode = os.getenv("ENV", "").lower()
             if env_mode != "simulation":
@@ -68,25 +87,16 @@ class ToolMock:
             logger.info(f"Simulating function: {func.__name__}")
             sig = inspect.signature(func)
             type_hints = get_type_hints(func)
-            
+
             # Extract return type object (not just the name)
             return_type_obj = type_hints.pop("return", Any)
-            
+
             # Create parameter info
             params_info = {}
             bound_args = sig.bind(*args, **kwargs)
             bound_args.apply_defaults()
-            
-            for param_name, param_value in bound_args.arguments.items():
-                params_info[param_name] = {
-                    "value": param_value,
-                    "type": type_hints.get(param_name, Any).__name__,
-                }
-
-            # Get function docstring
-            docstring = inspect.getdoc(func) or ""
 
-            ctx = bound_args.arguments.pop('ctx', None)
+            ctx = bound_args.arguments.pop("ctx", None)
             session_id = None
             if ctx:
                 try:
@@ -94,35 +104,41 @@ class ToolMock:
                 except AttributeError:
                     logger.warning("Cannot get session_id from context.")
 
+            for param_name, param_value in bound_args.arguments.items():
+                params_info[param_name] = {
+                    "value": param_value,
+                    "type": type_hints.get(param_name, Any).__name__,
+                }
+
+            # Get function docstring
+            docstring = inspect.getdoc(func) or ""
             # Prepare payload
             payload = {
                 "session_id": session_id,
                 "tool_call": {
-                    'function_name': func.__name__,
-                    'parameters': params_info,
-                    'return_type': return_type_obj.__name__,
-                    'docstring': docstring,
-                }
+                    "function_name": func.__name__,
+                    "parameters": params_info,
+                    "return_type": return_type_obj.__name__,
+                    "docstring": docstring,
+                },
             }
 
             # Send request to endpoint with timeout
-            async with httpx.AsyncClient(timeout=self.timeout) as client:
-                response = await client.post(self.endpoint, json=payload)
+            async with httpx.AsyncClient(timeout=timeout) as client:
+                response = await client.post(endpoint, json=payload)
                 response.raise_for_status()
                 mock_result = response.json()["result"]
                 logger.info(f"Mock response: {mock_result}")
-                
+
                 # Parse the mock result if it's a string
                 if isinstance(mock_result, str):
-                    try:
+                    with suppress(json.JSONDecodeError):
                         mock_result = json.loads(mock_result)
-                    except json.JSONDecodeError:
-                        # If it's not valid JSON, treat it as a raw string
-                        pass
-                
+
                 # Convert the mock result to the expected return type
                 return self._convert_to_type(mock_result, return_type_obj)
 
         return wrapper
 
+
 veris = ToolMock()

{veris_ai-0.1.0.dist-info → veris_ai-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 0.1.0
+Version: 0.2.0
 Summary: A Python package for Veris AI tools
 Project-URL: Homepage, https://github.com/veris-ai/veris-python-sdk
 Project-URL: Bug Tracker, https://github.com/veris-ai/veris-python-sdk/issues
@@ -9,8 +9,6 @@ License-Expression: MIT
 License-File: LICENSE
 Requires-Python: >=3.11
 Requires-Dist: httpx>=0.24.0
-Requires-Dist: python-dotenv>=1.0.0
-Requires-Dist: typing-extensions>=4.0.0
 Provides-Extra: dev
 Requires-Dist: black>=23.7.0; extra == 'dev'
 Requires-Dist: mypy>=1.5.1; extra == 'dev'

veris_ai-0.2.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+veris_ai/__init__.py,sha256=7mEfVmqxQCn5FZN_1ua3iykCB1NUeJ4RfVXaAzdp4Hw,101
+veris_ai/tool_mock.py,sha256=7qFM35lz10k5i2X-XQVYcuFNF0bczEnHTr0vtineQrM,5172
+veris_ai-0.2.0.dist-info/METADATA,sha256=rD-TyXPbDZfJcQCtNuUUA4KGYD6PAFXQIhB5SL1lyNg,3419
+veris_ai-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+veris_ai-0.2.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
+veris_ai-0.2.0.dist-info/RECORD,,

veris_ai-0.1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-veris_ai/__init__.py,sha256=7mEfVmqxQCn5FZN_1ua3iykCB1NUeJ4RfVXaAzdp4Hw,101
-veris_ai/tool_mock.py,sha256=NTaBr8xkAp3tiB3naWCcm27KWO5JSzcr9BcurfhlCfs,5125
-veris_ai-0.1.0.dist-info/METADATA,sha256=DzCmToKvOycKfE3ilpcEEl7mlz0a0w5yITXemoc4DWU,3495
-veris_ai-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-veris_ai-0.1.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
-veris_ai-0.1.0.dist-info/RECORD,,