mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Giles Odigwe
7c2dae8855
* Fix sample bugs: incorrect API params, wrong client types, and invalid options - typed_options.py: Fix AnthropicClient model->model_id, wrap raw strings in Message objects for get_response(), fix reasoning_effort->reasoning dict, fix budget_tokens minimum (1024), use OpenAIChatClient not FoundryChatClient, remove unused import - client_reasoning.py: Fix deprecated model_id to model param - client_with_hosted_mcp.py: Remove invalid store=True kwarg from Agent.run() - code_defined_skill.py: Fix precision kwarg to use function_invocation_kwargs - Various other samples: Fix deprecated API usage and incorrect params Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Address PR review comments - client_with_hosted_mcp.py: Fix remaining store=True kwarg on line 68 to use options dict - client_with_session.py: Change store=True to store=False to match in-memory persistence demo intent - typed_options.py: Remove non-existent import and model key from docstring example Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * new sample fixes --------- Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
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_id" 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", 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())
|