wehub-resource-sync
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
"""
|
|
Minimal Agent API example.
|
|
|
|
Set the provider API key in your environment before running, for example:
|
|
|
|
export OPENAI_API_KEY="..."
|
|
python examples/agents/simple_agent.py
|
|
"""
|
|
|
|
import aisuite as ai
|
|
|
|
|
|
def get_weather(city: str) -> str:
|
|
"""Get the current weather for a city."""
|
|
return f"The weather in {city} is sunny."
|
|
|
|
|
|
def allow_safe_tools(context: ai.ToolPolicyContext) -> bool:
|
|
"""Allow only the tools this example expects."""
|
|
return context.tool_name in {"get_weather"}
|
|
|
|
|
|
agent = ai.Agent(
|
|
name="weather_assistant",
|
|
model="openai:gpt-4o",
|
|
instructions="Answer briefly. Use tools when they help.",
|
|
tools=[get_weather],
|
|
model_settings={"temperature": 0.2},
|
|
tags=["example", "weather"],
|
|
metadata={"app": "simple_agent_example"},
|
|
)
|
|
|
|
result = ai.Runner.run_sync(
|
|
agent,
|
|
"What is the weather in San Francisco?",
|
|
max_turns=3,
|
|
run_name="weather_lookup",
|
|
group_id="example_conversation_1",
|
|
metadata={"request_id": "example_request_1", "user_id": "example_user"},
|
|
tool_policy=allow_safe_tools,
|
|
)
|
|
|
|
print(result.final_output)
|
|
result.print_trace()
|
|
result.write_trace_jsonl(".aisuite/runs.jsonl")
|
|
|
|
next_result = ai.Runner.continue_sync(
|
|
result,
|
|
"What about Oakland?",
|
|
tool_policy=allow_safe_tools,
|
|
)
|
|
|
|
print(next_result.final_output)
|
|
next_result.print_trace()
|
|
next_result.write_trace_jsonl(".aisuite/runs.jsonl")
|