JatinAutonomousLabs commited on
Commit
f1c4c17
·
verified ·
1 Parent(s): 4a7e904

Update graph.py

Browse files
Files changed (1) hide show
  1. graph.py +19 -2
graph.py CHANGED
@@ -582,11 +582,28 @@ def run_experimenter_agent(state: AgentState):
582
  }
583
 
584
  # Detect language from plan or request
585
- detected = detect_requested_output_types_enhanced(pm.get('experiment_goal', ''))
586
- language = detected.get('language', 'python')
587
  exp_type = normalize_experiment_type(pm.get('experiment_type'), pm.get('experiment_goal',''))
588
  goal = pm.get('experiment_goal', 'No goal')
589
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
590
  # Get language config
591
  from multi_language_support import LANGUAGES
592
  lang_config = LANGUAGES.get(language)
 
582
  }
583
 
584
  # Detect language from plan or request
585
+ #detected = detect_requested_output_types_enhanced(pm.get('experiment_goal', ''))
586
+ #language = detected.get('language', 'python')
587
  exp_type = normalize_experiment_type(pm.get('experiment_type'), pm.get('experiment_goal',''))
588
  goal = pm.get('experiment_goal', 'No goal')
589
 
590
+ # BUILD RICH CONTEXT (this was missing!)
591
+ context_parts = [
592
+ f"=== USER REQUEST ===\n{state.get('userInput', '')}",
593
+ f"\n=== OBJECTIVE ===\n{state.get('coreObjectivePrompt', '')}",
594
+ f"\n=== PLAN ===\n{json.dumps(pm.get('plan_steps', []), indent=2)}",
595
+ f"\n=== REQUIREMENTS ===\n{json.dumps(pm.get('key_requirements', []), indent=2)}",
596
+ ]
597
+
598
+ if state.get('retrievedMemory'):
599
+ context_parts.append(f"\n=== CONTEXT ===\n{state.get('retrievedMemory', '')}")
600
+
601
+ if state.get('qaFeedback'):
602
+ context_parts.append(f"\n=== FEEDBACK TO ADDRESS ===\n{state.get('qaFeedback', '')}")
603
+
604
+ full_context = "\n".join(context_parts) # This line was missing!
605
+
606
+
607
  # Get language config
608
  from multi_language_support import LANGUAGES
609
  lang_config = LANGUAGES.get(language)