Python: [BREAKING] Python: move Azure AI embeddings to Foundry (#5056)

* renamed AzureAIINferenceEmbeddings and lazy load azure-cosmos and env var rename

* updated coverage

* fix readme

python/packages/core/README.md

@@ -34,8 +34,8 @@ FOUNDRY_PROJECT_ENDPOINT=...
 FOUNDRY_MODEL=...
 ...
 OPENAI_API_KEY=sk-...
+OPENAI_CHAT_COMPLETION_MODEL=...
 OPENAI_CHAT_MODEL=...
-OPENAI_RESPONSES_MODEL=...
 ...
 AZURE_OPENAI_API_KEY=...
 AZURE_OPENAI_ENDPOINT=...

python/packages/core/agent_framework/azure/__init__.py

@@ -14,13 +14,7 @@ _IMPORTS: dict[str, tuple[str, str]] = {
     "AgentResponseCallbackProtocol": ("agent_framework_durabletask", "agent-framework-durabletask"),
     "AzureAISearchContextProvider": ("agent_framework_azure_ai_search", "agent-framework-azure-ai-search"),
     "AzureAISearchSettings": ("agent_framework_azure_ai_search", "agent-framework-azure-ai-search"),
-    "AzureAISettings": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
-    "AzureAIInferenceEmbeddingClient": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
-    "AzureAIInferenceEmbeddingOptions": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
-    "AzureAIInferenceEmbeddingSettings": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
-    "RawAzureAIInferenceEmbeddingClient": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
-    "AzureCredentialTypes": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
-    "AzureTokenProvider": ("agent_framework_azure_ai", "agent-framework-azure-ai"),
+    "CosmosHistoryProvider": ("agent_framework_azure_cosmos", "agent-framework-azure-cosmos"),
     "DurableAIAgent": ("agent_framework_durabletask", "agent-framework-durabletask"),
     "DurableAIAgentClient": ("agent_framework_durabletask", "agent-framework-durabletask"),
     "DurableAIAgentOrchestrationContext": ("agent_framework_durabletask", "agent-framework-durabletask"),

python/packages/core/agent_framework/azure/__init__.pyi

@@ -3,19 +3,11 @@
 # Type stubs for the agent_framework.azure lazy-loading namespace.
 # Install the relevant packages for full type support.
 
-from agent_framework_azure_ai import (
-    AzureAIInferenceEmbeddingClient,
-    AzureAIInferenceEmbeddingOptions,
-    AzureAIInferenceEmbeddingSettings,
-    AzureAISettings,
-    AzureCredentialTypes,
-    AzureTokenProvider,
-    RawAzureAIInferenceEmbeddingClient,
-)
 from agent_framework_azure_ai_search import (
     AzureAISearchContextProvider,
     AzureAISearchSettings,
 )
+from agent_framework_azure_cosmos import CosmosHistoryProvider
 from agent_framework_azurefunctions import AgentFunctionApp
 from agent_framework_durabletask import (
     AgentCallbackContext,
@@ -30,17 +22,11 @@ __all__ = [
     "AgentCallbackContext",
     "AgentFunctionApp",
     "AgentResponseCallbackProtocol",
-    "AzureAIInferenceEmbeddingClient",
-    "AzureAIInferenceEmbeddingOptions",
-    "AzureAIInferenceEmbeddingSettings",
     "AzureAISearchContextProvider",
     "AzureAISearchSettings",
-    "AzureAISettings",
-    "AzureCredentialTypes",
-    "AzureTokenProvider",
+    "CosmosHistoryProvider",
     "DurableAIAgent",
     "DurableAIAgentClient",
     "DurableAIAgentOrchestrationContext",
     "DurableAIAgentWorker",
-    "RawAzureAIInferenceEmbeddingClient",
 ]

python/packages/core/agent_framework/foundry/__init__.py

@@ -16,6 +16,9 @@ _IMPORTS: dict[str, tuple[str, str]] = {
     "FoundryAgent": ("agent_framework_foundry", "agent-framework-foundry"),
     "FoundryChatClient": ("agent_framework_foundry", "agent-framework-foundry"),
     "FoundryChatOptions": ("agent_framework_foundry", "agent-framework-foundry"),
+    "FoundryEmbeddingClient": ("agent_framework_foundry", "agent-framework-foundry"),
+    "FoundryEmbeddingOptions": ("agent_framework_foundry", "agent-framework-foundry"),
+    "FoundryEmbeddingSettings": ("agent_framework_foundry", "agent-framework-foundry"),
     "FoundryEvals": ("agent_framework_foundry", "agent-framework-foundry"),
     "FoundryMemoryProvider": ("agent_framework_foundry", "agent-framework-foundry"),
     "FoundryLocalChatOptions": ("agent_framework_foundry_local", "agent-framework-foundry-local"),
@@ -25,6 +28,7 @@ _IMPORTS: dict[str, tuple[str, str]] = {
     "RawFoundryAgent": ("agent_framework_foundry", "agent-framework-foundry"),
     "RawFoundryAgentChatClient": ("agent_framework_foundry", "agent-framework-foundry"),
     "RawFoundryChatClient": ("agent_framework_foundry", "agent-framework-foundry"),
+    "RawFoundryEmbeddingClient": ("agent_framework_foundry", "agent-framework-foundry"),
     "evaluate_foundry_target": ("agent_framework_foundry", "agent-framework-foundry"),
     "evaluate_traces": ("agent_framework_foundry", "agent-framework-foundry"),
 }

python/packages/core/agent_framework/foundry/__init__.pyi

@@ -8,11 +8,15 @@ from agent_framework_foundry import (
     FoundryAgent,
     FoundryChatClient,
     FoundryChatOptions,
+    FoundryEmbeddingClient,
+    FoundryEmbeddingOptions,
+    FoundryEmbeddingSettings,
     FoundryEvals,
     FoundryMemoryProvider,
     RawFoundryAgent,
     RawFoundryAgentChatClient,
     RawFoundryChatClient,
+    RawFoundryEmbeddingClient,
     evaluate_foundry_target,
     evaluate_traces,
 )
@@ -27,6 +31,9 @@ __all__ = [
     "FoundryAgent",
     "FoundryChatClient",
     "FoundryChatOptions",
+    "FoundryEmbeddingClient",
+    "FoundryEmbeddingOptions",
+    "FoundryEmbeddingSettings",
     "FoundryEvals",
     "FoundryLocalChatOptions",
     "FoundryLocalClient",
@@ -36,6 +43,7 @@ __all__ = [
     "RawFoundryAgent",
     "RawFoundryAgentChatClient",
     "RawFoundryChatClient",
+    "RawFoundryEmbeddingClient",
     "evaluate_foundry_target",
     "evaluate_traces",
 ]

python/packages/core/agent_framework/microsoft/__init__.py

@@ -5,7 +5,6 @@
 This module lazily re-exports objects from:
 - ``agent-framework-copilotstudio``
 - ``agent-framework-purview``
-- ``agent-framework-foundry-local``
 
 Supported classes:
 - CopilotStudioAgent
@@ -20,9 +19,6 @@ Supported classes:
 - PurviewRequestError
 - PurviewServiceError
 - CacheProvider
-- FoundryLocalChatOptions
-- FoundryLocalClient
-- FoundryLocalSettings
 
 """
 
@@ -43,9 +39,6 @@ _IMPORTS: dict[str, tuple[str, str]] = {
     "PurviewRequestError": ("agent_framework_purview", "agent-framework-purview"),
     "PurviewServiceError": ("agent_framework_purview", "agent-framework-purview"),
     "CacheProvider": ("agent_framework_purview", "agent-framework-purview"),
-    "FoundryLocalChatOptions": ("agent_framework_foundry_local", "agent-framework-foundry-local"),
-    "FoundryLocalClient": ("agent_framework_foundry_local", "agent-framework-foundry-local"),
-    "FoundryLocalSettings": ("agent_framework_foundry_local", "agent-framework-foundry-local"),
 }
 
 

python/packages/core/agent_framework/microsoft/__init__.pyi

@@ -4,11 +4,6 @@ from agent_framework_copilotstudio import (
     CopilotStudioAgent,
     acquire_token,
 )
-from agent_framework_foundry_local import (
-    FoundryLocalChatOptions,
-    FoundryLocalClient,
-    FoundryLocalSettings,
-)
 from agent_framework_purview import (
     CacheProvider,
     PurviewAppLocation,
@@ -26,9 +21,6 @@ from agent_framework_purview import (
 __all__ = [
     "CacheProvider",
     "CopilotStudioAgent",
-    "FoundryLocalChatOptions",
-    "FoundryLocalClient",
-    "FoundryLocalSettings",
     "PurviewAppLocation",
     "PurviewAuthenticationError",
     "PurviewChatPolicyMiddleware",

python/packages/core/pyproject.toml

@@ -35,10 +35,10 @@ all = [
     "agent-framework-a2a",
     "agent-framework-ag-ui",
     "agent-framework-azure-ai-search",
+    "agent-framework-azure-cosmos",
     "agent-framework-anthropic",
     "agent-framework-openai",
     "agent-framework-claude",
-    "agent-framework-azure-ai",
     "agent-framework-azurefunctions",
     "agent-framework-bedrock",
     "agent-framework-chatkit",

python/packages/core/tests/core/test_azure_namespace.py

@@ -0,0 +1,10 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+from agent_framework_azure_cosmos import CosmosHistoryProvider
+
+import agent_framework.azure as azure
+
+
+def test_azure_namespace_exposes_cosmos_history_provider() -> None:
+    assert azure.CosmosHistoryProvider is CosmosHistoryProvider
+    assert "CosmosHistoryProvider" in dir(azure)