fix(ChatData): ChatData (#694)

1.empty table show bug fix;
eosphoros-ai · Oct 19, 2023 · 5282208 · 5282208
2 parents a833e8b + 365eb0f
commit 5282208
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pilot/scene/chat_db/auto_execute/out_parser.py b/pilot/scene/chat_db/auto_execute/out_parser.py
@@ -36,8 +36,8 @@ def parse_view_response(self, speak, data) -> str:
 
         ### tool out data to table view
         data_loader = DbDataLoader()
-        if len(data) <= 1:
-            data.insert(0, ["result"])
+        if len(data) < 1:
+            data.insert(0, [])
         df = pd.DataFrame(data[1:], columns=data[0])
         if not CFG.NEW_SERVER_MODE and not CFG.SERVER_LIGHT_MODE:
             table_style = """<style> 

diff --git a/pilot/scene/chat_db/data_loader.py b/pilot/scene/chat_db/data_loader.py
@@ -3,7 +3,7 @@ def get_table_view_by_conn(self, data, speak):
         import pandas as pd
 
         ### tool out data to table view
-        if len(data) <= 1:
+        if len(data) < 1:
             data.insert(0, ["result"])
         df = pd.DataFrame(data[1:], columns=data[0])
         html_table = df.to_html(index=False, escape=False, sparsify=False)