veris-ai 1.6.0__py3-none-any.whl → 1.7.0__py3-none-any.whl

veris_ai/tool_mock.py

@@ -172,67 +172,82 @@ class VerisSDK:
                     # If not in simulation mode, execute the original function
                     return await func(*args, **kwargs)
 
-                # Handle spy mode - execute original function and log
-                if mode == "spy":
-                    logger.info(f"Spying on function: {func.__name__}")
-
-                    # Log the tool call
-                    sig = inspect.signature(func)
-                    bound_args = sig.bind(*args, **kwargs)
-                    bound_args.apply_defaults()
-                    _ = bound_args.arguments.pop("ctx", None)
-                    _ = bound_args.arguments.pop("self", None)
-                    _ = bound_args.arguments.pop("cls", None)
-
-                    await log_tool_call_async(
-                        session_id=self.session_id,
-                        function_name=func.__name__,
-                        parameters=bound_args.arguments,
-                        docstring=inspect.getdoc(func) or "",
-                    )
-
-                    # Execute the original function
-                    result = await func(*args, **kwargs)
-
-                    # Log the response
-                    await log_tool_response_async(session_id=self.session_id, response=result)
-
-                    return result
-
-                # Regular mock mode
-                base_url = os.getenv("VERIS_ENDPOINT_URL")
-                if not base_url:
-                    error_msg = "VERIS_ENDPOINT_URL environment variable is not set"
-                    raise ValueError(error_msg)
-                endpoint = f"{base_url.rstrip('/')}/api/v2/tool_mock"
-                # 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)
-
-                # Send request to endpoint with timeout
-                # Inject current trace context headers if OpenTelemetry is available
-                headers: dict[str, str] | None = None
+                async def _execute_mock_logic(session_id: str) -> object:
+                    # Handle spy mode - execute original function and log
+                    if mode == "spy":
+                        logger.info(f"Spying on function: {func.__name__}")
+
+                        # Log the tool call
+                        sig = inspect.signature(func)
+                        bound_args = sig.bind(*args, **kwargs)
+                        bound_args.apply_defaults()
+                        _ = bound_args.arguments.pop("ctx", None)
+                        _ = bound_args.arguments.pop("self", None)
+                        _ = bound_args.arguments.pop("cls", None)
+
+                        await log_tool_call_async(
+                            session_id=session_id,
+                            function_name=func.__name__,
+                            parameters=bound_args.arguments,
+                            docstring=inspect.getdoc(func) or "",
+                        )
+
+                        # Execute the original function
+                        result = await func(*args, **kwargs)
+
+                        # Log the response
+                        await log_tool_response_async(session_id=session_id, response=result)
+
+                        return result
+
+                    # Regular mock mode
+                    base_url = os.getenv("VERIS_ENDPOINT_URL")
+                    if not base_url:
+                        error_msg = "VERIS_ENDPOINT_URL environment variable is not set"
+                        raise ValueError(error_msg)
+                    endpoint = f"{base_url.rstrip('/')}/api/v2/tool_mock"
+                    # 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)
+
+                    # Send request to endpoint with timeout
+                    # Inject current trace context headers if OpenTelemetry is available
+                    headers: dict[str, str] | None = None
+                    try:
+                        from opentelemetry.propagate import get_global_textmap  # type: ignore[import-not-found]
+
+                        headers = {}
+                        get_global_textmap().inject(headers)
+                    except Exception:  # pragma: no cover - otel optional
+                        headers = None
+
+                    async with httpx.AsyncClient(timeout=timeout) as client:
+                        response = await client.post(endpoint, json=payload, headers=headers)
+                        response.raise_for_status()
+                        mock_result = response.json()
+                        logger.info(f"Mock response: {mock_result}")
+
+                    if isinstance(mock_result, str):
+                        with suppress(json.JSONDecodeError):
+                            mock_result = json.loads(mock_result)
+                            return convert_to_type(mock_result, return_type_obj)
+                    return convert_to_type(mock_result, return_type_obj)
+
+                # Create a top-level span for the simulated mock call if OpenTelemetry is available
                 try:
-                    from opentelemetry.propagate import get_global_textmap  # type: ignore[import-not-found]
-
-                    headers = {}
-                    get_global_textmap().inject(headers)
-                except Exception:  # pragma: no cover - otel optional
-                    headers = None
-
-                async with httpx.AsyncClient(timeout=timeout) as client:
-                    response = await client.post(endpoint, json=payload, headers=headers)
-                    response.raise_for_status()
-                    mock_result = response.json()
-                    logger.info(f"Mock response: {mock_result}")
-
-                if isinstance(mock_result, str):
-                    with suppress(json.JSONDecodeError):
-                        mock_result = json.loads(mock_result)
-                        return convert_to_type(mock_result, return_type_obj)
-                return convert_to_type(mock_result, return_type_obj)
+                    from opentelemetry import trace  # type: ignore[import-not-found]
+
+                    tracer = trace.get_tracer("veris_ai.tool_mock")
+                    span_name = f"mock.{func.__name__}"
+                    with tracer.start_as_current_span(span_name) as span:  # type: ignore[attr-defined]
+                        span.set_attribute("veris_ai.session.id", self.session_id or "")  # type: ignore[attr-defined]
+                        span.set_attribute("veris_ai.mock.mode", mode)  # type: ignore[attr-defined]
+                        return await _execute_mock_logic(self.session_id)
+                except Exception:
+                    # If OpenTelemetry is not available, run without span
+                    return await _execute_mock_logic(self.session_id)
 
             @wraps(func)
             def sync_wrapper(
@@ -244,67 +259,82 @@ class VerisSDK:
                     # If not in simulation mode, execute the original function
                     return func(*args, **kwargs)
 
-                # Handle spy mode - execute original function and log
-                if mode == "spy":
-                    logger.info(f"Spying on function: {func.__name__}")
-
-                    # Log the tool call
-                    sig = inspect.signature(func)
-                    bound_args = sig.bind(*args, **kwargs)
-                    bound_args.apply_defaults()
-                    _ = bound_args.arguments.pop("ctx", None)
-                    _ = bound_args.arguments.pop("self", None)
-                    _ = bound_args.arguments.pop("cls", None)
-
-                    log_tool_call_sync(
-                        session_id=self.session_id,
-                        function_name=func.__name__,
-                        parameters=bound_args.arguments,
-                        docstring=inspect.getdoc(func) or "",
-                    )
-
-                    # Execute the original function
-                    result = func(*args, **kwargs)
-
-                    # Log the response
-                    log_tool_response_sync(session_id=self.session_id, response=result)
-
-                    return result
-
-                # Regular mock mode
-                base_url = os.getenv("VERIS_ENDPOINT_URL")
-                if not base_url:
-                    error_msg = "VERIS_ENDPOINT_URL environment variable is not set"
-                    raise ValueError(error_msg)
-                endpoint = f"{base_url.rstrip('/')}/api/v2/tool_mock"
-                # 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)
-
-                # Send request to endpoint with timeout (synchronous)
-                # Inject current trace context headers if OpenTelemetry is available
-                headers: dict[str, str] | None = None
+                def _execute_mock_logic(session_id: str) -> object:
+                    # Handle spy mode - execute original function and log
+                    if mode == "spy":
+                        logger.info(f"Spying on function: {func.__name__}")
+
+                        # Log the tool call
+                        sig = inspect.signature(func)
+                        bound_args = sig.bind(*args, **kwargs)
+                        bound_args.apply_defaults()
+                        _ = bound_args.arguments.pop("ctx", None)
+                        _ = bound_args.arguments.pop("self", None)
+                        _ = bound_args.arguments.pop("cls", None)
+
+                        log_tool_call_sync(
+                            session_id=session_id,
+                            function_name=func.__name__,
+                            parameters=bound_args.arguments,
+                            docstring=inspect.getdoc(func) or "",
+                        )
+
+                        # Execute the original function
+                        result = func(*args, **kwargs)
+
+                        # Log the response
+                        log_tool_response_sync(session_id=session_id, response=result)
+
+                        return result
+
+                    # Regular mock mode
+                    base_url = os.getenv("VERIS_ENDPOINT_URL")
+                    if not base_url:
+                        error_msg = "VERIS_ENDPOINT_URL environment variable is not set"
+                        raise ValueError(error_msg)
+                    endpoint = f"{base_url.rstrip('/')}/api/v2/tool_mock"
+                    # 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)
+
+                    # Send request to endpoint with timeout (synchronous)
+                    # Inject current trace context headers if OpenTelemetry is available
+                    headers: dict[str, str] | None = None
+                    try:
+                        from opentelemetry.propagate import get_global_textmap  # type: ignore[import-not-found]
+
+                        headers = {}
+                        get_global_textmap().inject(headers)
+                    except Exception:  # pragma: no cover - otel optional
+                        headers = None
+
+                    with httpx.Client(timeout=timeout) as client:
+                        response = client.post(endpoint, json=payload, headers=headers)
+                        response.raise_for_status()
+                        mock_result = response.json()
+                        logger.info(f"Mock response: {mock_result}")
+
+                    if isinstance(mock_result, str):
+                        with suppress(json.JSONDecodeError):
+                            mock_result = json.loads(mock_result)
+                            return convert_to_type(mock_result, return_type_obj)
+                    return convert_to_type(mock_result, return_type_obj)
+
+                # Create a top-level span for the simulated mock call if OpenTelemetry is available
                 try:
-                    from opentelemetry.propagate import get_global_textmap  # type: ignore[import-not-found]
-
-                    headers = {}
-                    get_global_textmap().inject(headers)
-                except Exception:  # pragma: no cover - otel optional
-                    headers = None
-
-                with httpx.Client(timeout=timeout) as client:
-                    response = client.post(endpoint, json=payload, headers=headers)
-                    response.raise_for_status()
-                    mock_result = response.json()
-                    logger.info(f"Mock response: {mock_result}")
-
-                if isinstance(mock_result, str):
-                    with suppress(json.JSONDecodeError):
-                        mock_result = json.loads(mock_result)
-                        return convert_to_type(mock_result, return_type_obj)
-                return convert_to_type(mock_result, return_type_obj)
+                    from opentelemetry import trace  # type: ignore[import-not-found]
+
+                    tracer = trace.get_tracer("veris_ai.tool_mock")
+                    span_name = f"mock.{func.__name__}"
+                    with tracer.start_as_current_span(span_name) as span:  # type: ignore[attr-defined]
+                        span.set_attribute("veris_ai.session.id", self.session_id or "")  # type: ignore[attr-defined]
+                        span.set_attribute("veris_ai.mock.mode", mode)  # type: ignore[attr-defined]
+                        return _execute_mock_logic(self.session_id)
+                except Exception:
+                    # If OpenTelemetry is not available, run without span
+                    return _execute_mock_logic(self.session_id)
 
             # Return the appropriate wrapper based on whether the function is async
             return async_wrapper if is_async else sync_wrapper

{veris_ai-1.6.0.dist-info → veris_ai-1.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.6.0
+Version: 1.7.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.6.0.dist-info → veris_ai-1.7.0.dist-info}/RECORD

@@ -3,13 +3,13 @@ veris_ai/__init__.py,sha256=Gh52-XfFpsxX37uqp8vuX0V3Np7f-Rlf5k3MMANu6e0,425
 veris_ai/logging.py,sha256=srEdVimcQSCsnwGhyzCydehD2JW1cQmwSRd3X20NQs0,5233
 veris_ai/models.py,sha256=6HINPxNFCakCVPcyEbUswWkXwb2K4lF0A8g8EvTMal4,213
 veris_ai/observability.py,sha256=fG5ixAiVDykLCdHUScKQNCsePLphKQ7PcaKkAGBVdF0,5113
-veris_ai/tool_mock.py,sha256=GTgHFXJ1Z0SvZje2Ws4bbCNjznFl5IFCJCaCSXkzK1Y,14184
+veris_ai/tool_mock.py,sha256=C1BANT3LCMKOIwJm_nOFuNdHnaO9_tbjeDTtBcI41RI,16512
 veris_ai/utils.py,sha256=aqFFNuNiBehil6874nOHtU7G_bWHbFpVuubcz2AIx6I,9267
 veris_ai/jaeger_interface/README.md,sha256=kd9rKcE5xf3EyNaiHu0tjn-0oES9sfaK6Ih-OhhTyCM,2821
 veris_ai/jaeger_interface/__init__.py,sha256=KD7NSiMYRG_2uF6dOLKkGG5lNQe4K9ptEwucwMT4_aw,1128
 veris_ai/jaeger_interface/client.py,sha256=yJrh86wRR0Dk3Gq12DId99WogcMIVbL0QQFqVSevvlE,8772
 veris_ai/jaeger_interface/models.py,sha256=e64VV6IvOEFuzRUgvDAMQFyOZMRb56I-PUPZLBZ3rX0,1864
-veris_ai-1.6.0.dist-info/METADATA,sha256=c4BgajmFhmFI-DPmiC4sCJoeLJeZvM6CAY_5Bde2Lkg,8679
-veris_ai-1.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-veris_ai-1.6.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
-veris_ai-1.6.0.dist-info/RECORD,,
+veris_ai-1.7.0.dist-info/METADATA,sha256=dmWFBpwU2MK9tjnJtlrkQdqnR9C0yn0Lhb1TMWb-XvA,8679
+veris_ai-1.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+veris_ai-1.7.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
+veris_ai-1.7.0.dist-info/RECORD,,