veris-ai 0.1.0__py3-none-any.whl

veris_ai/__init__.py

@@ -0,0 +1,7 @@
+"""Veris AI Python SDK."""
+
+__version__ = "0.1.0"
+
+from .tool_mock import veris
+
+__all__ = ["veris"]

veris_ai/tool_mock.py

@@ -0,0 +1,128 @@
+import inspect
+import json
+import logging
+import os
+from collections.abc import Callable
+from functools import wraps
+from typing import Any, Union, get_type_hints
+
+import httpx
+
+logger = logging.getLogger(__name__)
+
+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
+        """Convert a value to the specified type."""
+        if target_type == 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:
+            if not isinstance(value, list):
+                raise ValueError(f"Expected list but got {type(value)}")
+            item_type = target_type.__args__[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 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()}
+            
+        # Handle Union types
+        if hasattr(target_type, "__origin__") and target_type.__origin__ == Union:
+            for possible_type in target_type.__args__:
+                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
+            
+        # For other types, try direct conversion
+        return target_type(value)
+
+    def mock(self, func: Callable) -> Callable:
+        """Decorator for mocking tool calls."""
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            # Check if we're in simulation mode
+            env_mode = os.getenv("ENV", "").lower()
+            if env_mode != "simulation":
+                # If not in simulation mode, execute the original function
+                return await func(*args, **kwargs)
+
+            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)
+            session_id = None
+            if ctx:
+                try:
+                    session_id = ctx.request_context.lifespan_context.session_id
+                except AttributeError:
+                    logger.warning("Cannot get session_id from context.")
+
+            # Prepare payload
+            payload = {
+                "session_id": session_id,
+                "tool_call": {
+                    '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)
+                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:
+                        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/METADATA

@@ -0,0 +1,139 @@
+Metadata-Version: 2.4
+Name: veris-ai
+Version: 0.1.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
+Author-email: Mehdi Jamei <mehdi@veris.ai>
+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'
+Requires-Dist: pre-commit>=3.3.3; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21.1; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1.0; extra == 'dev'
+Requires-Dist: pytest>=7.4.0; extra == 'dev'
+Requires-Dist: ruff>=0.11.4; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# Veris AI Python SDK
+
+A Python package for Veris AI tools with simulation capabilities.
+
+## Installation
+
+You can install the package using `uv`:
+
+```bash
+# Install the package
+uv add veris-ai
+
+# Install with development dependencies
+uv add "veris-ai[dev]"
+```
+
+## Environment Setup
+
+The package requires the following environment variables:
+
+```bash
+# Required: URL for the mock endpoint
+VERIS_MOCK_ENDPOINT_URL=http://your-mock-endpoint.com
+
+# Optional: Timeout in seconds (default: 30.0)
+VERIS_MOCK_TIMEOUT=30.0
+
+# Optional: Set to "simulation" to enable mock mode
+ENV=simulation
+```
+
+## Python Version
+
+This project requires Python 3.11 or higher. We use [pyenv](https://github.com/pyenv/pyenv) for Python version management.
+
+To set up the correct Python version:
+
+```bash
+# Install Python 3.11.0 using pyenv
+pyenv install 3.11.0
+
+# Set the local Python version for this project
+pyenv local 3.11.0
+```
+
+## Usage
+
+```python
+from veris_ai import veris
+
+@veris.mock()
+async def your_function(param1: str, param2: int) -> dict:
+    """
+    Your function documentation here.
+    
+    Args:
+        param1: Description of param1
+        param2: Description of param2
+        
+    Returns:
+        A dictionary containing the results
+    """
+    # Your implementation here
+    return {"result": "actual implementation"}
+```
+
+When `ENV=simulation` is set, the decorator will:
+1. Capture the function signature, type hints, and docstring
+2. Send this information to the mock endpoint
+3. Convert the mock response to the expected return type
+4. Return the mock result
+
+When not in simulation mode, the original function will be executed normally.
+
+## Development
+
+This project uses `pyproject.toml` for dependency management and `uv` for package installation.
+
+### Development Dependencies
+
+To install the package with development dependencies:
+
+```bash
+uv add "veris-ai[dev]"
+```
+
+This will install the following development tools:
+- **Ruff**: Fast Python linter
+- **pytest**: Testing framework
+- **pytest-asyncio**: Async support for pytest
+- **pytest-cov**: Coverage reporting for pytest
+- **black**: Code formatter
+- **mypy**: Static type checker
+- **pre-commit**: Git hooks for code quality
+
+### Code Quality
+
+This project uses [Ruff](https://github.com/charliermarsh/ruff) for linting and code quality checks. Ruff is a fast Python linter written in Rust.
+
+To run Ruff:
+
+```bash
+ruff check .
+```
+
+To automatically fix issues:
+
+```bash
+ruff check --fix .
+```
+
+The Ruff configuration is defined in `pyproject.toml` under the `[tool.ruff]` section.
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

veris_ai-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+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,,

veris_ai-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

veris_ai-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Veris AI
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.