⚡️ Speed up method AstraDBVectorStoreComponent.reset_database_list by 42% in PR #6028 (PlaygroundPage)

src/backend/base/langflow/components/vectorstores/astradb.py

@@ -444,14 +444,13 @@
 
     def _initialize_database_options(self):
         try:
-            return [
-                {
-                    "name": name,
-                    "collections": info["collections"],
-                    "api_endpoint": info["api_endpoint"],
-                }
-                for name, info in self.get_database_list().items()
-            ]
+            database_list = self.get_database_list()
+            database_options = []
+            for name, info in database_list.items():
+                database_options.append(
+                    {"name": name, "collections": info["collections"], "api_endpoint": info["api_endpoint"]}
+                )
+            return database_options
         except Exception as e:
             msg = f"Error fetching database options: {e}"
             raise ValueError(msg) from e
@@ -495,16 +494,17 @@
         return build_config
 
     def reset_database_list(self, build_config: dict):
-        # Get the list of options we have based on the token provided
         database_options = self._initialize_database_options()
 
-        # If we retrieved options based on the token, show the dropdown
-        build_config["api_endpoint"]["options"] = [db["name"] for db in database_options]
-        build_config["api_endpoint"]["options_metadata"] = [
-            {k: v for k, v in db.items() if k not in ["name"]} for db in database_options
-        ]
+        api_endpoint_options = []
+        api_endpoint_metadata = []
+
+        for db in database_options:
+            api_endpoint_options.append(db["name"])
+            api_endpoint_metadata.append({"collections": db["collections"], "api_endpoint": db["api_endpoint"]})
 
-        # Reset the selected database
+        build_config["api_endpoint"]["options"] = api_endpoint_options
+        build_config["api_endpoint"]["options_metadata"] = api_endpoint_metadata
         build_config["api_endpoint"]["value"] = ""
 
         return build_config