From 5e6aef1e3797b34be1bc844815ef4a1f771fe57d Mon Sep 17 00:00:00 2001
From: gasolin <gasolin@gmail.com>
Date: Fri, 7 Jul 2023 18:38:14 +0800
Subject: [PATCH] feat: pass QueryConfig to dry_run (#173)

---
 embedchain/embedchain.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/embedchain/embedchain.py b/embedchain/embedchain.py
index cc5e62de..15f0c5b6 100644
--- a/embedchain/embedchain.py
+++ b/embedchain/embedchain.py
@@ -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.
@@ -387,4 +387,4 @@ class OpenSourceApp(EmbedChain):
         response = gpt4all_model.generate(
             prompt=prompt,
         )
-        return response
\ No newline at end of file
+        return response