veris-ai 1.2.0__py3-none-any.whl → 1.3.0__py3-none-any.whl

veris_ai/__init__.py

@@ -5,9 +5,9 @@ from typing import Any
 __version__ = "0.1.0"
 
 # Import lightweight modules that only use base dependencies
-from .tool_mock import veris
 from .jaeger_interface import JaegerClient
 from .models import ResponseExpectation
+from .tool_mock import veris
 
 # Lazy import for modules with heavy dependencies
 _instrument = None
@@ -34,9 +34,4 @@ def instrument(*args: Any, **kwargs: Any) -> Any:  # noqa: ANN401
     return _instrument(*args, **kwargs)
 
 
-__all__ = [
-    "veris",
-    "JaegerClient",
-    "instrument",
-    "ResponseExpectation"
-]
+__all__ = ["veris", "JaegerClient", "instrument", "ResponseExpectation"]

veris_ai/jaeger_interface/client.py

@@ -5,12 +5,12 @@ This implementation keeps dependencies minimal while providing fully-typed
 """
 
 import json
-from typing import Any, Dict, List, Optional, Self
+import types
+from typing import Any, Self
 
 import requests
 
-from .models import GetTraceResponse, SearchResponse, Span, Tag, Trace
-
+from .models import GetTraceResponse, SearchResponse, Span, Trace
 
 __all__ = ["JaegerClient"]
 
@@ -60,30 +60,27 @@ class JaegerClient:  # noqa: D101
         session.headers.update(self._headers)
         return session, True
 
-    def _span_matches_tags(self, span: Span, span_tags: Dict[str, Any]) -> bool:
+    def _span_matches_tags(self, span: Span, span_tags: dict[str, Any]) -> bool:
         """Check if a span matches any of the provided tags (OR logic)."""
         if not span.tags or not span_tags:
             return False
-        
+
         # Convert span tags to a dict for easier lookup
         span_tag_dict = {tag.key: tag.value for tag in span.tags}
-        
+
         # OR logic: return True if ANY tag matches
-        for key, value in span_tags.items():
-            if span_tag_dict.get(key) == value:
-                return True
-        
-        return False
+        return any(span_tag_dict.get(key) == value for key, value in span_tags.items())
 
     def _filter_spans(
         self,
-        traces: List[Trace],
-        span_tags: Optional[Dict[str, Any]],
-        span_operations: Optional[List[str]] = None
-    ) -> List[Trace]:
-        """
-        Filter spans within traces based on span_tags (OR logic) and/or span_operations (OR logic).
-        If both are provided, a span must match at least one tag AND at least one operation.
+        traces: list[Trace],
+        span_tags: dict[str, Any] | None,
+        span_operations: list[str] | None = None,
+    ) -> list[Trace]:
+        """Filter spans within traces based on span_tags and/or span_operations.
+
+        Uses OR logic within each filter type. If both are provided, a span must
+        match at least one tag AND at least one operation.
         """
         if not span_tags and not span_operations:
             return traces
@@ -109,7 +106,7 @@ class JaegerClient:  # noqa: D101
                     traceID=trace.traceID,
                     spans=filtered_spans,
                     process=trace.process,
-                    warnings=trace.warnings
+                    warnings=trace.warnings,
                 )
                 filtered_traces.append(filtered_trace)
 
@@ -119,16 +116,16 @@ class JaegerClient:  # noqa: D101
     # Public API
     # ---------------------------------------------------------------------
 
-    def search(
+    def search(  # noqa: PLR0913
         self,
-        service: Optional[str] = None,
+        service: str | None = None,
         *,
-        limit: Optional[int] = None,
-        tags: Optional[Dict[str, Any]] = None,
-        operation: Optional[str] = None,
-        span_tags: Optional[Dict[str, Any]] = None,
-        span_operations: Optional[List[str]] = None,
-        **kwargs: Any
+        limit: int | None = None,
+        tags: dict[str, Any] | None = None,
+        operation: str | None = None,
+        span_tags: dict[str, Any] | None = None,
+        span_operations: list[str] | None = None,
+        **kwargs: Any,  # noqa: ANN401
     ) -> SearchResponse:  # noqa: D401
         """Search traces using the *v1* ``/api/traces`` endpoint with optional span filtering.
 
@@ -137,9 +134,11 @@ class JaegerClient:  # noqa: D101
             limit: Maximum number of traces to return.
             tags: Dictionary of tag filters for trace-level filtering (AND-combined).
             operation: Operation name to search for.
-            span_tags: Dictionary of tag filters for span-level filtering (OR-combined AND-combined with span_operations).
+            span_tags: Dictionary of tag filters for span-level filtering.
+                      Uses OR logic. Combined with span_operations using AND.
                       Applied client-side after retrieving traces.
-            span_operations: List of operation names to search for (OR-combined AND-combined with span_tags).
+            span_operations: List of operation names to search for.
+                            Uses OR logic. Combined with span_tags using AND.
             **kwargs: Additional parameters to pass to the Jaeger API.
 
         Returns:
@@ -147,24 +146,24 @@ class JaegerClient:  # noqa: D101
             with spans filtered according to span_tags if provided.
         """
         # Build params for the Jaeger API (excluding span_tags)
-        params: Dict[str, Any] = {}
-        
+        params: dict[str, Any] = {}
+
         if service is not None:
             params["service"] = service
-        
+
         if limit is not None:
             params["limit"] = limit
-        
+
         if operation is not None:
             params["operation"] = operation
-        
+
         if tags:
             # Convert tags to JSON string as expected by Jaeger API
             params["tags"] = json.dumps(tags)
-        
+
         # Add any additional parameters
         params.update(kwargs)
-        
+
         session, should_close = self._make_session()
         try:
             url = f"{self._base_url}/api/traces"
@@ -174,18 +173,22 @@ class JaegerClient:  # noqa: D101
         finally:
             if should_close:
                 session.close()
-        
+
         # Parse the response
         search_response = SearchResponse.model_validate(data)  # type: ignore[arg-type]
-        
+
         # Apply span-level filtering if span_tags is provided
-        if span_tags or span_operations and search_response.data and isinstance(search_response.data, list):
+        if (
+            (span_tags or span_operations)
+            and search_response.data
+            and isinstance(search_response.data, list)
+        ):
             filtered_traces = self._filter_spans(search_response.data, span_tags, span_operations)
             search_response.data = filtered_traces
             # Update the total to reflect filtered results
             if search_response.total is not None:
                 search_response.total = len(filtered_traces)
-        
+
         return search_response
 
     def get_trace(self, trace_id: str) -> GetTraceResponse:  # noqa: D401
@@ -225,7 +228,7 @@ class JaegerClient:  # noqa: D101
         self,
         exc_type: type[BaseException] | None,
         exc: BaseException | None,
-        tb: Any | None,
+        tb: types.TracebackType | None,
     ) -> None:
         """Exit the context manager."""
         # Only close if we created the session

veris_ai/jaeger_interface/models.py

@@ -1,9 +1,8 @@
 from __future__ import annotations
 
-import json
 from typing import Any
 
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
 __all__ = [
     "Tag",

veris_ai/models.py

@@ -5,7 +5,7 @@ from enum import Enum
 
 class ResponseExpectation(str, Enum):
     """Expected response behavior for tool mocking."""
-    
+
     AUTO = "auto"
     REQUIRED = "required"
-    NONE = "none"
+    NONE = "none"

veris_ai/tool_mock.py

@@ -92,7 +92,7 @@ class VerisSDK:
             **params_dict,
         )
 
-    def mock(
+    def mock(  # noqa: C901, PLR0915
         self,
         mode: Literal["tool", "function"] = "tool",
         expects_response: bool | None = None,
@@ -100,15 +100,8 @@ class VerisSDK:
     ) -> Callable:
         """Decorator for mocking tool calls."""
 
-        def decorator(func: Callable) -> Callable:
+        def decorator(func: Callable) -> Callable:  # noqa: C901, PLR0915
             """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", "90.0"))
-            
             # Check if the original function is async
             is_async = inspect.iscoroutinefunction(func)
 
@@ -148,7 +141,7 @@ class VerisSDK:
                     response_expectation = ResponseExpectation.REQUIRED
                 else:
                     response_expectation = ResponseExpectation.AUTO
-                
+
                 payload = {
                     "session_id": self.session_id,
                     "response_expectation": response_expectation.value,
@@ -169,11 +162,16 @@ class VerisSDK:
                 **kwargs: dict[str, object],
             ) -> object:
                 # Check if we're in simulation mode
-                env_mode = os.getenv("ENV", "").lower()
-                if env_mode != "simulation":
+                if not self.session_id:
                     # If not in simulation mode, execute the original function
                     return await func(*args, **kwargs)
-                
+                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", "90.0"))
+
                 logger.info(f"Simulating function: {func.__name__}")
                 payload, return_type_obj = create_mock_payload(*args, **kwargs)
 
@@ -196,11 +194,16 @@ class VerisSDK:
                 **kwargs: dict[str, object],
             ) -> object:
                 # Check if we're in simulation mode
-                env_mode = os.getenv("ENV", "").lower()
-                if env_mode != "simulation":
+                if not self.session_id:
                     # If not in simulation mode, execute the original function
                     return func(*args, **kwargs)
-                
+                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", "90.0"))
+
                 logger.info(f"Simulating function: {func.__name__}")
                 payload, return_type_obj = create_mock_payload(*args, **kwargs)
 
@@ -228,23 +231,21 @@ class VerisSDK:
         def decorator(func: Callable) -> Callable:
             # Check if the original function is async
             is_async = inspect.iscoroutinefunction(func)
-            
+
             @wraps(func)
             async def async_wrapper(
                 *args: tuple[object, ...],
                 **kwargs: dict[str, object],
             ) -> object:
-                env_mode = os.getenv("ENV", "").lower()
-                if env_mode != "simulation":
+                if not self.session_id:
                     # If not in simulation mode, execute the original function
-                    return func(*args, **kwargs)
+                    return await func(*args, **kwargs)
                 logger.info(f"Simulating function: {func.__name__}")
                 return return_value
-            
+
             @wraps(func)
             def sync_wrapper(*args: tuple[object, ...], **kwargs: dict[str, object]) -> object:
-                env_mode = os.getenv("ENV", "").lower()
-                if env_mode != "simulation":
+                if not self.session_id:
                     # If not in simulation mode, execute the original function
                     return func(*args, **kwargs)
                 logger.info(f"Simulating function: {func.__name__}")

{veris_ai-1.2.0.dist-info → veris_ai-1.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.2.0
+Version: 1.3.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

veris_ai-1.3.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+veris_ai/__init__.py,sha256=Vp5yf9ZRchw0mmPwrRuNebI5cb2NGwpJGfr41l86q1U,1177
+veris_ai/braintrust_tracing.py,sha256=0i-HR6IuK3-Q5ujMjT1FojxESRZLgqEvJ44bfJfDHaw,11938
+veris_ai/models.py,sha256=6HINPxNFCakCVPcyEbUswWkXwb2K4lF0A8g8EvTMal4,213
+veris_ai/tool_mock.py,sha256=U4xjINEZbtjutb_Vtr_So1ENNrU9kDROSyNUU5RL1Jc,10610
+veris_ai/utils.py,sha256=3R2J9ko5t1UATiF1R6Hox9IPbtUz59EHsMDFg1-i7sk,9208
+veris_ai/jaeger_interface/README.md,sha256=te5z3MWLqd2ECVV-a0MImBwTKRgQuuSuDB3bwIIsHi0,4437
+veris_ai/jaeger_interface/__init__.py,sha256=d873a0zq3eUYU2Y77MtdjCwIARjAsAP7WDqGXDMWpYs,1158
+veris_ai/jaeger_interface/client.py,sha256=yJrh86wRR0Dk3Gq12DId99WogcMIVbL0QQFqVSevvlE,8772
+veris_ai/jaeger_interface/models.py,sha256=e64VV6IvOEFuzRUgvDAMQFyOZMRb56I-PUPZLBZ3rX0,1864
+veris_ai-1.3.0.dist-info/METADATA,sha256=FD5DuzpYTZKjDmsPOb897HIMdOzS4l1UNTKh_68d29Y,13832
+veris_ai-1.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+veris_ai-1.3.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
+veris_ai-1.3.0.dist-info/RECORD,,

veris_ai-1.2.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-veris_ai/__init__.py,sha256=FkpF1jAEX9y5mZjICuxv0-uc_pTgWuCzK5D5fEQ-b8s,1195
-veris_ai/braintrust_tracing.py,sha256=0i-HR6IuK3-Q5ujMjT1FojxESRZLgqEvJ44bfJfDHaw,11938
-veris_ai/models.py,sha256=wj1avahszKrErFbWVar9xvFrp_pVkQUCaJUc_gFzu3k,216
-veris_ai/tool_mock.py,sha256=UO-6tYvfreLXfOxHFKFl-GJ1nI7bLoz93uuv_yBehWc,10507
-veris_ai/utils.py,sha256=3R2J9ko5t1UATiF1R6Hox9IPbtUz59EHsMDFg1-i7sk,9208
-veris_ai/jaeger_interface/README.md,sha256=te5z3MWLqd2ECVV-a0MImBwTKRgQuuSuDB3bwIIsHi0,4437
-veris_ai/jaeger_interface/__init__.py,sha256=d873a0zq3eUYU2Y77MtdjCwIARjAsAP7WDqGXDMWpYs,1158
-veris_ai/jaeger_interface/client.py,sha256=AU9qP_pzutwantcLiKaml5JW_AHrXHH7X-ALYVkbNZc,8777
-veris_ai/jaeger_interface/models.py,sha256=ZYL2vlJbObAiAowsashuAuT6tUV9HaEmGKuxLPi_ye8,1876
-veris_ai-1.2.0.dist-info/METADATA,sha256=ZmgJqKZUnL2G0ANQxX6SiYX2UK3HqeG7FjGgCttAQPM,13832
-veris_ai-1.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-veris_ai-1.2.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
-veris_ai-1.2.0.dist-info/RECORD,,