Update rag_routerv2.py

rag_routerv2.py

@@ -176,26 +176,30 @@ 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.*, 
-            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
+   db = get_db()
+   tables = db.execute('''
+       SELECT 
+           t.table_id,
+           t.table_name,
+           t.created_time as created_at,
+           GROUP_CONCAT(tf.filename) as filenames
+       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)
+       result.append({
+           'table_id': table_dict['table_id'],
+           'table_name': table_dict['table_name'],
+           'created_at': table_dict['created_at'],
+           'documents': [filename for filename in table_dict['filenames'].split(',') if filename] if table_dict['filenames'] else []
+       })
+   
+   return result
 
 
 @router.delete("/delete_table/{table_id}")