From 33397f958316198211b9f98d2e1743d75f066084 Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Thu, 10 Aug 2023 15:53:30 +0800 Subject: [PATCH] fix:knowledge arguments prompt override --- pilot/scene/chat_knowledge/v1/chat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pilot/scene/chat_knowledge/v1/chat.py b/pilot/scene/chat_knowledge/v1/chat.py index ef6b281c7..b7e439ca2 100644 --- a/pilot/scene/chat_knowledge/v1/chat.py +++ b/pilot/scene/chat_knowledge/v1/chat.py @@ -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 )