Further fix get_user_api_tokens for missing columns

vidai/database.py

@@ -1161,13 +1161,27 @@ def get_user_api_tokens(user_id: int) -> List[Dict[str, Any]]:
             FROM user_api_tokens
             WHERE user_id = ? ORDER BY created_at DESC
         ''', (user_id,))
-    except sqlite3.OperationalError:
-        # Fallback if 'name' column doesn't exist
-        cursor.execute('''
-            SELECT id, created_at, last_used, active
-            FROM user_api_tokens
-            WHERE user_id = ? ORDER BY created_at DESC
-        ''', (user_id,))
+    except sqlite3.OperationalError as e:
+        if 'name' in str(e):
+            # Fallback if 'name' column doesn't exist
+            try:
+                cursor.execute('''
+                    SELECT id, created_at, last_used, active
+                    FROM user_api_tokens
+                    WHERE user_id = ? ORDER BY created_at DESC
+                ''', (user_id,))
+            except sqlite3.OperationalError as e2:
+                if 'last_used' in str(e2):
+                    # Fallback if 'last_used' also doesn't exist
+                    cursor.execute('''
+                        SELECT id, created_at, active
+                        FROM user_api_tokens
+                        WHERE user_id = ? ORDER BY created_at DESC
+                    ''', (user_id,))
+                else:
+                    raise e2
+        else:
+            raise e
     rows = cursor.fetchall()
     conn.close()
     tokens = []
@@ -1175,6 +1189,8 @@ def get_user_api_tokens(user_id: int) -> List[Dict[str, Any]]:
         token_dict = dict(row)
         if 'name' not in token_dict:
             token_dict['name'] = 'Unnamed Token'  # Default name
+        if 'last_used' not in token_dict:
+            token_dict['last_used'] = None
         tokens.append(token_dict)
     return tokens