Move activity_as_tool function to DurableAIAgentContext class

Author: AnatoliB

samples-v2/openai_agents/durable_ai_agent_context.py

@@ -1,7 +1,16 @@
 import json
+from typing import Any, Callable, Optional
+
 from azure.durable_functions.models.DurableOrchestrationContext import DurableOrchestrationContext
 from yield_exception import YieldException
 
+from agents import (
+    RunContextWrapper,
+    Tool,
+)
+from agents.function_schema import function_schema
+from agents.tool import FunctionTool
+
 class DurableAIAgentContext:
     def __init__(self, context: DurableOrchestrationContext):
         self.context = context
@@ -42,3 +51,42 @@ def yield_and_clear_tasks(self):
         for task in self.tasks_to_yield:
             yield task
         self.tasks_to_yield.clear()
+
+    def activity_as_tool(
+        self,
+        activity_func: Callable,
+        *,
+        description: Optional[str] = None,
+    ) -> Tool:
+        """
+        Convert an Azure Durable Functions activity to an OpenAI Agents SDK Tool.
+
+        Args:
+            activity_func: The Azure Functions activity function to convert
+
+        Returns:
+            Tool: An OpenAI Agents SDK Tool object
+        """
+
+        activity_name = activity_func._function._name
+
+        async def run_activity(ctx: RunContextWrapper[Any], input: str) -> Any:
+            result = self.get_activity_call_result(activity_name, input)
+            return result
+
+        schema = function_schema(
+            func=activity_func._function._func,
+            name_override=activity_name,
+            docstring_style=None,
+            description_override=description,
+            use_docstring_info=True,
+            strict_json_schema=False,
+        )
+
+        return FunctionTool(
+            name=schema.name,
+            description=schema.description or "",
+            params_json_schema=schema.params_json_schema,
+            on_invoke_tool=run_activity,
+            strict_json_schema=False,
+        )

samples-v2/openai_agents/function_app.py

@@ -7,7 +7,6 @@
 from agents.model_settings import ModelSettings
 from durable_decorators import durable_openai_agent_orchestrator
 from azure.identity import DefaultAzureCredential
-from tools import activity_as_tool
 from pydantic import BaseModel
 
 
@@ -126,7 +125,7 @@ def weather_expert(context):
         name="Hello world",
         instructions="You are a helpful agent.",
         tools=[
-            activity_as_tool(context, get_weather)
+            context.activity_as_tool(get_weather)
         ],
     )