#!/usr/bin/env python3
"""
TokioAI Engine -- Custom Tools Example

Shows how to add your own tools to the engine.
Modify engine/schemas.py to add tool definitions,
or pass them through the API.
"""

from engine.pipeline import Pipeline, EngineConfig
from engine.schemas import TOOL_DEFINITIONS

# You can inspect the built-in tools
print("Built-in tools:")
for tool in TOOL_DEFINITIONS:
    name = tool["function"]["name"]
    desc = tool["function"]["description"][:60]
    print(f"  {name:20s} -- {desc}")

print()

# Use the engine normally -- it already knows all TokioAI tools
config = EngineConfig(
    model_name="qwen3-coder",
    verbose=True,
)
engine = Pipeline(config)

# The engine's router will classify this as tool_call
# and the guided generator will constrain output to valid tool names
result = engine.process("search for TODO comments in all python files")

print(f"\nResult: {result}")
if result.tool_calls:
    print(f"Tool: {result.tool_calls[0]['name']}")
    print(f"Args: {result.tool_calls[0]['arguments']}")