oai_agents.py•728 B
from agents import Agent, Runner, ModelSettings
from openai.types.shared import Reasoning
from ingest.prompt import SUMMARY_PROMPT, VERIFY_PROMPT
class DataPrepAgent:
def __init__(self):
self.summarizer_agent = Agent(
name = 'Data Summarizer Agent',
instructions = SUMMARY_PROMPT,
model = 'gpt-5',
output_type = str
)
self.verifier_agent = Agent(
name = 'Data Verifier Agent',
instructions = VERIFY_PROMPT,
model = 'o3',
model_settings = ModelSettings(
reasoning = Reasoning(
effort = 'high'
)
),
output_type = str
)