Set MODEL_ID

src/workflows_v2.py

@@ -19,9 +19,12 @@ load_dotenv(env_path)
 # --- Atla Insights Configuration ---
 from atla_insights import configure, instrument, instrument_agno, instrument_openai, mark_success, mark_failure, tool
 
+# Model configuration - using same model across all agents but you can change them to any model you want
+MODEL_ID = "gpt-4o"
+
 # Define metadata for tracing
 metadata = {
-    "model": "gpt-4o",
+    "model": MODEL_ID,
     "prompt": "v1.1",
     "environment": "prod",
     "agent_name": "Startup Idea Validator"
@@ -94,9 +97,10 @@ class ValidationReport(BaseModel):
 
 
 # --- Agents ---
+# NOTE: All agents use the same model (MODEL_ID) for consistent performance and cost tracking
 idea_clarifier_agent = Agent(
     name="Idea Clarifier",
-    model=OpenAIChat(id="gpt-4o"),
+    model=OpenAIChat(id=MODEL_ID),
     instructions=[
         "Given a user's startup idea, your goal is to refine that idea.",
         "Evaluate the originality of the idea by comparing it with existing concepts.",
@@ -111,7 +115,7 @@ idea_clarifier_agent = Agent(
 
 market_research_agent = Agent(
     name="Market Research Agent",
-    model=OpenAIChat(id="gpt-4o"),
+    model=OpenAIChat(id=MODEL_ID),  # Same model as other agents
     tools=[GoogleSearchTools()],
     instructions=[
         "You are provided with a startup idea and the company's mission and objectives.",
@@ -127,7 +131,7 @@ market_research_agent = Agent(
 
 competitor_analysis_agent = Agent(
     name="Competitor Analysis Agent",
-    model=OpenAIChat(id="gpt-4o"),
+    model=OpenAIChat(id=MODEL_ID),  # Same model as other agents
     tools=[GoogleSearchTools()],
     instructions=[
         "You are provided with a startup idea and market research data.",
@@ -144,7 +148,7 @@ competitor_analysis_agent = Agent(
 
 report_agent = Agent(
     name="Report Generator",
-    model=OpenAIChat(id="gpt-4o"),
+    model=OpenAIChat(id=MODEL_ID),  # Same model as other agents
     instructions=[
         "You are provided with comprehensive data about a startup idea including clarification, market research, and competitor analysis.",
         "Synthesize all information into a comprehensive validation report.",