PydanticAI | Flow AI Docs

Use PydanticAI’s OpenAI model with a provider pointed at Flow AI.

pip install pydantic-ai

import os
from pydantic import BaseModel
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider
model = OpenAIChatModel(
    "openai/gpt-6-luna",
    provider=OpenAIProvider(
        base_url="https://ai.hyperflow.fun/v1",
        api_key=os.environ["HYPERFLOW_API_KEY"],
    ),
)
class City(BaseModel):
    name: str
    country: str
agent = Agent(model, output_type=City)
result = agent.run_sync("Which city is the capital of Vietnam?")
print(result.output)

Tools work as usual with @agent.tool_plain or @agent.tool, on models that support tool calling.