Addressed PR feedback

python/samples/getting_started/agents/azure_ai/azure_ai_with_explicit_settings.py

@@ -45,7 +45,7 @@ async def main() -> None:
     ):
         query = "What's the weather like in New York?"
         print(f"User: {query}")
-        result = await agent.run("What's the weather like in New York?")
+        result = await agent.run(query)
         print(f"Agent: {result}\n")
 
 

python/samples/getting_started/agents/azure_ai/azure_ai_with_hosted_mcp.py

@@ -3,7 +3,7 @@
 import asyncio
 from typing import Any
 
-from agent_framework import AgentProtocol, AgentThread, ChatMessage, HostedMCPTool
+from agent_framework import AgentProtocol, AgentRunResponse, AgentThread, ChatMessage, HostedMCPTool
 from agent_framework.azure import AzureAIClient
 from azure.identity.aio import AzureCliCredential
 
@@ -14,7 +14,7 @@ This sample demonstrates integrating hosted Model Context Protocol (MCP) tools w
 """
 
 
-async def handle_approvals_without_thread(query: str, agent: "AgentProtocol"):
+async def handle_approvals_without_thread(query: str, agent: "AgentProtocol") -> AgentRunResponse:
     """When we don't have a thread, we need to ensure we return with the input, approval request and approval."""
 
     result = await agent.run(query, store=False)
@@ -35,7 +35,7 @@ async def handle_approvals_without_thread(query: str, agent: "AgentProtocol"):
     return result
 
 
-async def handle_approvals_with_thread(query: str, agent: "AgentProtocol", thread: "AgentThread"):
+async def handle_approvals_with_thread(query: str, agent: "AgentProtocol", thread: "AgentThread") -> AgentRunResponse:
     """Here we let the thread deal with the previous responses, and we just rerun with the approval."""
 
     result = await agent.run(query, thread=thread)

python/samples/getting_started/agents/azure_ai_agent/azure_ai_with_file_search.py

@@ -4,7 +4,7 @@ import asyncio
 from pathlib import Path
 
 from agent_framework import ChatAgent, HostedFileSearchTool, HostedVectorStoreContent
-from agent_framework_azure_ai import AzureAIAgentClient
+from agent_framework.azure import AzureAIAgentClient
 from azure.ai.agents.models import FileInfo, VectorStore
 from azure.identity.aio import AzureCliCredential
 

python/samples/getting_started/agents/azure_ai_agent/azure_ai_with_hosted_mcp.py

@@ -3,7 +3,7 @@
 import asyncio
 from typing import Any
 
-from agent_framework import AgentProtocol, AgentThread, HostedMCPTool
+from agent_framework import AgentProtocol, AgentRunResponse, AgentThread, HostedMCPTool
 from agent_framework.azure import AzureAIAgentClient
 from azure.identity.aio import AzureCliCredential
 
@@ -15,7 +15,7 @@ servers, including user approval workflows for function call security.
 """
 
 
-async def handle_approvals_with_thread(query: str, agent: "AgentProtocol", thread: "AgentThread"):
+async def handle_approvals_with_thread(query: str, agent: "AgentProtocol", thread: "AgentThread") -> AgentRunResponse:
     """Here we let the thread deal with the previous responses, and we just rerun with the approval."""
     from agent_framework import ChatMessage