update sth. at 2024-12-24 00:00:30

getm.py

@@ -1,38 +0,0 @@
-import json
-
-# 保持不变的get_models方法
-def get_models():
-    models = {
-        "object": "list",
-        "data": [
-            {"id": "Qwen2.5-72B", "object": "model", "created": 0, "owned_by": "Qwen"},
-            {"id": "Llama-3.1-Nemotron-70B", "object": "model", "created": 0, "owned_by": "Nemotron"},
-            {"id": "NVLM-D-72B", "object": "model", "created": 0, "owned_by": "NVDIA"},
-            {"id": "DeepSeek-Coder-V2", "object": "model", "created": 0, "owned_by": "DeepSeek"},
-            {"id": "Qwen2.5-Coder-32B", "object": "model", "created": 0, "owned_by": "Qwen"},
-        ]
-    }
-    return json.dumps(models)
-
-# 新的方法，检查model是否有效并返回id
-def get_model_by_id(model_id=None):
-    # 获取所有模型数据
-    models_data = json.loads(get_models())["data"]
-    
-    # 提取所有有效的模型ID
-    valid_ids = [model["id"] for model in models_data]
-    
-    # 如果model_id无效，默认返回Qwen2.5-72B
-    if model_id not in valid_ids:
-        model_id = "Qwen2.5-72B"
-    
-    # 根据model_id获取对应的模型数据
-    model_data = next((model for model in models_data if model["id"] == model_id), None)
-    
-    # 返回model_data的id字段
-    return model_data["id"] if model_data else None
-
-
-# 示例调用
-print(get_model_by_id("Llama-3.1-Nemotron-70B"))  # 返回指定模型
-print(get_model_by_id("Invalid-ID"))  # 返回默认模型 Qwen2.5-72B