Workers AI

export CLOUDFLARE_ACCOUNT_ID="..."
export CLOUDFLARE_API_TOKEN="..."

from ai_query import generate_text
from ai_query.providers import workers_ai

result = await generate_text(
    model=workers_ai("@cf/moonshotai/kimi-k2.5"),
    prompt="Explain edge computing in simple terms.",
)

from ai_query import embed
from ai_query.providers import workers_ai

result = await embed(
    model=workers_ai.embedding("@cf/baai/bge-large-en-v1.5"),
    value="Cloudflare runs workloads close to users.",
)

from ai_query.providers import workers_ai

model = workers_ai(
    "@cf/moonshotai/kimi-k2.5",
    api_key="...",
    base_url="https://api.cloudflare.com/client/v4/accounts/<account_id>/ai/v1",
)

model = workers_ai(
    "@cf/moonshotai/kimi-k2.5",
    api_key="...",
    account_id="...",
)

from ai_query import generate_text, tool, Field
from ai_query.providers import workers_ai

@tool(description="Search docs")
async def search(query: str = Field(description="Search query")) -> str:
    return f"Results for: {query}"

result = await generate_text(
    model=workers_ai("@cf/moonshotai/kimi-k2.5"),
    prompt="Search for Workers AI pricing.",
    tools={"search": search},
)