diff --git a/pilot/scene/base_chat.py b/pilot/scene/base_chat.py index 92a018794..245c03b34 100644 --- a/pilot/scene/base_chat.py +++ b/pilot/scene/base_chat.py @@ -174,8 +174,8 @@ class BaseChat(ABC): def stream_plugin_call(self, text): return text - def knowledge_reference_call(self, text): - return text + # def knowledge_reference_call(self, text): + # return text async def check_iterator_end(iterator): try: @@ -218,7 +218,7 @@ class BaseChat(ABC): view_msg = view_msg.replace("\n", "\\n") yield view_msg self.current_message.add_ai_message(msg) - view_msg = self.knowledge_reference_call(msg) + # view_msg = self.knowledge_reference_call(msg) self.current_message.add_view_message(view_msg) span.end() except Exception as e: diff --git a/pilot/scene/chat_data/chat_excel/excel_analyze/chat.py b/pilot/scene/chat_data/chat_excel/excel_analyze/chat.py index 57f5b06bf..a2cf586f1 100644 --- a/pilot/scene/chat_data/chat_excel/excel_analyze/chat.py +++ b/pilot/scene/chat_data/chat_excel/excel_analyze/chat.py @@ -14,6 +14,7 @@ from pilot.scene.chat_data.chat_excel.excel_learning.chat import ExcelLearning from pilot.common.path_utils import has_path from pilot.configs.model_config import LLM_MODEL_CONFIG, KNOWLEDGE_UPLOAD_ROOT_PATH from pilot.base_modules.agent.common.schema import Status +from pilot.utils.tracer import root_tracer, trace CFG = Config() diff --git a/pilot/scene/chat_knowledge/v1/chat.py b/pilot/scene/chat_knowledge/v1/chat.py index a9c63b268..e4e373cfe 100644 --- a/pilot/scene/chat_knowledge/v1/chat.py +++ b/pilot/scene/chat_knowledge/v1/chat.py @@ -89,7 +89,7 @@ class ChatKnowledge(BaseChat): last_output = last_output + reference yield last_output - def knowledge_reference_call(self, text): + def stream_plugin_call(self, text): """return reference""" return text + f"\n\n{self.parse_source_view(self.sources)}" diff --git a/pilot/server/static/404.html b/pilot/server/static/404.html index 8bfda489e..8efd87b48 100644 --- a/pilot/server/static/404.html +++ b/pilot/server/static/404.html @@ -1 +1 @@ -
No data |
No data |