mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 17:38:10 +00:00
1f1233b823
- P0-1: ProviderError type + ReportStatus for precise 401/403 detection - P0-2: Remove -config flag from deploy/homeagent.service - P2-1: 5s debounce on context.go Save() - P2-2→C1: Delete output_set_channel entirely - P2-3: Extract mediaDataURL/mediaChat helpers - P2-4: Dedup defaultSources var - P3: Delete dead packages (embed/tokenizer/container/snapshot) - P3: Delete dead functions (messagesToMap, RunStageAll) - CL: Update .gitignore, docs, Makefile, gojieba removal - Config: Delete config/config.yaml, update docs - Arch: Remove EmitOutputTo from emitResponse - CL-1: go.work 1.19→1.21 - Docs: Add Mermaid architecture diagrams to README - Docs: Add kernel-rebuild requires plugin-rebuild note to PLUGIN_DEV.md
93 lines
2.9 KiB
Go
93 lines
2.9 KiB
Go
package sdk
|
|
|
|
import (
|
|
"gitcode.com/JianFeeeee/HomeAgent/internal/memory"
|
|
doc "gitcode.com/JianFeeeee/HomeAgent/internal/memory/document"
|
|
"gitcode.com/JianFeeeee/HomeAgent/internal/memory/text"
|
|
)
|
|
|
|
type graphMemory struct{ db *memory.GraphDB }
|
|
|
|
func NewGraphMemory(db *memory.GraphDB) MemoryAPI { return &graphMemory{db: db} }
|
|
|
|
func (m *graphMemory) Recall(query []string, depth int) ([]Entity, []Relation, error) {
|
|
if m.db == nil { return nil, nil, nil }
|
|
result, err := m.db.Recall(query, nil, depth, "")
|
|
if err != nil { return nil, nil, err }
|
|
entities := make([]Entity, len(result.Entities))
|
|
for i, e := range result.Entities {
|
|
entities[i] = Entity{Name: e.Name, Type: e.Type, MentionCount: e.MentionCount}
|
|
}
|
|
relations := make([]Relation, len(result.Relations))
|
|
for i, r := range result.Relations {
|
|
relations[i] = Relation{SourceName: r.SourceName, TargetName: r.TargetName, RelationType: r.RelationType}
|
|
}
|
|
return entities, relations, nil
|
|
}
|
|
|
|
func (m *graphMemory) Commit(triples []Triple) error {
|
|
if m.db == nil { return nil }
|
|
ts := make([]memory.Triple, len(triples))
|
|
for i, t := range triples {
|
|
ts[i] = memory.Triple{Subject: t.Subject, Relation: t.Relation, Object: t.Object}
|
|
}
|
|
_, _, err := m.db.Commit(ts, "plugin", 0)
|
|
return err
|
|
}
|
|
|
|
func (m *graphMemory) Introspect() (map[string]interface{}, error) {
|
|
if m.db == nil { return map[string]interface{}{}, nil }
|
|
return m.db.Introspect()
|
|
}
|
|
|
|
func (m *graphMemory) MergeEntities(source, target string) (int, error) {
|
|
if m.db == nil { return 0, nil }
|
|
return m.db.MergeEntities(source, target)
|
|
}
|
|
|
|
func (m *graphMemory) Purge(criteria map[string]string, mode string) (int, error) {
|
|
if m.db == nil { return 0, nil }
|
|
return m.db.Purge(criteria, mode)
|
|
}
|
|
|
|
type textMemoryImpl struct{ tm *text.Memory }
|
|
|
|
func NewTextMemory(tm *text.Memory) TextMemoryAPI { return &textMemoryImpl{tm: tm} }
|
|
|
|
func (m *textMemoryImpl) Append(evt TextEvent) error {
|
|
if m.tm == nil { return nil }
|
|
return m.tm.Append(text.Event{
|
|
Timestamp: evt.Timestamp, Source: evt.Role, Input: evt.Content, AgentID: evt.Channel,
|
|
})
|
|
}
|
|
|
|
type docMemoryImpl struct{ ds *doc.Store }
|
|
|
|
func NewDocMemory(ds *doc.Store) DocMemoryAPI { return &docMemoryImpl{ds: ds} }
|
|
|
|
func (m *docMemoryImpl) Query(text string, topK int) []*Doc {
|
|
if m.ds == nil { return nil }
|
|
got := m.ds.Query(text, topK)
|
|
out := make([]*Doc, len(got))
|
|
for i, d := range got {
|
|
out[i] = &Doc{ID: d.ID, Title: d.Summary, Content: d.Content}
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (m *docMemoryImpl) Insert(d *Doc) error {
|
|
if m.ds == nil { return nil }
|
|
return m.ds.Insert(&doc.Doc{ID: d.ID, Summary: d.Title, Content: d.Content})
|
|
}
|
|
|
|
func (m *docMemoryImpl) Remove(id string) { if m.ds != nil { m.ds.Remove(id) } }
|
|
|
|
func (m *docMemoryImpl) Stats() map[string]interface{} {
|
|
if m.ds == nil { return map[string]interface{}{} }
|
|
return m.ds.Stats()
|
|
}
|
|
|
|
var _ MemoryAPI = (*graphMemory)(nil)
|
|
var _ TextMemoryAPI = (*textMemoryImpl)(nil)
|
|
var _ DocMemoryAPI = (*docMemoryImpl)(nil)
|