kaikaidai commited on
Commit
69ccac3
·
1 Parent(s): a1b46ea

Fix metadata of trace

Browse files
Files changed (1) hide show
  1. src/workflows_v2.py +5 -0
src/workflows_v2.py CHANGED
@@ -13,6 +13,7 @@ from dotenv import load_dotenv
13
  import os
14
 
15
  script_dir = os.path.dirname(os.path.abspath(__file__))
 
16
  env_path = os.path.join(script_dir, '.env')
17
  load_dotenv(env_path)
18
 
@@ -32,6 +33,7 @@ def get_metadata(model_id):
32
  }
33
 
34
  # Configure Atla Insights with metadata - REQUIRED FIRST
 
35
  configure(token=os.getenv("ATLA_INSIGHTS_TOKEN"), metadata=get_metadata(DEFAULT_MODEL_ID))
36
 
37
  # Instrument based on detected framework and LLM provider
@@ -180,6 +182,9 @@ async def startup_validation_execution(
180
  ) -> str:
181
  """Execute the complete startup idea validation workflow"""
182
 
 
 
 
183
  # Create agents with the specified model
184
  idea_clarifier_agent, market_research_agent, competitor_analysis_agent, report_agent = create_agents(model_id)
185
 
 
13
  import os
14
 
15
  script_dir = os.path.dirname(os.path.abspath(__file__))
16
+
17
  env_path = os.path.join(script_dir, '.env')
18
  load_dotenv(env_path)
19
 
 
33
  }
34
 
35
  # Configure Atla Insights with metadata - REQUIRED FIRST
36
+ # Note: Initial configuration will be updated dynamically in the workflow execution
37
  configure(token=os.getenv("ATLA_INSIGHTS_TOKEN"), metadata=get_metadata(DEFAULT_MODEL_ID))
38
 
39
  # Instrument based on detected framework and LLM provider
 
182
  ) -> str:
183
  """Execute the complete startup idea validation workflow"""
184
 
185
+ # Update Atla Insights configuration with the correct model_id for this execution
186
+ configure(token=os.getenv("ATLA_INSIGHTS_TOKEN"), metadata=get_metadata(model_id))
187
+
188
  # Create agents with the specified model
189
  idea_clarifier_agent, market_research_agent, competitor_analysis_agent, report_agent = create_agents(model_id)
190