eyre
248236ce6d
问题： 探针拦截（suggestion mode / warmup / max_tokens=1 haiku）的模拟响应以及 Gemini/Antigravity 兼容层生成的 message ID 不符合 Anthropic 官方 API 格式， 容易被客户端识别为非正宗响应。 修复： 1. generateRealisticMsgID()：msg_bdrk_ + 24字符 → msg_01 + 22位 Base62 （与官方 API 返回的 msg_011CdS6b8gAhoKWdW9jE87Zs 格式一致） 2. 去掉固定的 msg_mock_suggestion / msg_mock_warmup，统一使用随机 ID 3. 流式响应格式对齐官方： - message_start 增加 stop_details/cache token 字段 - content_block_start 字段顺序修正 - message_delta.usage 只含 output_tokens 4. 非流式响应：增加 stop_details:null，移除非标准 total_tokens 5. Gemini Messages/ChatCompletions 兼容层：msg_ + hex → msg_01 + Base62 6. Antigravity response/stream transformer：msg_ + 12位 → msg_01 + 22位 Base62 验证方式：对照 Anthropic 官方 API 实际响应格式确认。
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDetectInterceptType_MaxTokensOneHaikuRequiresClaudeCodeClient(t *testing.T) {
|
|
body := []byte(`{"messages":[{"role":"user","content":[{"type":"text","text":"hello"}]}]}`)
|
|
|
|
notClaudeCode := detectInterceptType(body, "claude-haiku-4-5", 1, false)
|
|
require.Equal(t, InterceptTypeNone, notClaudeCode)
|
|
|
|
isClaudeCode := detectInterceptType(body, "claude-haiku-4-5", 1, true)
|
|
require.Equal(t, InterceptTypeMaxTokensOneHaiku, isClaudeCode)
|
|
}
|
|
|
|
func TestDetectInterceptType_SuggestionModeUnaffected(t *testing.T) {
|
|
body := []byte(`{
|
|
"messages":[{
|
|
"role":"user",
|
|
"content":[{"type":"text","text":"[SUGGESTION MODE:foo]"}]
|
|
}],
|
|
"system":[]
|
|
}`)
|
|
|
|
got := detectInterceptType(body, "claude-sonnet-4-5", 256, false)
|
|
require.Equal(t, InterceptTypeSuggestionMode, got)
|
|
}
|
|
|
|
func TestSendMockInterceptResponse_MaxTokensOneHaiku(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
rec := httptest.NewRecorder()
|
|
ctx, _ := gin.CreateTestContext(rec)
|
|
|
|
sendMockInterceptResponse(ctx, "claude-haiku-4-5", InterceptTypeMaxTokensOneHaiku)
|
|
|
|
require.Equal(t, http.StatusOK, rec.Code)
|
|
|
|
var response map[string]any
|
|
require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &response))
|
|
require.Equal(t, "max_tokens", response["stop_reason"])
|
|
|
|
id, ok := response["id"].(string)
|
|
require.True(t, ok)
|
|
require.True(t, strings.HasPrefix(id, "msg_01"))
|
|
|
|
content, ok := response["content"].([]any)
|
|
require.True(t, ok)
|
|
require.NotEmpty(t, content)
|
|
|
|
firstBlock, ok := content[0].(map[string]any)
|
|
require.True(t, ok)
|
|
require.Equal(t, "#", firstBlock["text"])
|
|
|
|
usage, ok := response["usage"].(map[string]any)
|
|
require.True(t, ok)
|
|
require.Equal(t, float64(1), usage["output_tokens"])
|
|
}
|