Changing the architecture of the AI Foundry Agent cell

Author: nourshaker-msft

labs/mcp-from-api/mcp-from-api.ipynb

@@ -278,77 +278,81 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Copyright (c) Microsoft. All rights reserved.\n",
-    "\n",
-    "import asyncio\n",
-    "\n",
-    "from azure.ai.agents.models import McpTool\n",
-    "from azure.identity.aio import DefaultAzureCredential\n",
-    "\n",
-    "from semantic_kernel.agents import AzureAIAgent, AzureAIAgentSettings, AzureAIAgentThread\n",
-    "from semantic_kernel.contents import ChatMessageContent, FunctionCallContent, FunctionResultContent\n",
-    "\n",
-    "TASK = \"What's the current weather in Lisbon?\"\n",
-    "\n",
-    "async def handle_intermediate_messages(message: ChatMessageContent) -> None:\n",
-    "    for item in message.items or []:\n",
-    "        if isinstance(item, FunctionResultContent):\n",
-    "            print(f\"Function Result:> {item.result} for function: {item.name}\")\n",
-    "        elif isinstance(item, FunctionCallContent):\n",
-    "            print(f\"Function Call:> {item.name} with arguments: {item.arguments}\")\n",
-    "        else:\n",
-    "            print(f\"{item}\")\n",
-    "\n",
-    "\n",
-    "async def main() -> None:\n",
-    "    async with (\n",
-    "        DefaultAzureCredential() as creds,\n",
-    "        AzureAIAgent.create_client(credential=creds) as client,\n",
-    "    ):\n",
-    "        # 1. Define the MCP tool with the server URL\n",
-    "        mcp_tool = McpTool(\n",
-    "            server_label=\"weather\",\n",
-    "            server_url=f\"{apim_resource_gateway_url}/weather-mcp/mcp\",\n",
-    "            allowed_tools=[],  # Specify allowed tools if needed\n",
-    "        )\n",
-    "\n",
-    "        # Optionally you may configure to require approval\n",
-    "        # Allowed values are \"never\" or \"always\"\n",
-    "        mcp_tool.set_approval_mode(\"never\")\n",
-    "\n",
-    "        agent_settings = AzureAIAgentSettings(\n",
-    "            model_deployment_name=models_config[0]['name']\n",
-    "        )\n",
-    "\n",
-    "        # 2. Create an agent with the MCP tool on the Azure AI agent service\n",
-    "        agent_definition = await client.agents.create_agent(\n",
-    "            model=agent_settings.model_deployment_name,\n",
-    "            tools=mcp_tool.definitions,\n",
-    "            instructions=\"You are a helpful agent that can use MCP tools to assist users.\",\n",
-    "        )\n",
-    "\n",
-    "        # 3. Create a Semantic Kernel agent for the Azure AI agent\n",
-    "        agent = AzureAIAgent(\n",
-    "            client=client,\n",
-    "            definition=agent_definition,\n",
-    "        )\n",
-    "\n",
-    "        # 4. Create a thread for the agent\n",
-    "        # If no thread is provided, a new thread will be\n",
-    "        # created and returned with the initial response\n",
-    "        thread: AzureAIAgentThread | None = None\n",
-    "\n",
-    "        print(f\"# User: '{TASK}'\")\n",
-    "        # 5. Invoke the agent for the specified thread for response\n",
-    "        async for response in agent.invoke(\n",
-    "            messages=TASK, thread=thread, on_intermediate_message=handle_intermediate_messages\n",
-    "        ):\n",
-    "            print(f\"# Agent: {response}\")\n",
-    "\n",
-    "\n",
-    "\n",
-    "if __name__ == \"__main__\":\n",
-    "    asyncio.run(main())"
+    "from azure.ai.agents.models import ListSortOrder, MessageTextContent, McpTool, RequiredMcpToolCall, SubmitToolApprovalAction, ToolApproval\n",
+    "from azure.ai.projects import AIProjectClient\n",
+    "from azure.identity import DefaultAzureCredential\n",
+    "import time\n",
+    "\n",
+    "project_client = AIProjectClient(endpoint=foundry_project_endpoint,\n",
+    "            credential=DefaultAzureCredential())\n",
+    "agents_client = project_client.agents\n",
+    "\n",
+    "# MCP tool definition\n",
+    "mcp_tool = McpTool(\n",
+    "    server_label=\"weather\",\n",
+    "    server_url=f\"{apim_resource_gateway_url}/weather-mcp/sse\",\n",
+    "    #allowed_tools=[],          # Optional initial allow‑list\n",
+    ")\n",
+    "\n",
+    "#mcp_tool.update_headers({\"Authorization\": f\"Bearer {apim_subscription_key}\"})\n",
+    "\n",
+    "prompt = \"What's the weather in Lisbon, Cairo and London?\"\n",
+    "\n",
+    "# Agent creation\n",
+    "agent = agents_client.create_agent(\n",
+    "    model=str(models_config[0].get('name')),\n",
+    "    name=\"agent-mcp\",\n",
+    "    instructions=\"You are a sarcastic AI agent. Use the tools provided to answer the user's questions. Be sure to cite your sources and answer in details.\",\n",
+    "    tools=mcp_tool.definitions\n",
+    ")\n",
+    "print(f\"🎉 Created agent, agent ID: {agent.id}\")\n",
+    "print(f\"✨ MCP Server: {mcp_tool.server_label} at {mcp_tool.server_url}\")\n",
+    "\n",
+    "# Thread creation\n",
+    "thread = agents_client.threads.create()\n",
+    "print(f\"🧵 Created thread, thread ID: {thread.id}\")\n",
+    "\n",
+    "# Message creation\n",
+    "message = agents_client.messages.create(\n",
+    "    thread_id=thread.id,\n",
+    "    role=\"user\",\n",
+    "    content=prompt,\n",
+    ")\n",
+    "print(f\"💬 Created message, message ID: {message.id}\")\n",
+    "\n",
+    "mcp_tool.set_approval_mode(\"never\")          # Disable human approval\n",
+    "\n",
+    "# Run\n",
+    "run = agents_client.runs.create(thread_id=thread.id, agent_id=agent.id, tool_resources=mcp_tool.resources)\n",
+    "while run.status in [\"queued\", \"in_progress\", \"requires_action\"]:\n",
+    "    time.sleep(2)\n",
+    "    run = agents_client.runs.get(thread_id=thread.id, run_id=run.id)\n",
+    "    print(f\"⏳ Run status: {run.status}\")\n",
+    "if run.status == \"failed\":\n",
+    "    print(f\"❌ Run error: {run.last_error}\")\n",
+    "\n",
+    "# Get Run steps\n",
+    "run_steps = agents_client.run_steps.list(thread_id=thread.id, run_id=run.id)\n",
+    "print()\n",
+    "\n",
+    "for step in run_steps:\n",
+    "    print(f\"🔄 Run step: {step.id}, status: {step.status}, type: {step.type}\")\n",
+    "    if step.type == \"tool_calls\":\n",
+    "        print(f\"🛠️ Tool call details:\")\n",
+    "        for tool_call in step.step_details.tool_calls:\n",
+    "            print(json.dumps(tool_call.as_dict(), indent=5))\n",
+    "\n",
+    "# Get the messages in the thread\n",
+    "print(\"\\n📜 Messages in the thread:\")\n",
+    "messages = agents_client.messages.list(thread_id=thread.id, order=ListSortOrder.ASCENDING)\n",
+    "\n",
+    "for item in messages:\n",
+    "    last_message_content = item.content[-1]\n",
+    "    if isinstance(last_message_content, MessageTextContent):\n",
+    "        print(f\"🗨️ {item.role}: {last_message_content.text.value}\")\n",
+    "\n",
+    "# Clean up resources\n",
+    "# agents_client.delete_agent(agent.id)\n"
    ]
   },
   {
@@ -516,7 +520,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": ".venv",
+   "display_name": "myenv",
    "language": "python",
    "name": "python3"
   },
@@ -530,7 +534,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.10"
+   "version": "3.13.5"
   }
  },
  "nbformat": 4,