Function testDetailedVision

examples/custom_claude_api/test_vision.go:155–204

testDetailedVision 测试详细图片分析

(cp *provider.CustomClaudeProvider, imageData []byte, mediaType string)

// testDetailedVision 测试详细图片分析
func testDetailedVision(cp *provider.CustomClaudeProvider, imageData []byte, mediaType string) {
	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
	defer cancel()

	// Base64 编码
	base64Data := base64.StdEncoding.EncodeToString(imageData)

	// 构造消息
	messages := []types.Message{
		{
			Role: types.MessageRoleUser,
			ContentBlocks: []types.ContentBlock{
				&types.ImageContent{
					Type:     "base64",
					Source:   base64Data,
					MimeType: mediaType,
				},
				&types.TextBlock{
					Text: "请详细描述这张图片的：1) 主要内容 2) 颜色 3) 风格。用中文回答，不超过100字。",
				},
			},
		},
	}

	fmt.Println("💬 问题: 请详细描述图片的内容、颜色和风格")
	fmt.Print("🤖 回复: ")

	// 调用
	opts := &provider.StreamOptions{
		MaxTokens: 1000,
	}
	response, err := cp.Complete(ctx, messages, opts)
	if err != nil {
		fmt.Printf("\n❌ 调用失败: %v\n", err)
		return
	}

	// 输出结果
	var content string
	for _, block := range response.Message.ContentBlocks {
		if textBlock, ok := block.(*types.TextBlock); ok {
			content += textBlock.Text
		}
	}
	fmt.Printf("%s\n", content)
	if response.Usage != nil {
		fmt.Printf("📊 Token: 输入=%d, 输出=%d\n",
			response.Usage.InputTokens, response.Usage.OutputTokens)
	}
}