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

feat(translator): improve signature handling by associating with model name in cache functions

Browse Source
This commit is contained in:
Luis Pater
2026-01-20 13:31:36 +08:00
parent 6184c43319
commit 020e61d0da
3 changed files with 31 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/cache/signature_cache.go vendored
View File

@@ -3,6 +3,7 @@ package cache
import (
"crypto/sha256"
"encoding/hex"
"fmt"
"sync"
"time"
)
@@ -94,7 +95,7 @@ func purgeExpiredSessions() {
// CacheSignature stores a thinking signature for a given session and text.
// Used for Claude models that require signed thinking blocks in multi-turn conversations.
func CacheSignature(sessionID, text, signature string) {
func CacheSignature(modelName, sessionID, text, signature string) {
if sessionID == "" || text == "" || signature == "" {
return
}
@@ -102,7 +103,7 @@ func CacheSignature(sessionID, text, signature string) {
return
}
sc := getOrCreateSession(sessionID)
sc := getOrCreateSession(fmt.Sprintf("%s#%s", modelName, sessionID))
textHash := hashText(text)
sc.mu.Lock()
@@ -116,12 +117,12 @@ func CacheSignature(sessionID, text, signature string) {
// GetCachedSignature retrieves a cached signature for a given session and text.
// Returns empty string if not found or expired.
func GetCachedSignature(sessionID, text string) string {
func GetCachedSignature(modelName, sessionID, text string) string {
if sessionID == "" || text == "" {
return ""
}
val, ok := signatureCache.Load(sessionID)
val, ok := signatureCache.Load(fmt.Sprintf("%s#%s", modelName, sessionID))
if !ok {
return ""
}