kaikaidai commited on
Commit
3edb367
·
1 Parent(s): 7e2f346
Files changed (1) hide show
  1. src/workflows_v2.py +5 -6
src/workflows_v2.py CHANGED
@@ -20,7 +20,7 @@ load_dotenv(env_path)
20
  from atla_insights import configure, instrument, instrument_agno, instrument_openai, mark_success, mark_failure, tool
21
 
22
  # Model configuration - using same model across all agents but you can change them to any model you want
23
- MODEL_ID = "gpt-4o"
24
 
25
  # Define metadata for tracing
26
  metadata = {
@@ -35,7 +35,6 @@ configure(token=os.getenv("ATLA_INSIGHTS_TOKEN"), metadata=metadata)
35
 
36
  # Instrument based on detected framework and LLM provider
37
  instrument_agno("openai") # Agno framework with OpenAI
38
- instrument_openai() # OpenAI LLM provider
39
 
40
  # --- Response models ---
41
  class IdeaClarification(BaseModel):
@@ -97,7 +96,7 @@ class ValidationReport(BaseModel):
97
 
98
 
99
  # --- Agents ---
100
- # NOTE: All agents use the same model (MODEL_ID) for consistent performance and cost tracking
101
  idea_clarifier_agent = Agent(
102
  name="Idea Clarifier",
103
  model=OpenAIChat(id=MODEL_ID),
@@ -115,7 +114,7 @@ idea_clarifier_agent = Agent(
115
 
116
  market_research_agent = Agent(
117
  name="Market Research Agent",
118
- model=OpenAIChat(id=MODEL_ID), # Same model as other agents
119
  tools=[GoogleSearchTools()],
120
  instructions=[
121
  "You are provided with a startup idea and the company's mission and objectives.",
@@ -131,7 +130,7 @@ market_research_agent = Agent(
131
 
132
  competitor_analysis_agent = Agent(
133
  name="Competitor Analysis Agent",
134
- model=OpenAIChat(id=MODEL_ID), # Same model as other agents
135
  tools=[GoogleSearchTools()],
136
  instructions=[
137
  "You are provided with a startup idea and market research data.",
@@ -148,7 +147,7 @@ competitor_analysis_agent = Agent(
148
 
149
  report_agent = Agent(
150
  name="Report Generator",
151
- model=OpenAIChat(id=MODEL_ID), # Same model as other agents
152
  instructions=[
153
  "You are provided with comprehensive data about a startup idea including clarification, market research, and competitor analysis.",
154
  "Synthesize all information into a comprehensive validation report.",
 
20
  from atla_insights import configure, instrument, instrument_agno, instrument_openai, mark_success, mark_failure, tool
21
 
22
  # Model configuration - using same model across all agents but you can change them to any model you want
23
+ MODEL_ID = "o3-mini"
24
 
25
  # Define metadata for tracing
26
  metadata = {
 
35
 
36
  # Instrument based on detected framework and LLM provider
37
  instrument_agno("openai") # Agno framework with OpenAI
 
38
 
39
  # --- Response models ---
40
  class IdeaClarification(BaseModel):
 
96
 
97
 
98
  # --- Agents ---
99
+
100
  idea_clarifier_agent = Agent(
101
  name="Idea Clarifier",
102
  model=OpenAIChat(id=MODEL_ID),
 
114
 
115
  market_research_agent = Agent(
116
  name="Market Research Agent",
117
+ model=OpenAIChat(id=MODEL_ID),
118
  tools=[GoogleSearchTools()],
119
  instructions=[
120
  "You are provided with a startup idea and the company's mission and objectives.",
 
130
 
131
  competitor_analysis_agent = Agent(
132
  name="Competitor Analysis Agent",
133
+ model=OpenAIChat(id=MODEL_ID),
134
  tools=[GoogleSearchTools()],
135
  instructions=[
136
  "You are provided with a startup idea and market research data.",
 
147
 
148
  report_agent = Agent(
149
  name="Report Generator",
150
+ model=OpenAIChat(id=MODEL_ID),
151
  instructions=[
152
  "You are provided with comprehensive data about a startup idea including clarification, market research, and competitor analysis.",
153
  "Synthesize all information into a comprehensive validation report.",