ai-query

await generate_text(...)        # call a model
stream_text(...)                # stream a model response
await agent.chat(...)           # persistent conversation
await agent.run(...)            # structured result with steps and usage
agent.turn(...)                 # live events, abort, and steering

import asyncio

from ai_query.agents import Agent, SQLiteStorage
from ai_query.providers import openai


async def main():
    agent = Agent(
        "assistant",
        model=openai("gpt-4o"),
        storage=SQLiteStorage("agents.sqlite3"),
    )

    async with agent:
        print(await agent.chat("Hi, I'm Ada."))
        print(await agent.chat("What's my name?"))


asyncio.run(main())