diff --git a/python/samples/getting_started/agents/openai_assistants_client/openai_assistants_with_file_search.py b/python/samples/getting_started/agents/openai_assistants_client/openai_assistants_with_file_search.py index 0840b7e1eb..dfb802ca38 100644 --- a/python/samples/getting_started/agents/openai_assistants_client/openai_assistants_with_file_search.py +++ b/python/samples/getting_started/agents/openai_assistants_client/openai_assistants_with_file_search.py @@ -1,11 +1,13 @@ # Copyright (c) Microsoft. All rights reserved. import asyncio + from agent_framework import ChatClientAgent, HostedFileSearchTool, HostedVectorStoreContent from agent_framework.openai import OpenAIAssistantsClient # Helper functions + async def create_vector_store(client: OpenAIAssistantsClient) -> tuple[str, HostedVectorStoreContent]: """Create a vector store with sample documents.""" file = await client.client.files.create( @@ -42,12 +44,12 @@ async def main() -> None: print(f"User: {query}") print("Agent: ", end="", flush=True) async for chunk in agent.run_streaming( - query, - tool_resources={"file_search": {"vector_store_ids": [vector_store.vector_store_id]}} + query, tool_resources={"file_search": {"vector_store_ids": [vector_store.vector_store_id]}} ): if chunk.text: print(chunk.text, end="", flush=True) await delete_vector_store(client, file_id, vector_store.vector_store_id) + if __name__ == "__main__": asyncio.run(main()) diff --git a/python/samples/getting_started/agents/openai_chat_client/openai_chat_client_with_web_search.py b/python/samples/getting_started/agents/openai_chat_client/openai_chat_client_with_web_search.py index e2ba52474d..f83ca14aba 100644 --- a/python/samples/getting_started/agents/openai_chat_client/openai_chat_client_with_web_search.py +++ b/python/samples/getting_started/agents/openai_chat_client/openai_chat_client_with_web_search.py @@ -1,9 +1,11 @@ # Copyright (c) Microsoft. All rights reserved. import asyncio + from agent_framework import HostedWebSearchTool from agent_framework.openai import OpenAIChatClient + async def main() -> None: client = OpenAIChatClient(ai_model_id="gpt-4o-search-preview") @@ -35,5 +37,6 @@ async def main() -> None: ) print(f"Assistant: {response}") + if __name__ == "__main__": asyncio.run(main()) diff --git a/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_file_search.py b/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_file_search.py index 369aae1d43..a41f6ac29c 100644 --- a/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_file_search.py +++ b/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_file_search.py @@ -1,11 +1,13 @@ # Copyright (c) Microsoft. All rights reserved. import asyncio + from agent_framework import HostedFileSearchTool, HostedVectorStoreContent from agent_framework.openai import OpenAIResponsesClient # Helper functions + async def create_vector_store(client: OpenAIResponsesClient) -> tuple[str, HostedVectorStoreContent]: """Create a vector store with sample documents.""" file = await client.client.files.create( @@ -56,5 +58,6 @@ async def main() -> None: print(f"Assistant: {response}") await delete_vector_store(client, file_id, vector_store.vector_store_id) + if __name__ == "__main__": asyncio.run(main()) diff --git a/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_web_search.py b/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_web_search.py index b6c3511039..bfe0e1086c 100644 --- a/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_web_search.py +++ b/python/samples/getting_started/agents/openai_responses_client/openai_responses_client_with_web_search.py @@ -1,9 +1,11 @@ # Copyright (c) Microsoft. All rights reserved. import asyncio + from agent_framework import HostedWebSearchTool from agent_framework.openai import OpenAIResponsesClient + async def main() -> None: client = OpenAIResponsesClient() @@ -35,5 +37,6 @@ async def main() -> None: ) print(f"Assistant: {response}") + if __name__ == "__main__": asyncio.run(main()) diff --git a/python/samples/getting_started/chat_client/chat_response_cancellation.py b/python/samples/getting_started/chat_client/chat_response_cancellation.py index 2c1239559e..fcd43d2f0b 100644 --- a/python/samples/getting_started/chat_client/chat_response_cancellation.py +++ b/python/samples/getting_started/chat_client/chat_response_cancellation.py @@ -4,7 +4,16 @@ import asyncio from agent_framework.openai import OpenAIChatClient -async def main(): + +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 "ai_model_id" parameter or "OPENAI_CHAT_MODEL_ID" environment variable + - OpenAI API key: Use "api_key" parameter or "OPENAI_API_KEY" environment variable + """ chat_client = OpenAIChatClient() try: @@ -15,5 +24,6 @@ async def main(): except asyncio.CancelledError: print("Request was cancelled") + if __name__ == "__main__": asyncio.run(main())