mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-06 17:27:31 +00:00
fix(memory): 召回可用性——L2 片段带正文 + WebUI 上下文走双路
① L2 预填充此前只给摘要行('已回复老大:原来是这个谐音梗'),模型读得懂 '我回复过'却不知道梗是什么 ⇒ 召回命中仍答不知道。现取两段候选 (事件文本块 + 工具回执),JSON 回执先提取字符串值再压成单行。 ② WebUI「上下文」页只显示 L3,而 L3 有 72h 冷化延迟 ⇒ 冷化没跑起来时 面板永远'无上下文',看着像召回彻底坏了(实际 L2 一直有命中)。 现 IndexerAPI 带上 L2 片段(与模型预填充同一 QueryScored),前端两路并列。 判据:internal/agent/core/crossmodal_body_test.go
This commit is contained in:
@ -2,19 +2,28 @@ package sdk
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/JianFeeeee/HologramAgent/internal/memory"
|
||||
"github.com/JianFeeeee/HologramAgent/internal/memory/document"
|
||||
)
|
||||
|
||||
// indexerImpl 桥接 memory.Indexer 到中立的 IndexerAPI。
|
||||
type indexerImpl struct {
|
||||
idx *memory.Indexer
|
||||
idx *memory.Indexer
|
||||
docs *document.Store // 可选:给「上下文」诊断面提供 L2 那一路
|
||||
}
|
||||
|
||||
func NewIndexer(idx *memory.Indexer) IndexerAPI {
|
||||
return &indexerImpl{idx: idx}
|
||||
}
|
||||
|
||||
// NewIndexerWithDocs 让诊断面同时覆盖 L2 冷上下文与 L3 图记忆(双路召回)。
|
||||
func NewIndexerWithDocs(idx *memory.Indexer, docs *document.Store) IndexerAPI {
|
||||
return &indexerImpl{idx: idx, docs: docs}
|
||||
}
|
||||
|
||||
func (i *indexerImpl) BuildContext(q string) (*IndexContext, error) {
|
||||
if i.idx == nil {
|
||||
return nil, errors.New("indexer not available")
|
||||
@ -31,12 +40,14 @@ func (i *indexerImpl) BuildContext(q string) (*IndexContext, error) {
|
||||
for j, r := range ctx.Relations {
|
||||
relations[j] = Relation{SourceName: r.SourceName, TargetName: r.TargetName, RelationType: r.RelationType}
|
||||
}
|
||||
return &IndexContext{
|
||||
out := &IndexContext{
|
||||
Entities: entities,
|
||||
Relations: relations,
|
||||
Summary: ctx.Summary,
|
||||
TokenEstimate: ctx.TokenEstimate,
|
||||
}, nil
|
||||
}
|
||||
out.Docs = i.queryDocs(q, 8)
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (i *indexerImpl) FormatContext(ctx *IndexContext) string {
|
||||
@ -54,6 +65,55 @@ func (i *indexerImpl) FormatContext(ctx *IndexContext) string {
|
||||
})
|
||||
}
|
||||
|
||||
// queryDocs 取 L2 那一路的命中片段(与模型预填充同一检索路径)。
|
||||
func (i *indexerImpl) queryDocs(q string, limit int) []DocSnippet {
|
||||
if i.docs == nil || strings.TrimSpace(q) == "" {
|
||||
return nil
|
||||
}
|
||||
hits := i.docs.QueryScored(q, limit)
|
||||
out := make([]DocSnippet, 0, len(hits))
|
||||
for _, h := range hits {
|
||||
if h.Doc == nil {
|
||||
continue
|
||||
}
|
||||
when := h.Doc.CreatedAt.Format(time.DateTime)
|
||||
if h.Doc.HasEvents() {
|
||||
when = h.Doc.Events[0].Timestamp.Format(time.DateTime)
|
||||
}
|
||||
// 正文在事件的**文本块**里(DocEvent 只有 Blocks,没有 Input/Response)。
|
||||
text := docFirstText(h.Doc)
|
||||
out = append(out, DocSnippet{ID: h.Doc.ID, When: when, Text: truncateRunes(text, 120), Score: h.Score})
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// docFirstText 取文档可用于展示的首段正文(跳过媒体块)。
|
||||
func docFirstText(d *document.Doc) string {
|
||||
if d == nil {
|
||||
return ""
|
||||
}
|
||||
for _, ev := range d.Events {
|
||||
for _, b := range ev.Blocks {
|
||||
if b.Modality != "" && b.Modality != "text" {
|
||||
continue
|
||||
}
|
||||
if t := strings.TrimSpace(b.Text); t != "" {
|
||||
return t
|
||||
}
|
||||
}
|
||||
}
|
||||
return d.Content
|
||||
}
|
||||
|
||||
func truncateRunes(s string, n int) string {
|
||||
s = strings.Join(strings.Fields(s), " ")
|
||||
r := []rune(s)
|
||||
if len(r) <= n {
|
||||
return s
|
||||
}
|
||||
return string(r[:n]) + "…"
|
||||
}
|
||||
|
||||
func (i *indexerImpl) GetToolDefinitions() []map[string]interface{} {
|
||||
if i.idx == nil {
|
||||
return nil
|
||||
|
||||
Reference in New Issue
Block a user