Function main

examples/custom_claude_api/test_vision.go:18–68

()

)

func main() {
	apiKey := os.Getenv("CLAUDE_API_KEY")
	baseURL := os.Getenv("CLAUDE_BASE_URL")

	if apiKey == "" {
		fmt.Println("❌ 请设置 CLAUDE_API_KEY 环境变量")
		os.Exit(1)
	}
	if baseURL == "" {
		baseURL = "https://api.anthropic.com"
	}

	fmt.Println("=== Claude API 图片识别测试 ===")
	fmt.Printf("API 端点: %s\n", baseURL)
	fmt.Printf("模型: claude-sonnet-4-5-20250929\n\n")

	// 下载测试图片
	fmt.Println("📥 下载测试图片...")
	imageData, mediaType, err := downloadTestImage()
	if err != nil {
		fmt.Printf("❌ 下载失败: %v\n", err)
		os.Exit(1)
	}
	fmt.Printf("✅ 下载成功 (类型: %s, 大小: %d bytes)\n\n", mediaType, len(imageData))

	// 创建 Provider
	config := &types.ModelConfig{
		Provider: "anthropic",
		Model:    "claude-sonnet-4-5-20250929",
		APIKey:   apiKey,
		BaseURL:  baseURL,
	}

	cp, err := provider.NewCustomClaudeProvider(config)
	if err != nil {
		fmt.Printf("❌ 创建 Provider 失败: %v\n", err)
		os.Exit(1)
	}

	// 测试 1: 基础图片识别
	fmt.Println("【测试 1】基础图片识别")
	fmt.Println("----------------------------------------")
	testBasicVision(cp, imageData, mediaType)

	// 测试 2: 详细分析
	fmt.Println("\n【测试 2】详细图片分析")
	fmt.Println("----------------------------------------")
	testDetailedVision(cp, imageData, mediaType)

	fmt.Println("\n✅ 图片识别测试完成！")
}

// downloadTestImage 下载测试图片
func downloadTestImage() ([]byte, string, error) {