Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
5.1 kB · 125 lines
Go
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126package test
import ( "context" "strings" "testing"
_ "github.com/router-for-me/CLIProxyAPI/v7/internal/translator" sdktranslator "github.com/router-for-me/CLIProxyAPI/v7/sdk/translator" "github.com/tidwall/gjson")
func TestCodexToClaudeParallelFunctionCallsHaveValidLifecycle(t *testing.T) { chunks := [][]byte{ []byte(`data: {"type":"response.created","response":{"id":"resp_parallel","model":"gpt-5"}}`), []byte(`data: {"type":"response.output_item.added","item":{"type":"function_call","call_id":"call_a","name":"Read"},"output_index":1}`), []byte(`data: {"type":"response.output_item.added","item":{"type":"function_call","call_id":"call_b","name":"Read"},"output_index":2}`), []byte(`data: {"type":"response.function_call_arguments.delta","delta":"{\"file_path\":\"a\"}","output_index":1}`), []byte(`data: {"type":"response.function_call_arguments.done","arguments":"{\"file_path\":\"a\"}","output_index":1}`), []byte(`data: {"type":"response.output_item.done","item":{"type":"function_call","call_id":"call_a","name":"Read","arguments":"{\"file_path\":\"a\"}"},"output_index":1}`), []byte(`data: {"type":"response.function_call_arguments.delta","delta":"{\"file_path\":\"b\"}","output_index":2}`), []byte(`data: {"type":"response.function_call_arguments.done","arguments":"{\"file_path\":\"b\"}","output_index":2}`), []byte(`data: {"type":"response.output_item.done","item":{"type":"function_call","call_id":"call_b","name":"Read","arguments":"{\"file_path\":\"b\"}"},"output_index":2}`), []byte(`data: {"type":"response.completed","response":{"usage":{"input_tokens":1,"output_tokens":1},"output":[{"type":"function_call","call_id":"call_a","name":"Read","arguments":"{\"file_path\":\"a\"}"},{"type":"function_call","call_id":"call_b","name":"Read","arguments":"{\"file_path\":\"b\"}"}]}}`), }
originalRequest := []byte(`{"stream":true,"tools":[{"name":"Read"}]}`) var state any open := make(map[int64]struct{}) started := make(map[int64]struct{}) toolIDs := make(map[int64]string) arguments := make(map[int64]string) var startIndices []int64 var stopIndices []int64 messageState := 0
for _, chunk := range chunks { outputs := sdktranslator.TranslateStream( context.Background(), sdktranslator.FormatCodex, sdktranslator.FormatClaude, "gpt-5", originalRequest, nil, chunk, &state, ) for _, output := range outputs { for _, line := range strings.Split(string(output), "\n") { if !strings.HasPrefix(line, "data: ") { continue } event := gjson.Parse(strings.TrimPrefix(line, "data: ")) if messageState == 2 { t.Fatalf("event emitted after message_stop: %s", event.Raw) } index := event.Get("index").Int() switch event.Get("type").String() { case "content_block_start": if messageState != 0 { t.Fatalf("content block started after message terminal events: %s", event.Raw) } if len(open) != 0 { t.Fatalf("content block start emitted while another block remains open: %v", open) } if _, exists := started[index]; exists { t.Fatalf("content block index %d was reused", index) } open[index] = struct{}{} started[index] = struct{}{} startIndices = append(startIndices, index) toolIDs[index] = event.Get("content_block.id").String() case "content_block_delta": if _, exists := open[index]; !exists { t.Fatalf("content block delta targets unopened index %d", index) } if event.Get("delta.type").String() == "input_json_delta" { arguments[index] += event.Get("delta.partial_json").String() } case "content_block_stop": if _, exists := open[index]; !exists { t.Fatalf("content block stop targets unopened index %d", index) } delete(open, index) stopIndices = append(stopIndices, index) case "message_delta": if len(open) != 0 { t.Fatalf("message_delta emitted while content blocks remain open: %v", open) } if messageState != 0 { t.Fatalf("duplicate or out-of-order message_delta: %s", event.Raw) } messageState = 1 case "message_stop": if len(open) != 0 { t.Fatalf("message_stop emitted while content blocks remain open: %v", open) } if messageState != 1 { t.Fatalf("message_stop emitted before message_delta: %s", event.Raw) } messageState = 2 } } } }
if len(open) != 0 { t.Fatalf("content blocks remain open: %v", open) } if messageState != 2 { t.Fatalf("terminal message event state = %d, want message_delta followed by message_stop", messageState) } if len(startIndices) != 2 || startIndices[0] != 0 || startIndices[1] != 1 { t.Fatalf("start indices = %v, want [0 1]", startIndices) } if len(stopIndices) != 2 || stopIndices[0] != 0 || stopIndices[1] != 1 { t.Fatalf("stop indices = %v, want [0 1]", stopIndices) } if toolIDs[0] != "call_a" || toolIDs[1] != "call_b" { t.Fatalf("tool IDs = %v, want call_a and call_b", toolIDs) } if arguments[0] != `{"file_path":"a"}` || arguments[1] != `{"file_path":"b"}` { t.Fatalf("tool arguments = %v", arguments) }}