Fix metadata of trace

src/workflows_v2.py

@@ -13,7 +13,6 @@ from dotenv import load_dotenv
 import os
 
 script_dir = os.path.dirname(os.path.abspath(__file__))
-
 env_path = os.path.join(script_dir, '.env')
 load_dotenv(env_path)
 
@@ -33,7 +32,6 @@ def get_metadata(model_id):
     }
 
 # Configure Atla Insights with metadata - REQUIRED FIRST
-# Note: Initial configuration will be updated dynamically in the workflow execution
 configure(token=os.getenv("ATLA_INSIGHTS_TOKEN"), metadata=get_metadata(DEFAULT_MODEL_ID))
 
 # Instrument based on detected framework and LLM provider
@@ -171,8 +169,29 @@ def create_agents(model_id: str = DEFAULT_MODEL_ID):
 
 
 # --- Execution function ---
-@instrument("Startup Idea Validation Workflow")
 async def startup_validation_execution(
+    workflow: Workflow,
+    execution_input: WorkflowExecutionInput,
+    startup_idea: str,
+    model_id: str = DEFAULT_MODEL_ID,
+    progress_callback=None,
+    **kwargs: Any,
+) -> str:
+    """Wrapper function that applies instrument decorator with dynamic metadata"""
+    
+    # Apply the instrument decorator with the dynamic model_id metadata
+    instrumented_func = instrument(
+        "Startup Idea Validation Workflow", 
+        metadata=get_metadata(model_id)
+    )(_startup_validation_execution_impl)
+    
+    # Execute the instrumented function
+    return await instrumented_func(
+        workflow, execution_input, startup_idea, model_id, progress_callback, **kwargs
+    )
+
+
+async def _startup_validation_execution_impl(
     workflow: Workflow,
     execution_input: WorkflowExecutionInput, # This is a Pydantic model to ensure type safety
     startup_idea: str,
@@ -182,9 +201,6 @@ async def startup_validation_execution(
 ) -> str:
     """Execute the complete startup idea validation workflow"""
 
-    # Update Atla Insights configuration with the correct model_id for this execution
-    configure(token=os.getenv("ATLA_INSIGHTS_TOKEN"), metadata=get_metadata(model_id))
-
     # Create agents with the specified model
     idea_clarifier_agent, market_research_agent, competitor_analysis_agent, report_agent = create_agents(model_id)