Function testBasicVision

examples/custom_claude_api/test_vision.go:103–152

testBasicVision 测试基础图片识别

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

// testBasicVision 测试基础图片识别
func testBasicVision(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: "这张图片里有什么？请用中文简短描述（不超过30字）。",
				},
			},
		},
	}

	fmt.Println("💬 问题: 这张图片里有什么？")
	fmt.Print("🤖 回复: ")

	// 调用 Provider
	opts := &provider.StreamOptions{
		MaxTokens: 500,
	}
	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)
	}
}

// testDetailedVision 测试详细图片分析
func testDetailedVision(cp *provider.CustomClaudeProvider, imageData []byte, mediaType string) {