agent-openai-agents-sdk: improve startup script

agent-openai-agents-sdk/agent_server/agent.py

@@ -41,7 +41,7 @@ def create_coding_agent(mcp_server: McpServer) -> Agent:
 
 
 @invoke()
-async def invoke(request: ResponsesAgentRequest) -> ResponsesAgentResponse:
+async def invoke_handler(request: ResponsesAgentRequest) -> ResponsesAgentResponse:
     # Optionally use the user's workspace client for on-behalf-of authentication
     # user_workspace_client = get_user_workspace_client()
     async with await init_mcp_server() as mcp_server:
@@ -52,7 +52,7 @@ async def invoke(request: ResponsesAgentRequest) -> ResponsesAgentResponse:
 
 
 @stream()
-async def stream(request: dict) -> AsyncGenerator[ResponsesAgentStreamEvent, None]:
+async def stream_handler(request: dict) -> AsyncGenerator[ResponsesAgentStreamEvent, None]:
     # Optionally use the user's workspace client for on-behalf-of authentication
     # user_workspace_client = get_user_workspace_client()
     async with await init_mcp_server() as mcp_server:

agent-openai-agents-sdk/agent_server/app.py

@@ -0,0 +1,28 @@
+from fastapi.responses import StreamingResponse
+from aiohttp.web import Request
+from mlflow.genai.agent_server import AgentServer
+
+
+class MlflowAgentServer(AgentServer):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._initialized = False
+
+    async def _handle_invocations_request(self, *args, **kwargs):
+        if not self._initialized:
+            self._initialized = True
+            from mlflow.genai.agent_server import setup_mlflow_git_based_version_tracking
+            setup_mlflow_git_based_version_tracking()
+        return await super()._handle_invocations_request(*args, **kwargs)
+
+
+def create_app():
+    # Need to import the agent to register the functions with the server
+    import agent_server.agent
+    from dotenv import load_dotenv
+
+    # Load env vars from .env before importing the agent for proper auth
+    load_dotenv(dotenv_path=".env", override=True)
+
+    server = MlflowAgentServer("ResponsesAgent", enable_chat_proxy=True)
+    return server.app

agent-openai-agents-sdk/app.yaml

@@ -1,4 +1,4 @@
-command: ["uv", "run", "start-app"]
+command: ["python", "scripts/start_app.py"]
 # databricks apps listen by default on port 8000
 
 env:

agent-openai-agents-sdk/pyproject.toml

@@ -9,7 +9,7 @@ authors = [
 requires-python = ">=3.11"
 dependencies = [
     "fastapi>=0.115.12",
-    "uvicorn>=0.34.2",
+    "uvicorn[standard]>=0.34.2",
     "databricks-openai>=0.9.0",
     "mlflow>=3.9.0",
     "openai-agents>=0.4.1",
@@ -33,3 +33,6 @@ start-app = "scripts.start_app:main"
 start-server = "agent_server.start_server:main"
 agent-evaluate = "agent_server.evaluate_agent:evaluate"
 discover-tools = "scripts.discover_tools:main"
+
+[tool.uv.pip]
+output-file = "requirements.txt"