Update rag_routerv2.py

rag_routerv2.py

@@ -81,26 +81,6 @@ async def create_embedding_table(
         with open(file_path, "wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
 
-    # Store user_id and table_name in a JSON file
-    try:
-        tables_file_path = './data/tables.json'
-        os.makedirs(os.path.dirname(tables_file_path), exist_ok=True)
-        # Load existing tables or create a new file if it doesn't exist
-        try:
-            with open(tables_file_path, 'r') as f:
-                tables = json.load(f)
-        except (FileNotFoundError, json.JSONDecodeError):
-            tables = {}
-
-        # Update the tables dictionary
-        if user_id not in tables:
-            tables[user_id] = []
-        if table_id not in [table['table_id'] for table in tables[user_id]]:
-            tables[user_id].append({"table_id": table_id, "table_name": table_name})
-
-        # Write the updated tables back to the JSON file
-        with open(tables_file_path, 'w') as f:
-            json.dump(tables, f)
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to update tables file: {str(e)}")
@@ -123,6 +103,28 @@ async def create_embedding_table(
         )
         index.storage_context.persist(persist_dir=f"./lancedb/index/{table_id}")
 
+
+        # Store user_id and table_name in a JSON file
+        try:
+            tables_file_path = './data/tables.json'
+            os.makedirs(os.path.dirname(tables_file_path), exist_ok=True)
+            # Load existing tables or create a new file if it doesn't exist
+            try:
+                with open(tables_file_path, 'r') as f:
+                    tables = json.load(f)
+            except (FileNotFoundError, json.JSONDecodeError):
+                tables = {}
+    
+            # Update the tables dictionary
+            if user_id not in tables:
+                tables[user_id] = []
+            if table_id not in [table['table_id'] for table in tables[user_id]]:
+                tables[user_id].append({"table_id": table_id, "table_name": table_name})
+    
+            # Write the updated tables back to the JSON file
+            with open(tables_file_path, 'w') as f:
+                json.dump(tables, f)
+        
         return CreateTableResponse(
             table_id=table_id,
             message="Table created and documents indexed successfully",