Update model_mapper.py

model_mapper.py

@@ -1,51 +1,51 @@
-"""
-模型映射模块
-管理OpenAI模型名称到Warp模型名称的映射
-"""
-from typing import List
-
-
-class ModelMapper:
-    """模型名称映射管理"""
-    
-    MODEL_MAPPING = {
-        "claude-4-sonnet": "claude-4-sonnet",
-        "claude-3-7-sonnet": "claude-3-7-sonnet", 
-        "claude-3-5-sonnet": "claude-3-5-sonnet",
-        "claude-3-5-haiku": "claude-3-5-haiku",
-        "gpt-4o": "gpt-4o",
-        "gpt-4.1": "gpt-4.1",
-        "o4-mini": "o4-mini", 
-        "o3": "o3",
-        "o3-mini": "o3-mini",
-        "gemini-2.0-flash": "gemini-2.0-flash",
-        "gemini-2.5-pro": "gemini-2.5-pro"
-    }
-    
-    DEFAULT_MODEL = "gemini-2.0-flash"
-    
-    @classmethod
-    def get_warp_model(cls, openai_model: str) -> str:
-        """将OpenAI模型名转换为Warp模型名"""
-        return cls.MODEL_MAPPING.get(openai_model, cls.DEFAULT_MODEL)
-    
-    @classmethod
-    def get_available_models(cls) -> List[str]:
-        """获取所有可用模型列表"""
-        return list(cls.MODEL_MAPPING.keys())
-    
-    @classmethod
-    def is_valid_model(cls, model: str) -> bool:
-        """检查模型是否有效"""
-        return model in cls.MODEL_MAPPING
-    
-    @classmethod
-    def add_model_mapping(cls, openai_model: str, warp_model: str):
-        """添加新的模型映射"""
-        cls.MODEL_MAPPING[openai_model] = warp_model
-    
-    @classmethod
-    def remove_model_mapping(cls, openai_model: str):
-        """移除模型映射"""
-        if openai_model in cls.MODEL_MAPPING:
+"""
+模型映射模块
+管理OpenAI模型名称到Warp模型名称的映射
+"""
+from typing import List
+
+
+class ModelMapper:
+    """模型名称映射管理"""
+    
+    MODEL_MAPPING = {
+        "claude-sonnet-4-20250514": "claude-4-sonnet",
+        "claude-3-7-sonnet-20250219": "claude-3-7-sonnet", 
+        "claude-3-5-sonnet-20241022": "claude-3-5-sonnet",
+        "claude-3-5-haiku-20241022": "claude-3-5-haiku",
+        "gpt-4o": "gpt-4o",
+        "gpt-4.1": "gpt-4.1",
+        "o4-mini": "o4-mini", 
+        "o3": "o3",
+        "o3-mini": "o3-mini",
+        "gemini-2.0-flash": "gemini-2.0-flash",
+        "gemini-2.5-pro": "gemini-2.5-pro"
+    }
+    
+    DEFAULT_MODEL = "gemini-2.0-flash"
+    
+    @classmethod
+    def get_warp_model(cls, openai_model: str) -> str:
+        """将OpenAI模型名转换为Warp模型名"""
+        return cls.MODEL_MAPPING.get(openai_model, cls.DEFAULT_MODEL)
+    
+    @classmethod
+    def get_available_models(cls) -> List[str]:
+        """获取所有可用模型列表"""
+        return list(cls.MODEL_MAPPING.keys())
+    
+    @classmethod
+    def is_valid_model(cls, model: str) -> bool:
+        """检查模型是否有效"""
+        return model in cls.MODEL_MAPPING
+    
+    @classmethod
+    def add_model_mapping(cls, openai_model: str, warp_model: str):
+        """添加新的模型映射"""
+        cls.MODEL_MAPPING[openai_model] = warp_model
+    
+    @classmethod
+    def remove_model_mapping(cls, openai_model: str):
+        """移除模型映射"""
+        if openai_model in cls.MODEL_MAPPING:
             del cls.MODEL_MAPPING[openai_model]