diff --git a/go/internal/api/routes/core.go b/go/internal/api/routes/core.go index 06c70cf..7884568 100644 --- a/go/internal/api/routes/core.go +++ b/go/internal/api/routes/core.go @@ -24,11 +24,13 @@ type API struct { } func NewAPI(ldb storage.LanceDB, emb *storage.Embedder, rerank *storage.Reranker) *API { + pipeline := storage.NewRecallPipeline(emb, ldb, rerank) + pipeline.SetPrefetchPusher(&WSPrefetchAdapter{}) return &API{ LanceDB: ldb, Embedder: emb, Reranker: rerank, - Pipeline: storage.NewRecallPipeline(emb, ldb, rerank), + Pipeline: pipeline, } } diff --git a/go/internal/api/routes/ws_events.go b/go/internal/api/routes/ws_events.go index 8329e92..99241b5 100644 --- a/go/internal/api/routes/ws_events.go +++ b/go/internal/api/routes/ws_events.go @@ -1,6 +1,19 @@ // 织忆 MemoryWeave — WebSocket 事件推送函数 package routes +import "github.com/xiaoxue/memoryweave/internal/models" + +// WSPrefetchAdapter 实现 storage.PrefetchPusher 接口 +type WSPrefetchAdapter struct{} + +func (a *WSPrefetchAdapter) PushPrefetch(agentID string, memories []models.RecallResult) { + if agentID != "" { + WSBus.Push(agentID, "prefetch.push", memories) + } else { + WSBus.Broadcast("prefetch.push", memories) + } +} + // PushPrefetch 预取推送(recall 管道调用) func PushPrefetch(agentID string, memories interface{}) { WSBus.Push(agentID, "prefetch.push", memories)