Spaces:
Running
Running
o3-mini
Browse files- 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 = "
|
| 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 |
-
|
| 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),
|
| 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),
|
| 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),
|
| 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.",
|