Method textSearch

pkg/knowledge/rag.go:362–406

textSearch 文本搜索

(ctx context.Context, query string, opts *RAGOptions)

// textSearch 文本搜索
func (r *RAG) textSearch(ctx context.Context, query string, opts *RAGOptions) ([]*RetrievalItem, error) {
	searchQuery := &SearchQuery{
		Query:      query,
		Strategy:   StrategyText,
		MaxResults: opts.MaxResults,
		Namespace:  opts.Namespace,
	}

	// 应用选项过滤
	if opts.KnowledgeType != "" {
		searchQuery.Type = opts.KnowledgeType
	}
	if opts.Category != "" {
		searchQuery.Category = opts.Category
	}
	if len(opts.Tags) > 0 {
		searchQuery.Tags = opts.Tags
	}
	if opts.TimeRange != nil {
		searchQuery.After = &opts.TimeRange.Start
		searchQuery.Before = &opts.TimeRange.End
	}
	if opts.QualityFilter != nil {
		searchQuery.MinConfidence = opts.QualityFilter.MinConfidence
		searchQuery.MinQuality = opts.QualityFilter.MinQuality
	}

	results, err := r.manager.Search(ctx, searchQuery)
	if err != nil {
		return nil, err
	}

	items := make([]*RetrievalItem, len(results))
	for i, result := range results {
		items[i] = &RetrievalItem{
			Item:        &result.Item,
			Score:       result.Score * r.config.TextSearchWeight,
			Source:      "text",
			Explanation: "Text-based search match",
			Rank:        i + 1,
		}
	}

	return items, nil
}

// vectorSearch 向量搜索
func (r *RAG) vectorSearch(_ context.Context, _ string, _ *RAGOptions) ([]*RetrievalItem, error) {