feat: publish sanitized custom.97 source

relaykit/relayconvert/reasoning/suffix.go

@@ -13,7 +13,7 @@ var EffortSuffixes = []string{"-max", "-xhigh", "-high", "-medium", "-low", "-mi
 
 var OpenAIEffortSuffixes = []string{"-max", "-xhigh", "-high", "-medium", "-low", "-minimal", "-none"}
 
-var DeepSeekV4EffortSuffixes = []string{"-none", "-max"}
+var DeepSeekV4EffortSuffixes = []string{"-none", "-low", "-max"}
 
 var (
 	legacyOpenAIModelPattern = regexp.MustCompile(`^(gpt-[a-z0-9][a-z0-9._-]*|o[1-9][a-z0-9._-]*)$`)
@@ -136,12 +136,16 @@ func ParseOpenAIReasoningEffortFromModelSuffix(modelName string, preserveEffortT
 	return effort, baseModel
 }
 
-func ParseClaudeModelSuffix(modelName string, allowThinkingAlias bool) (string, Intent, bool, error) {
+// ParseClaudeModelSuffix folds the naked claude-* aliases (-thinking,
+// -nothinking, -thinking-<budget>, and effort tails such as -high) into an
+// Intent. allowSuffixAlias is the host thinking-adapter toggle; when false the
+// name is returned verbatim and no intent is produced.
+func ParseClaudeModelSuffix(modelName string, allowSuffixAlias bool) (string, Intent, bool, error) {
 	prefix, bare := splitModelNamespace(modelName)
 	if !strings.HasPrefix(bare, "claude-") {
 		return modelName, Intent{}, false, nil
 	}
-	base, intent, found, err := parseProviderModelSuffix(bare, "claude-", allowThinkingAlias, true)
+	base, intent, found, err := parseProviderModelSuffix(bare, "claude-", allowSuffixAlias, true)
 	if err != nil || !found || !legacyClaudeModelPattern.MatchString(base) {
 		return modelName, Intent{}, false, err
 	}
@@ -157,7 +161,7 @@ func isKnownClaudeModel(modelName string) bool {
 	}
 	knownPrefixes := []string{
 		"claude-fable-5", "claude-mythos-5", "claude-mythos-preview",
-		"claude-opus-5", "claude-sonnet-5", "claude-opus-4-8",
+		"claude-opus-5", "claude-sonnet-5", "claude-haiku-5", "claude-opus-4-8",
 		"claude-opus-4-7", "claude-opus-4-6", "claude-sonnet-4-6",
 		"claude-opus-4-5", "claude-sonnet-4-5", "claude-haiku-4-5",
 		"claude-opus-4-1", "claude-opus-4-", "claude-sonnet-4-",
@@ -171,12 +175,14 @@ func isKnownClaudeModel(modelName string) bool {
 	return false
 }
 
-func ParseGeminiModelSuffix(modelName string, allowThinkingAlias bool) (string, Intent, bool, error) {
+// ParseGeminiModelSuffix is the gemini-* counterpart of ParseClaudeModelSuffix
+// and honours allowSuffixAlias the same way.
+func ParseGeminiModelSuffix(modelName string, allowSuffixAlias bool) (string, Intent, bool, error) {
 	prefix, bare := splitModelNamespace(modelName)
 	if !strings.HasPrefix(bare, "gemini-") {
 		return modelName, Intent{}, false, nil
 	}
-	base, intent, found, err := parseProviderModelSuffix(bare, "gemini-", allowThinkingAlias, true)
+	base, intent, found, err := parseProviderModelSuffix(bare, "gemini-", allowSuffixAlias, true)
 	if err != nil || !found || !legacyGeminiModelPattern.MatchString(base) {
 		return modelName, Intent{}, false, err
 	}
@@ -186,13 +192,13 @@ func ParseGeminiModelSuffix(modelName string, allowThinkingAlias bool) (string,
 // ParseKnownProviderModelSuffix extracts a canonical intent only when the
 // origin identifies a provider family whose suffix vocabulary is defined by
 // relaykit. Unknown OpenAI-compatible model names are deliberately untouched.
-func ParseKnownProviderModelSuffix(modelName string, allowThinkingAlias bool) (string, Intent, bool, error) {
+func ParseKnownProviderModelSuffix(modelName string, allowSuffixAlias bool) (string, Intent, bool, error) {
 	bare := lastModelPathSegment(modelName)
 	if strings.HasPrefix(bare, "claude-") {
-		return ParseClaudeModelSuffix(modelName, allowThinkingAlias)
+		return ParseClaudeModelSuffix(modelName, allowSuffixAlias)
 	}
 	if strings.HasPrefix(bare, "gemini-") {
-		return ParseGeminiModelSuffix(modelName, allowThinkingAlias)
+		return ParseGeminiModelSuffix(modelName, allowSuffixAlias)
 	}
 	return modelName, Intent{}, false, nil
 }
@@ -214,37 +220,39 @@ func TrimGeminiThinkingSuffix(modelName string) (string, bool) {
 	return baseModel, ok && err == nil
 }
 
-func parseProviderModelSuffix(modelName string, requiredPrefix string, allowThinkingAlias bool, includeThoughts bool) (string, Intent, bool, error) {
-	if allowThinkingAlias {
-		if marker := strings.LastIndex(modelName, "-thinking-"); marker >= 0 {
-			baseModel := modelName[:marker]
-			if !strings.HasPrefix(baseModel, requiredPrefix) {
-				return modelName, Intent{}, false, nil
-			}
-			budget, err := strconv.Atoi(modelName[marker+len("-thinking-"):])
-			if err != nil {
-				return modelName, Intent{}, false, fmt.Errorf("invalid thinking budget suffix on model %q: %w", modelName, err)
-			}
-			intent := Intent{BudgetTokens: &budget, Source: SourceSuffix, BudgetSource: SourceSuffix}
-			if includeThoughts {
-				value := true
-				intent.IncludeThoughts = &value
-			}
-			return baseModel, intent, true, nil
+// parseProviderModelSuffix recognises a provider family's naked model-name
+// aliases. allowSuffixAlias gates all of them, including effort tails, so a
+// disabled thinking adapter leaves the model name untouched.
+func parseProviderModelSuffix(modelName string, requiredPrefix string, allowSuffixAlias bool, includeThoughts bool) (string, Intent, bool, error) {
+	if !allowSuffixAlias {
+		return modelName, Intent{}, false, nil
+	}
+	if marker := strings.LastIndex(modelName, "-thinking-"); marker >= 0 {
+		baseModel := modelName[:marker]
+		if !strings.HasPrefix(baseModel, requiredPrefix) {
+			return modelName, Intent{}, false, nil
 		}
-		if before, ok := strings.CutSuffix(modelName, "-nothinking"); ok {
-			baseModel := before
-			return baseModel, Intent{Mode: ModeDisabled, Effort: EffortNone, Source: SourceSuffix}, true, nil
+		budget, err := strconv.Atoi(modelName[marker+len("-thinking-"):])
+		if err != nil {
+			return modelName, Intent{}, false, fmt.Errorf("invalid thinking budget suffix on model %q: %w", modelName, err)
 		}
-		if before, ok := strings.CutSuffix(modelName, "-thinking"); ok {
-			baseModel := before
-			intent := Intent{Mode: ModeEnabled, Source: SourceSuffix}
-			if includeThoughts {
-				value := true
-				intent.IncludeThoughts = &value
-			}
-			return baseModel, intent, true, nil
+		intent := Intent{BudgetTokens: &budget, Source: SourceSuffix, BudgetSource: SourceSuffix}
+		if includeThoughts {
+			value := true
+			intent.IncludeThoughts = &value
 		}
+		return baseModel, intent, true, nil
+	}
+	if baseModel, ok := strings.CutSuffix(modelName, "-nothinking"); ok {
+		return baseModel, Intent{Mode: ModeDisabled, Effort: EffortNone, Source: SourceSuffix}, true, nil
+	}
+	if baseModel, ok := strings.CutSuffix(modelName, "-thinking"); ok {
+		intent := Intent{Mode: ModeEnabled, Source: SourceSuffix}
+		if includeThoughts {
+			value := true
+			intent.IncludeThoughts = &value
+		}
+		return baseModel, intent, true, nil
 	}
 
 	suffixes := []string{"-max", "-xhigh", "-high", "-medium", "-low", "-minimal", "-none"}
@@ -266,16 +274,19 @@ func parseProviderModelSuffix(modelName string, requiredPrefix string, allowThin
 	return baseModel, intent, true, nil
 }
 
+// ParseDeepSeekV4ThinkingSuffix maps the -none/-low/-max tails of
+// deepseek-flash and deepseek-v4-* onto DeepSeek's thinking switch and
+// effort. A bare name keeps DeepSeek's default (thinking on, effort high).
 func ParseDeepSeekV4ThinkingSuffix(modelName string) (baseModel string, thinkingType string, effort string, ok bool) {
 	baseModel, suffix, ok := TrimEffortSuffixWithSuffixes(modelName, DeepSeekV4EffortSuffixes)
-	if !ok || !strings.HasPrefix(baseModel, "deepseek-v4-") {
+	if !ok || baseModel != "deepseek-flash" && !strings.HasPrefix(baseModel, "deepseek-v4-") {
 		return modelName, "", "", false
 	}
 	switch suffix {
 	case "none":
 		return baseModel, "disabled", "", true
-	case "max":
-		return baseModel, "enabled", "max", true
+	case "low", "max":
+		return baseModel, "enabled", suffix, true
 	default:
 		return modelName, "", "", false
 	}