Method handleQueryRAG

pkg/tools/builtin/rag/rag.go:170–205

(ctx context.Context, args queryRAGArgs)

}

func (t *ToolSet) handleQueryRAG(ctx context.Context, args queryRAGArgs) (*tools.ToolCallResult, error) {
	if args.Query == "" {
		return nil, errors.New("query cannot be empty")
	}

	results, err := t.manager.Query(ctx, args.Query)
	if err != nil {
		return nil, fmt.Errorf("RAG query failed: %w", err)
	}

	out := make([]queryResult, 0, len(results))
	for _, r := range results {
		out = append(out, queryResult{
			SourcePath: r.Document.SourcePath,
			Content:    r.Document.Content,
			Similarity: r.Similarity,
			ChunkIndex: r.Document.ChunkIndex,
		})
	}

	slices.SortFunc(out, func(a, b queryResult) int {
		return cmp.Compare(b.Similarity, a.Similarity)
	})

	const maxResults = 10
	if len(out) > maxResults {
		out = out[:maxResults]
	}

	resultJSON, err := json.Marshal(out)
	if err != nil {
		return nil, fmt.Errorf("failed to marshal results: %w", err)
	}

	return tools.ResultSuccess(string(resultJSON)), nil
}