Method RetrieveMultiHop

pkg/knowledge/rag.go:189–240

RetrieveMultiHop 多跳检索

(ctx context.Context, query string, maxHops int)

// RetrieveMultiHop 多跳检索
func (r *RAG) RetrieveMultiHop(ctx context.Context, query string, maxHops int) (*RAGResult, error) {
	if !r.config.EnableMultiHop {
		return r.Retrieve(ctx, query)
	}

	if maxHops <= 0 {
		maxHops = r.config.MaxHops
	}

	var allItems []*RetrievalItem
	var allContextItems []*KnowledgeItem
	var expandedQuery = query

	for range maxHops {
		result, err := r.Retrieve(ctx, expandedQuery, WithMaxResults(r.config.MaxRetrievalResults/2))
		if err != nil {
			break
		}

		if len(result.Results) == 0 {
			break
		}

		// 添加到总结果
		allItems = append(allItems, result.Results...)
		allContextItems = append(allContextItems, result.ContextItems...)

		// 使用top结果扩展查询
		if len(result.Results) > 0 {
			expandedQuery = r.expandQuery(expandedQuery, result.Results[0].Item)
		}
	}

	// 去重和重新排序
	allItems = r.deduplicate(allItems)
	allItems = r.rerank(ctx, query, allItems)
	if len(allItems) > r.config.MaxRetrievalResults {
		allItems = allItems[:r.config.MaxRetrievalResults]
	}

	context, _, _ := r.generateContext(ctx, allItems, query)

	return &RAGResult{
		Query:         query,
		Results:       allItems,
		EnhancedQuery: expandedQuery,
		Context:       context,
		ContextItems:  allContextItems,
		Confidence:    r.calculateConfidence(allItems),
		TotalTime:     time.Since(time.Now()),
	}, nil
}

// RAGOption RAG选项
type RAGOption func(*RAGOptions)