Fix metadata of trace v3

src/workflows_v2.py

@@ -17,7 +17,7 @@ env_path = os.path.join(script_dir, '.env')
 load_dotenv(env_path)
 
 # --- Atla Insights Configuration ---
-from atla_insights import configure, instrument, instrument_agno, instrument_openai, mark_success, mark_failure, tool
+from atla_insights import configure, instrument, instrument_agno, instrument_openai, mark_success, mark_failure, tool, set_metadata
 
 # Model configuration - using same model across all agents but you can change them to any model you want
 DEFAULT_MODEL_ID = "gpt-4o"
@@ -169,29 +169,8 @@ 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,
@@ -201,6 +180,9 @@ async def _startup_validation_execution_impl(
 ) -> str:
     """Execute the complete startup idea validation workflow"""
 
+    # Set dynamic metadata for this execution
+    set_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)