Create/Get Agent API - fixes and example improvements (#3246)

python/packages/azure-ai/agent_framework_azure_ai/_project_provider.py

@@ -358,6 +358,7 @@ class AzureAIProjectAgentProvider(Generic[TOptions_co]):
             agent_name=details.name,
             agent_version=details.version,
             agent_description=details.description,
+            model_deployment_name=details.definition.model,
         )
 
         # Merge tools: hosted tools from definition + user-provided function tools

python/packages/azure-ai/agent_framework_azure_ai/_shared.py

@@ -501,6 +501,7 @@ def create_text_format_config(
         return ResponseTextFormatConfigurationJsonSchema(
             name=response_format.__name__,
             schema=schema,
+            strict=True,
         )
 
     if isinstance(response_format, Mapping):

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

@@ -317,11 +317,21 @@ def test_provider_as_agent(mock_project_client: MagicMock) -> None:
     mock_agent_version.definition.top_p = 0.9
     mock_agent_version.definition.tools = []
 
-    agent = provider.as_agent(mock_agent_version)
+    with patch("agent_framework_azure_ai._project_provider.AzureAIClient") as mock_azure_ai_client:
+        agent = provider.as_agent(mock_agent_version)
 
-    assert isinstance(agent, ChatAgent)
-    assert agent.name == "test-agent"
-    assert agent.description == "Test Agent"
+        assert isinstance(agent, ChatAgent)
+        assert agent.name == "test-agent"
+        assert agent.description == "Test Agent"
+
+        # Verify AzureAIClient was called with correct parameters
+        mock_azure_ai_client.assert_called_once()
+        call_kwargs = mock_azure_ai_client.call_args[1]
+        assert call_kwargs["project_client"] is mock_project_client
+        assert call_kwargs["agent_name"] == "test-agent"
+        assert call_kwargs["agent_version"] == "1.0"
+        assert call_kwargs["agent_description"] == "Test Agent"
+        assert call_kwargs["model_deployment_name"] == "gpt-4"
 
 
 async def test_provider_context_manager(mock_project_client: MagicMock) -> None:
@@ -370,6 +380,24 @@ async def test_provider_close_method(mock_project_client: MagicMock) -> None:
             mock_client.close.assert_called_once()
 
 
+def test_create_text_format_config_sets_strict_for_pydantic_models() -> None:
+    """Test that create_text_format_config sets strict=True for Pydantic models."""
+    from pydantic import BaseModel
+
+    from agent_framework_azure_ai._shared import create_text_format_config
+
+    class TestSchema(BaseModel):
+        subject: str
+        summary: str
+
+    result = create_text_format_config(TestSchema)
+
+    # Verify strict=True is set
+    assert result["strict"] is True
+    assert result["name"] == "TestSchema"
+    assert "schema" in result
+
+
 @pytest.mark.flaky
 @skip_if_azure_ai_integration_tests_disabled
 async def test_provider_create_and_get_agent_integration() -> None:

python/packages/core/agent_framework/openai/_assistants_client.py

@@ -660,6 +660,7 @@ class OpenAIAssistantsClient(
                     "json_schema": {
                         "name": response_format.__name__,
                         "schema": response_format.model_json_schema(),
+                        "strict": True,
                     },
                 }
 

python/packages/core/tests/openai/test_openai_assistants_client.py

@@ -784,6 +784,27 @@ def test_prepare_options_with_mapping_tool(mock_async_openai: MagicMock) -> None
     assert run_options["tool_choice"] == "auto"
 
 
+def test_prepare_options_with_pydantic_response_format(mock_async_openai: MagicMock) -> None:
+    """Test _prepare_options sets strict=True for Pydantic response_format."""
+    from pydantic import BaseModel, ConfigDict
+
+    class TestResponse(BaseModel):
+        name: str
+        value: int
+        model_config = ConfigDict(extra="forbid")
+
+    chat_client = create_test_openai_assistants_client(mock_async_openai)
+    messages = [ChatMessage(role=Role.USER, text="Test")]
+    options = {"response_format": TestResponse}
+
+    run_options, _ = chat_client._prepare_options(messages, options)  # type: ignore
+
+    assert "response_format" in run_options
+    assert run_options["response_format"]["type"] == "json_schema"
+    assert run_options["response_format"]["json_schema"]["name"] == "TestResponse"
+    assert run_options["response_format"]["json_schema"]["strict"] is True
+
+
 def test_prepare_options_with_system_message(mock_async_openai: MagicMock) -> None:
     """Test _prepare_options with system message converted to instructions."""
     chat_client = create_test_openai_assistants_client(mock_async_openai)