cmd/ocgo/main.go

@@ -188,16 +188,6 @@ func knownModelIDs() []string {
 	return []string{"glm-5.1", "glm-5", "kimi-k2.6", "kimi-k2.5", "mimo-v2.5-pro", "mimo-v2.5", "mimo-v2-pro", "mimo-v2-omni", "minimax-m2.7", "minimax-m2.5", "deepseek-v4-pro[1M]", "deepseek-v4-flash[1M]", "qwen3.6-plus", "qwen3.5-plus"}
 }
 
-func isKnownModelID(model string) bool {
-	cleaned := cleanModelName(model)
-	for _, m := range knownModelIDs() {
-		if cleanModelName(m) == cleaned {
-			return true
-		}
-	}
-	return false
-}
-
 func cleanModelName(model string) string {
 	model = strings.TrimSuffix(model, "[1M]")
 	model = strings.TrimSuffix(model, "[128K]")
@@ -458,7 +448,7 @@ func proxyChatCompletions(w http.ResponseWriter, r *http.Request, cfg Config) {
 		dm := cleanModelName(cfg.Model)
 		var reqMap map[string]any
 		if json.Unmarshal(body, &reqMap) == nil {
-			if m, _ := reqMap["model"].(string); m == "" || !isKnownModelID(m) {
+			if m, _ := reqMap["model"].(string); m == "" || strings.HasPrefix(m, "claude-") {
 				reqMap["model"] = dm
 				body, _ = json.Marshal(reqMap)
 			}
@@ -637,7 +627,7 @@ func stripRawChatImageDetails(req map[string]any) bool {
 
 func convertRequest(ar AnthropicRequest, defaultModel string) OAIRequest {
 	model := cleanModelName(ar.Model)
-	if model == "" || !isKnownModelID(model) {
+	if model == "" || strings.HasPrefix(model, "claude-") || defaultModel != "" {
 		model = cleanModelName(defaultModel)
 	}
 	if model == "" {
@@ -658,7 +648,7 @@ func convertRequest(ar AnthropicRequest, defaultModel string) OAIRequest {
 
 func responsesToChat(rr ResponsesRequest, defaultModel string) OAIRequest {
 	model := cleanModelName(rr.Model)
-	if model == "" || !isKnownModelID(model) {
+	if model == "" || defaultModel != "" {
 		model = cleanModelName(defaultModel)
 	}
 	if model == "" {