feat(ChatExcel): ChatExcel Use AntV Chart

1.Native Support AntvChart

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):
+    # def knowledge_reference_call(self, text):
-        return 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:

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()
 

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)}"