using BuildFlow.Domain.Entities.KnowledgeBase;
using BuildFlow.Domain.Interfaces;

namespace BuildFlow.Rag.Tests.Stub;

/// <summary>
/// Stub IRagRetrievalService:返回预设 sources,绕过 pgvector 真实召回。
/// POC 阶段用,后续 Testcontainers 集成测试覆盖真实召回。
/// </summary>
public sealed class StubRagRetrievalService : IRagRetrievalService
{
    /// <summary>预设的召回结果。</summary>
    public IReadOnlyList<RetrievalHit> Hits { get; init; } = [];

    public Task<List<RetrievalHit>> RetrieveAsync(
        KnowledgeBaseEntity knowledgeBase,
        string question,
        CancellationToken ct = default)
        => Task.FromResult(Hits.ToList());
}