|
@@ -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):
|