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

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: