fix:knowledge arguments prompt override

This commit is contained in:
aries_ckt 2023-08-10 15:53:30 +08:00
parent a7ec3400fe
commit 33397f9583

View File

@ -62,8 +62,8 @@ class ChatKnowledge(BaseChat):
def generate_input_values(self):
try:
if self.space_context:
prompt.template_define = self.space_context["prompt"]["scene"]
prompt.template = self.space_context["prompt"]["template"]
self.prompt_template.template_define = self.space_context["prompt"]["scene"]
self.prompt_template.template = self.space_context["prompt"]["template"]
docs = self.knowledge_embedding_client.similar_search(
self.current_user_input, self.top_k
)