Function TestOpenAIProvider

pkg/provider/openai_compatible_test.go:10–56

TestOpenAIProvider 测试 OpenAI Provider

(t *testing.T)

// TestOpenAIProvider 测试 OpenAI Provider
func TestOpenAIProvider(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 OpenAI provider: %v", err)
	}

	// 测试配置
	if provider.Config().Provider != "openai" {
		t.Errorf("Expected provider=openai, got %s", provider.Config().Provider)
	}

	if provider.Config().Model != "gpt-4o" {
		t.Errorf("Expected model=gpt-4o, got %s", provider.Config().Model)
	}

	// 测试能力
	caps := provider.Capabilities()
	if !caps.SupportStreaming {
		t.Error("OpenAI should support streaming")
	}
	if !caps.SupportToolCalling {
		t.Error("OpenAI should support tool calling")
	}
	if !caps.SupportVision {
		t.Error("OpenAI should support vision")
	}
	if !caps.SupportReasoning {
		t.Error("OpenAI should support reasoning models")
	}

	// 测试系统提示词
	err = provider.SetSystemPrompt("You are a helpful assistant")
	if err != nil {
		t.Errorf("Failed to set system prompt: %v", err)
	}

	prompt := provider.GetSystemPrompt()
	if prompt != "You are a helpful assistant" {
		t.Errorf("Expected prompt='You are a helpful assistant', got '%s'", prompt)
	}
}

// TestGroqProvider 测试 Groq Provider
func TestGroqProvider(t *testing.T) {