Agent Examples

from definable.agent import Agent
from definable.model.openai import OpenAIChat

agent = Agent(
    model=OpenAIChat(id="gpt-4o-mini"),
    instructions="You are a helpful assistant.",
)

output = agent.run("What is the capital of Japan?")
print(output.content)
print(f"Tokens: {output.metrics.total_tokens}")

python definable/examples/agents/01_simple_agent.py

from definable.agent import Agent
from definable.model.openai import OpenAIChat
from definable.tool.decorator import tool

@tool
def add(a: int, b: int) -> int:
    """Add two numbers together."""
    return a + b

@tool
def multiply(a: int, b: int) -> int:
    """Multiply two numbers together."""
    return a * b

agent = Agent(
    model=OpenAIChat(id="gpt-4o-mini"),
    tools=[add, multiply],
    instructions="You are a calculator. Use the provided tools.",
)

output = agent.run("What is 15 + 27, then multiply the result by 3?")
print(output.content)

python definable/examples/agents/02_agent_with_tools.py

from definable.agent import Agent, Toolkit
from definable.model.openai import OpenAIChat
from definable.tool.decorator import tool

class MathToolkit(Toolkit):
    @tool
    def add(self, a: int, b: int) -> int:
        """Add two numbers."""
        return a + b

    @tool
    def multiply(self, a: int, b: int) -> int:
        """Multiply two numbers."""
        return a * b

agent = Agent(
    model=OpenAIChat(id="gpt-4o-mini"),
    toolkits=[MathToolkit()],
)

output = agent.run("What is 5 + 3, then multiply by 2?")
print(output.content)

python definable/examples/agents/03_agent_with_toolkit.py

from definable.agent import Agent
from definable.model.openai import OpenAIChat

agent = Agent(
    model=OpenAIChat(id="gpt-4o-mini"),
    instructions="You are a math tutor.",
)

# First turn
output1 = agent.run("What is 15 * 23?")
print(f"Turn 1: {output1.content}")

# Second turn — carries forward conversation
output2 = agent.run("Now divide that by 5.", messages=output1.messages)
print(f"Turn 2: {output2.content}")

python definable/examples/agents/04_multi_turn.py

from definable.agent import Agent
from definable.model.openai import OpenAIChat

agent = Agent(
    model=OpenAIChat(id="gpt-4o-mini"),
    instructions="You are a storyteller.",
)

for event in agent.run_stream("Tell me a short story about a robot."):
    if event.event == "RunContent":
        print(event.content, end="", flush=True)
    elif event.event == "ToolCallStarted":
        print(f"\n[Calling {event.tool.tool_name}...]")
    elif event.event == "RunCompleted":
        print(f"\n\nDone. Tokens: {event.metrics.total_tokens if event.metrics else 'N/A'}")

python definable/examples/agents/05_streaming_agent.py

import asyncio
from definable.agent import Agent
from definable.model.openai import OpenAIChat

async def main():
    agent = Agent(
        model=OpenAIChat(id="gpt-4o-mini"),
        instructions="You are a helpful assistant.",
    )

    # Async run
    output = await agent.arun("Hello!")
    print(output.content)

    # Async streaming
    async for event in agent.arun_stream("Tell me a joke."):
        if event.event == "RunContent":
            print(event.content, end="", flush=True)

asyncio.run(main())

python definable/examples/agents/06_async_agent.py