Update app.py
Browse files
app.py
CHANGED
@@ -165,47 +165,6 @@ def process_key(key):
|
|
165 |
else:
|
166 |
return "valid"
|
167 |
|
168 |
-
def get_all_models(api_key, sub_type):
|
169 |
-
"""
|
170 |
-
获取所有模型列表。
|
171 |
-
"""
|
172 |
-
headers = {
|
173 |
-
"Authorization": f"Bearer {api_key}",
|
174 |
-
"Content-Type": "application/json"
|
175 |
-
}
|
176 |
-
try:
|
177 |
-
response = requests.get(
|
178 |
-
MODELS_ENDPOINT,
|
179 |
-
headers=headers,
|
180 |
-
params={"sub_type": sub_type}
|
181 |
-
)
|
182 |
-
response.raise_for_status()
|
183 |
-
data = response.json()
|
184 |
-
if (
|
185 |
-
isinstance(data, dict) and
|
186 |
-
'data' in data and
|
187 |
-
isinstance(data['data'], list)
|
188 |
-
):
|
189 |
-
return [
|
190 |
-
model.get("id") for model in data["data"]
|
191 |
-
if isinstance(model, dict) and "id" in model
|
192 |
-
]
|
193 |
-
else:
|
194 |
-
logging.error("获取模型列表失败:响应数据格式不正确")
|
195 |
-
return []
|
196 |
-
except requests.exceptions.RequestException as e:
|
197 |
-
logging.error(
|
198 |
-
f"获取模型列表失败,"
|
199 |
-
f"API Key:{api_key},错误信息:{e}"
|
200 |
-
)
|
201 |
-
return []
|
202 |
-
except (KeyError, TypeError) as e:
|
203 |
-
logging.error(
|
204 |
-
f"解析模型列表失败,"
|
205 |
-
f"API Key:{api_key},错误信息:{e}"
|
206 |
-
)
|
207 |
-
return []
|
208 |
-
|
209 |
def select_key(model_name):
|
210 |
"""
|
211 |
根据请求类型和模型名称选择合适的 KEY,
|
|
|
165 |
else:
|
166 |
return "valid"
|
167 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
168 |
def select_key(model_name):
|
169 |
"""
|
170 |
根据请求类型和模型名称选择合适的 KEY,
|