后端: 1. Memory Day1 链路打通(chat_history -> outbox -> memory_jobs) - 更新 service/events/chat_history_persist.go:聊天消息落库同事务追加 memory.extract.requested 事件(仅 user 消息,失败回滚后由 outbox 重试) - 新建 service/events/memory_extract_requested.go:消费 memory.extract.requested 并幂等入队 memory_jobs,补齐 payload 校验、文本截断与 idempotency key - 更新 cmd/start.go:注册 RegisterMemoryExtractRequestedHandler 2. Memory 模块骨架落地(先跑通状态机,再接入真实抽取) - 新建 memory/model、repo、service、orchestrator、worker、utils 目录与 Day1 mock 抽取执行链 - 新建 model/memory.go:补齐 memory_items / memory_jobs / memory_audit_logs / memory_user_settings 与事件 payload 模型 - 更新 inits/mysql.go:接入 4 张 memory 相关表 AutoMigrate 3. RAG 复用基础设施预埋(依赖可替换) - 新建 infra/rag:core pipeline + chunk/embed/retrieve/rerank/store/corpus/config 分层实现 - 默认接入 MockEmbedder + InMemoryStore,预留 Milvus / Eino 适配实现 - 新增 infra/rag/RAG复用接口实施计划.md 4. 本地依赖与交接文档同步 - 更新 docker-compose.yml:新增 etcd / minio / milvus / attu 服务与数据卷 - 删除 newAgent/HANDOFF_工具研究与运行态重置.md、newAgent/阶段3_上下文瘦身设计.md - 新增 newAgent/HANDOFF_WebSearch两阶段实施计划.md、memory/HANDOFF-RAG复用后续实施计划.md、memory/README.md 前端:无 仓库:无
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package embed
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/binary"
|
|
"strings"
|
|
)
|
|
|
|
const defaultDim = 16
|
|
|
|
// MockEmbedder 是本地可运行的占位向量化实现。
|
|
//
|
|
// 说明:
|
|
// 1. 该实现用于开发阶段打通链路,不代表真实语义向量质量;
|
|
// 2. 后续可替换为 Eino embedding 实现,接口保持不变。
|
|
type MockEmbedder struct {
|
|
dim int
|
|
}
|
|
|
|
func NewMockEmbedder(dim int) *MockEmbedder {
|
|
if dim <= 0 {
|
|
dim = defaultDim
|
|
}
|
|
return &MockEmbedder{dim: dim}
|
|
}
|
|
|
|
func (e *MockEmbedder) Embed(_ context.Context, texts []string, _ string) ([][]float32, error) {
|
|
result := make([][]float32, 0, len(texts))
|
|
for _, text := range texts {
|
|
result = append(result, e.embedOne(text))
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (e *MockEmbedder) embedOne(text string) []float32 {
|
|
normalized := strings.TrimSpace(strings.ToLower(text))
|
|
sum := sha256.Sum256([]byte(normalized))
|
|
vec := make([]float32, e.dim)
|
|
for i := 0; i < e.dim; i++ {
|
|
offset := (i * 4) % len(sum)
|
|
v := binary.BigEndian.Uint32(sum[offset : offset+4])
|
|
vec[i] = float32(v%1000) / 1000
|
|
}
|
|
return vec
|
|
}
|