fix: change gleaning prompt to validation_prompt

ucbepic · Oct 13, 2024 · 652efad · 652efad
1 parent 0aeb643
commit 652efad
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docetl/operations/utils.py b/docetl/operations/utils.py
@@ -470,7 +470,9 @@ def _cached_call_llm(
                 if gleaning_config:
                     # Retry gleaning prompt + regular LLM
                     num_gleaning_rounds = gleaning_config.get("num_rounds", 2)
-                    validator_prompt_template = Template(gleaning_config["prompt"])
+                    validator_prompt_template = Template(
+                        gleaning_config["validation_prompt"]
+                    )
 
                     parsed_output = self.parse_llm_response(
                         response, output_schema, tools