forgot the most important part

aelaguiz · Mar 21, 2024 · 1b89795 · 1b89795
1 parent b46d2d1
commit 1b89795
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/langdspy/prompt_runners.py b/langdspy/prompt_runners.py
@@ -200,7 +200,7 @@ def _invoke_with_retries(self, chain, input, max_tries=1, config: Optional[Runna
             except AttributeError:
                 model_name = config["llm"].model
 
-            self.prompt_history.add_entry(config["llm"].model_name, formatted_prompt, res, parsed_output, validation_err, start_time, end_time)
+            self.prompt_history.add_entry(model_name, formatted_prompt, res, parsed_output, validation_err, start_time, end_time)
 
             res = {attr_name: parsed_output.get(attr_name, None) for attr_name in self.template.output_variables.keys()}