mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
3446eb8d5d
* updates to final deprecated pieces and versions * fix mypy * fix readme links
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
|
|
from agent_framework import Message
|
|
from agent_framework.foundry import FoundryChatClient
|
|
from azure.identity import AzureCliCredential
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
"""
|
|
Chat Response Cancellation Example
|
|
|
|
Demonstrates proper cancellation of streaming chat responses during execution.
|
|
Shows asyncio task cancellation and resource cleanup techniques.
|
|
"""
|
|
|
|
|
|
async def main() -> None:
|
|
"""
|
|
Demonstrates cancelling a chat request after 1 second.
|
|
Creates a task for the chat request, waits briefly, then cancels it to show proper cleanup.
|
|
|
|
Configuration:
|
|
- OpenAI model ID: Use "model" parameter or "OPENAI_MODEL" environment variable
|
|
- OpenAI API key: Use "api_key" parameter or "OPENAI_API_KEY" environment variable
|
|
"""
|
|
client = FoundryChatClient(credential=AzureCliCredential())
|
|
|
|
try:
|
|
task = asyncio.create_task(
|
|
client.get_response(messages=[Message(role="user", contents=["Tell me a fantasy story."])])
|
|
)
|
|
await asyncio.sleep(1)
|
|
task.cancel()
|
|
await task
|
|
except asyncio.CancelledError:
|
|
print("Request was cancelled")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|