fix model list

server/app/vdr_session.py

@@ -37,6 +37,7 @@ class VDRSession:
         self.db_collection = None
         self.session_id = str(uuid.uuid4())[:5]
         self.indexed_images = []
+        self.model_name_list = []
         self.vector_db_client = None
 
     def set_api_key(self, api_key: str):
@@ -84,14 +85,16 @@ class VDRSession:
 
     def get_available_vlms(self) -> List[str]:
         assert self.client != None
-        model_name_list = []
+        #model_name_list = []
+        if self.model_name_list:
+            return self.model_name_list
         try:
             models = self.client.models.list()
             for model in models.data:
                 model_name = model.id
-                substrings = ['gemini','QWEN-VL2-7B']
+                substrings = ['gemini-2.0','claude','Qwen2.5-VL-72B-Instruct']
                 if any(substring in model_name for substring in substrings):
-                    model_name_list.append(model.id)
+                    self.model_name_list.append(model.id)
                 
         except Exception as e:
             logger.error(f"Error while query all models: {e}")
@@ -99,14 +102,14 @@ class VDRSession:
         
         # Prioritize name
         # Remove the item if it exists in the list
-        priority_item = "gemini-2.0-flash-exp-US"
-        if priority_item in model_name_list:
-            model_name_list.remove(priority_item)
+        # priority_item = "gemini-2.0-flash"
+        # if priority_item in model_name_list:
+        #     model_name_list.remove(priority_item)
 
-        # Insert the item at the beginning of the list
-        model_name_list.insert(0, priority_item)
+        # # Insert the item at the beginning of the list
+        # model_name_list.insert(0, priority_item)
             
-        return model_name_list
+        return self.model_name_list
 
     def get_available_image_embeds(self) -> List[str]:
         assert self.client != None