Spaces:
Paused
Paused
Update graph.py
Browse files
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)
|