LLM/CLIProxyAPIPlus
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPIPlus.git synced 2026-04-17 20:03:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(codex): preserve continuity and safe affinity fallback

Browse Source 
Restore Claude continuity after the continuity refactor, keep auth-affinity keys out of upstream Codex session identifiers, and only persist affinity after successful execution so retries can still rotate to healthy credentials when the first auth fails.
This commit is contained in:
VooDisss
2026-03-27 18:27:33 +02:00
parent 62b17f40a1
commit 26eca8b6ba
6 changed files with 79 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
internal/runtime/executor/codex_executor_cache_test.go
View File

@@ -151,3 +151,45 @@ func TestCodexExecutorCacheHelper_OpenAIChatCompletions_FallsBackToStableAuthID(
t.Fatalf("session_id = %q, want %q", got, expected)
}
}
func TestCodexExecutorCacheHelper_ClaudePreservesCacheContinuity(t *testing.T) {
executor := &CodexExecutor{}
req := cliproxyexecutor.Request{
Model: "claude-3-7-sonnet",
Payload: []byte(`{"metadata":{"user_id":"user-1"}}`),
}
rawJSON := []byte(`{"model":"gpt-5.4","stream":true}`)
httpReq, continuity, err := executor.cacheHelper(context.Background(), nil, sdktranslator.FromString("claude"), "https://example.com/responses", req, cliproxyexecutor.Options{}, rawJSON)
if err != nil {
t.Fatalf("cacheHelper error: %v", err)
}
if continuity.Key == "" {
t.Fatal("continuity.Key = empty, want non-empty")
}
body, err := io.ReadAll(httpReq.Body)
if err != nil {
t.Fatalf("read request body: %v", err)
}
if got := gjson.GetBytes(body, "prompt_cache_key").String(); got != continuity.Key {
t.Fatalf("prompt_cache_key = %q, want %q", got, continuity.Key)
}
if got := httpReq.Header.Get("session_id"); got != continuity.Key {
t.Fatalf("session_id = %q, want %q", got, continuity.Key)
}
}
func TestResolveCodexContinuity_DoesNotForwardAuthAffinityKey(t *testing.T) {
req := cliproxyexecutor.Request{Payload: []byte(`{"model":"gpt-5.4"}`)}
opts := cliproxyexecutor.Options{Metadata: map[string]any{"auth_affinity_key": "principal:raw-client-secret"}}
auth := &cliproxyauth.Auth{ID: "codex-auth-1", Provider: "codex"}
continuity := resolveCodexContinuity(context.Background(), auth, req, opts)
if continuity.Source != "auth_id" {
t.Fatalf("continuity.Source = %q, want %q", continuity.Source, "auth_id")
}
if continuity.Key == "principal:raw-client-secret" {
t.Fatal("continuity.Key leaked raw auth affinity key")
}
}