PydanticAI

pip install pydantic-ai

from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIModel

model = OpenAIModel(
    'gpt-4o',
    base_url='https://api.ohmygpt.com/v1',
    api_key='<OHMYGPT_API_KEY>',
)

agent = Agent(model)

result = agent.run_sync('What is the capital of France?')
print(result.data)

from pydantic import BaseModel

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

agent = Agent(model, result_type=CityInfo)
result = agent.run_sync('Tell me about Tokyo')
print(result.data.name)  # Tokyo
print(result.data.population)  # 13960000

from pydantic_ai import Agent, RunContext

agent = Agent(model)

@agent.tool
def get_weather(ctx: RunContext, location: str) -> str:
    return f"The weather in {location} is sunny, 72°F"

result = agent.run_sync('What is the weather in Paris?')