Update app.py

app.py

@@ -414,6 +414,48 @@ class TextToSQLSystem:
         
         return relevant_tables[:3]  # 最多返回3個相關表格
 
+    # 請將這整個函數複製到您的 TextToSQLSystem class 內部
+
+    def _format_relevant_schema(self, table_names: List[str]) -> str:
+        """
+        生成一個簡化的、不易被模型錯誤模仿的 Schema 字符串。
+        """
+        if not self.schema:
+            return "No schema available.\n"
+        
+        actual_table_names_map = {name.lower(): name for name in self.schema.keys()}
+        real_table_names = []
+        for table in table_names:
+            actual_name = actual_table_names_map.get(table.lower())
+            if actual_name:
+                real_table_names.append(actual_name)
+            elif table in self.schema:
+                real_table_names.append(table)
+
+        if not real_table_names:
+            self._log("未識別到相關表格，使用預設核心表格。", "WARNING")
+            real_table_names = ['TSR53SampleDescription', 'JobTimeline', 'JobsInProgress']
+
+        formatted = ""
+        for table in real_table_names:
+            if table in self.schema:
+                # 使用簡單的 "Table: ..." 和 "Columns: ..." 格式
+                formatted += f"Table: {table}\n"
+                cols_str = []
+                # 只顯示前 10 個關鍵欄位
+                for col in self.schema[table][:10]:
+                    col_name = col['name']
+                    col_type = col['type']
+                    col_desc = col.get('description', '').replace('\n', ' ')
+                    # 將描述信息放在括號裡
+                    if col_desc:
+                        cols_str.append(f"{col_name} ({col_type}, {col_desc})")
+                    else:
+                        cols_str.append(f"{col_name} ({col_type})")
+                formatted += f"Columns: {', '.join(cols_str)}\n\n"
+        
+        return formatted.strip()
+    
     def _validate_and_fix_sql(self, sql: str, question: str) -> str:
         """
         (V10 / 組別查詢版)