.

Author: spike-spiegel-21

deepeval/integrations/pydantic_ai/agent.py

@@ -84,13 +84,15 @@ def __init__(
     async def run(
         self,
         *args,
-        trace_name: Optional[str] = None,
-        trace_tags: Optional[List[str]] = None,
-        trace_user_id: Optional[str] = None,
-        trace_metadata: Optional[dict] = None,
-        trace_thread_id: Optional[str] = None,
-        trace_metrics: Optional[List[BaseMetric]] = None,
-        trace_metric_collection: Optional[str] = None,
+        
+        name: Optional[str] = None,
+        tags: Optional[List[str]] = None,
+        user_id: Optional[str] = None,
+        metadata: Optional[dict] = None,
+        thread_id: Optional[str] = None,
+        metrics: Optional[List[BaseMetric]] = None,
+        metric_collection: Optional[str] = None,
+        
         **kwargs
     ):
         sig = inspect.signature(super().run)
@@ -111,13 +113,13 @@ async def run(
             observer.result = result.output
             update_trace_context(
 
-                trace_name=trace_name if trace_name is not None else self.trace_name,
-                trace_tags=trace_tags if trace_tags is not None else self.trace_tags,
-                trace_metadata=trace_metadata if trace_metadata is not None else self.trace_metadata,
-                trace_thread_id=trace_thread_id if trace_thread_id is not None else self.trace_thread_id,
-                trace_user_id=trace_user_id if trace_user_id is not None else self.trace_user_id,
-                trace_metric_collection=trace_metric_collection if trace_metric_collection is not None else self.trace_metric_collection,
-                trace_metrics=trace_metrics if trace_metrics is not None else self.trace_metrics,
+                trace_name=name if name is not None else self.trace_name,
+                trace_tags=tags if tags is not None else self.trace_tags,
+                trace_metadata=metadata if metadata is not None else self.trace_metadata,
+                trace_thread_id=thread_id if thread_id is not None else self.trace_thread_id,
+                trace_user_id=user_id if user_id is not None else self.trace_user_id,
+                trace_metric_collection=metric_collection if metric_collection is not None else self.trace_metric_collection,
+                trace_metrics=metrics if metrics is not None else self.trace_metrics,
 
                 trace_input=input,
                 trace_output=result.output,
@@ -134,15 +136,17 @@ async def run(
         return result
 
     def run_sync(
-        self,
+        self,   
         *args,
-        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,
+        
+        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
     ):
         sig = inspect.signature(super().run_sync)
@@ -169,13 +173,13 @@ def run_sync(
             observer.result = result.output
             update_trace_context(
 
-                trace_name=trace_name if trace_name is not None else self.trace_name,
-                trace_tags=trace_tags if trace_tags is not None else self.trace_tags,
-                trace_metadata=trace_metadata if trace_metadata is not None else self.trace_metadata,
-                trace_thread_id=trace_thread_id if trace_thread_id is not None else self.trace_thread_id,
-                trace_user_id=trace_user_id if trace_user_id is not None else self.trace_user_id,
-                trace_metric_collection=trace_metric_collection if trace_metric_collection is not None else self.trace_metric_collection,
-                trace_metrics=trace_metrics if trace_metrics is not None else self.trace_metrics,
+                trace_name=name if name is not None else self.trace_name,
+                trace_tags=tags if tags is not None else self.trace_tags,
+                trace_metadata=metadata if metadata is not None else self.trace_metadata,
+                trace_thread_id=thread_id if thread_id is not None else self.trace_thread_id,
+                trace_user_id=user_id if user_id is not None else self.trace_user_id,
+                trace_metric_collection=metric_collection if metric_collection is not None else self.trace_metric_collection,
+                trace_metrics=metrics if metrics is not None else self.trace_metrics,
 
                 trace_input=input,
                 trace_output=result.output,
@@ -196,13 +200,15 @@ def run_sync(
     async def run_stream(
         self,
         *args,
-        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,
+        
+        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
     ):
         sig = inspect.signature(super().run_stream)
@@ -233,13 +239,15 @@ async def run_stream(
                         pass
 
                     update_trace_context(
-                        trace_name=trace_name if trace_name is not None else self.trace_name,
-                        trace_tags=trace_tags if trace_tags is not None else self.trace_tags,
-                        trace_metadata=trace_metadata if trace_metadata is not None else self.trace_metadata,
-                        trace_thread_id=trace_thread_id if trace_thread_id is not None else self.trace_thread_id,
-                        trace_user_id=trace_user_id if trace_user_id is not None else self.trace_user_id,
-                        trace_metric_collection=trace_metric_collection if trace_metric_collection is not None else self.trace_metric_collection,
-                        trace_metrics=trace_metrics if trace_metrics is not None else self.trace_metrics,
+                        
+                        trace_name=name if name is not None else self.trace_name,
+                        trace_tags=tags if tags is not None else self.trace_tags,
+                        trace_metadata=metadata if metadata is not None else self.trace_metadata,
+                        trace_thread_id=thread_id if thread_id is not None else self.trace_thread_id,
+                        trace_user_id=user_id if user_id is not None else self.trace_user_id,
+                        trace_metric_collection=metric_collection if metric_collection is not None else self.trace_metric_collection,
+                        trace_metrics=metrics if metrics is not None else self.trace_metrics,
+                        
                         trace_input=input,
                         trace_output=(final_result if final_result is not None else None),
                     )

tests/test_integrations/test_pydanticai/pydantic_all_tests.py

@@ -27,17 +27,19 @@ def get_weather(city: str) -> str:
 )
 
 async def execute_agent_stream():
-    async with agent.run_stream("What is the weather in London?", trace_name="test_name_2") as result:
+    async with agent.run_stream("What is the weather in London?", name="test_name_2") as result:
         async for chunk in result.stream_text(delta=True):
             print(chunk, end="", flush=True)
         final = await result.get_output()
         print("\n\nFinal:", final)        
 
 async def execute_agent_run():
-    result = await agent.run("What is the weather in London?", trace_name="test_name_4")
+    result = await agent.run("What is the weather in London?", name="test_name_4")
     print(result.output)
 
 def execute_all():
     asyncio.run(execute_agent_stream())
-    agent.run_sync("What is the weather in London?", trace_name="test_name_3")
-    asyncio.run(execute_agent_run())
+    agent.run_sync("What is the weather in London?", name="test_name_3")
+    asyncio.run(execute_agent_run())
+
+execute_all()

tests/test_integrations/test_pydanticai/pydanticai_app.py

@@ -82,11 +82,11 @@ async def run_agent(input_query: str):
             input_query,
             deps=deps,
             # metric_collection="test_collection_1",
-            trace_name="test_trace_2",
+            name="test_trace_2",
             # tags=["test_tag_1"],
             # metadata={"test_metadata_1": "test_metadata_1"},
             # thread_id="test_thread_id_1",
-            trace_user_id="test_user_id_2",
+            user_id="test_user_id_2",
         )
 
         return result.output
@@ -95,3 +95,5 @@ async def run_agent(input_query: str):
 def execute_agent():
     output = asyncio.run(run_agent("What's the weather in Paris?"))
     return output
+
+execute_agent()