fix: use custom prompt in dry run (#203)

2023-07-08 22:01:06 +02:00
parent 046b76798b
commit feda6b347a
1 changed files with 1 additions and 1 deletions
--- 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):