from pydantic_ai import Agent
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=[],
)