AI-Studio/app/MindWork AI Studio/Tools/RAG/RetrievalTextContext.cs

namespace AIStudio.Tools.RAG;

/// <summary>
/// The retrieval context for text data.
/// </summary>
public sealed class RetrievalTextContext : IRetrievalContext
{
    #region Implementation of IRetrievalContext
    
    /// <inheritdoc />
    public required string DataSourceName { get; init; }
    
    /// <inheritdoc />
    public required RetrievalContentCategory Category { get; init; }
    
    /// <inheritdoc />
    public required RetrievalContentType Type { get; init; }
    
    /// <inheritdoc />
    public required string Path { get; init; }

    /// <inheritdoc />
    public IReadOnlyList<string> Links { get; init; } = [];

    #endregion

    /// <summary>
    /// The text content.
    /// </summary>
    /// <remarks>
    /// Should contain the matched text and some small context around it.
    /// </remarks>
    public required string MatchedText { get; set; }

    /// <summary>
    /// The surrounding content of the matched text.
    /// </summary>
    /// <remarks>
    /// Might give the user some context about the matched text.
    /// For example, one sentence or paragraph before and after the matched text.
    /// </remarks>
    public IReadOnlyList<string> SurroundingContent { get; set; } = [];

    /// <summary>
    /// Optional title used when this context is displayed as a source reference.
    /// </summary>
    public string ReferenceTitle { get; init; } = string.Empty;

    /// <summary>
    /// Optional link used when this context is displayed as a source reference.
    /// </summary>
    public string ReferenceLink { get; init; } = string.Empty;

    /// <summary>
    /// The page this passage was found on, or null when it has none.
    /// </summary>
    /// <remarks>
    /// Kept as a number rather than only inside the reference title: the AI is told the page so it
    /// can say where an answer comes from, and a source has to name a page a program can be sent
    /// to. A slide or a sheet has no page and leaves this empty.
    /// </remarks>
    public int? PageNumber { get; init; }
}