mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
3a49b1d6dd
* [BREAKING] Remove deprecated Python OpenAI/Azure AI surfaces Also clean up follow-on docs, environment guidance, package metadata, and lab test stability. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Fix deleted semantic-kernel sample links Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Address PR review feedback Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * improve foundry language * Fix A2A Foundry sample regression Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --------- Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
# ruff: noqa: T201
|
|
|
|
import asyncio
|
|
import os
|
|
|
|
from agent_framework.foundry import FoundryChatClient
|
|
from azure.identity.aio import AzureCliCredential
|
|
from dotenv import load_dotenv
|
|
|
|
from agent_framework_azure_cosmos import CosmosHistoryProvider
|
|
|
|
# Load environment variables from .env file.
|
|
load_dotenv()
|
|
|
|
"""
|
|
This sample demonstrates CosmosHistoryProvider as an agent context provider.
|
|
|
|
Key components:
|
|
- FoundryChatClient configured with an Azure AI project endpoint
|
|
- CosmosHistoryProvider configured for Cosmos DB-backed message history
|
|
- Provider-configured container name with session_id as partition key
|
|
|
|
Environment variables:
|
|
FOUNDRY_PROJECT_ENDPOINT
|
|
FOUNDRY_MODEL
|
|
AZURE_COSMOS_ENDPOINT
|
|
AZURE_COSMOS_DATABASE_NAME
|
|
AZURE_COSMOS_CONTAINER_NAME
|
|
Optional:
|
|
AZURE_COSMOS_KEY
|
|
"""
|
|
|
|
|
|
async def main() -> None:
|
|
"""Run the Cosmos history provider sample with an Agent."""
|
|
project_endpoint = os.getenv("FOUNDRY_PROJECT_ENDPOINT")
|
|
deployment_name = os.getenv("FOUNDRY_MODEL")
|
|
cosmos_endpoint = os.getenv("AZURE_COSMOS_ENDPOINT")
|
|
cosmos_database_name = os.getenv("AZURE_COSMOS_DATABASE_NAME")
|
|
cosmos_container_name = os.getenv("AZURE_COSMOS_CONTAINER_NAME")
|
|
cosmos_key = os.getenv("AZURE_COSMOS_KEY")
|
|
|
|
if (
|
|
not project_endpoint
|
|
or not deployment_name
|
|
or not cosmos_endpoint
|
|
or not cosmos_database_name
|
|
or not cosmos_container_name
|
|
):
|
|
print(
|
|
"Please set FOUNDRY_PROJECT_ENDPOINT, FOUNDRY_MODEL, "
|
|
"AZURE_COSMOS_ENDPOINT, AZURE_COSMOS_DATABASE_NAME, and AZURE_COSMOS_CONTAINER_NAME."
|
|
)
|
|
return
|
|
|
|
# 1. Create an Azure credential and Foundry chat client using project endpoint auth.
|
|
async with AzureCliCredential() as credential:
|
|
client = FoundryChatClient(
|
|
project_endpoint=project_endpoint,
|
|
model=deployment_name,
|
|
credential=credential,
|
|
)
|
|
|
|
# 2. Create an agent that uses the history provider as a context provider.
|
|
async with (
|
|
CosmosHistoryProvider(
|
|
endpoint=cosmos_endpoint,
|
|
database_name=cosmos_database_name,
|
|
container_name=cosmos_container_name,
|
|
credential=cosmos_key or credential,
|
|
) as history_provider,
|
|
client.as_agent(
|
|
name="CosmosHistoryAgent",
|
|
instructions="You are a helpful assistant that remembers prior turns.",
|
|
context_providers=[history_provider],
|
|
default_options={"store": False},
|
|
) as agent,
|
|
):
|
|
# 3. Create a session (session_id is used as the partition key).
|
|
session = agent.create_session()
|
|
|
|
# 4. Run a multi-turn conversation; history is persisted by CosmosHistoryProvider.
|
|
response1 = await agent.run("My name is Ada and I enjoy distributed systems.", session=session)
|
|
print(f"Assistant: {response1.text}")
|
|
|
|
response2 = await agent.run("What do you remember about me?", session=session)
|
|
print(f"Assistant: {response2.text}")
|
|
print(f"Container: {history_provider.container_name}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|
|
|
|
"""
|
|
Sample output:
|
|
Assistant: Nice to meet you, Ada! Distributed systems are a fascinating area.
|
|
Assistant: You told me your name is Ada and that you enjoy distributed systems.
|
|
Container: <AZURE_COSMOS_CONTAINER_NAME>
|
|
"""
|