Method indexInternal

pkg/memory/semantic.go:138–175

indexInternal 内部索引方法。

(ctx context.Context, docID string, text string, meta map[string]any, provenance *MemoryProvenance)

// indexInternal 内部索引方法。
func (sm *SemanticMemory) indexInternal(ctx context.Context, docID string, text string, meta map[string]any, provenance *MemoryProvenance) error {
	if docID == "" || text == "" {
		return errors.New("docID and text are required")
	}

	vecs, err := sm.cfg.Embedder.EmbedText(ctx, []string{text})
	if err != nil {
		return fmt.Errorf("embed text: %w", err)
	}
	if len(vecs) == 0 {
		return errors.New("embedder returned empty vectors")
	}

	// 将文本复制到 metadata 中, 方便检索结果直接携带原文片段。
	metaCopy := make(map[string]any, len(meta)+2)
	for k, v := range meta {
		metaCopy[k] = v
	}
	metaCopy["text"] = text

	// 添加 Provenance 到 metadata
	if provenance != nil {
		provenanceMeta := provenance.ToMetadata()
		for k, v := range provenanceMeta {
			metaCopy[k] = v
		}
	}

	return sm.cfg.Store.Upsert(ctx, []vector.Document{
		{
			ID:        docID,
			Text:      text,
			Embedding: vecs[0],
			Metadata:  metaCopy,
			Namespace: sm.namespaceFromMeta(meta),
		},
	})
}

// Search 在指定命名空间内执行向量检索。
// query 为用户自然语言查询, meta 用于构造 namespace/过滤。