lanyuanxiaoyao
b7e205f4b6
## 主要变更 **核心修改**： - 路由定义：/:protocol/v1/*path → /:protocol/*path - proxy_handler：nativePath 直接使用 path 参数，不添加 /v1 前缀 - OpenAI 适配器：DetectInterfaceType 和 BuildUrl 去掉 /v1 前缀 - Anthropic 适配器：保持 /v1 前缀（Claude Code 兼容） **URL 格式变化**： - OpenAI: /openai/v1/chat/completions → /openai/chat/completions - Anthropic: /anthropic/v1/messages (保持不变) **base_url 配置**： - OpenAI: 配置到版本路径，如 https://api.openai.com/v1 - Anthropic: 不配置版本路径，如 https://api.anthropic.com ## 测试验证 - 所有单元测试通过 - 所有集成测试通过 - 真实 API 测试验证成功 - 跨协议转换正常工作 ## 文档更新 - 更新 backend/README.md URL 格式说明 - 同步 OpenSpec 规范文件
634 lines
20 KiB
Go
634 lines
20 KiB
Go
package conversion
|
||
|
||
import (
|
||
"encoding/json"
|
||
"testing"
|
||
|
||
"nex/backend/internal/conversion/canonical"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
"github.com/stretchr/testify/require"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
// mockProtocolAdapter 模拟协议适配器
|
||
type mockProtocolAdapter struct {
|
||
protocolName string
|
||
passthrough bool
|
||
ifaceType InterfaceType
|
||
supportsIface map[InterfaceType]bool
|
||
decodeReqFn func([]byte) (*canonical.CanonicalRequest, error)
|
||
encodeReqFn func(*canonical.CanonicalRequest, *TargetProvider) ([]byte, error)
|
||
decodeRespFn func([]byte) (*canonical.CanonicalResponse, error)
|
||
encodeRespFn func(*canonical.CanonicalResponse) ([]byte, error)
|
||
streamDecoderFn func() StreamDecoder
|
||
streamEncoderFn func() StreamEncoder
|
||
rewriteReqFn func([]byte, string, InterfaceType) ([]byte, error)
|
||
rewriteRespFn func([]byte, string, InterfaceType) ([]byte, error)
|
||
}
|
||
|
||
func newMockAdapter(name string, passthrough bool) *mockProtocolAdapter {
|
||
return &mockProtocolAdapter{
|
||
protocolName: name,
|
||
passthrough: passthrough,
|
||
ifaceType: InterfaceTypeChat,
|
||
supportsIface: map[InterfaceType]bool{},
|
||
}
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) ProtocolName() string { return m.protocolName }
|
||
func (m *mockProtocolAdapter) ProtocolVersion() string { return "1.0" }
|
||
func (m *mockProtocolAdapter) SupportsPassthrough() bool { return m.passthrough }
|
||
|
||
func (m *mockProtocolAdapter) DetectInterfaceType(nativePath string) InterfaceType {
|
||
return m.ifaceType
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) BuildUrl(nativePath string, interfaceType InterfaceType) string {
|
||
return nativePath
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) BuildHeaders(provider *TargetProvider) map[string]string {
|
||
return map[string]string{"Authorization": "Bearer " + provider.APIKey}
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) SupportsInterface(interfaceType InterfaceType) bool {
|
||
if v, ok := m.supportsIface[interfaceType]; ok {
|
||
return v
|
||
}
|
||
return interfaceType == InterfaceTypeChat
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeRequest(raw []byte) (*canonical.CanonicalRequest, error) {
|
||
if m.decodeReqFn != nil {
|
||
return m.decodeReqFn(raw)
|
||
}
|
||
req := &canonical.CanonicalRequest{}
|
||
_ = json.Unmarshal(raw, req)
|
||
return req, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeRequest(req *canonical.CanonicalRequest, provider *TargetProvider) ([]byte, error) {
|
||
if m.encodeReqFn != nil {
|
||
return m.encodeReqFn(req, provider)
|
||
}
|
||
return json.Marshal(req)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeResponse(raw []byte) (*canonical.CanonicalResponse, error) {
|
||
if m.decodeRespFn != nil {
|
||
return m.decodeRespFn(raw)
|
||
}
|
||
resp := &canonical.CanonicalResponse{}
|
||
_ = json.Unmarshal(raw, resp)
|
||
return resp, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeResponse(resp *canonical.CanonicalResponse) ([]byte, error) {
|
||
if m.encodeRespFn != nil {
|
||
return m.encodeRespFn(resp)
|
||
}
|
||
return json.Marshal(resp)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) CreateStreamDecoder() StreamDecoder {
|
||
if m.streamDecoderFn != nil {
|
||
return m.streamDecoderFn()
|
||
}
|
||
return &noopStreamDecoder{}
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) CreateStreamEncoder() StreamEncoder {
|
||
if m.streamEncoderFn != nil {
|
||
return m.streamEncoderFn()
|
||
}
|
||
return &noopStreamEncoder{}
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeError(err *ConversionError) ([]byte, int) {
|
||
return []byte(`{"error":"mock"}`), 400
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeModelsResponse(raw []byte) (*canonical.CanonicalModelList, error) {
|
||
return &canonical.CanonicalModelList{}, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeModelsResponse(list *canonical.CanonicalModelList) ([]byte, error) {
|
||
return json.Marshal(list)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeModelInfoResponse(raw []byte) (*canonical.CanonicalModelInfo, error) {
|
||
return &canonical.CanonicalModelInfo{}, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeModelInfoResponse(info *canonical.CanonicalModelInfo) ([]byte, error) {
|
||
return json.Marshal(info)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeEmbeddingRequest(raw []byte) (*canonical.CanonicalEmbeddingRequest, error) {
|
||
return &canonical.CanonicalEmbeddingRequest{}, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeEmbeddingRequest(req *canonical.CanonicalEmbeddingRequest, provider *TargetProvider) ([]byte, error) {
|
||
return json.Marshal(req)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeEmbeddingResponse(raw []byte) (*canonical.CanonicalEmbeddingResponse, error) {
|
||
return &canonical.CanonicalEmbeddingResponse{}, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeEmbeddingResponse(resp *canonical.CanonicalEmbeddingResponse) ([]byte, error) {
|
||
return json.Marshal(resp)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeRerankRequest(raw []byte) (*canonical.CanonicalRerankRequest, error) {
|
||
return &canonical.CanonicalRerankRequest{}, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeRerankRequest(req *canonical.CanonicalRerankRequest, provider *TargetProvider) ([]byte, error) {
|
||
return json.Marshal(req)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) DecodeRerankResponse(raw []byte) (*canonical.CanonicalRerankResponse, error) {
|
||
return &canonical.CanonicalRerankResponse{}, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) EncodeRerankResponse(resp *canonical.CanonicalRerankResponse) ([]byte, error) {
|
||
return json.Marshal(resp)
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) ExtractUnifiedModelID(nativePath string) (string, error) {
|
||
return "", nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) ExtractModelName(body []byte, ifaceType InterfaceType) (string, error) {
|
||
return "", nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) RewriteRequestModelName(body []byte, newModel string, ifaceType InterfaceType) ([]byte, error) {
|
||
if m.rewriteReqFn != nil {
|
||
return m.rewriteReqFn(body, newModel, ifaceType)
|
||
}
|
||
return body, nil
|
||
}
|
||
|
||
func (m *mockProtocolAdapter) RewriteResponseModelName(body []byte, newModel string, ifaceType InterfaceType) ([]byte, error) {
|
||
if m.rewriteRespFn != nil {
|
||
return m.rewriteRespFn(body, newModel, ifaceType)
|
||
}
|
||
return body, nil
|
||
}
|
||
|
||
// noopStreamDecoder 空流式解码器
|
||
type noopStreamDecoder struct{}
|
||
|
||
func (d *noopStreamDecoder) ProcessChunk(rawChunk []byte) []canonical.CanonicalStreamEvent { return nil }
|
||
func (d *noopStreamDecoder) Flush() []canonical.CanonicalStreamEvent { return nil }
|
||
|
||
// noopStreamEncoder 空流式编码器
|
||
type noopStreamEncoder struct{}
|
||
|
||
func (e *noopStreamEncoder) EncodeEvent(event canonical.CanonicalStreamEvent) [][]byte { return nil }
|
||
func (e *noopStreamEncoder) Flush() [][]byte { return nil }
|
||
|
||
// ============ 测试用例 ============
|
||
|
||
func TestNewConversionEngine(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
assert.NotNil(t, engine)
|
||
assert.Equal(t, registry, engine.GetRegistry())
|
||
}
|
||
|
||
func TestNewConversionEngine_LoggerInjection(t *testing.T) {
|
||
t.Run("nil_logger_uses_global", func(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
assert.NotNil(t, engine.logger)
|
||
})
|
||
|
||
t.Run("custom_logger", func(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
customLogger := zap.NewNop()
|
||
engine := NewConversionEngine(registry, customLogger)
|
||
assert.Equal(t, customLogger, engine.logger)
|
||
})
|
||
}
|
||
|
||
func TestRegisterAdapter(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
adapter := newMockAdapter("test-proto", true)
|
||
err := engine.RegisterAdapter(adapter)
|
||
require.NoError(t, err)
|
||
|
||
protocols := registry.ListProtocols()
|
||
assert.Contains(t, protocols, "test-proto")
|
||
}
|
||
|
||
func TestIsPassthrough_SameProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
adapter := newMockAdapter("openai", true)
|
||
_ = engine.RegisterAdapter(adapter)
|
||
|
||
assert.True(t, engine.IsPassthrough("openai", "openai"))
|
||
}
|
||
|
||
func TestIsPassthrough_DifferentProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("openai", true))
|
||
_ = engine.RegisterAdapter(newMockAdapter("anthropic", true))
|
||
|
||
assert.False(t, engine.IsPassthrough("openai", "anthropic"))
|
||
}
|
||
|
||
func TestIsPassthrough_NoPassthrough(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("custom", false))
|
||
|
||
assert.False(t, engine.IsPassthrough("custom", "custom"))
|
||
}
|
||
|
||
func TestDetectInterfaceType(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
adapter := newMockAdapter("test", true)
|
||
adapter.ifaceType = InterfaceTypeChat
|
||
_ = engine.RegisterAdapter(adapter)
|
||
|
||
ifaceType, err := engine.DetectInterfaceType("/chat/completions", "test")
|
||
require.NoError(t, err)
|
||
assert.Equal(t, InterfaceTypeChat, ifaceType)
|
||
}
|
||
|
||
func TestDetectInterfaceType_NonExistentProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
_, err := engine.DetectInterfaceType("/v1/chat", "nonexistent")
|
||
assert.Error(t, err)
|
||
}
|
||
|
||
func TestConvertHttpRequest_Passthrough(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("openai", true))
|
||
|
||
provider := NewTargetProvider("https://api.openai.com/v1", "sk-test", "gpt-4")
|
||
spec := HTTPRequestSpec{
|
||
URL: "/chat/completions",
|
||
Method: "POST",
|
||
Body: []byte(`{"model":"gpt-4","messages":[{"role":"user","content":"hi"}]}`),
|
||
}
|
||
|
||
result, err := engine.ConvertHttpRequest(spec, "openai", "openai", provider)
|
||
require.NoError(t, err)
|
||
assert.Equal(t, "https://api.openai.com/v1/chat/completions", result.URL)
|
||
assert.Equal(t, spec.Body, result.Body)
|
||
}
|
||
|
||
func TestConvertHttpRequest_CrossProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
clientAdapter := newMockAdapter("client-proto", false)
|
||
clientAdapter.decodeReqFn = func(raw []byte) (*canonical.CanonicalRequest, error) {
|
||
return &canonical.CanonicalRequest{
|
||
Model: "test-model",
|
||
Messages: []canonical.CanonicalMessage{{Role: canonical.RoleUser, Content: []canonical.ContentBlock{canonical.NewTextBlock("hi")}}},
|
||
}, nil
|
||
}
|
||
_ = engine.RegisterAdapter(clientAdapter)
|
||
|
||
providerAdapter := newMockAdapter("provider-proto", false)
|
||
providerAdapter.encodeReqFn = func(req *canonical.CanonicalRequest, p *TargetProvider) ([]byte, error) {
|
||
return json.Marshal(map[string]any{"model": p.ModelName})
|
||
}
|
||
_ = engine.RegisterAdapter(providerAdapter)
|
||
|
||
provider := NewTargetProvider("https://example.com", "key", "my-model")
|
||
spec := HTTPRequestSpec{
|
||
URL: "/v1/chat",
|
||
Method: "POST",
|
||
Body: []byte(`{"model":"test"}`),
|
||
}
|
||
|
||
result, err := engine.ConvertHttpRequest(spec, "client-proto", "provider-proto", provider)
|
||
require.NoError(t, err)
|
||
assert.Contains(t, result.URL, "https://example.com")
|
||
assert.NotNil(t, result.Body)
|
||
}
|
||
|
||
func TestConvertHttpResponse_Passthrough(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("openai", true))
|
||
|
||
spec := HTTPResponseSpec{
|
||
StatusCode: 200,
|
||
Body: []byte(`{"id":"123"}`),
|
||
}
|
||
|
||
result, err := engine.ConvertHttpResponse(spec, "openai", "openai", InterfaceTypeChat, "")
|
||
require.NoError(t, err)
|
||
assert.Equal(t, 200, result.StatusCode)
|
||
assert.Equal(t, spec.Body, result.Body)
|
||
}
|
||
|
||
func TestCreateStreamConverter_Passthrough(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("openai", true))
|
||
|
||
converter, err := engine.CreateStreamConverter("openai", "openai", "", InterfaceTypeChat)
|
||
require.NoError(t, err)
|
||
_, ok := converter.(*PassthroughStreamConverter)
|
||
assert.True(t, ok)
|
||
}
|
||
|
||
func TestCreateStreamConverter_Canonical(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("client", false))
|
||
_ = engine.RegisterAdapter(newMockAdapter("provider", false))
|
||
|
||
converter, err := engine.CreateStreamConverter("client", "provider", "", InterfaceTypeChat)
|
||
require.NoError(t, err)
|
||
_, ok := converter.(*CanonicalStreamConverter)
|
||
assert.True(t, ok)
|
||
}
|
||
|
||
func TestEncodeError(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
_ = engine.RegisterAdapter(newMockAdapter("openai", true))
|
||
|
||
convErr := NewConversionError(ErrorCodeInvalidInput, "测试错误")
|
||
body, statusCode, err := engine.EncodeError(convErr, "openai")
|
||
require.NoError(t, err)
|
||
assert.Equal(t, 400, statusCode)
|
||
assert.NotNil(t, body)
|
||
}
|
||
|
||
func TestEncodeError_NonExistentProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
convErr := NewConversionError(ErrorCodeInvalidInput, "测试错误")
|
||
body, statusCode, err := engine.EncodeError(convErr, "nonexistent")
|
||
require.NoError(t, err)
|
||
assert.Equal(t, 500, statusCode)
|
||
assert.Contains(t, string(body), "测试错误")
|
||
}
|
||
|
||
func TestRegistry_DuplicateRegistration(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
adapter := newMockAdapter("openai", true)
|
||
|
||
err := registry.Register(adapter)
|
||
require.NoError(t, err)
|
||
|
||
err = registry.Register(adapter)
|
||
assert.Error(t, err)
|
||
assert.Contains(t, err.Error(), "适配器已注册")
|
||
}
|
||
|
||
func TestRegistry_GetNonExistent(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
|
||
_, err := registry.Get("nonexistent")
|
||
assert.Error(t, err)
|
||
assert.Contains(t, err.Error(), "未找到适配器")
|
||
}
|
||
|
||
// ============ modelOverride 测试 ============
|
||
|
||
func TestConvertHttpResponse_ModelOverride_CrossProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
clientAdapter := newMockAdapter("client", false)
|
||
clientAdapter.encodeRespFn = func(resp *canonical.CanonicalResponse) ([]byte, error) {
|
||
return json.Marshal(map[string]any{"model": resp.Model})
|
||
}
|
||
_ = engine.RegisterAdapter(clientAdapter)
|
||
|
||
providerAdapter := newMockAdapter("provider", false)
|
||
providerAdapter.decodeRespFn = func(raw []byte) (*canonical.CanonicalResponse, error) {
|
||
return &canonical.CanonicalResponse{ID: "test", Model: "native-model", Content: []canonical.ContentBlock{canonical.NewTextBlock("hi")}}, nil
|
||
}
|
||
_ = engine.RegisterAdapter(providerAdapter)
|
||
|
||
spec := HTTPResponseSpec{
|
||
StatusCode: 200,
|
||
Body: []byte(`{"model":"native-model"}`),
|
||
}
|
||
|
||
result, err := engine.ConvertHttpResponse(spec, "client", "provider", InterfaceTypeChat, "provider/gpt-4")
|
||
require.NoError(t, err)
|
||
|
||
var resp map[string]interface{}
|
||
require.NoError(t, json.Unmarshal(result.Body, &resp))
|
||
assert.Equal(t, "provider/gpt-4", resp["model"])
|
||
}
|
||
|
||
func TestConvertHttpResponse_ModelOverride_SameProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
// 使用真实 OpenAI adapter 验证 Smart Passthrough 改写
|
||
openaiAdapter := newMockAdapter("openai", true)
|
||
openaiAdapter.rewriteRespFn = func(body []byte, newModel string, ifaceType InterfaceType) ([]byte, error) {
|
||
var m map[string]json.RawMessage
|
||
if err := json.Unmarshal(body, &m); err != nil {
|
||
return nil, err
|
||
}
|
||
m["model"], _ = json.Marshal(newModel)
|
||
return json.Marshal(m)
|
||
}
|
||
_ = engine.RegisterAdapter(openaiAdapter)
|
||
|
||
spec := HTTPResponseSpec{
|
||
StatusCode: 200,
|
||
Body: []byte(`{"id":"resp-1","model":"gpt-4"}`),
|
||
}
|
||
|
||
result, err := engine.ConvertHttpResponse(spec, "openai", "openai", InterfaceTypeChat, "openai/gpt-4")
|
||
require.NoError(t, err)
|
||
|
||
var resp map[string]interface{}
|
||
require.NoError(t, json.Unmarshal(result.Body, &resp))
|
||
assert.Equal(t, "openai/gpt-4", resp["model"])
|
||
assert.Equal(t, "resp-1", resp["id"])
|
||
}
|
||
|
||
func TestCreateStreamConverter_ModelOverride_SmartPassthrough(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
openaiAdapter := newMockAdapter("openai", true)
|
||
openaiAdapter.rewriteRespFn = func(body []byte, newModel string, ifaceType InterfaceType) ([]byte, error) {
|
||
var m map[string]json.RawMessage
|
||
if err := json.Unmarshal(body, &m); err != nil {
|
||
return nil, err
|
||
}
|
||
m["model"], _ = json.Marshal(newModel)
|
||
return json.Marshal(m)
|
||
}
|
||
_ = engine.RegisterAdapter(openaiAdapter)
|
||
|
||
converter, err := engine.CreateStreamConverter("openai", "openai", "openai/gpt-4", InterfaceTypeChat)
|
||
require.NoError(t, err)
|
||
|
||
_, ok := converter.(*SmartPassthroughStreamConverter)
|
||
assert.True(t, ok)
|
||
|
||
// 验证 chunk 改写
|
||
chunks := converter.ProcessChunk([]byte(`{"model":"gpt-4","choices":[]}`))
|
||
require.Len(t, chunks, 1)
|
||
|
||
var resp map[string]interface{}
|
||
require.NoError(t, json.Unmarshal(chunks[0], &resp))
|
||
assert.Equal(t, "openai/gpt-4", resp["model"])
|
||
}
|
||
|
||
func TestCreateStreamConverter_ModelOverride_CrossProtocol(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
// provider adapter 解码出含 model 的流式事件
|
||
providerAdapter := newMockAdapter("provider", false)
|
||
providerAdapter.streamDecoderFn = func() StreamDecoder {
|
||
return &engineTestStreamDecoder{
|
||
processFn: func(raw []byte) []canonical.CanonicalStreamEvent {
|
||
return []canonical.CanonicalStreamEvent{
|
||
canonical.NewMessageStartEvent("msg-1", "native-model"),
|
||
canonical.NewContentBlockStartEvent(0, canonical.StreamContentBlock{Type: "text", Text: "hi"}),
|
||
canonical.NewMessageStopEvent(),
|
||
}
|
||
},
|
||
}
|
||
}
|
||
_ = engine.RegisterAdapter(providerAdapter)
|
||
|
||
// client adapter 编码时输出 model 字段
|
||
clientAdapter := newMockAdapter("client", false)
|
||
clientAdapter.streamEncoderFn = func() StreamEncoder {
|
||
return &engineTestStreamEncoder{
|
||
encodeFn: func(event canonical.CanonicalStreamEvent) [][]byte {
|
||
if event.Message != nil {
|
||
data, _ := json.Marshal(map[string]string{
|
||
"type": string(event.Type),
|
||
"model": event.Message.Model,
|
||
})
|
||
return [][]byte{data}
|
||
}
|
||
data, _ := json.Marshal(map[string]string{"type": string(event.Type)})
|
||
return [][]byte{data}
|
||
},
|
||
}
|
||
}
|
||
_ = engine.RegisterAdapter(clientAdapter)
|
||
|
||
converter, err := engine.CreateStreamConverter("client", "provider", "provider/gpt-4", InterfaceTypeChat)
|
||
require.NoError(t, err)
|
||
|
||
// 验证类型是 CanonicalStreamConverter
|
||
_, ok := converter.(*CanonicalStreamConverter)
|
||
assert.True(t, ok)
|
||
|
||
// 处理一个 chunk，验证 model 被覆写为统一模型 ID
|
||
chunks := converter.ProcessChunk([]byte("raw"))
|
||
require.Len(t, chunks, 3) // message_start + content_block_start + message_stop
|
||
|
||
var startEvent map[string]string
|
||
require.NoError(t, json.Unmarshal(chunks[0], &startEvent))
|
||
assert.Equal(t, "provider/gpt-4", startEvent["model"], "跨协议流式中 modelOverride 应覆写 Message.Model")
|
||
}
|
||
|
||
func TestCreateStreamConverter_ModelOverride_CrossProtocol_Empty(t *testing.T) {
|
||
registry := NewMemoryRegistry()
|
||
engine := NewConversionEngine(registry, nil)
|
||
|
||
providerAdapter := newMockAdapter("provider", false)
|
||
providerAdapter.streamDecoderFn = func() StreamDecoder {
|
||
return &engineTestStreamDecoder{
|
||
processFn: func(raw []byte) []canonical.CanonicalStreamEvent {
|
||
return []canonical.CanonicalStreamEvent{
|
||
canonical.NewMessageStartEvent("msg-1", "native-model"),
|
||
}
|
||
},
|
||
}
|
||
}
|
||
_ = engine.RegisterAdapter(providerAdapter)
|
||
|
||
clientAdapter := newMockAdapter("client", false)
|
||
clientAdapter.streamEncoderFn = func() StreamEncoder {
|
||
return &engineTestStreamEncoder{
|
||
encodeFn: func(event canonical.CanonicalStreamEvent) [][]byte {
|
||
if event.Message != nil {
|
||
data, _ := json.Marshal(map[string]string{
|
||
"model": event.Message.Model,
|
||
})
|
||
return [][]byte{data}
|
||
}
|
||
return nil
|
||
},
|
||
}
|
||
}
|
||
_ = engine.RegisterAdapter(clientAdapter)
|
||
|
||
// modelOverride 为空，不应覆写
|
||
converter, err := engine.CreateStreamConverter("client", "provider", "", InterfaceTypeChat)
|
||
require.NoError(t, err)
|
||
|
||
chunks := converter.ProcessChunk([]byte("raw"))
|
||
require.Len(t, chunks, 1)
|
||
|
||
var resp map[string]string
|
||
require.NoError(t, json.Unmarshal(chunks[0], &resp))
|
||
assert.Equal(t, "native-model", resp["model"], "modelOverride 为空时不应覆写")
|
||
}
|
||
|
||
// engineTestStreamDecoder 可控的流式解码器（用于 engine_test）
|
||
type engineTestStreamDecoder struct {
|
||
processFn func([]byte) []canonical.CanonicalStreamEvent
|
||
flushFn func() []canonical.CanonicalStreamEvent
|
||
}
|
||
|
||
func (d *engineTestStreamDecoder) ProcessChunk(raw []byte) []canonical.CanonicalStreamEvent {
|
||
if d.processFn != nil {
|
||
return d.processFn(raw)
|
||
}
|
||
return nil
|
||
}
|
||
func (d *engineTestStreamDecoder) Flush() []canonical.CanonicalStreamEvent {
|
||
if d.flushFn != nil {
|
||
return d.flushFn()
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// engineTestStreamEncoder 可控的流式编码器（用于 engine_test）
|
||
type engineTestStreamEncoder struct {
|
||
encodeFn func(canonical.CanonicalStreamEvent) [][]byte
|
||
flushFn func() [][]byte
|
||
}
|
||
|
||
func (e *engineTestStreamEncoder) EncodeEvent(event canonical.CanonicalStreamEvent) [][]byte {
|
||
if e.encodeFn != nil {
|
||
return e.encodeFn(event)
|
||
}
|
||
return nil
|
||
}
|
||
func (e *engineTestStreamEncoder) Flush() [][]byte {
|
||
if e.flushFn != nil {
|
||
return e.flushFn()
|
||
}
|
||
return nil
|
||
}
|