Method Search

pkg/memory/semantic.go:179–204

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

(ctx context.Context, query string, meta map[string]any, topK int)

// Search 在指定命名空间内执行向量检索。
// query 为用户自然语言查询, meta 用于构造 namespace/过滤。
func (sm *SemanticMemory) Search(ctx context.Context, query string, meta map[string]any, topK int) ([]vector.Hit, error) {
	if sm == nil || sm.cfg.Store == nil || sm.cfg.Embedder == nil {
		return nil, nil
	}
	if query == "" {
		return nil, nil
	}
	if topK <= 0 {
		topK = sm.cfg.TopK
	}

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

	return sm.cfg.Store.Query(ctx, vector.Query{
		Vector:    vecs[0],
		TopK:      topK,
		Namespace: sm.namespaceFromMeta(meta),
		Filter:    meta,
	})
}

func (sm *SemanticMemory) namespaceFromMeta(meta map[string]any) string {
	if meta == nil {