Method Retrieve

pkg/knowledge/rag.go:114–162

Retrieve 检索相关知识

(ctx context.Context, query string, options ...RAGOption)

// Retrieve 检索相关知识
func (r *RAG) Retrieve(ctx context.Context, query string, options ...RAGOption) (*RAGResult, error) {
	startTime := time.Now()

	// 应用选项
	opts := &RAGOptions{}
	for _, opt := range options {
		opt(opts)
	}

	// 执行多源检索
	items, err := r.multiSourceRetrieval(ctx, query, opts)
	if err != nil {
		return nil, fmt.Errorf("knowledge: multi-source retrieval failed: %w", err)
	}

	retrievalTime := time.Since(startTime)
	processingStart := time.Now()

	// 重排序
	if r.config.Reranking {
		items = r.rerank(ctx, query, items)
	}

	// 应用过滤
	items = r.applyFilters(items, opts)

	// 生成上下文
	context, contextItems, err := r.generateContext(ctx, items, query)
	if err != nil {
		return nil, fmt.Errorf("knowledge: failed to generate context: %w", err)
	}

	// 查询增强
	enhancedQuery := r.enhanceQuery(query, items)

	processingTime := time.Since(processingStart)

	return &RAGResult{
		Query:          query,
		Results:        items,
		EnhancedQuery:  enhancedQuery,
		Context:        context,
		ContextItems:   contextItems,
		Confidence:     r.calculateConfidence(items),
		RetrievalTime:  retrievalTime,
		ProcessingTime: processingTime,
		TotalTime:      time.Since(startTime),
	}, nil
}

// RetrieveWithMetadata 带元数据的检索
func (r *RAG) RetrieveWithMetadata(ctx context.Context, query string, metadata map[string]any) (*RAGResult, error) {