deepeval 3.5.7__py3-none-any.whl → 3.5.9__py3-none-any.whl

{deepeval-3.5.7.dist-info → deepeval-3.5.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: deepeval
-Version: 3.5.7
+Version: 3.5.9
 Summary: The LLM Evaluation Framework
 Home-page: https://github.com/confident-ai/deepeval
 License: Apache-2.0
@@ -17,6 +17,7 @@ Requires-Dist: anthropic
 Requires-Dist: click (>=8.0.0,<8.3.0)
 Requires-Dist: google-genai (>=1.9.0,<2.0.0)
 Requires-Dist: grpcio (>=1.67.1,<2.0.0)
+Requires-Dist: jinja2
 Requires-Dist: nest_asyncio
 Requires-Dist: ollama
 Requires-Dist: openai

{deepeval-3.5.7.dist-info → deepeval-3.5.9.dist-info}/RECORD

@@ -1,5 +1,5 @@
 deepeval/__init__.py,sha256=6fsb813LD_jNhqR-xZnSdE5E-KsBbC3tc4oIg5ZMgTw,2115
-deepeval/_version.py,sha256=5PpKL25tWtYxTPc0_se2v49WDFVCzYaCu8yogWsx_qQ,27
+deepeval/_version.py,sha256=pEL83SMnaXGf-TcVHz9_kQx6-qWl_kZPXsJxqkdMIto,27
 deepeval/annotation/__init__.py,sha256=ZFhUVNNuH_YgQSZJ-m5E9iUb9TkAkEV33a6ouMDZ8EI,111
 deepeval/annotation/annotation.py,sha256=3j3-syeJepAcEj3u3e4T_BeRDzNr7yXGDIoNQGMKpwQ,2298
 deepeval/annotation/api.py,sha256=EYN33ACVzVxsFleRYm60KB4Exvff3rPJKt1VBuuX970,2147
@@ -179,11 +179,10 @@ deepeval/integrations/llama_index/__init__.py,sha256=zBwUFQXDp6QFtp1cfANy8ucV08r
 deepeval/integrations/llama_index/agent/patched.py,sha256=4JbH0WQmt4lct7xxIH0phj8_Y-V35dgVv7DEDXK0jZI,2149
 deepeval/integrations/llama_index/handler.py,sha256=eqI1n8E4MsvfKoFs5Zrm9IdCR7g9eBgNedISs7UkU_I,8947
 deepeval/integrations/llama_index/utils.py,sha256=mxW71-3PjvBvJpLIU0kNWuTzCidy5l_-roLt8ZyWYA0,2599
-deepeval/integrations/pydantic_ai/__init__.py,sha256=0-GZpWgCnFI-fVHI-3DosWQK85rk6CoRRhl4AiytBAw,258
-deepeval/integrations/pydantic_ai/agent.py,sha256=HxfeTLsdWGgRMy00ymgXdE6dcFDmFBsdgfl9BbvyJns,12311
-deepeval/integrations/pydantic_ai/otel.py,sha256=2DpO3RapdztXPlT9BWhQfF4dJDMyp2X7YvuplJ0SwC8,1661
-deepeval/integrations/pydantic_ai/patcher.py,sha256=yy4SZRmRhgYxh6qGVWWf8DnSMCDA9GLkFw1HbPToQ1w,17696
-deepeval/integrations/pydantic_ai/utils.py,sha256=734e9un-fn5V7MueAmVsXh304qgumv_fdcmdOC4HrJw,10998
+deepeval/integrations/pydantic_ai/__init__.py,sha256=UIkXn_g6h9LTQXG1PaWu1eCFkCssIwG48WSvN46UWgU,202
+deepeval/integrations/pydantic_ai/agent.py,sha256=4wRV25O1tC-txH2j3TNJWry6gDNBqqThj7zgFKBxJpw,606
+deepeval/integrations/pydantic_ai/instrumentator.py,sha256=tGHuP7nn4jL9bUFR2fZWbF4k_EhF5JXOnCwQm-tmDKc,6974
+deepeval/integrations/pydantic_ai/otel.py,sha256=0OuIpmaMtEt1dFWFZtYAiZ9hVCWweEWr1TRHYcDb4I8,1918
 deepeval/key_handler.py,sha256=damdQEBLGy4IVk5DR5-E3blIZdLbcMtyeGAFn_4_SG4,6505
 deepeval/metrics/__init__.py,sha256=nvO0Wv2JROjK1I9MDNIFUJlrRAZI2C0xbGYSBZK5q4g,4013
 deepeval/metrics/answer_relevancy/__init__.py,sha256=WbZUpoSg2GQoqJ4VIRirVVQ1JDx5xwT-RskwqNKfWGM,46
@@ -404,9 +403,9 @@ deepeval/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 deepeval/plugins/plugin.py,sha256=_dwsdx4Dg9DbXxK3f7zJY4QWTJQWc7QE1HmIg2Zjjag,1515
 deepeval/progress_context.py,sha256=ZSKpxrE9sdgt9G3REKnVeXAv7GJXHHVGgLynpG1Pudw,3557
 deepeval/prompt/__init__.py,sha256=M99QTWdxOfiNeySGCSqN873Q80PPxqRvjLq4_Mw-X1w,49
-deepeval/prompt/api.py,sha256=ccrMT6_Otuef9zrRm9iQsmZ-Apkjj3nypvw3wc8-eW0,1708
-deepeval/prompt/prompt.py,sha256=DlUPib0EAVC2T54SQB1xxEQBNSpEirL5x-CtrKE1rek,15463
-deepeval/prompt/utils.py,sha256=Gk0zj_9BK8MQccs8GmiC8o-YVtkou6ZJEz8kWgW5Mog,1678
+deepeval/prompt/api.py,sha256=kR3MkaHuU2wYILKVnvnXhQWxWp0XgtcWX-kIjpMJRl8,1728
+deepeval/prompt/prompt.py,sha256=192W5zFBx08nELxRHHDQscMM3psj8OUFV_JR85BZv8Q,15823
+deepeval/prompt/utils.py,sha256=Ermw9P-1-T5wQ5uYuj5yWgdj7pVB_JLw8D37Qvmh9ok,1938
 deepeval/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 deepeval/red_teaming/README.md,sha256=BY5rAdpp3-sMMToEKwq0Nsd9ivkGDzPE16DeDb8GY7U,154
 deepeval/scorer/__init__.py,sha256=hTvtoV3a4l0dSBjERm-jX7jveTtKZXK0c9JerQo0T_w,27
@@ -453,16 +452,16 @@ deepeval/tracing/offline_evals/span.py,sha256=pXqTVXs-WnjRVpCYYEbNe0zSM6Wz9GsKHs
 deepeval/tracing/offline_evals/thread.py,sha256=bcSGFcZJKnszArOLIlWvnCyt0zSmsd7Xsw5rl4RTVFg,1981
 deepeval/tracing/offline_evals/trace.py,sha256=vTflaTKysKRiYvKA-Nx6PUJ3J6NrRLXiIdWieVcm90E,1868
 deepeval/tracing/otel/__init__.py,sha256=HQsaF5yLPwyW5qg8AOV81_nG_7pFHnatOTHi9Wx3HEk,88
-deepeval/tracing/otel/exporter.py,sha256=dXQd834zm5rm1ss9pWkBBlk-JSdtiw7aFLso2hM53XY,26372
-deepeval/tracing/otel/utils.py,sha256=g8yAzhqbPh1fOKCWkfNekC6AVotLfu1SUcfNMo6zii8,9786
+deepeval/tracing/otel/exporter.py,sha256=YfzV2zyxRwD6FO-qV0PR-y1nk1kInPJp0TtIuWNHbh0,28606
+deepeval/tracing/otel/utils.py,sha256=xydJyUgR1s4hTGcz-onvVHyiT7F6lUXRuh_nprHkKbU,12396
 deepeval/tracing/patchers.py,sha256=DAPNkhrDtoeyJIVeQDUMhTz-xGcXu00eqjQZmov8FiU,3096
 deepeval/tracing/perf_epoch_bridge.py,sha256=iyAPddB6Op7NpMtPHJ29lDm53Btz9yLaN6xSCfTRQm4,1825
 deepeval/tracing/tracing.py,sha256=b-0T3W6lAEOEGhODx0e-yIwBkm5V46EDNAWS9lcWkD0,42306
 deepeval/tracing/types.py,sha256=l_utWKerNlE5H3mOKpeUJLsvpP3cMyjH7HRANNgTmSQ,5306
 deepeval/tracing/utils.py,sha256=w_kdhuyBCygllnbqLpDdKJqpJo42t3ZMlGhNicV2A8c,6467
 deepeval/utils.py,sha256=r8tV_NYJSi6ib-oQw6cLw3L7ZSe4KIJVJc1ng6-kDX4,17179
-deepeval-3.5.7.dist-info/LICENSE.md,sha256=0ATkuLv6QgsJTBODUHC5Rak_PArA6gv2t7inJzNTP38,11352
-deepeval-3.5.7.dist-info/METADATA,sha256=NldO1OinDSv_gGUP-kkFk1zpMGXKTceoYMtF92XGbgs,18721
-deepeval-3.5.7.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-deepeval-3.5.7.dist-info/entry_points.txt,sha256=fVr8UphXTfJe9I2rObmUtfU3gkSrYeM0pLy-NbJYg10,94
-deepeval-3.5.7.dist-info/RECORD,,
+deepeval-3.5.9.dist-info/LICENSE.md,sha256=0ATkuLv6QgsJTBODUHC5Rak_PArA6gv2t7inJzNTP38,11352
+deepeval-3.5.9.dist-info/METADATA,sha256=gsap42eEwklR_la4cilLkE2HKQxOP_rnSPmGtvjdqzg,18743
+deepeval-3.5.9.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+deepeval-3.5.9.dist-info/entry_points.txt,sha256=fVr8UphXTfJe9I2rObmUtfU3gkSrYeM0pLy-NbJYg10,94
+deepeval-3.5.9.dist-info/RECORD,,

deepeval/integrations/pydantic_ai/patcher.py

@@ -1,484 +0,0 @@
-# import inspect
-# import functools
-# import warnings
-# from typing import List, Callable, Optional, Any
-# from deepeval.tracing.types import LlmOutput, LlmToolCall
-# from pydantic_ai.agent import AgentRunResult
-# from deepeval.tracing.context import current_trace_context
-# from deepeval.tracing.types import AgentSpan, LlmSpan
-# from deepeval.tracing.tracing import Observer
-# from deepeval.test_case.llm_test_case import ToolCall
-# from deepeval.metrics.base_metric import BaseMetric
-# from deepeval.confident.api import get_confident_api_key
-# from deepeval.integrations.pydantic_ai.otel import instrument_pydantic_ai
-# from deepeval.telemetry import capture_tracing_integration
-# from deepeval.prompt import Prompt
-# import deepeval
-# # from contextvars import ContextVar
-
-# try:
-#     from pydantic_ai.agent import Agent
-#     from pydantic_ai.models import Model
-#     from pydantic_ai.messages import (
-#         ModelResponse,
-#         ModelRequest,
-#         ModelResponsePart,
-#         TextPart,
-#         ToolCallPart,
-#         SystemPromptPart,
-#         ToolReturnPart,
-#         UserPromptPart,
-#     )
-#     from pydantic_ai._run_context import RunContext
-#     from deepeval.integrations.pydantic_ai.utils import (
-#         extract_tools_called_from_llm_response,
-#         extract_tools_called,
-#         sanitize_run_context,
-#     )
-
-#     pydantic_ai_installed = True
-# except:
-#     pydantic_ai_installed = True
-
-# # _IN_RUN_SYNC = ContextVar("deepeval_in_run_sync", default=False)
-# # _INSTRUMENTED = False
-
-
-import warnings
-from typing import Optional
-
-
-def instrument(otel: Optional[bool] = False, api_key: Optional[str] = None):
-    """
-    DEPRECATED: This function is deprecated and will be removed in a future version.
-    Please deepeval.integrations.pydantic_ai.Agent to instrument instead.
-    """
-    warnings.warn(
-        "The 'instrument_pydantic_ai()' function is deprecated and will be removed in a future version. "
-        "Please use deepeval.integrations.pydantic_ai.Agent to instrument instead. Refer to the documentation [link]",  # TODO: add the link,
-        UserWarning,
-        stacklevel=2,
-    )
-
-    # Don't execute the original functionality
-    return
-
-    # Original code below (commented out to prevent execution)
-    # global _INSTRUMENTED
-    # if api_key:
-    #     deepeval.login(api_key)
-    #
-    # api_key = get_confident_api_key()
-    #
-    # if not api_key:
-    #     raise ValueError("No api key provided.")
-    #
-    # if otel:
-    #     instrument_pydantic_ai(api_key)
-    # else:
-    #     with capture_tracing_integration("pydantic_ai"):
-    #         if _INSTRUMENTED:
-    #             return
-    #         _patch_agent_init()
-    #         _patch_agent_tool_decorator()
-    #         _INSTRUMENTED = True
-
-
-# ################### Init Patches ###################
-
-
-# # def _patch_agent_init():
-# #     original_init = Agent.__init__
-
-# #     @functools.wraps(original_init)
-# #     def wrapper(
-# #         *args,
-# #         llm_metric_collection: Optional[str] = None,
-# #         llm_metrics: Optional[List[BaseMetric]] = None,
-# #         llm_prompt: Optional[Prompt] = None,
-# #         agent_metric_collection: Optional[str] = None,
-# #         agent_metrics: Optional[List[BaseMetric]] = None,
-# #         name: Optional[str] = None,
-# #         tags: Optional[List[str]] = None,
-# #         metadata: Optional[dict] = None,
-# #         thread_id: Optional[str] = None,
-# #         user_id: Optional[str] = None,
-# #         metric_collection: Optional[str] = None,
-# #         metrics: Optional[List[BaseMetric]] = None,
-# #         **kwargs
-# #     ):
-# #         result = original_init(*args, **kwargs)
-# #         _patch_llm_model(args[0]._model, llm_metric_collection, llm_metrics, llm_prompt)  # runtime patch of the model
-# #         _patch_agent_run(
-# #             agent=args[0],
-# #             agent_metric_collection=agent_metric_collection,
-# #             agent_metrics=agent_metrics,
-# #             init_trace_name=name,
-# #             init_trace_tags=tags,
-# #             init_trace_metadata=metadata,
-# #             init_trace_thread_id=thread_id,
-# #             init_trace_user_id=user_id,
-# #             init_trace_metric_collection=metric_collection,
-# #             init_trace_metrics=metrics,
-# #         )
-# #         _patch_agent_run_sync(
-# #             agent=args[0],
-# #             agent_metric_collection=agent_metric_collection,
-# #             agent_metrics=agent_metrics,
-# #             init_trace_name=name,
-# #             init_trace_tags=tags,
-# #             init_trace_metadata=metadata,
-# #             init_trace_thread_id=thread_id,
-# #             init_trace_user_id=user_id,
-# #             init_trace_metric_collection=metric_collection,
-# #             init_trace_metrics=metrics,
-# #         )
-# #         return result
-
-# #     Agent.__init__ = wrapper
-
-
-# # def _patch_agent_tool_decorator():
-# #     original_tool = Agent.tool
-
-# #     @functools.wraps(original_tool)
-# #     def wrapper(
-# #         *args,
-# #         metrics: Optional[List[BaseMetric]] = None,
-# #         metric_collection: Optional[str] = None,
-# #         **kwargs
-# #     ):
-# #         # Case 1: Direct decoration - @agent.tool
-# #         if args and callable(args[0]):
-# #             patched_func = _create_patched_tool(
-# #                 args[0], metrics, metric_collection
-# #             )
-# #             new_args = (patched_func,) + args[1:]
-# #             return original_tool(*new_args, **kwargs)
-
-# #         # Case 2: Decoration with arguments - @agent.tool(metrics=..., metric_collection=...)
-# #         else:
-# #             # Return a decorator function that will receive the actual function
-# #             def decorator(func):
-# #                 patched_func = _create_patched_tool(
-# #                     func, metrics, metric_collection
-# #                 )
-# #                 return original_tool(*args, **kwargs)(patched_func)
-
-# #             return decorator
-
-# #     Agent.tool = wrapper
-
-
-# ################### Runtime Patches ###################
-
-
-# # def _patch_agent_run_sync(
-# #     agent: Agent,
-# #     agent_metric_collection: Optional[str] = None,
-# #     agent_metrics: Optional[List[BaseMetric]] = None,
-# #     init_trace_name: Optional[str] = None,
-# #     init_trace_tags: Optional[List[str]] = None,
-# #     init_trace_metadata: Optional[dict] = None,
-# #     init_trace_thread_id: Optional[str] = None,
-# #     init_trace_user_id: Optional[str] = None,
-# #     init_trace_metric_collection: Optional[str] = None,
-# #     init_trace_metrics: Optional[List[BaseMetric]] = None,
-# # ):
-# #     original_run_sync = agent.run_sync
-
-# #     @functools.wraps(original_run_sync)
-# #     def wrapper(
-# #         *args,
-# #         metric_collection: Optional[str] = None,
-# #         metrics: Optional[List[BaseMetric]] = None,
-# #         name: Optional[str] = None,
-# #         tags: Optional[List[str]] = None,
-# #         metadata: Optional[dict] = None,
-# #         thread_id: Optional[str] = None,
-# #         user_id: Optional[str] = None,
-# #         **kwargs
-# #     ):
-
-# #         sig = inspect.signature(original_run_sync)
-# #         bound = sig.bind_partial(*args, **kwargs)
-# #         bound.apply_defaults()
-# #         input = bound.arguments.get("user_prompt", None)
-
-# #         with Observer(
-# #             span_type="agent",
-# #             func_name="Agent",
-# #             function_kwargs={"input": input},
-# #             metrics=agent_metrics,
-# #             metric_collection=agent_metric_collection,
-# #         ) as observer:
-
-# #             token = _IN_RUN_SYNC.set(True)
-# #             try:
-# #                 result = original_run_sync(*args, **kwargs)
-# #             finally:
-# #                 _IN_RUN_SYNC.reset(token)
-
-# #             observer.update_span_properties = (
-# #                 lambda agent_span: set_agent_span_attributes(agent_span, result)
-# #             )
-# #             observer.result = result.output
-
-# #             _update_trace_context(
-# #                 trace_name=init_trace_name if init_trace_name else name,
-# #                 trace_tags=init_trace_tags if init_trace_tags else tags,
-# #                 trace_metadata=init_trace_metadata if init_trace_metadata else metadata,
-# #                 trace_thread_id=init_trace_thread_id if init_trace_thread_id else thread_id,
-# #                 trace_user_id=init_trace_user_id if init_trace_user_id else user_id,
-# #                 trace_metric_collection=init_trace_metric_collection if init_trace_metric_collection else metric_collection,
-# #                 trace_metrics=init_trace_metrics if init_trace_metrics else metrics,
-# #                 trace_input=input,
-# #                 trace_output=result.output,
-# #             )
-
-# #         return result
-
-# #     agent.run_sync = wrapper
-
-
-# # def _patch_agent_run(
-# #     agent: Agent,
-# #     agent_metric_collection: Optional[str] = None,
-# #     agent_metrics: Optional[List[BaseMetric]] = None,
-# #     init_trace_name: Optional[str] = None,
-# #     init_trace_tags: Optional[List[str]] = None,
-# #     init_trace_metadata: Optional[dict] = None,
-# #     init_trace_thread_id: Optional[str] = None,
-# #     init_trace_user_id: Optional[str] = None,
-# #     init_trace_metric_collection: Optional[str] = None,
-# #     init_trace_metrics: Optional[List[BaseMetric]] = None,
-# # ):
-# #     original_run = agent.run
-
-# #     @functools.wraps(original_run)
-# #     async def wrapper(
-# #         *args,
-# #         metric_collection: Optional[str] = None,
-# #         metrics: Optional[List[BaseMetric]] = None,
-# #         name: Optional[str] = None,
-# #         tags: Optional[List[str]] = None,
-# #         metadata: Optional[dict] = None,
-# #         thread_id: Optional[str] = None,
-# #         user_id: Optional[str] = None,
-# #         **kwargs
-# #     ):
-# #         sig = inspect.signature(original_run)
-# #         bound = sig.bind_partial(*args, **kwargs)
-# #         bound.apply_defaults()
-# #         input = bound.arguments.get("user_prompt", None)
-
-# #         in_sync = _IN_RUN_SYNC.get()
-# #         with Observer(
-# #             span_type="agent" if not in_sync else "custom",
-# #             func_name="Agent" if not in_sync else "run",
-# #             function_kwargs={"input": input},
-# #             metrics=agent_metrics if not in_sync else None,
-# #             metric_collection=agent_metric_collection if not in_sync else None,
-# #         ) as observer:
-# #             print(args)
-# #             print(kwargs)
-# #             result = await original_run(*args, **kwargs)
-# #             observer.update_span_properties = (
-# #                 lambda agent_span: set_agent_span_attributes(agent_span, result)
-# #             )
-# #             observer.result = result.output
-
-# #             _update_trace_context(
-# #                 trace_name=init_trace_name if init_trace_name else name,
-# #                 trace_tags=init_trace_tags if init_trace_tags else tags,
-# #                 trace_metadata=init_trace_metadata if init_trace_metadata else metadata,
-# #                 trace_thread_id=init_trace_thread_id if init_trace_thread_id else thread_id,
-# #                 trace_user_id=init_trace_user_id if init_trace_user_id else user_id,
-# #                 trace_metric_collection=init_trace_metric_collection if init_trace_metric_collection else metric_collection,
-# #                 trace_metrics=init_trace_metrics if init_trace_metrics else metrics,
-# #                 trace_input=input,
-# #                 trace_output=result.output,
-# #             )
-
-# #         return result
-
-# #     agent.run = wrapper
-
-
-# def patch_llm_model(
-#     model: Model,
-#     llm_metric_collection: Optional[str] = None,
-#     llm_metrics: Optional[List[BaseMetric]] = None,
-#     llm_prompt: Optional[Prompt] = None,
-# ):
-#     original_func = model.request
-#     sig = inspect.signature(original_func)
-
-#     try:
-#         model_name = model.model_name
-#     except Exception:
-#         model_name = "unknown"
-
-#     @functools.wraps(original_func)
-#     async def wrapper(*args, **kwargs):
-#         bound = sig.bind_partial(*args, **kwargs)
-#         bound.apply_defaults()
-#         request = bound.arguments.get("messages", [])
-
-#         with Observer(
-#             span_type="llm",
-#             func_name="LLM",
-#             observe_kwargs={"model": model_name},
-#             metrics=llm_metrics,
-#             metric_collection=llm_metric_collection,
-#         ) as observer:
-#             result = await original_func(*args, **kwargs)
-#             observer.update_span_properties = (
-#                 lambda llm_span: set_llm_span_attributes(
-#                     llm_span, request, result, llm_prompt
-#                 )
-#             )
-#             observer.result = result
-#             return result
-
-#     model.request = wrapper
-
-
-# ################### Helper Functions ###################
-
-
-# def create_patched_tool(
-#     func: Callable,
-#     metrics: Optional[List[BaseMetric]] = None,
-#     metric_collection: Optional[str] = None,
-# ):
-#     import asyncio
-
-#     original_func = func
-
-#     is_async = asyncio.iscoroutinefunction(original_func)
-
-#     if is_async:
-
-#         @functools.wraps(original_func)
-#         async def async_wrapper(*args, **kwargs):
-#             sanitized_args = sanitize_run_context(args)
-#             sanitized_kwargs = sanitize_run_context(kwargs)
-#             with Observer(
-#                 span_type="tool",
-#                 func_name=original_func.__name__,
-#                 metrics=metrics,
-#                 metric_collection=metric_collection,
-#                 function_kwargs={"args": sanitized_args, **sanitized_kwargs},
-#             ) as observer:
-#                 result = await original_func(*args, **kwargs)
-#                 observer.result = result
-
-#             return result
-
-#         return async_wrapper
-#     else:
-
-#         @functools.wraps(original_func)
-#         def sync_wrapper(*args, **kwargs):
-#             sanitized_args = sanitize_run_context(args)
-#             sanitized_kwargs = sanitize_run_context(kwargs)
-#             with Observer(
-#                 span_type="tool",
-#                 func_name=original_func.__name__,
-#                 metrics=metrics,
-#                 metric_collection=metric_collection,
-#                 function_kwargs={"args": sanitized_args, **sanitized_kwargs},
-#             ) as observer:
-#                 result = original_func(*args, **kwargs)
-#                 observer.result = result
-
-#             return result
-
-#         return sync_wrapper
-
-
-# def update_trace_context(
-#     trace_name: Optional[str] = None,
-#     trace_tags: Optional[List[str]] = None,
-#     trace_metadata: Optional[dict] = None,
-#     trace_thread_id: Optional[str] = None,
-#     trace_user_id: Optional[str] = None,
-#     trace_metric_collection: Optional[str] = None,
-#     trace_metrics: Optional[List[BaseMetric]] = None,
-#     trace_input: Optional[Any] = None,
-#     trace_output: Optional[Any] = None,
-# ):
-
-#     current_trace = current_trace_context.get()
-
-#     if trace_name:
-#         current_trace.name = trace_name
-#     if trace_tags:
-#         current_trace.tags = trace_tags
-#     if trace_metadata:
-#         current_trace.metadata = trace_metadata
-#     if trace_thread_id:
-#         current_trace.thread_id = trace_thread_id
-#     if trace_user_id:
-#         current_trace.user_id = trace_user_id
-#     if trace_metric_collection:
-#         current_trace.metric_collection = trace_metric_collection
-#     if trace_metrics:
-#         current_trace.metrics = trace_metrics
-#     if trace_input:
-#         current_trace.input = trace_input
-#     if trace_output:
-#         current_trace.output = trace_output
-
-
-# def set_llm_span_attributes(
-#     llm_span: LlmSpan,
-#     requests: List[ModelRequest],
-#     result: ModelResponse,
-#     llm_prompt: Optional[Prompt] = None,
-# ):
-#     llm_span.prompt = llm_prompt
-
-#     input = []
-#     for request in requests:
-#         for part in request.parts:
-#             if isinstance(part, SystemPromptPart):
-#                 input.append({"role": "System", "content": part.content})
-#             elif isinstance(part, UserPromptPart):
-#                 input.append({"role": "User", "content": part.content})
-#             elif isinstance(part, ToolCallPart):
-#                 input.append(
-#                     {
-#                         "role": "Tool Call",
-#                         "name": part.tool_name,
-#                         "content": part.args_as_json_str(),
-#                     }
-#                 )
-#             elif isinstance(part, ToolReturnPart):
-#                 input.append(
-#                     {
-#                         "role": "Tool Return",
-#                         "name": part.tool_name,
-#                         "content": part.model_response_str(),
-#                     }
-#                 )
-#     llm_span.input = input
-
-#     content = ""
-#     tool_calls = []
-#     for part in result.parts:
-#         if isinstance(part, TextPart):
-#             content += part.content + "\n"
-#         elif isinstance(part, ToolCallPart):
-#             tool_calls.append(
-#                 LlmToolCall(name=part.tool_name, args=part.args_as_dict())
-#             )
-#     llm_span.output = LlmOutput(
-#         role="Assistant", content=content, tool_calls=tool_calls
-#     )
-#     llm_span.tools_called = extract_tools_called_from_llm_response(result.parts)
-
-
-# def set_agent_span_attributes(agent_span: AgentSpan, result: AgentRunResult):
-#     agent_span.tools_called = extract_tools_called(result)