klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
cachho
2023-07-08 22:01:06 +02:00
committed by GitHub
parent 046b76798b
commit feda6b347a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
embedchain/embedchain.py
View File

@@ -243,7 +243,7 @@ class EmbedChain:
if config is None: if config is None:
config = QueryConfig() config = QueryConfig()
context = self.retrieve_from_database(input_query) 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 return prompt
def count(self): def count(self):