fix(translator): update HasValidSignature to require modelName parameter for improved validation
This commit is contained in:
4
internal/cache/signature_cache.go
vendored
4
internal/cache/signature_cache.go
vendored
@@ -186,8 +186,8 @@ func ClearSignatureCache(sessionID string) {
|
||||
}
|
||||
|
||||
// HasValidSignature checks if a signature is valid (non-empty and long enough)
|
||||
func HasValidSignature(signature string) bool {
|
||||
return signature != "" && len(signature) >= MinValidSignatureLen
|
||||
func HasValidSignature(modelName, signature string) bool {
|
||||
return (signature != "" && len(signature) >= MinValidSignatureLen) || (signature == "skip_thought_signature_validator" && GetModelGroup(modelName) == "gemini")
|
||||
}
|
||||
|
||||
func GetModelGroup(modelName string) string {
|
||||
|
||||
2
internal/cache/signature_cache_test.go
vendored
2
internal/cache/signature_cache_test.go
vendored
@@ -132,7 +132,7 @@ func TestHasValidSignature(t *testing.T) {
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := HasValidSignature(tt.signature)
|
||||
result := HasValidSignature("claude-sonnet-4-5-thinking", tt.signature)
|
||||
if result != tt.expected {
|
||||
t.Errorf("HasValidSignature(%q) = %v, expected %v", tt.signature, result, tt.expected)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user