From 20f4456dedaa7c577a75ac220850a4ba8c543e2b Mon Sep 17 00:00:00 2001
From: ulleo <ulleo.wang@gmail.com>
Date: Fri, 14 Nov 2025 11:54:03 +0800
Subject: [PATCH] fix(mcp-chat): prevent bigint/float data from displaying in
 scientific notation

Add zero-width space (\u200b) prefix to all numeric values in Markdown tables to maintain formatting.
---
 backend/apps/chat/task/llm.py | 37 ++++++++++++++++++-----------------
 1 file changed, 19 insertions(+), 18 deletions(-)

diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py
index 6de56a33..242f10cc 100644
--- a/backend/apps/chat/task/llm.py
+++ b/backend/apps/chat/task/llm.py
@@ -1131,14 +1131,19 @@ def run_task(self, in_chat: bool = True, stream: bool = True,
                 yield 'data:' + orjson.dumps({'type': 'finish'}).decode() + '\n\n'
             else:
                 # todo generate picture
-                if chart['type'] != 'table':
-                    yield '### generated chart picture\n\n'
-                    image_url = request_picture(self.record.chat_id, self.record.id, chart, format_json_data(result))
-                    SQLBotLogUtil.info(image_url)
+                try:
+                    if chart['type'] != 'table':
+                        yield '### generated chart picture\n\n'
+                        image_url = request_picture(self.record.chat_id, self.record.id, chart,
+                                                    format_json_data(result))
+                        SQLBotLogUtil.info(image_url)
+                        if stream:
+                            yield f'![{chart["type"]}]({image_url})'
+                        else:
+                            json_result['image_url'] = image_url
+                except Exception as e:
                     if stream:
-                        yield f'![{chart["type"]}]({image_url})'
-                    else:
-                        json_result['image_url'] = image_url
+                        raise e
 
             if not stream:
                 yield json_result
@@ -1173,20 +1178,16 @@ def run_task(self, in_chat: bool = True, stream: bool = True,
 
     @staticmethod
     def safe_convert_to_string(df):
-        """
-        安全地将数值列转换为字符串，避免科学记数法
-        """
         df_copy = df.copy()
 
+        def format_value(x):
+            if pd.isna(x):
+                return ""
+
+            return "\u200b" + str(x)
+
         for col in df_copy.columns:
-            # 只处理数值类型的列
-            if pd.api.types.is_numeric_dtype(df_copy[col]):
-                try:
-                    df_copy[col] = df_copy[col].astype(str)
-                except Exception as e:
-                    print(f"列 {col} 转换失败: {e}")
-                    # 如果转换失败，保持原样
-                    continue
+            df_copy[col] = df_copy[col].apply(format_value)
 
         return df_copy