Function TestMessageConversion

pkg/provider/openai_compatible_test.go:366–444

TestMessageConversion 测试消息转换

(t *testing.T)

// TestMessageConversion 测试消息转换
func TestMessageConversion(t *testing.T) {
	config := &types.ModelConfig{
		Provider: "openai",
		Model:    "gpt-4o",
		APIKey:   "test-key",
	}

	provider, err := NewOpenAIProvider(config)
	if err != nil {
		t.Fatalf("Failed to create provider: %v", err)
	}

	// 类型断言获取基础 Provider
	openaiProvider, ok := provider.(*OpenAIProvider)
	if !ok {
		t.Fatal("Failed to cast to OpenAIProvider")
	}
	baseProvider := openaiProvider.OpenAICompatibleProvider

	// 测试简单消息
	messages := []types.Message{
		{
			Role:    types.RoleUser,
			Content: "Hello",
		},
	}

	converted := baseProvider.convertMessages(messages)
	if len(converted) != 1 {
		t.Errorf("Expected 1 message, got %d", len(converted))
	}

	if converted[0]["role"] != "user" {
		t.Errorf("Expected role=user, got %s", converted[0]["role"])
	}

	if converted[0]["content"] != "Hello" {
		t.Errorf("Expected content=Hello, got %s", converted[0]["content"])
	}

	// 测试多模态消息
	messagesWithImage := []types.Message{
		{
			Role: types.RoleUser,
			ContentBlocks: []types.ContentBlock{
				&types.TextBlock{Text: "What's in this image?"},
				&types.ImageContent{
					Type:   "url",
					Source: "https://example.com/image.jpg",
					Detail: "high",
				},
			},
		},
	}

	converted = baseProvider.convertMessages(messagesWithImage)
	if len(converted) != 1 {
		t.Errorf("Expected 1 message, got %d", len(converted))