diff --git a/src/backend/base/langflow/components/astra_assistants/util.py b/src/backend/base/langflow/components/astra_assistants/util.py
index 3c6f48755..784880813 100644
--- a/src/backend/base/langflow/components/astra_assistants/util.py
+++ b/src/backend/base/langflow/components/astra_assistants/util.py
@@ -28,11 +28,7 @@ response = requests.get(url)
 data = json.loads(response.text)
 
 # Extract the model names into a Python list
-litellm_model_names = []
-for model, _ in data.items():
-    if model != "sample_spec":
-        # litellm_model_names.append(f"{details['litellm_provider']}/{model}")
-        litellm_model_names.append(model)
+litellm_model_names = [model for model, _ in data.items() if model != "sample_spec"]
 
 
 # To store the class names that extend ToolInterface