Pydantic AI

pip install pydantic-ai

poetry add pydantic-ai

from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

model = OpenAIChatModel(
    "openai/gpt-5.5",
    provider=OpenAIProvider(
        api_key="YOUR_EDEN_AI_API_KEY",  # Get from https://app.edenai.run
        base_url="https://api.edenai.run/v3",
    ),
)

agent = Agent(model)
result = agent.run_sync("Hello! How are you?")
print(result.output)

from pydantic import BaseModel
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

class City(BaseModel):
    name: str
    country: str
    population: int

model = OpenAIChatModel(
    "anthropic/claude-sonnet-5",
    provider=OpenAIProvider(
        api_key="YOUR_EDEN_AI_API_KEY",
        base_url="https://api.edenai.run/v3",
    ),
)

agent = Agent(model, output_type=City)
result = agent.run_sync("Tell me about the capital of France.")
print(result.output)  # City(name='Paris', country='France', population=...)

from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

model = OpenAIChatModel(
    "anthropic/claude-sonnet-5",
    provider=OpenAIProvider(
        api_key="YOUR_EDEN_AI_API_KEY",
        base_url="https://api.edenai.run/v3",
    ),
)
agent = Agent(model, system_prompt="You are a helpful assistant.")

result = agent.run_sync("What is the capital of France?")
print(result.output)

# Continue the conversation with prior context
result = agent.run_sync(
    "What's its population?",
    message_history=result.all_messages(),
)
print(result.output)

from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider
from pydantic_ai.exceptions import ModelHTTPError, UnexpectedModelBehavior

model = OpenAIChatModel(
    "openai/gpt-5.5",
    provider=OpenAIProvider(
        api_key="YOUR_EDEN_AI_API_KEY",
        base_url="https://api.edenai.run/v3",
    ),
)
agent = Agent(model)

try:
    result = agent.run_sync("Hello!")
    print(result.output)
except ModelHTTPError as e:
    print(f"Model HTTP error (auth, rate limit, bad request): {e}")
except UnexpectedModelBehavior as e:
    print(f"Unexpected model behavior: {e}")

EDEN_AI_API_KEY=your_api_key_here

import os
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

model = OpenAIChatModel(
    "openai/gpt-5.5",
    provider=OpenAIProvider(
        api_key=os.getenv("EDEN_AI_API_KEY"),
        base_url="https://api.edenai.run/v3",
    ),
)