quickstart.py

"""Build a small custom agent loop from CapSolver Agent JSON schemas."""

import asyncio
import json
import os
from typing import Any

from capsolver_agent import create_executor, get_all_tools
from openai import AsyncOpenAI


async def main() -> None:
    client = AsyncOpenAI()
    executor = create_executor()
    tools = [tool.to_openai_function() for tool in get_all_tools()]
    messages: list[dict[str, Any]] = [
        {
            "role": "system",
            "content": (
                "Use CapSolver only for lawful, user-authorized workflows. "
                "Never invent target details or call a solve tool without required inputs."
            ),
        },
        {
            "role": "user",
            "content": os.getenv(
                "DEMO_PROMPT",
                "Check my CapSolver balance with the available tools.",
            ),
        },
    ]

    first = await client.chat.completions.create(
        model=os.getenv("OPENAI_MODEL", "gpt-4.1-mini"),
        messages=messages,
        tools=tools,
    )
    assistant = first.choices[0].message
    messages.append(assistant.model_dump(exclude_none=True))

    if not assistant.tool_calls:
        print(assistant.content)
        return

    for call in assistant.tool_calls:
        arguments = json.loads(call.function.arguments)
        result = await executor.execute(call.function.name, arguments)
        messages.append(
            {
                "role": "tool",
                "tool_call_id": call.id,
                "content": json.dumps(result, ensure_ascii=False),
            }
        )

    final = await client.chat.completions.create(
        model=os.getenv("OPENAI_MODEL", "gpt-4.1-mini"),
        messages=messages,
        tools=tools,
    )
    print(final.choices[0].message.content)


if __name__ == "__main__":
    asyncio.run(main())