.

Author: spike-spiegel-21

deepeval/openai_agents/runner.py

@@ -3,21 +3,32 @@
 from dataclasses import replace
 from typing import List, Any
 
-from agents import (
-    RunConfig,
-    RunResult,
-    RunResultStreaming,
-    Runner as AgentsRunner,
-)
-from agents.agent import Agent
-from agents.models.interface import ModelProvider
-from agents.items import TResponseInputItem
-from agents.lifecycle import RunHooks
-from agents.memory import Session
-from agents.run import DEFAULT_MAX_TURNS
-from agents.run import AgentRunner
-from agents.run_context import TContext
-from agents.models.interface import Model
+try:
+    from agents import (
+        RunConfig,
+        RunResult,
+        RunResultStreaming,
+        Runner as AgentsRunner,
+    )
+    from agents.agent import Agent
+    from agents.models.interface import ModelProvider
+    from agents.items import TResponseInputItem
+    from agents.lifecycle import RunHooks
+    from agents.memory import Session
+    from agents.run import DEFAULT_MAX_TURNS
+    from agents.run import AgentRunner
+    from agents.run_context import TContext
+    from agents.models.interface import Model
+    agents_available = True
+except:
+    agents_available = False
+
+def is_agents_available():
+    if not agents_available:
+        raise ImportError(
+            "agents is required for this integration. Install it via your package manager"
+        )
+
 
 from deepeval.tracing.tracing import Observer
 from deepeval.tracing.context import current_span_context, current_trace_context
@@ -57,8 +68,8 @@ def patched_get_model(cls, agent: Agent[Any], run_config: RunConfig) -> Model:
     AgentRunner._get_model = patched_get_model
     _PATCHED_DEFAULT_GET_MODEL = True
 
-
-_patch_default_agent_runner_get_model()
+if agents_available:
+    _patch_default_agent_runner_get_model()
 
 class Runner(AgentsRunner):
 
@@ -85,7 +96,7 @@ async def run(
         user_id: str | None = None,
         **kwargs, # backwards compatibility
     ) -> RunResult:
-        
+        is_agents_available()
         # _patch_default_agent_runner_get_model()
 
         with Observer(
@@ -154,6 +165,7 @@ def run_sync(
         user_id: str | None = None,
         **kwargs,
     ) -> RunResult:
+        is_agents_available()
         input_val = input
 
         update_trace_attributes(
@@ -223,7 +235,7 @@ def run_streamed(
         user_id: str | None = None,
         **kwargs, # backwards compatibility
     ) -> RunResultStreaming:
-
+        is_agents_available()
         # Manually enter observer; we'll exit when streaming finishes
         observer = Observer(
             span_type="custom",

tests/test_integrations/test_openai_agents/multi_agents.py

@@ -1,5 +1,5 @@
-from deepeval.openai_agents import Agent, Runner
 import asyncio
+from deepeval.openai_agents import Agent, Runner
 from deepeval.prompt import Prompt
 from deepeval.openai_agents import DeepEvalTracingProcessor
 
@@ -27,15 +27,18 @@
     instructions="Handoff to the appropriate agent based on the language of the request.",
     handoffs=[spanish_agent, english_agent],
 )
-
 async def main():
-    result = await Runner.run(
-        triage_agent, input="Hola, ¿cómo estás?",
-        metric_collection="test_collection_1",
-        tags=["test"],
-        thread_id="test",
-    )
-
-    print(result.final_output)
-
-asyncio.run(main())
+    # result = await Runner.run(
+    #     triage_agent, input="Hola, ¿cómo estás?",
+    #     metric_collection="test_collection_1",
+    #     tags=["test"],
+    #     thread_id="test",
+    # )
+    
+    runner = Runner()
+    result = runner.run_streamed(triage_agent, "Hola, ¿cómo estás?", metric_collection="test_collection_1", thread_id="test")
+    async for chunk in result.stream_events():
+        print(chunk, end="", flush=True)
+        print("="*50)
+
+# asyncio.run(main())

tests/test_integrations/test_openai_agents/streaming.py

@@ -10,4 +10,4 @@ async def main():
         print(chunk, end="", flush=True)
         print("="*50)
 
-asyncio.run(main())
+# asyncio.run(main())