Files
Eduard van Valkenburg 1e350ea22f Python: [BREAKING] PR2 — Wire context provider pipeline, remove old types, update all consumers (#3850)
* PR2: Wire context provider pipeline and update all internal consumers

- Replace AgentThread with AgentSession across all packages
- Replace ContextProvider with BaseContextProvider across all packages
- Replace context_provider param with context_providers (Sequence)
- Replace thread= with session= in run() signatures
- Replace get_new_thread() with create_session()
- Add get_session(service_session_id) to agent interface
- DurableAgentThread -> DurableAgentSession
- Remove _notify_thread_of_new_messages from WorkflowAgent
- Wire before_run/after_run context provider pipeline in RawAgent
- Auto-inject InMemoryHistoryProvider when no providers configured

* fix: update all tests for context provider pipeline, fix lazy-loaders, remove old test files

* refactor: update all sample files for context provider pipeline (AgentThread→AgentSession, ContextProvider→BaseContextProvider)

* fix: update remaining ag-ui references (client docstring, getting_started sample)

* fix: make get_session service_session_id keyword-only to avoid confusion with session_id

* refactor: rename _RunContext.thread_messages to session_messages

* refactor: remove _threads.py, _memory.py, and old provider files; migrate devui to use plain message lists

* rename: remove _new_ prefix from test files

* refactor: rewrite SlidingWindowChatMessageStore as SlidingWindowHistoryProvider(InMemoryHistoryProvider)

* fix: read full history from session state directly instead of reaching into provider internals

* fix: update stale .pyi stubs, sample imports, and README references for new provider types

* fix: remove stale message_store, _notify_thread_of_new_messages, and session_id.key references in samples

* refactor: merge context_providers and sessions sample folders into sessions, remove aggregate_context_provider

* refactor: UserInfoMemory stores state in session.state instead of instance attributes

* feat: add Pydantic BaseModel support to session state serialization

Pydantic models stored in session.state are now automatically serialized
via model_dump() and restored via model_validate() during to_dict()/from_dict()
round-trips. Models are auto-registered on first serialization; use
register_state_type() for cold-start deserialization.

Also export register_state_type as a public API.

* fix mem0

* Update sample README links and descriptions for session terminology

- Replace 'thread' with 'session' in sample descriptions across all READMEs
- Update file links for renamed samples (mem0_sessions, redis_sessions, etc.)
- Fix Threads section → Sessions section in main samples/README.md
- Update tools, middleware, workflows, durabletask, azure_functions READMEs
- Update architecture diagrams in concepts/tools/README.md
- Update migration guides (autogen, semantic-kernel)

* Fix broken Redis README link to renamed sample

* Fix Mem0 OSS client search: pass scoping params as direct kwargs

AsyncMemory (OSS) expects user_id/agent_id/run_id as direct kwargs,
while AsyncMemoryClient (Platform) expects them in a filters dict.
Adds tests for both client types.

Port of fix from #3844 to new Mem0ContextProvider.

* Fix rebase issues: restore missing _conversation_state.py and checkpoint decode logic

- Add back _conversation_state.py (encode/decode_chat_messages) lost in rebase
- Fix on_checkpoint_restore to decode cache/conversation with decode_chat_messages
- Fix on_checkpoint_restore to use decode_checkpoint_value for pending requests
- Add tests/workflow/__init__.py for relative import support
- Fix test_agent_executor checkpoint selection (checkpoints[1] not superstep)

* Add STORES_BY_DEFAULT ClassVar to skip redundant InMemoryHistoryProvider injection

Chat clients that store history server-side by default (OpenAI Responses API,
Azure AI Agent) now declare STORES_BY_DEFAULT = True. The agent checks this
during auto-injection and skips InMemoryHistoryProvider unless the user
explicitly sets store=False.

* Fix broken markdown links in azure_ai and redis READMEs

* Fix getting-started samples to use session API instead of removed thread/ContextProvider API

* updates to workflow as agent

* fix group chat import

* Rename Thread→Session throughout, fix service_session_id propagation, remove stale AGUIThread

- Fix: Propagate conversation_id from ChatResponse back to session.service_session_id
  in both streaming and non-streaming paths in _agents.py
- Rename AgentThreadException → AgentSessionException
- Remove stale AGUIThread from ag_ui lazy-loader
- Rename use_service_thread → use_service_session in ag-ui package
- Rename test functions from *_thread_* to *_session_*
- Rename sample files from *_thread* to *_session*
- Update docstrings and comments: thread → session
- Update _mcp.py kwargs filter: add 'session' alongside 'thread'
- Fix ContinuationToken docstring example: thread=thread → session=session
- Fix _clients.py docstring: 'Agent threads' → 'Agent sessions'

* Fix broken markdown links after thread→session file renames

* fix azure ai test
2026-02-12 21:00:32 +00:00

8.5 KiB

Azure AI Agent Examples

This folder contains examples demonstrating different ways to create and use agents with Azure AI using the AzureAIAgentsProvider from the agent_framework.azure package. These examples use the azure-ai-agents 1.x (V1) API surface. For updated V2 (azure-ai-projects 2.x) samples, see the Azure AI V2 examples folder.

Provider Pattern

All examples in this folder use the AzureAIAgentsProvider class which provides a high-level interface for agent operations:

  • create_agent() - Create a new agent on the Azure AI service
  • get_agent() - Retrieve an existing agent by ID or from a pre-fetched Agent object
  • as_agent() - Wrap an SDK Agent object as a Agent without HTTP calls
from agent_framework.azure import AzureAIAgentsProvider
from azure.identity.aio import AzureCliCredential

async with (
    AzureCliCredential() as credential,
    AzureAIAgentsProvider(credential=credential) as provider,
):
    agent = await provider.create_agent(
        name="MyAgent",
        instructions="You are a helpful assistant.",
        tools=my_function,
    )
    result = await agent.run("Hello!")

Examples

File Description
azure_ai_provider_methods.py Comprehensive example demonstrating all AzureAIAgentsProvider methods: create_agent(), get_agent(), as_agent(), and managing multiple agents from a single provider.
azure_ai_basic.py The simplest way to create an agent using AzureAIAgentsProvider. It automatically handles all configuration using environment variables. Shows both streaming and non-streaming responses.
azure_ai_with_bing_custom_search.py Shows how to use Bing Custom Search with Azure AI agents to find real-time information from the web using custom search configurations. Demonstrates how to use AzureAIAgentClient.get_web_search_tool() with custom search instances.
azure_ai_with_bing_grounding.py Shows how to use Bing Grounding search with Azure AI agents to find real-time information from the web. Demonstrates AzureAIAgentClient.get_web_search_tool() with proper source citations and comprehensive error handling.
azure_ai_with_bing_grounding_citations.py Demonstrates how to extract and display citations from Bing Grounding search responses. Shows how to collect citation annotations (title, URL, snippet) during streaming responses, enabling users to verify sources and access referenced content.
azure_ai_with_code_interpreter_file_generation.py Shows how to retrieve file IDs from code interpreter generated files using both streaming and non-streaming approaches.
azure_ai_with_code_interpreter.py Shows how to use AzureAIAgentClient.get_code_interpreter_tool() with Azure AI agents to write and execute Python code. Includes helper methods for accessing code interpreter data from response chunks.
azure_ai_with_existing_agent.py Shows how to work with an existing SDK Agent object using provider.as_agent(). This wraps the agent without making HTTP calls.
azure_ai_with_existing_session.py Shows how to work with a pre-existing session by providing the session ID. Demonstrates proper cleanup of manually created sessions.
azure_ai_with_explicit_settings.py Shows how to create an agent with explicitly configured provider settings, including project endpoint and model deployment name.
azure_ai_with_azure_ai_search.py Demonstrates how to use Azure AI Search with Azure AI agents. Shows how to create an agent with search tools using the SDK directly and wrap it with provider.get_agent().
azure_ai_with_file_search.py Demonstrates how to use AzureAIAgentClient.get_file_search_tool() with Azure AI agents to search through uploaded documents. Shows file upload, vector store creation, and querying document content.
azure_ai_with_function_tools.py Demonstrates how to use function tools with agents. Shows both agent-level tools (defined when creating the agent) and query-level tools (provided with specific queries).
azure_ai_with_hosted_mcp.py Shows how to use AzureAIAgentClient.get_mcp_tool() with hosted Model Context Protocol (MCP) servers for enhanced functionality and tool integration. Demonstrates remote MCP server connections and tool discovery.
azure_ai_with_local_mcp.py Shows how to integrate Azure AI agents with local Model Context Protocol (MCP) servers for enhanced functionality and tool integration. Demonstrates both agent-level and run-level tool configuration.
azure_ai_with_multiple_tools.py Demonstrates how to use multiple tools together with Azure AI agents, including web search, MCP servers, and function tools using client static methods. Shows coordinated multi-tool interactions and approval workflows.
azure_ai_with_openapi_tools.py Demonstrates how to use OpenAPI tools with Azure AI agents to integrate external REST APIs. Shows OpenAPI specification loading, anonymous authentication, session context management, and coordinated multi-API conversations.
azure_ai_with_response_format.py Demonstrates how to use structured outputs with Azure AI agents using Pydantic models.
azure_ai_with_session.py Demonstrates session management with Azure AI agents, including automatic session creation for stateless conversations and explicit session management for maintaining conversation context across multiple interactions.

Environment Variables

Before running the examples, you need to set up your environment variables. You can do this in one of two ways:

  1. Copy the .env.example file from the python directory to create a .env file:

    cp ../../.env.example ../../.env
    
  2. Edit the .env file and add your values:

    AZURE_AI_PROJECT_ENDPOINT="your-project-endpoint"
    AZURE_AI_MODEL_DEPLOYMENT_NAME="your-model-deployment-name"
    
  3. For samples using Bing Grounding search (like azure_ai_with_bing_grounding.py and azure_ai_with_multiple_tools.py), you'll also need:

    BING_CONNECTION_ID="your-bing-connection-id"
    

    To get your Bing connection details:

    • Go to Azure AI Foundry portal
    • Navigate to your project's "Connected resources" section
    • Add a new connection for "Grounding with Bing Search"
    • Copy the ID
  4. For samples using Bing Custom Search (like azure_ai_with_bing_custom_search.py), you'll also need:

    BING_CUSTOM_CONNECTION_ID="your-bing-custom-connection-id"
    BING_CUSTOM_INSTANCE_NAME="your-bing-custom-instance-name"
    

    To get your Bing Custom Search connection details:

    • Go to Azure AI Foundry portal
    • Navigate to your project's "Connected resources" section
    • Add a new connection for "Grounding with Bing Custom Search"
    • Copy the connection ID and instance name

Option 2: Using environment variables directly

Set the environment variables in your shell:

export AZURE_AI_PROJECT_ENDPOINT="your-project-endpoint"
export AZURE_AI_MODEL_DEPLOYMENT_NAME="your-model-deployment-name"
export BING_CONNECTION_ID="your-bing-connection-id"
export BING_CUSTOM_CONNECTION_ID="your-bing-custom-connection-id"
export BING_CUSTOM_INSTANCE_NAME="your-bing-custom-instance-name"

Required Variables

  • AZURE_AI_PROJECT_ENDPOINT: Your Azure AI project endpoint (required for all examples)
  • AZURE_AI_MODEL_DEPLOYMENT_NAME: The name of your model deployment (required for all examples)

Optional Variables

  • BING_CONNECTION_ID: Your Bing connection ID (required for azure_ai_with_bing_grounding.py and azure_ai_with_multiple_tools.py)
  • BING_CUSTOM_CONNECTION_ID: Your Bing Custom Search connection ID (required for azure_ai_with_bing_custom_search.py)
  • BING_CUSTOM_INSTANCE_NAME: Your Bing Custom Search instance name (required for azure_ai_with_bing_custom_search.py)