src/agents/generic_agent.py

from pydantic_ai import Agent, RunContext
from pydantic_ai.models.openai import OpenAIModel
from dotenv import load_dotenv
import os

load_dotenv()

model = OpenAIModel(
    "gpt-4o",
    api_key=os.environ.get("OPENAI_API_KEY")
)

system_prompt = """
You are a helpful assistant that can answer questions and help with tasks.
"""

generic_agent = Agent(
    model=model,
    system_prompt=system_prompt,
    tools=[],
)