From 652efad63ee32ac5d4ca8e2d704a65cd86760dba Mon Sep 17 00:00:00 2001 From: Shreya Shankar Date: Sun, 13 Oct 2024 00:43:26 -0400 Subject: [PATCH] fix: change gleaning prompt to validation_prompt --- docetl/operations/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docetl/operations/utils.py b/docetl/operations/utils.py index f7313690..bd9899dc 100644 --- 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