From ee7bf067794c35b8e7c486a74b03833b92921b53 Mon Sep 17 00:00:00 2001
From: eavanvalkenburg <github@vanvalkenburg.eu>
Date: Fri, 13 Feb 2026 16:09:51 +0100
Subject: [PATCH] Align Python hosting get-started sample with Azure Functions

---
 .../01-get-started/06_host_your_agent.py      | 64 +++++++------------
 python/samples/01-get-started/README.md       |  2 +-
 2 files changed, 24 insertions(+), 42 deletions(-)

diff --git a/python/samples/01-get-started/06_host_your_agent.py b/python/samples/01-get-started/06_host_your_agent.py
index 916944cdb6..7f80607dde 100644
--- a/python/samples/01-get-started/06_host_your_agent.py
+++ b/python/samples/01-get-started/06_host_your_agent.py
@@ -1,60 +1,42 @@
 # Copyright (c) Microsoft. All rights reserved.
 
-import asyncio
-import os
+"""Host your agent with Azure Functions.
 
-from agent_framework.azure import AzureOpenAIResponsesClient
-from azure.identity import AzureCliCredential
-
-"""
-Host Your Agent — Minimal A2A hosting stub
-
-This sample shows the pattern for exposing an agent via the Agent-to-Agent
-(A2A) protocol. It creates the agent and demonstrates how to wrap it with
-the A2A hosting layer.
+This sample shows the Python hosting pattern used in docs:
+- Create an agent with `AzureOpenAIChatClient`
+- Register it with `AgentFunctionApp`
+- Run with Azure Functions Core Tools (`func start`)
 
 Prerequisites:
-  pip install agent-framework[a2a] --pre
+  pip install agent-framework-azurefunctions --pre
 
 Environment variables:
-  AZURE_AI_PROJECT_ENDPOINT        — Your Azure AI Foundry project endpoint
-  AZURE_OPENAI_RESPONSES_DEPLOYMENT_NAME — Model deployment name (e.g. gpt-4o)
-
-To run a full A2A server, see samples/04-hosting/a2a/ for a complete example.
+  AZURE_OPENAI_ENDPOINT
+  AZURE_OPENAI_CHAT_DEPLOYMENT_NAME
 """
 
+from typing import Any
+
+from agent_framework.azure import AgentFunctionApp, AzureOpenAIChatClient
+from azure.identity import AzureCliCredential
 
-async def main() -> None:
-    # <create_agent>
-    credential = AzureCliCredential()
-    client = AzureOpenAIResponsesClient(
-        project_endpoint=os.environ["AZURE_AI_PROJECT_ENDPOINT"],
-        deployment_name=os.environ["AZURE_OPENAI_RESPONSES_DEPLOYMENT_NAME"],
-        credential=credential,
-    )
 
-    agent = client.as_agent(
+# <create_agent>
+def _create_agent() -> Any:
+    """Create a hosted agent backed by Azure OpenAI."""
+    return AzureOpenAIChatClient(credential=AzureCliCredential()).as_agent(
         name="HostedAgent",
-        instructions="You are a helpful assistant exposed via A2A.",
+        instructions="You are a helpful assistant hosted in Azure Functions.",
     )
-    # </create_agent>
 
-    # <host_agent>
-    # The A2A hosting integration wraps your agent behind an HTTP endpoint.
-    # Import is gated so this sample can run without the a2a extra installed.
-    try:
-        from agent_framework.a2a import A2AAgent  # noqa: F401
 
-        print("A2A support is available.")
-        print("See samples/04-hosting/a2a/ for a runnable A2A server example.")
-    except ImportError:
-        print("Install a2a extras: pip install agent-framework[a2a] --pre")
+# </create_agent>
 
-    # Quick smoke-test: run the agent locally to verify it works
-    result = await agent.run("Hello! What can you do?")
-    print(f"Agent: {result}")
-    # </host_agent>
+# <host_agent>
+app = AgentFunctionApp(agents=[_create_agent()], enable_health_check=True, max_poll_retries=50)
+# </host_agent>
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    print("Start the Functions host with: func start")
+    print("Then call: POST /api/agents/HostedAgent/run")
diff --git a/python/samples/01-get-started/README.md b/python/samples/01-get-started/README.md
index f13eeb26c5..774368e15f 100644
--- a/python/samples/01-get-started/README.md
+++ b/python/samples/01-get-started/README.md
@@ -25,7 +25,7 @@ export OPENAI_RESPONSES_MODEL_ID="gpt-4o"   # optional, defaults to gpt-4o
 | 3 | [03_multi_turn.py](03_multi_turn.py) | Keep conversation history across turns with `AgentThread`. |
 | 4 | [04_memory.py](04_memory.py) | Add dynamic context with a custom `ContextProvider`. |
 | 5 | [05_first_workflow.py](05_first_workflow.py) | Chain executors into a workflow with edges. |
-| 6 | [06_host_your_agent.py](06_host_your_agent.py) | Prepare your agent for A2A hosting. |
+| 6 | [06_host_your_agent.py](06_host_your_agent.py) | Host a single agent with Azure Functions. |
 
 Run any sample with: