yangtb24 commited on
Commit
8ea742a
1 Parent(s): b34c36b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -1
app.py CHANGED
@@ -100,12 +100,31 @@ def refresh_models():
100
  """
101
  global text_models, free_text_models
102
  global embedding_models, free_embedding_models
103
-
104
  text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
105
  embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
106
  free_text_models = []
107
  free_embedding_models = []
108
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
109
  with concurrent.futures.ThreadPoolExecutor(
110
  max_workers=10
111
  ) as executor:
 
100
  """
101
  global text_models, free_text_models
102
  global embedding_models, free_embedding_models
103
+
104
  text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
105
  embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
106
  free_text_models = []
107
  free_embedding_models = []
108
 
109
+ ban_models_str = os.environ.get("BAN_MODELS")
110
+ ban_models = []
111
+ if ban_models_str:
112
+ try:
113
+ ban_models = json.loads(ban_models_str)
114
+ if not isinstance(ban_models, list):
115
+ logging.warning(
116
+ "环境变量 BAN_MODELS 格式不正确,应为 JSON 数组。"
117
+ )
118
+ ban_models = []
119
+ except json.JSONDecodeError:
120
+ logging.warning(
121
+ "环境变量 BAN_MODELS JSON 解析失败,请检查格式。"
122
+ )
123
+ ban_models = []
124
+
125
+ text_models = [model for model in text_models if model not in ban_models]
126
+ embedding_models = [model for model in embedding_models if model not in ban_models]
127
+
128
  with concurrent.futures.ThreadPoolExecutor(
129
  max_workers=10
130
  ) as executor: