From feda6b347a9987395ac76cb10a2c0775580da8aa Mon Sep 17 00:00:00 2001
From: cachho <admin@ch-webdev.com>
Date: Sat, 8 Jul 2023 22:01:06 +0200
Subject: [PATCH] fix: use custom prompt in dry run (#203)

---
 embedchain/embedchain.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/embedchain/embedchain.py b/embedchain/embedchain.py
index 14d24de4..ff4d94c1 100644
--- a/embedchain/embedchain.py
+++ b/embedchain/embedchain.py
@@ -243,7 +243,7 @@ class EmbedChain:
         if config is None:
             config = QueryConfig()
         context = self.retrieve_from_database(input_query)
-        prompt = self.generate_prompt(input_query, context)
+        prompt = self.generate_prompt(input_query, context, config.template)
         return prompt
 
     def count(self):