Update app.py

app.py

@@ -99,11 +99,12 @@ def refresh_models():
     """
     刷新模型列表和免费模型列表。
     """
-    global all_models, free_models
+    global text_models, free_text_models
     global embedding_models, free_embedding_models
 
-    all_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
+    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")
     free_models = []
     free_embedding_models = []
 
@@ -115,14 +116,14 @@ def refresh_models():
                 test_model_availability,
                 FREE_MODEL_TEST_KEY,
                 model
-            ): model for model in all_models
+            ): model for model in text_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_models.append(model)
+                    free_text_models.append(model)
             except Exception as exc:
                 logging.error(f"模型 {model} 测试生成异常: {exc}")
 
@@ -144,8 +145,26 @@ def refresh_models():
             except Exception as exc:
                 logging.error(f"模型 {model} 测试生成异常: {exc}")
 
-    logging.info(f"所有文本模型列表：{all_models}")
-    logging.info(f"免费文本模型列表：{free_models}")
+    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}")
     logging.info(f"免费向量模型列表：{free_embedding_models}")