Toolkits Overview

from definable.agent import Toolkit
from definable.tool.decorator import tool

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

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

    @tool
    def divide(self, a: float, b: float) -> str:
        """Divide two numbers."""
        if b == 0:
            return "Error: Division by zero"
        return str(a / b)

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

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

output = agent.run("What is 15 * 23, then divide the result by 5?")
print(output.content)

class DatabaseToolkit(Toolkit):
    def __init__(self, connection_string: str):
        self.db = Database(connection_string)
        super().__init__()

    @tool
    def run_query(self, sql: str) -> str:
        """Execute a SQL query."""
        return str(self.db.execute(sql))

    @tool
    def list_tables(self) -> str:
        """List all database tables."""
        return str(self.db.get_tables())

# Usage
toolkit = DatabaseToolkit("postgresql://localhost/mydb")
agent = Agent(model=model, toolkits=[toolkit])

@tool
def get_current_time() -> str:
    """Get the current time."""
    from datetime import datetime
    return datetime.now().isoformat()

agent = Agent(
    model=OpenAIChat(id="gpt-4o"),
    tools=[get_current_time],
    toolkits=[MathToolkit(), DatabaseToolkit(conn_str)],
)