Update rag_routerv2.py

rag_routerv2.py

@@ -171,28 +171,26 @@ async def query_table(
 
 @router.get("/get_tables/{user_id}")
 async def get_tables(user_id: str):
-   db = get_db()
-   tables = db.execute('''
-       SELECT t.*, t.created_time, GROUP_CONCAT(tf.filename) as filenames, GROUP_CONCAT(tf.file_path) as file_paths
-       FROM tables t
-       LEFT JOIN table_files tf ON t.table_id = tf.table_id
-       WHERE t.user_id = ?
-       GROUP BY t.table_id
-   ''', (user_id,)).fetchall()
-   
-   result = []
-   for table in tables:
-       table_dict = dict(table)
-       table_dict['files'] = [
-           {'filename': f, 'file_path': p} 
-           for f, p in zip(
-               table_dict.pop('filenames').split(',') if table_dict['filenames'] else [],
-               table_dict.pop('file_paths').split(',') if table_dict['file_paths'] else []
-           )
-       ]
-       result.append(table_dict)
-   
-   return result
+    db = get_db()
+    tables = db.execute('''
+        SELECT t.*, 
+            GROUP_CONCAT(DISTINCT tf.filename) as filenames,
+            GROUP_CONCAT(DISTINCT tf.file_path) as file_paths
+        FROM tables t
+        LEFT JOIN table_files tf ON t.table_id = tf.table_id 
+        WHERE t.user_id = ?
+        GROUP BY t.id, t.user_id, t.table_id, t.table_name, t.created_time
+    ''', (user_id,)).fetchall()
+    
+    result = []
+    for table in tables:
+        table_dict = dict(table)
+        filenames = table_dict.pop('filenames', '').split(',') if table_dict.get('filenames') else []
+        filepaths = table_dict.pop('file_paths', '').split(',') if table_dict.get('file_paths') else []
+        table_dict['files'] = [{'filename': f, 'file_path': p} for f, p in zip(filenames, filepaths)]
+        result.append(table_dict)
+    
+    return result
 
 
 @router.delete("/delete_table/{table_id}")