mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
cc0cfaaac8
* Python: fix OpenAI Azure routing and provider samples Prefer OpenAI when OPENAI_API_KEY is present unless Azure is explicitly requested. Clarify constructor docs, keep deprecated Azure wrappers compatible with stricter settings validation, and refresh the provider samples and tests to use the current client patterns. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * fix bandit * Python: align OpenAI embedding Azure routing Extend the shared OpenAI-vs-Azure routing and credential behavior to the embedding client, add Azure embedding regression coverage, and refresh the embedding samples to use the generic client path. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: fix embedding client pyright check Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: thin OpenAI embedding wrapper Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: document embedding overload routing Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: fix callable OpenAI key routing Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: fix Azure credential routing tests Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: address OpenAI review feedback Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: narrow Azure routing markers Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: refine OpenAI model fallback order Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: narrow Azure deployment docs Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: remove embedding routing wording Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: run embedding Azure integration tests Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * changed variable name * Python: expand OpenAI package README Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * clarified readme * Python: fix Azure OpenAI integration setup Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: correct Azure integration env mapping Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * updated code to fix int tests * test updates * test fix * fix test setup * updates to tests and setup * remove openai assistants int tests * improvements in int tests * fix env var * fix env vars * fix azure responses test * trigger actions --------- Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
"""OpenAI Agent factory for GAIA benchmark.
|
|
|
|
This module provides a factory function to create an OpenAI agent
|
|
configured for GAIA benchmark tasks using the OpenAI Responses API.
|
|
|
|
Required Environment Variables:
|
|
OPENAI_API_KEY: Your OpenAI API key
|
|
OPENAI_RESPONSES_MODEL: Model to use with Responses API (e.g., gpt-4o, gpt-4o-mini)
|
|
|
|
Optional Environment Variables:
|
|
OPENAI_BASE_URL: Custom API base URL if using a proxy or compatible service
|
|
OPENAI_ORG_ID: Organization ID for OpenAI API (if applicable)
|
|
|
|
Authentication:
|
|
Uses OPENAI_API_KEY environment variable.
|
|
Get your API key from: https://platform.openai.com/api-keys
|
|
|
|
Example:
|
|
export OPENAI_API_KEY="sk-..."
|
|
export OPENAI_RESPONSES_MODEL="gpt-4o"
|
|
"""
|
|
|
|
from collections.abc import AsyncIterator
|
|
from contextlib import asynccontextmanager
|
|
|
|
from agent_framework import Agent
|
|
from agent_framework.openai import OpenAIResponsesClient
|
|
|
|
|
|
@asynccontextmanager
|
|
async def create_gaia_agent() -> AsyncIterator[Agent]:
|
|
"""Create an OpenAI agent configured for GAIA benchmark tasks.
|
|
|
|
Uses OpenAI Responses API for enhanced capabilities.
|
|
|
|
The agent is configured with:
|
|
- Web Search tool for information retrieval
|
|
- Code Interpreter tool for calculations and data analysis
|
|
|
|
Yields:
|
|
Agent: A configured agent ready to run GAIA tasks.
|
|
|
|
Example:
|
|
async with create_gaia_agent() as agent:
|
|
result = await agent.run("What is the capital of France?")
|
|
print(result.text)
|
|
"""
|
|
client = OpenAIResponsesClient()
|
|
|
|
async with client.as_agent(
|
|
name="GaiaAgent",
|
|
instructions="Solve tasks to your best ability. Use Web Search to find "
|
|
"information and Code Interpreter to perform calculations and data analysis.",
|
|
tools=[
|
|
OpenAIResponsesClient.get_web_search_tool(),
|
|
OpenAIResponsesClient.get_code_interpreter_tool(),
|
|
],
|
|
) as agent:
|
|
yield agent
|