chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-19 04:40:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(translator): preserve thinking blocks by skipping signature

Browse Source
This commit is contained in:
hkfires
2026-01-21 10:49:03 +08:00
parent 6dcbbf64c3
commit 9332316383
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
internal/translator/antigravity/gemini/antigravity_gemini_request.go
View File

@@ -98,19 +98,20 @@ func ConvertGeminiRequestToAntigravity(modelName string, inputRawJSON []byte, _
} }
} }
// Gemini-specific handling: add skip_thought_signature_validator to functionCall parts // Gemini-specific handling for non-Claude models:
// and remove thinking blocks entirely (Gemini doesn't need to preserve them) // - Add skip_thought_signature_validator to functionCall parts so upstream can bypass signature validation.
// - Also mark thinking parts with the same sentinel when present (we keep the parts; we only annotate them).
if !strings.Contains(modelName, "claude") { if !strings.Contains(modelName, "claude") {
const skipSentinel = "skip_thought_signature_validator" const skipSentinel = "skip_thought_signature_validator"
gjson.GetBytes(rawJSON, "request.contents").ForEach(func(contentIdx, content gjson.Result) bool { gjson.GetBytes(rawJSON, "request.contents").ForEach(func(contentIdx, content gjson.Result) bool {
if content.Get("role").String() == "model" { if content.Get("role").String() == "model" {
// First pass: collect indices of thinking parts to remove // First pass: collect indices of thinking parts to mark with skip sentinel
var thinkingIndicesToRemove []int64 var thinkingIndicesToSkipSignature []int64
content.Get("parts").ForEach(func(partIdx, part gjson.Result) bool { content.Get("parts").ForEach(func(partIdx, part gjson.Result) bool {
// Mark thinking blocks for removal // Collect indices of thinking blocks to mark with skip sentinel
if part.Get("thought").Bool() { if part.Get("thought").Bool() {
thinkingIndicesToRemove = append(thinkingIndicesToRemove, partIdx.Int()) thinkingIndicesToSkipSignature = append(thinkingIndicesToSkipSignature, partIdx.Int())
} }
// Add skip sentinel to functionCall parts // Add skip sentinel to functionCall parts
if part.Get("functionCall").Exists() { if part.Get("functionCall").Exists() {
@@ -122,10 +123,10 @@ func ConvertGeminiRequestToAntigravity(modelName string, inputRawJSON []byte, _
return true return true
}) })
// Remove thinking blocks in reverse order to preserve indices // Add skip_thought_signature_validator sentinel to thinking blocks in reverse order to preserve indices
for i := len(thinkingIndicesToRemove) - 1; i >= 0; i-- { for i := len(thinkingIndicesToSkipSignature) - 1; i >= 0; i-- {
idx := thinkingIndicesToRemove[i] idx := thinkingIndicesToSkipSignature[i]
rawJSON, _ = sjson.DeleteBytes(rawJSON, fmt.Sprintf("request.contents.%d.parts.%d", contentIdx.Int(), idx)) rawJSON, _ = sjson.SetBytes(rawJSON, fmt.Sprintf("request.contents.%d.parts.%d.thoughtSignature", contentIdx.Int(), idx), skipSentinel)
} }
} }
return true return true