Update app/core/agents.py
Browse files- app/core/agents.py +1 -1
app/core/agents.py
CHANGED
|
@@ -1277,7 +1277,7 @@ NO extra text.
|
|
| 1277 |
system_prompt = sql_judge_prompt
|
| 1278 |
elif plan_type == "ml_workflow":
|
| 1279 |
logger.info("🧠 Using ML Judge Prompt")
|
| 1280 |
-
system_prompt =
|
| 1281 |
else:
|
| 1282 |
logger.info("🧠 Using General QA Judge Prompt")
|
| 1283 |
system_prompt = general_qa_judge_prompt
|
|
|
|
| 1277 |
system_prompt = sql_judge_prompt
|
| 1278 |
elif plan_type == "ml_workflow":
|
| 1279 |
logger.info("🧠 Using ML Judge Prompt")
|
| 1280 |
+
system_prompt = ml_judge_prompt
|
| 1281 |
else:
|
| 1282 |
logger.info("🧠 Using General QA Judge Prompt")
|
| 1283 |
system_prompt = general_qa_judge_prompt
|