Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
11 kB · 303 lines
Go
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304package executor
import ( "context" "fmt" "io" "net/http" "net/http/httptest" "strings" "testing"
"github.com/gin-gonic/gin" "github.com/router-for-me/CLIProxyAPI/v7/internal/config" "github.com/router-for-me/CLIProxyAPI/v7/internal/registry" cliproxyauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth" cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor" sdktranslator "github.com/router-for-me/CLIProxyAPI/v7/sdk/translator" "github.com/tidwall/gjson")
func TestCodexExecutorOptimizeMultiAgentV2(t *testing.T) { modelID := "codex-executor-spawn-agent-test-model" clientID := "codex-executor-spawn-agent-test-client" modelRegistry := registry.GetGlobalRegistry() modelRegistry.RegisterClient(clientID, "codex", []*registry.ModelInfo{{ ID: modelID, Description: "Executor test model.", Thinking: ®istry.ThinkingSupport{ Levels: []string{"low", "medium", "high"}, }, }}) defer modelRegistry.UnregisterClient(clientID)
var upstreamBody []byte server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, request *http.Request) { upstreamBody, _ = io.ReadAll(request.Body) if request.URL.Path == "/responses/compact" { w.Header().Set("Content-Type", "application/json") namespace := gjson.GetBytes(upstreamBody, "input.0.tools.0.name").String() compact := fmt.Sprintf(`{"id":"resp_1","object":"response.compaction","output":[{"type":"function_call","name":"spawn_agent","namespace":%q,"arguments":"{}","call_id":"call_1"}]}`, namespace) _, _ = w.Write([]byte(compact)) return } w.Header().Set("Content-Type", "text/event-stream") namespace := gjson.GetBytes(upstreamBody, "input.0.tools.0.name").String() completed := fmt.Sprintf(`data: {"type":"response.completed","response":{"id":"resp_1","object":"response","status":"completed","output":[{"type":"function_call","name":"spawn_agent","namespace":%q,"arguments":"{}","call_id":"call_1"}]}}`+"\n\n", namespace) _, _ = w.Write([]byte(completed)) })) defer server.Close()
payload := codexSpawnAgentTestPayload() auth := &cliproxyauth.Auth{Attributes: map[string]string{ "base_url": server.URL, "api_key": "test", }}
tests := []struct { name string enabled bool mode string }{ {name: "execute enabled", enabled: true, mode: "execute"}, {name: "execute disabled", enabled: false, mode: "execute"}, {name: "stream enabled", enabled: true, mode: "stream"}, {name: "stream disabled", enabled: false, mode: "stream"}, {name: "compact enabled", enabled: true, mode: "compact"}, {name: "compact disabled", enabled: false, mode: "compact"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { upstreamBody = nil executor := NewCodexExecutor(&config.Config{Codex: config.CodexConfig{OptimizeMultiAgentV2: tt.enabled}}) ctx := codexSpawnAgentTestContext() headers := http.Header{"User-Agent": []string{"overridden-client/1.0"}} req := cliproxyexecutor.Request{Model: "gpt-5.4", Payload: payload} opts := cliproxyexecutor.Options{SourceFormat: sdktranslator.FromString("openai-response"), Headers: headers}
var clientPayload []byte switch tt.mode { case "stream": result, errExecute := executor.ExecuteStream(ctx, auth, req, opts) if errExecute != nil { t.Fatalf("ExecuteStream() error = %v", errExecute) } for chunk := range result.Chunks { clientPayload = append(clientPayload, chunk.Payload...) } case "compact": opts.Alt = "responses/compact" response, errExecute := executor.Execute(ctx, auth, req, opts) if errExecute != nil { t.Fatalf("compact Execute() error = %v", errExecute) } clientPayload = response.Payload default: response, errExecute := executor.Execute(ctx, auth, req, opts) if errExecute != nil { t.Fatalf("Execute() error = %v", errExecute) } clientPayload = response.Payload }
assertCodexSpawnAgentOptimization(t, upstreamBody, modelID, tt.enabled) assertCodexSpawnAgentRequestMessage(t, upstreamBody, tt.enabled) assertCodexSpawnAgentClientNamespace(t, clientPayload) }) }}
func TestCodexExecutorIsCompatConvertsAgentMessage(t *testing.T) { var upstreamBody []byte server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, request *http.Request) { upstreamBody, _ = io.ReadAll(request.Body) w.Header().Set("Content-Type", "text/event-stream") _, _ = w.Write([]byte(`data: {"type":"response.completed","response":{"id":"resp_1","object":"response","status":"completed","output":[]}}` + "\n\n")) })) defer server.Close()
payload := codexSpawnAgentTestPayload() baseCfg := config.Config{ Codex: config.CodexConfig{OptimizeMultiAgentV2: true}, CodexKey: []config.CodexKey{{ APIKey: "test", BaseURL: server.URL, Models: []config.CodexModel{ {Name: "deepseek-v4-flash", Alias: "deepseek-alias", IsCompat: true}, {Name: "gpt-5.4", Alias: "codex-native"}, }, }}, } auth := &cliproxyauth.Auth{ Provider: "codex", Attributes: map[string]string{ "base_url": server.URL, "api_key": "test", }, }
tests := []struct { name string model string enabled bool wantType string wantRole string wantRoleExists bool }{ { name: "is-compat converts agent_message", model: "deepseek-v4-flash", enabled: true, wantType: "message", wantRole: "user", wantRoleExists: true, }, { name: "native model keeps agent_message", model: "gpt-5.4", enabled: true, wantType: "agent_message", wantRoleExists: false, }, { name: "optimize disabled keeps agent_message", model: "deepseek-v4-flash", enabled: false, wantType: "agent_message", wantRoleExists: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { upstreamBody = nil cfg := baseCfg cfg.Codex.OptimizeMultiAgentV2 = tt.enabled executor := NewCodexExecutor(&cfg) ctx := codexSpawnAgentTestContext() req := cliproxyexecutor.Request{Model: tt.model, Payload: payload} opts := cliproxyexecutor.Options{ SourceFormat: sdktranslator.FromString("openai-response"), Headers: http.Header{"User-Agent": []string{"overridden-client/1.0"}}, } if _, errExecute := executor.Execute(ctx, auth, req, opts); errExecute != nil { t.Fatalf("Execute() error = %v", errExecute) } message := gjson.GetBytes(upstreamBody, "input.1") if message.Get("type").String() != tt.wantType { t.Fatalf("input.1.type = %q, want %q; body=%s", message.Get("type").String(), tt.wantType, upstreamBody) } if tt.wantRoleExists { if message.Get("role").String() != tt.wantRole { t.Fatalf("input.1.role = %q, want %q; body=%s", message.Get("role").String(), tt.wantRole, upstreamBody) } if message.Get("content.1.type").String() != "input_text" || message.Get("content.1.text").String() != "delegated task" { t.Fatalf("compat conversion did not normalize content: %s", upstreamBody) } return } if message.Get("role").Exists() { t.Fatalf("input.1.role unexpectedly present: %s", upstreamBody) } }) }}
func codexSpawnAgentTestPayload() []byte { return []byte(`{ "model":"gpt-5.4", "input":[{ "type":"additional_tools", "role":"developer", "tools":[{ "type":"namespace", "name":"collaboration", "tools":[{ "type":"function", "name":"spawn_agent", "description":"Available model overrides (optional; inherited parent model is preferred):\n- old-model\nSpawns an agent.", "parameters":{"type":"object","properties":{"message":{"type":"string","encrypted":true}}} }] }] },{ "type":"agent_message", "id":"amsg_1", "author":"/root", "recipient":"/root/worker", "content":[ {"type":"input_text","text":"Payload:\n"}, {"type":"encrypted_content","encrypted_content":"delegated task"} ], "internal_chat_message_metadata_passthrough":{"turn_id":"turn_1"} }] }`)}
func codexSpawnAgentTestContext() context.Context { request := httptest.NewRequest(http.MethodPost, "/v1/responses", nil) request.Header.Set("User-Agent", "codex-tui/0.153.3") ginCtx, _ := gin.CreateTestContext(httptest.NewRecorder()) ginCtx.Request = request return context.WithValue(context.Background(), "gin", ginCtx)}
func assertCodexSpawnAgentClientNamespace(t *testing.T, payload []byte) { t.Helper() if strings.Contains(string(payload), "collaboration-optimize") { t.Fatalf("optimized namespace leaked to client: %s", payload) } if !strings.Contains(string(payload), `"namespace":"collaboration"`) { t.Fatalf("restored collaboration namespace missing from client payload: %s", payload) }}
func assertCodexSpawnAgentRequestMessage(t *testing.T, payload []byte, enabled bool) { t.Helper() message := gjson.GetBytes(payload, "input.1") if message.Get("type").String() != "agent_message" || message.Get("role").Exists() { t.Fatalf("Codex executor changed outer agent message: %s", payload) } if message.Get("author").String() != "/root" || message.Get("recipient").String() != "/root/worker" || message.Get("internal_chat_message_metadata_passthrough.turn_id").String() != "turn_1" { t.Fatalf("Codex executor changed agent message metadata: %s", payload) } if enabled { if message.Get("content.1.type").String() != "input_text" || message.Get("content.1.text").String() != "delegated task" { t.Fatalf("Codex executor did not normalize agent message content: %s", payload) } if message.Get("content.1.encrypted_content").Exists() { t.Fatalf("Codex executor preserved encrypted_content: %s", payload) } return } if message.Get("content.1.type").String() != "encrypted_content" || message.Get("content.1.encrypted_content").String() != "delegated task" { t.Fatalf("disabled optimization changed agent message content: %s", payload) }}
func assertCodexSpawnAgentOptimization(t *testing.T, payload []byte, modelID string, enabled bool) { t.Helper() namespace := gjson.GetBytes(payload, "input.0.tools.0.name").String() description := gjson.GetBytes(payload, "input.0.tools.0.tools.0.description").String() encrypted := gjson.GetBytes(payload, "input.0.tools.0.tools.0.parameters.properties.message.encrypted") if enabled { if namespace != "collaboration-optimize" { t.Fatalf("optimized namespace = %q, want collaboration-optimize", namespace) } wantModel := "- `" + modelID + "`: Executor test model. Reasoning efforts: low, medium (default), high." if !strings.Contains(description, wantModel) { t.Fatalf("description does not contain model metadata: %q", description) } if encrypted.Exists() { t.Fatalf("message encrypted was not removed: %s", encrypted.Raw) } return } if namespace != "collaboration" { t.Fatalf("disabled namespace = %q, want collaboration", namespace) } if !strings.Contains(description, "- old-model") { t.Fatalf("disabled optimization changed description: %q", description) } if !encrypted.Bool() { t.Fatalf("disabled optimization removed message encrypted: %s", encrypted.Raw) }}