letterm commited on
Commit
10387a6
·
verified ·
1 Parent(s): 37d13d8

Delete model_mapper.py

Browse files
Files changed (1) hide show
  1. model_mapper.py +0 -52
model_mapper.py DELETED
@@ -1,52 +0,0 @@
1
- """
2
- 模型映射模块
3
- 管理OpenAI模型名称到Warp模型名称的映射
4
- """
5
- from typing import List
6
-
7
-
8
- class ModelMapper:
9
- """模型名称映射管理"""
10
-
11
- MODEL_MAPPING = {
12
- "claude-opus-4-20250514": "claude-4-opus",
13
- "claude-sonnet-4-20250514": "claude-4-sonnet",
14
- "claude-3-7-sonnet-20250219": "claude-3-7-sonnet",
15
- "claude-3-5-sonnet-20241022": "claude-3-5-sonnet",
16
- "claude-3-5-haiku-20241022": "claude-3-5-haiku",
17
- "gpt-4o": "gpt-4o",
18
- "gpt-4.1": "gpt-4.1",
19
- "o4-mini": "o4-mini",
20
- "o3": "o3",
21
- "o3-mini": "o3-mini",
22
- "gemini-2.0-flash": "gemini-2.0-flash",
23
- "gemini-2.5-pro": "gemini-2.5-pro"
24
- }
25
-
26
- DEFAULT_MODEL = "gemini-2.0-flash"
27
-
28
- @classmethod
29
- def get_warp_model(cls, openai_model: str) -> str:
30
- """将OpenAI模型名转换为Warp模型名"""
31
- return cls.MODEL_MAPPING.get(openai_model, cls.DEFAULT_MODEL)
32
-
33
- @classmethod
34
- def get_available_models(cls) -> List[str]:
35
- """获取所有可用模型列表"""
36
- return list(cls.MODEL_MAPPING.keys())
37
-
38
- @classmethod
39
- def is_valid_model(cls, model: str) -> bool:
40
- """检查模型是否有效"""
41
- return model in cls.MODEL_MAPPING
42
-
43
- @classmethod
44
- def add_model_mapping(cls, openai_model: str, warp_model: str):
45
- """添加新的模型映射"""
46
- cls.MODEL_MAPPING[openai_model] = warp_model
47
-
48
- @classmethod
49
- def remove_model_mapping(cls, openai_model: str):
50
- """移除模型映射"""
51
- if openai_model in cls.MODEL_MAPPING:
52
- del cls.MODEL_MAPPING[openai_model]