chore: import upstream snapshot with attribution

python/samples/04-hosting/durabletask/02_multi_agent/client.py

@@ -0,0 +1,120 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+"""Client application for interacting with multiple hosted agents.
+
+This client connects to the Durable Task Scheduler and interacts with two different
+agents (WeatherAgent and MathAgent), demonstrating how to work with multiple agents
+each with their own specialized capabilities and tools.
+
+Prerequisites:
+- The worker must be running with both agents registered
+- Set AZURE_OPENAI_ENDPOINT and AZURE_OPENAI_MODEL when running the worker
+- Sign in with Azure CLI for AzureCliCredential authentication
+- Durable Task Scheduler must be running
+"""
+
+import asyncio
+import logging
+import os
+
+from agent_framework.azure import DurableAIAgentClient
+from azure.identity import AzureCliCredential
+from dotenv import load_dotenv
+from durabletask.azuremanaged.client import DurableTaskSchedulerClient
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+def get_client(
+    taskhub: str | None = None, endpoint: str | None = None, log_handler: logging.Handler | None = None
+) -> DurableAIAgentClient:
+    """Create a configured DurableAIAgentClient.
+
+    Args:
+        taskhub: Task hub name (defaults to TASKHUB env var or "default")
+        endpoint: Scheduler endpoint (defaults to ENDPOINT env var or "http://localhost:8080")
+        log_handler: Optional logging handler for client logging
+
+    Returns:
+        Configured DurableAIAgentClient instance
+    """
+    taskhub_name = taskhub or os.getenv("TASKHUB", "default")
+    endpoint_url = endpoint or os.getenv("ENDPOINT", "http://localhost:8080")
+
+    logger.debug(f"Using taskhub: {taskhub_name}")
+    logger.debug(f"Using endpoint: {endpoint_url}")
+
+    credential = None if endpoint_url == "http://localhost:8080" else AzureCliCredential()
+
+    dts_client = DurableTaskSchedulerClient(
+        host_address=endpoint_url,
+        secure_channel=endpoint_url != "http://localhost:8080",
+        taskhub=taskhub_name,
+        token_credential=credential,
+        log_handler=log_handler,
+    )
+
+    return DurableAIAgentClient(dts_client)
+
+
+def run_client(agent_client: DurableAIAgentClient) -> None:
+    """Run client interactions with both WeatherAgent and MathAgent.
+
+    Args:
+        agent_client: The DurableAIAgentClient instance
+    """
+    logger.debug("Testing WeatherAgent")
+
+    # Get reference to WeatherAgent
+    weather_agent = agent_client.get_agent("WeatherAgent")
+    weather_session = weather_agent.create_session()
+
+    logger.debug(f"Created weather conversation session: {weather_session.session_id}")
+
+    # Test WeatherAgent
+    weather_message = "What is the weather in Seattle?"
+    logger.info(f"User: {weather_message}")
+
+    weather_response = weather_agent.run(weather_message, session=weather_session)
+    logger.info(f"WeatherAgent: {weather_response.text} \n")
+
+    logger.debug("Testing MathAgent")
+
+    # Get reference to MathAgent
+    math_agent = agent_client.get_agent("MathAgent")
+    math_session = math_agent.create_session()
+
+    logger.debug(f"Created math conversation session: {math_session.session_id}")
+
+    # Test MathAgent
+    math_message = "Calculate a 20% tip on a $50 bill"
+    logger.info(f"User: {math_message}")
+
+    math_response = math_agent.run(math_message, session=math_session)
+    logger.info(f"MathAgent: {math_response.text} \n")
+
+    logger.debug("Both agents completed successfully!")
+
+
+async def main() -> None:
+    """Main entry point for the client application."""
+    logger.debug("Starting Durable Task Multi-Agent Client...")
+
+    # Create client using helper function
+    agent_client = get_client()
+
+    try:
+        run_client(agent_client)
+    except Exception as e:
+        logger.exception(f"Error during agent interaction: {e}")
+    finally:
+        logger.debug("Client shutting down")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())