Method generateContext

pkg/knowledge/rag.go:494–534

generateContext 生成上下文

(_ context.Context, items []*RetrievalItem, _ string)

// generateContext 生成上下文
func (r *RAG) generateContext(_ context.Context, items []*RetrievalItem, _ string) (string, []*KnowledgeItem, error) {
	var contextItems []*KnowledgeItem
	var contextParts []string

	switch r.config.ContextStrategy {
	case ContextStrategyAll:
		for _, item := range items {
			contextItems = append(contextItems, item.Item)
			contextParts = append(contextParts, r.formatContextItem(item.Item))
		}

	case ContextStrategyTopK:
		topK := min(len(items), 5)
		for i := range topK {
			contextItems = append(contextItems, items[i].Item)
			contextParts = append(contextParts, r.formatContextItem(items[i].Item))
		}

	case ContextStrategyDiverse:
		contextItems = r.selectDiverseItems(items, 5)
		for _, item := range contextItems {
			contextParts = append(contextParts, r.formatContextItem(item))
		}

	case ContextStrategyChain:
		// TODO: 实现推理链生成
		for _, item := range items {
			contextItems = append(contextItems, item.Item)
			contextParts = append(contextParts, r.formatContextItem(item.Item))
		}
	}

	context := strings.Join(contextParts, "\n\n")

	// 上下文压缩
	if r.config.ContextCompression && len(context) > r.config.MaxContextLength {
		context = r.compressContext(context)
	}

	return context, contextItems, nil
}

// selectDiverseItems 选择多样化的项目
func (r *RAG) selectDiverseItems(items []*RetrievalItem, maxCount int) []*KnowledgeItem {