media abstraction: add structured MediaItem model and expose media-source audit in task queue detail

This commit is contained in:
DBT
2026-02-28 02:49:13 +00:00
parent f274acf818
commit 46a0403b0f
5 changed files with 71 additions and 0 deletions

View File

@@ -374,6 +374,8 @@ func (al *AgentLoop) appendTaskAuditEvent(taskID string, msg bus.InboundMessage,
"retry_count": 0,
"log": logText,
"input_preview": truncate(strings.ReplaceAll(msg.Content, "\n", " "), 180),
"media_count": len(msg.MediaItems),
"media_items": msg.MediaItems,
}
b, _ := json.Marshal(row)
f, err := os.OpenFile(path, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)

View File

@@ -1,11 +1,20 @@
package bus
type MediaItem struct {
Source string `json:"source,omitempty"`
Type string `json:"type,omitempty"`
Ref string `json:"ref,omitempty"`
Path string `json:"path,omitempty"`
Channel string `json:"channel,omitempty"`
}
type InboundMessage struct {
Channel string `json:"channel"`
SenderID string `json:"sender_id"`
ChatID string `json:"chat_id"`
Content string `json:"content"`
Media []string `json:"media,omitempty"`
MediaItems []MediaItem `json:"media_items,omitempty"`
SessionKey string `json:"session_key"`
Metadata map[string]string `json:"metadata,omitempty"`
}

View File

@@ -3,6 +3,7 @@ package channels
import (
"context"
"fmt"
"path/filepath"
"strings"
"sync/atomic"
@@ -96,6 +97,7 @@ func (c *BaseChannel) HandleMessage(senderID, chatID, content string, media []st
ChatID: chatID,
Content: content,
Media: media,
MediaItems: toMediaItems(c.name, media),
Metadata: metadata,
SessionKey: sessionKey,
}
@@ -103,6 +105,46 @@ func (c *BaseChannel) HandleMessage(senderID, chatID, content string, media []st
c.bus.PublishInbound(msg)
}
func toMediaItems(channel string, media []string) []bus.MediaItem {
if len(media) == 0 {
return nil
}
out := make([]bus.MediaItem, 0, len(media))
for _, m := range media {
item := bus.MediaItem{Channel: channel, Ref: m, Source: "raw", Type: "unknown"}
switch {
case strings.HasPrefix(m, "feishu:image:"):
item.Source = "feishu"
item.Type = "image"
case strings.HasPrefix(m, "feishu:file:"):
item.Source = "feishu"
item.Type = "file"
case strings.HasPrefix(m, "telegram:"):
item.Source = "telegram"
item.Type = "remote"
case strings.HasPrefix(m, "http://") || strings.HasPrefix(m, "https://"):
item.Source = "url"
item.Type = "remote"
default:
ext := strings.ToLower(filepath.Ext(m))
item.Path = m
switch ext {
case ".jpg", ".jpeg", ".png", ".gif", ".webp", ".bmp":
item.Type = "image"
case ".mp4", ".mov", ".webm", ".avi":
item.Type = "video"
case ".mp3", ".wav", ".ogg", ".m4a":
item.Type = "audio"
default:
item.Type = "file"
}
item.Source = "local"
}
out = append(out, item)
}
return out
}
func (c *BaseChannel) setRunning(running bool) {
c.running.Store(running)
}