1
0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-12 14:06:35 +08:00

feat: support image upload

This commit is contained in:
源文雨
2025-09-24 00:18:32 +08:00
parent e3d1b92cc3
commit 8757564fe7
10 changed files with 376 additions and 98 deletions

33
model/content_test.go Normal file
View File

@@ -0,0 +1,33 @@
package model
import (
"encoding/base64"
"strings"
"testing"
)
const contentTextSmallImage = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII="
func TestContentImageDataBytes(t *testing.T) {
// 从 smallimg 中提取 base64 数据
parts := strings.Split(contentTextSmallImage, ",")
if len(parts) != 2 {
t.Fatal("Invalid data URL format")
}
// 解码 base64 数据
data, err := base64.StdEncoding.DecodeString(parts[1])
if err != nil {
t.Fatalf("Failed to decode base64: %v", err)
}
s, err := NewContentImageDataBase64URL(data)
if err != nil {
t.Fatalf("NewContentImageDataBytes failed: %v", err)
}
// 比较结果
if s != contentTextSmallImage {
t.Errorf("Expected %s, got %s", contentTextSmallImage, s)
}
}