Addressed PR feedback

python/packages/azure-ai/pyproject.toml

@@ -24,7 +24,7 @@ classifiers = [
 ]
 dependencies = [
     "agent-framework-core",
-    "azure-ai-projects == 2.0.0b1",
+    "azure-ai-projects >= 2.0.0b1",
     "azure-ai-agents == 1.2.0b5",
     "aiohttp",
 ]

python/packages/azure-ai/tests/test_azure_ai_client.py

@@ -691,10 +691,6 @@ def test_get_conversation_id_with_parsed_response_and_store_true() -> None:
     """Test get_conversation_id works with ParsedResponse when store is True."""
     client = create_test_azure_ai_client(MagicMock())
 
-    # Create a simple BaseModel for testing
-    class TestModel(BaseModel):
-        content: str = "test"
-
     # Mock ParsedResponse with conversation
     mock_response = MagicMock(spec=ParsedResponse[BaseModel])
     mock_response.id = "resp_parsed_12345"
@@ -711,10 +707,6 @@ def test_get_conversation_id_with_parsed_response_no_conversation() -> None:
     """Test get_conversation_id returns response ID with ParsedResponse when no conversation exists."""
     client = create_test_azure_ai_client(MagicMock())
 
-    # Create a simple BaseModel for testing
-    class TestModel(BaseModel):
-        content: str = "test"
-
     # Mock ParsedResponse without conversation
     mock_response = MagicMock(spec=ParsedResponse[BaseModel])
     mock_response.id = "resp_parsed_12345"

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