add max turns in REPL (#2431)

Author: xju2

src/agents/repl.py

@@ -7,13 +7,17 @@
 from .agent import Agent
 from .items import TResponseInputItem
 from .result import RunResultBase
-from .run import Runner
+from .run import DEFAULT_MAX_TURNS, Runner
 from .run_context import TContext
 from .stream_events import AgentUpdatedStreamEvent, RawResponsesStreamEvent, RunItemStreamEvent
 
 
 async def run_demo_loop(
-    agent: Agent[Any], *, stream: bool = True, context: TContext | None = None
+    agent: Agent[Any],
+    *,
+    stream: bool = True,
+    context: TContext | None = None,
+    max_turns: int = DEFAULT_MAX_TURNS,
 ) -> None:
     """Run a simple REPL loop with the given agent.
 
@@ -25,6 +29,7 @@ async def run_demo_loop(
         agent: The starting agent to run.
         stream: Whether to stream the agent output.
         context: Additional context information to pass to the runner.
+        max_turns: Maximum number of turns for the runner to iterate.
     """
 
     current_agent = agent
@@ -44,7 +49,9 @@ async def run_demo_loop(
 
         result: RunResultBase
         if stream:
-            result = Runner.run_streamed(current_agent, input=input_items, context=context)
+            result = Runner.run_streamed(
+                current_agent, input=input_items, context=context, max_turns=max_turns
+            )
             async for event in result.stream_events():
                 if isinstance(event, RawResponsesStreamEvent):
                     if isinstance(event.data, ResponseTextDeltaEvent):
@@ -58,7 +65,9 @@ async def run_demo_loop(
                     print(f"\n[Agent updated: {event.new_agent.name}]", flush=True)
             print()
         else:
-            result = await Runner.run(current_agent, input_items, context=context)
+            result = await Runner.run(
+                current_agent, input_items, context=context, max_turns=max_turns
+            )
             if result.final_output is not None:
                 print(result.final_output)