34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
import asyncio
|
|
|
|
from agents import Agent, CodeInterpreterTool, Runner, trace
|
|
|
|
|
|
async def main():
|
|
agent = Agent(
|
|
name="Code interpreter",
|
|
instructions="You love doing math.",
|
|
tools=[
|
|
CodeInterpreterTool(
|
|
tool_config={"type": "code_interpreter", "container": {"type": "auto"}},
|
|
)
|
|
],
|
|
)
|
|
|
|
with trace("Code interpreter example"):
|
|
print("Solving math problem...")
|
|
result = Runner.run_streamed(agent, "What is the square root of273 * 312821 plus 1782?")
|
|
async for event in result.stream_events():
|
|
if (
|
|
event.type == "run_item_stream_event"
|
|
and event.item.type == "tool_call_item"
|
|
and event.item.raw_item.type == "code_interpreter_call"
|
|
):
|
|
print(f"Code interpreter code:\n```\n{event.item.raw_item.code}\n```\n")
|
|
elif event.type == "run_item_stream_event":
|
|
print(f"Other event: {event.item.type}")
|
|
|
|
print(f"Final output: {result.final_output}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|