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
             )