Substitute the agent runner with DurableOpenAIRunner

Author: AnatoliB

samples-v2/openai_agents/durable_openai_runner.py

@@ -0,0 +1,89 @@
+import asyncio
+import logging
+from typing import Any, Callable, Optional
+import azure.functions as func
+
+import json
+import typing
+from dataclasses import replace
+from typing import Any, Union
+
+from agents import (
+    Agent,
+    RunConfig,
+    RunResult,
+    RunResultStreaming,
+    SQLiteSession,
+    TContext,
+    Tool,
+    TResponseInputItem,
+)
+from agents.run import DEFAULT_AGENT_RUNNER, DEFAULT_MAX_TURNS, AgentRunner
+from pydantic_core import to_json
+
+logger = logging.getLogger(__name__)
+
+class DurableOpenAIRunner:
+    def __init__(self) -> None:
+        self._runner = DEFAULT_AGENT_RUNNER or AgentRunner()
+
+    def run_sync(
+        self,
+        starting_agent: Agent[TContext],
+        input: Union[str, list[TResponseInputItem]],
+        **kwargs: Any,
+    ) -> RunResult:
+        # workaround for https://github.com/pydantic/pydantic/issues/9541
+        # ValidatorIterator returned
+        input_json = to_json(input)
+        input = json.loads(input_json)
+
+        context = kwargs.get("context")
+        max_turns = kwargs.get("max_turns", DEFAULT_MAX_TURNS)
+        hooks = kwargs.get("hooks")
+        run_config = kwargs.get("run_config")
+        previous_response_id = kwargs.get("previous_response_id")
+        session = kwargs.get("session")
+
+        if run_config is None:
+            run_config = RunConfig()
+
+        model_name = run_config.model or starting_agent.model
+        if model_name is not None and not isinstance(model_name, str):
+            raise ValueError(
+                "Durable Functions require a model name to be a string in the run config and/or agent."
+            )
+        # updated_run_config = replace(
+        #     run_config,
+        #     model=_TemporalModelStub(
+        #         model_name=model_name
+        #     ),
+        # )
+        updated_run_config = run_config
+
+        return self._runner.run_sync(
+            starting_agent=starting_agent,
+            input=input,
+            context=context,
+            max_turns=max_turns,
+            hooks=hooks,
+            run_config=updated_run_config,
+            previous_response_id=previous_response_id,
+            session=session,
+        )
+
+    def run(
+        self,
+        starting_agent: Agent[TContext],
+        input: Union[str, list[TResponseInputItem]],
+        **kwargs: Any,
+    ) -> RunResult:
+        raise RuntimeError("Durable Functions do not support asynchronous runs.")
+
+    def run_streamed(
+        self,
+        starting_agent: Agent[TContext],
+        input: Union[str, list[TResponseInputItem]],
+        **kwargs: Any,
+    ) -> RunResultStreaming:
+        raise RuntimeError("Durable Functions do not support streaming.")

samples-v2/openai_agents/function_app.py

@@ -1,6 +1,11 @@
 import azure.functions as func
 import logging
 
+from agents.run import set_default_agent_runner
+from durable_openai_runner import DurableOpenAIRunner
+
+set_default_agent_runner(DurableOpenAIRunner())
+
 
 app = func.FunctionApp(http_auth_level=func.AuthLevel.FUNCTION)