소스 검색

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

cachho 2 년 전
부모
커밋
feda6b347a
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      embedchain/embedchain.py

+ 1 - 1
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):