Method Index

pkg/memory/semantic.go:81–111

Index 将一段文本写入向量索引。 docID 应全局唯一, meta 中可包含 user_id/project_id/resource_id 等信息。 如果启用 Provenance, 将自动创建溯源信息。

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

// docID 应全局唯一, meta 中可包含 user_id/project_id/resource_id 等信息。
// 如果启用 Provenance, 将自动创建溯源信息。
func (sm *SemanticMemory) Index(ctx context.Context, docID string, text string, meta map[string]any) error {
	if sm.cfg.EnableProvenance {
		// 自动创建 Provenance
		sourceID := docID // 使用 docID 作为默认 sourceID
		if sid, ok := meta["source_id"].(string); ok && sid != "" {
			sourceID = sid
		}

		sourceType := sm.cfg.DefaultSourceType
		if st, ok := meta["source_type"].(string); ok && st != "" {
			sourceType = SourceType(st)
		}

		isExplicit := false
		if exp, ok := meta["is_explicit"].(bool); ok {
			isExplicit = exp
		}

		var provenance *MemoryProvenance
		if isExplicit {
			provenance = NewExplicitProvenance(sourceType, sourceID)
		} else {
			provenance = NewProvenance(sourceType, sourceID)
		}

		return sm.IndexWithProvenance(ctx, docID, text, meta, provenance, nil)
	}

	// 不启用 Provenance 时的原始逻辑
	return sm.indexInternal(ctx, docID, text, meta, nil)
}

// IndexWithProvenance 使用指定的 Provenance 索引文本。
// derivedFromIDs 表示该记忆派生自哪些其他记忆。