feat: pass QueryConfig to dry_run (#173)

2023-07-07 16:08:14 +05:30
parent 0c2653ccf5
commit 5e6aef1e37
1 changed files with 2 additions and 2 deletions
@@ -279,7 +279,7 @@ class EmbedChain:
        memory.chat_memory.add_ai_message(answer)
        return answer

-    def dry_run(self, input_query):
+    def dry_run(self, input_query, config: QueryConfig = None):
        """
        A dry run does everything except send the resulting prompt to
        the LLM. The purpose is to test the prompt, not the response.