Add max turns parameter & more oai tool support

- Also adds corresponding unit tests

Author: chakrris

src/agentex/lib/adk/providers/_modules/openai.py

@@ -88,6 +88,7 @@ async def run_agent(
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> SerializableRunResult | RunResult:
         """
         Run an agent without streaming or TaskMessage creation.
@@ -114,6 +115,7 @@ async def run_agent(
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
             input_guardrails: Optional list of input guardrails to run on initial user input.
             output_guardrails: Optional list of output guardrails to run on final agent output.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             Union[SerializableRunResult, RunResult]: SerializableRunResult when in Temporal, RunResult otherwise.
@@ -136,6 +138,7 @@ async def run_agent(
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
             return await ActivityHelpers.execute_activity(
                 activity_name=OpenAIActivityName.RUN_AGENT,
@@ -163,6 +166,7 @@ async def run_agent(
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
 
     async def run_agent_auto_send(
@@ -191,6 +195,7 @@ async def run_agent_auto_send(
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> SerializableRunResult | RunResult:
         """
         Run an agent with automatic TaskMessage creation.
@@ -216,6 +221,7 @@ async def run_agent_auto_send(
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
             input_guardrails: Optional list of input guardrails to run on initial user input.
             output_guardrails: Optional list of output guardrails to run on final agent output.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             Union[SerializableRunResult, RunResult]: SerializableRunResult when in Temporal, RunResult otherwise.
@@ -239,6 +245,7 @@ async def run_agent_auto_send(
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
             return await ActivityHelpers.execute_activity(
                 activity_name=OpenAIActivityName.RUN_AGENT_AUTO_SEND,
@@ -267,6 +274,7 @@ async def run_agent_auto_send(
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
 
     async def run_agent_streamed(
@@ -291,6 +299,7 @@ async def run_agent_streamed(
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> RunResultStreaming:
         """
         Run an agent with streaming enabled but no TaskMessage creation.
@@ -320,6 +329,7 @@ async def run_agent_streamed(
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
             input_guardrails: Optional list of input guardrails to run on initial user input.
             output_guardrails: Optional list of output guardrails to run on final agent output.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             RunResultStreaming: The result of the agent run with streaming.
@@ -352,6 +362,7 @@ async def run_agent_streamed(
             mcp_timeout_seconds=mcp_timeout_seconds,
             input_guardrails=input_guardrails,
             output_guardrails=output_guardrails,
+            max_turns=max_turns,
         )
 
     async def run_agent_streamed_auto_send(
@@ -380,6 +391,7 @@ async def run_agent_streamed_auto_send(
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> SerializableRunResultStreaming | RunResultStreaming:
         """
         Run an agent with streaming enabled and automatic TaskMessage creation.
@@ -405,6 +417,7 @@ async def run_agent_streamed_auto_send(
             output_type: Optional output type.
             tool_use_behavior: Optional tool use behavior.
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             Union[SerializableRunResultStreaming, RunResultStreaming]: SerializableRunResultStreaming when in Temporal, RunResultStreaming otherwise.
@@ -428,6 +441,7 @@ async def run_agent_streamed_auto_send(
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns
             )
             return await ActivityHelpers.execute_activity(
                 activity_name=OpenAIActivityName.RUN_AGENT_STREAMED_AUTO_SEND,
@@ -456,4 +470,5 @@ async def run_agent_streamed_auto_send(
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )