Skip to content

Python: Alternative fix for #1546 - Add comprehensive tests and fix CI issues for Azure AI Search Context Provider #2247

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
hayato-kishikawa wants to merge 6 commits into from
Closed

Python: Alternative fix for #1546 - Add comprehensive tests and fix CI issues for Azure AI Search Context Provider #2247

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
dc7b011
Python: Fix pyright errors and move search provider to core (#1546)
Oct 29, 2025
8461562
address pablo coments
Nov 2, 2025
9c47994
fix(azure): remove DEFAULT_CONTEXT_PROMPT override to fix mypy error
Nov 15, 2025
01b662f
test(azure): improve search provider coverage to 84%
Nov 15, 2025
e6b496a
test(azure): add vector auto-discovery tests for 91% coverage
Nov 15, 2025
968acb3
test(azure): achieve 98% test coverage for search provider
Nov 15, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions python/.env.example
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,14 @@ AZURE_AI_PROJECT_ENDPOINT=""
AZURE_AI_MODEL_DEPLOYMENT_NAME=""
# Bing connection for web search (optional, used by samples with web search)
BING_CONNECTION_ID=""
# Azure AI Search (optional, used by AzureAISearchContextProvider samples)
AZURE_SEARCH_ENDPOINT=""
AZURE_SEARCH_API_KEY=""
AZURE_SEARCH_INDEX_NAME=""
AZURE_SEARCH_SEMANTIC_CONFIG=""
AZURE_SEARCH_KNOWLEDGE_BASE_NAME=""
# Note: For agentic mode Knowledge Bases, also set AZURE_OPENAI_ENDPOINT below
# (different from AZURE_AI_PROJECT_ENDPOINT - Knowledge Base needs OpenAI endpoint for model calls)
# OpenAI
OPENAI_API_KEY=""
OPENAI_CHAT_MODEL_ID=""
Expand Down
1 change: 1 addition & 0 deletions python/packages/core/agent_framework/azure/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
"AgentResponseCallbackProtocol": ("agent_framework_azurefunctions", "azurefunctions"),
"AzureAIAgentClient": ("agent_framework_azure_ai", "azure-ai"),
"AzureAIClient": ("agent_framework_azure_ai", "azure-ai"),
"AzureAISearchContextProvider": ("agent_framework.azure._search_provider", "core"),
"AzureOpenAIAssistantsClient": ("agent_framework.azure._assistants_client", "core"),
"AzureOpenAIChatClient": ("agent_framework.azure._chat_client", "core"),
"AzureAISettings": ("agent_framework_azure_ai", "azure-ai"),
Expand Down
631 changes: 631 additions & 0 deletions python/packages/core/agent_framework/azure/_search_provider.py
View file
Open in desktop

Large diffs are not rendered by default.

1 change: 1 addition & 0 deletions python/packages/core/pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ dependencies = [
"azure-identity>=1,<2",
"mcp[ws]>=1.13",
"packaging>=24.1",
"azure-search-documents>=11.7.0b1",
]

[project.optional-dependencies]
Expand Down
Loading