Update app.py

app.py

@@ -101,10 +101,12 @@ def refresh_models():
     """
     global text_models, free_text_models
     global embedding_models, free_embedding_models
-
+    
     text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
-    embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
-    reranker_models = get_all_models(FREE_MODEL_TEST_KEY, "reranker")
+    embedding_models = {
+        sub_type: get_all_models(FREE_MODEL_TEST_KEY, sub_type)
+        for sub_type in ["embedding", "reranker"]
+    }
     free_text_models = []
     free_embedding_models = []
 
@@ -145,24 +147,6 @@ def refresh_models():
             except Exception as exc:
                 logging.error(f"模型 {model} 测试生成异常: {exc}")
 
-    with concurrent.futures.ThreadPoolExecutor(
-        max_workers=10
-    ) as executor:
-        future_to_model = {
-            executor.submit(
-                test_embedding_model_availability,
-                FREE_MODEL_TEST_KEY, model
-            ): model for model in reranker_models
-        }
-        for future in concurrent.futures.as_completed(future_to_model):
-            model = future_to_model[future]
-            try:
-                is_free = future.result()
-                if is_free:
-                    free_embedding_models.append(model)
-            except Exception as exc:
-                logging.error(f"模型 {model} 测试生成异常: {exc}")
-
     logging.info(f"所有文本模型列表：{text_models}")
     logging.info(f"免费文本模型列表：{free_text_models}")
     logging.info(f"所有向量模型列表：{embedding_models}")