StudioTools

from agno.agent import Agent
from agno.db.sqlite import SqliteDb
from agno.models.anthropic import Claude
from agno.models.openai import OpenAIResponses
from agno.os import AgentOS
from agno.registry import Registry
from agno.tools.calculator import CalculatorTools
from agno.tools.duckduckgo import DuckDuckGoTools
from agno.tools.hackernews import HackerNewsTools
from agno.tools.studio import StudioTools
from agno.tools.user_control_flow import UserControlFlowTools
from agno.tools.user_feedback import UserFeedbackTools

db = SqliteDb(id="studio-hitl-db", db_file="tmp/studio_hitl.db")

registry = Registry(
    name="Studio HITL Registry",
    tools=[DuckDuckGoTools(), HackerNewsTools(), CalculatorTools()],
    models=[
        OpenAIResponses(id="gpt-5.5"),
        Claude(id="claude-sonnet-4-6"),
    ],
    dbs=[db],
)

tools = [
    StudioTools(
        registry=registry,
        db=db,
        default_model_id="gpt-5.5",
        requires_confirmation_tools=["create_agent", "delete_agent"],
    ),
    UserFeedbackTools(),
    UserControlFlowTools(),
]

studio_agent = Agent(
    id="studio-hitl-agent",
    name="Studio HITL",
    model=OpenAIResponses(id="gpt-5.5"),
    tools=tools,
    db=db,
    instructions=[
        "Create and update Studio components only after you have the required details.",
        "Ask for clarification before making destructive changes.",
        "Return the component id, version, and next action after each Studio change.",
    ],
    markdown=True,
)

agent_os = AgentOS(
    id="studio-hitl-agent-os",
    description="Studio agent with human-in-the-loop composition",
    agents=[studio_agent],
    registry=registry,
    db=db,
)

app = agent_os.get_app()

if __name__ == "__main__":
    agent_os.serve(app="studio_tool:app", port=7777, reload=True)