mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-27 07:16:19 +09:00
BBQ
bc374fe8cd
* refactor(attachment): multimodal attachment refactor with snapshot schema and storage layer - Add snapshot schema migration (0008) and update init/versions/snapshots - Add internal/attachment and internal/channel normalize for unified attachment handling - Move containerfs provider from internal/media to internal/storage - Update agent types, channel adapters (Telegram/Feishu), inbound and handlers - Add containerd snapshot lineage and local_channel tests - Regenerate sqlc, swagger and SDK * refactor(media): content-addressed asset system with unified naming - Replace asset_id foreign key with content_hash as sole identifier for bot_history_message_assets (pure soft-link model) - Remove mime, size_bytes, storage_key from DB; derive at read time via media.Resolve from actual storage - Merge migrations 0008/0009 into single 0008; keep 0001 as canonical schema - Add Docker initdb script for deterministic migration execution order - Fix cross-channel real-time image display (Telegram → WebUI SSE) - Fix message disappearing on refresh (null assets fallback) - Fix file icon instead of image preview (mime derivation from storage) - Unify AssetID → ContentHash naming across Go, Agent, and Frontend - Change storage key prefix from 4-char to 2-char for directory sharding - Add server-entrypoint.sh for Docker deployment migration handling * refactor(infra): embedded migrations, Docker simplification, and config consolidation - Embed SQL migrations into Go binary, removing shell-based migration scripts - Consolidate config files into conf/ directory (app.example.toml, app.docker.toml, app.dev.toml) - Simplify Docker setup: remove initdb.d scripts, streamline nginx config and entrypoint - Remove legacy CLI, feishu-echo commands, and obsolete incremental migration files - Update install script and docs to require sudo for one-click install - Add mise tasks for dev environment orchestration * chore: recover migrations --------- Co-authored-by: Acbox <acbox0328@gmail.com>
92 lines
3.4 KiB
Go
92 lines
3.4 KiB
Go
package flow
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRouteAttachmentsByCapability_AllSupported(t *testing.T) {
|
|
modalities := []string{"text", "image", "audio"}
|
|
attachments := []gatewayAttachment{
|
|
{Type: "image", Transport: gatewayTransportInlineDataURL, Payload: "data:image/png;base64,abc"},
|
|
{Type: "audio", Transport: gatewayTransportToolFileRef, Payload: "/data/voice.wav"},
|
|
}
|
|
result := routeAttachmentsByCapability(modalities, attachments)
|
|
assert.Len(t, result.Native, 1)
|
|
assert.Len(t, result.Fallback, 1)
|
|
assert.Equal(t, "image", result.Native[0].Type)
|
|
assert.Equal(t, "audio", result.Fallback[0].Type)
|
|
}
|
|
|
|
func TestRouteAttachmentsByCapability_TextOnly(t *testing.T) {
|
|
modalities := []string{"text"}
|
|
attachments := []gatewayAttachment{
|
|
{Type: "image", Transport: gatewayTransportInlineDataURL, Payload: "data:image/png;base64,abc"},
|
|
{Type: "video", Transport: gatewayTransportToolFileRef, Payload: "/data/video.mp4"},
|
|
}
|
|
result := routeAttachmentsByCapability(modalities, attachments)
|
|
assert.Len(t, result.Native, 0)
|
|
assert.Len(t, result.Fallback, 2)
|
|
}
|
|
|
|
func TestRouteAttachmentsByCapability_Mixed(t *testing.T) {
|
|
modalities := []string{"text", "image"}
|
|
attachments := []gatewayAttachment{
|
|
{Type: "image", Transport: gatewayTransportInlineDataURL, Payload: "data:image/png;base64,abc"},
|
|
{Type: "video", Transport: gatewayTransportToolFileRef, Payload: "/data/video.mp4"},
|
|
{Type: "audio", Transport: gatewayTransportToolFileRef, Payload: "/data/audio.mp3"},
|
|
}
|
|
result := routeAttachmentsByCapability(modalities, attachments)
|
|
assert.Len(t, result.Native, 1)
|
|
assert.Equal(t, "image", result.Native[0].Type)
|
|
assert.Len(t, result.Fallback, 2)
|
|
}
|
|
|
|
func TestRouteAttachmentsByCapability_ImagePathOnlyFallsBack(t *testing.T) {
|
|
modalities := []string{"text", "image"}
|
|
attachments := []gatewayAttachment{
|
|
{Type: "image", Transport: gatewayTransportToolFileRef, Payload: "/data/image.png"},
|
|
}
|
|
result := routeAttachmentsByCapability(modalities, attachments)
|
|
assert.Len(t, result.Native, 0)
|
|
assert.Len(t, result.Fallback, 1)
|
|
assert.Equal(t, "image", result.Fallback[0].Type)
|
|
}
|
|
|
|
func TestRouteAttachmentsByCapability_ImageURLIsNative(t *testing.T) {
|
|
modalities := []string{"text", "image"}
|
|
attachments := []gatewayAttachment{
|
|
{Type: "image", Transport: gatewayTransportPublicURL, Payload: "https://example.com/image.png"},
|
|
}
|
|
result := routeAttachmentsByCapability(modalities, attachments)
|
|
assert.Len(t, result.Native, 1)
|
|
assert.Len(t, result.Fallback, 0)
|
|
assert.Equal(t, "image", result.Native[0].Type)
|
|
}
|
|
|
|
func TestRouteAttachmentsByCapability_UnknownType(t *testing.T) {
|
|
modalities := []string{"text", "image"}
|
|
attachments := []gatewayAttachment{
|
|
{Type: "hologram", Transport: gatewayTransportToolFileRef, Payload: "/data/holo.dat"},
|
|
}
|
|
result := routeAttachmentsByCapability(modalities, attachments)
|
|
assert.Len(t, result.Native, 0)
|
|
assert.Len(t, result.Fallback, 1)
|
|
}
|
|
|
|
func TestRouteAttachmentsByCapability_Empty(t *testing.T) {
|
|
result := routeAttachmentsByCapability([]string{"text"}, nil)
|
|
assert.Len(t, result.Native, 0)
|
|
assert.Len(t, result.Fallback, 0)
|
|
}
|
|
|
|
func TestAttachmentsToAny(t *testing.T) {
|
|
atts := []gatewayAttachment{
|
|
{Type: "image", Transport: gatewayTransportInlineDataURL, Payload: "data:image/png;base64,abc"},
|
|
{Type: "file", Transport: gatewayTransportToolFileRef, Payload: "/data/doc.pdf"},
|
|
}
|
|
result := attachmentsToAny(atts)
|
|
assert.Len(t, result, 2)
|
|
}
|