mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
f48c4512d3
* updated automation tasks and commands, with alias for the time being * Restore aggregate test exclusions Preserve the legacy all-tests scope for test --all by excluding lab and devui from the default aggregate sweep, while still allowing explicit package selection. Also ignore hidden/generated test directories such as .mypy_cache during aggregate discovery. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * updated versions in pre-commit --------- Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
|
|
from agent_framework import Message
|
|
from agent_framework.openai import OpenAIChatClient
|
|
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_id" parameter or "OPENAI_CHAT_MODEL_ID" environment variable
|
|
- OpenAI API key: Use "api_key" parameter or "OPENAI_API_KEY" environment variable
|
|
"""
|
|
client = OpenAIChatClient()
|
|
|
|
try:
|
|
task = asyncio.create_task(
|
|
client.get_response(messages=[Message(role="user", text="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())
|