Python: Create/Get Agent API for Azure V2

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

@@ -4,6 +4,7 @@
 
 from ._chat_client import AzureAIAgentClient, AzureAIAgentOptions
 from ._client import AzureAIClient
+from ._provider import AzureAIProjectAgentProvider
 from ._shared import AzureAISettings
 
 try:
@@ -15,6 +16,7 @@
     "AzureAIAgentClient",
     "AzureAIAgentOptions",
     "AzureAIClient",
+    "AzureAIProjectAgentProvider",
     "AzureAISettings",
     "__version__",
 ]

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

@@ -2,7 +2,7 @@
 
 import sys
 from collections.abc import Mapping, MutableSequence
-from typing import TYPE_CHECKING, Any, ClassVar, Generic, TypedDict, cast
+from typing import TYPE_CHECKING, Any, ClassVar, Generic, TypedDict, TypeVar, cast
 
 from agent_framework import (
     AGENT_FRAMEWORK_USER_AGENT,
@@ -13,23 +13,20 @@
     use_chat_middleware,
     use_function_invocation,
 )
-from agent_framework.exceptions import ServiceInitializationError, ServiceInvalidRequestError
+from agent_framework.exceptions import ServiceInitializationError
 from agent_framework.observability import use_instrumentation
 from agent_framework.openai._responses_client import OpenAIBaseResponsesClient
 from azure.ai.projects.aio import AIProjectClient
 from azure.ai.projects.models import (
     MCPTool,
     PromptAgentDefinition,
     PromptAgentDefinitionText,
-    ResponseTextFormatConfigurationJsonObject,
-    ResponseTextFormatConfigurationJsonSchema,
-    ResponseTextFormatConfigurationText,
 )
 from azure.core.credentials_async import AsyncTokenCredential
 from azure.core.exceptions import ResourceNotFoundError
-from pydantic import BaseModel, ValidationError
+from pydantic import ValidationError
 
-from ._shared import AzureAISettings
+from ._shared import AzureAISettings, create_text_format_config
 
 if TYPE_CHECKING:
     from agent_framework.openai import OpenAIResponsesOptions
@@ -286,47 +283,6 @@ async def close(self) -> None:
         """Close the project_client."""
         await self._close_client_if_needed()
 
-    def _create_text_format_config(
-        self, response_format: type[BaseModel] | Mapping[str, Any]
-    ) -> (
-        ResponseTextFormatConfigurationJsonSchema
-        | ResponseTextFormatConfigurationJsonObject
-        | ResponseTextFormatConfigurationText
-    ):
-        """Convert response_format into Azure text format configuration."""
-        if isinstance(response_format, type) and issubclass(response_format, BaseModel):
-            schema = response_format.model_json_schema()
-            # Ensure additionalProperties is explicitly false to satisfy Azure validation
-            if isinstance(schema, dict):
-                schema.setdefault("additionalProperties", False)
-            return ResponseTextFormatConfigurationJsonSchema(
-                name=response_format.__name__,
-                schema=schema,
-            )
-
-        if isinstance(response_format, Mapping):
-            format_config = self._convert_response_format(response_format)
-            format_type = format_config.get("type")
-            if format_type == "json_schema":
-                # Ensure schema includes additionalProperties=False to satisfy Azure validation
-                schema = dict(format_config.get("schema", {}))  # type: ignore[assignment]
-                schema.setdefault("additionalProperties", False)
-                config_kwargs: dict[str, Any] = {
-                    "name": format_config.get("name") or "response",
-                    "schema": schema,
-                }
-                if "strict" in format_config:
-                    config_kwargs["strict"] = format_config["strict"]
-                if "description" in format_config:
-                    config_kwargs["description"] = format_config["description"]
-                return ResponseTextFormatConfigurationJsonSchema(**config_kwargs)
-            if format_type == "json_object":
-                return ResponseTextFormatConfigurationJsonObject()
-            if format_type == "text":
-                return ResponseTextFormatConfigurationText()
-
-        raise ServiceInvalidRequestError("response_format must be a Pydantic model or mapping.")
-
     async def _get_agent_reference_or_create(
         self,
         run_options: dict[str, Any],
@@ -380,7 +336,7 @@ async def _get_agent_reference_or_create(
             # response_format is accessed from chat_options or additional_properties
             # since the base class excludes it from run_options
             if chat_options and (response_format := chat_options.get("response_format")):
-                args["text"] = PromptAgentDefinitionText(format=self._create_text_format_config(response_format))
+                args["text"] = PromptAgentDefinitionText(format=create_text_format_config(response_format))
 
             # Combine instructions from messages and options
             combined_instructions = [