- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
27 lines
897 B
C#
27 lines
897 B
C#
using BuildFlow.Application.AiModel.DTOs;
|
||
using BuildFlow.Application.AiModel.Handlers;
|
||
using BuildFlow.Application.AiModel.Queries;
|
||
using BuildFlow.Domain.Enums;
|
||
using FluentAssertions;
|
||
using Xunit;
|
||
|
||
namespace BuildFlow.Rag.Tests;
|
||
|
||
/// <summary>
|
||
/// 查询 AI 模型供应商预设处理器测试。纯逻辑(GetProviderPresets),无外部依赖。
|
||
/// </summary>
|
||
public class GetAiModelProviderPresetsQueryHandlerTests
|
||
{
|
||
[Fact]
|
||
public async Task Handle_ReturnsNonEmptyPresetsIncludingLmStudio()
|
||
{
|
||
var handler = new GetAiModelProviderPresetsQueryHandler();
|
||
|
||
var result = await handler.Handle(new GetAiModelProviderPresetsQuery(), CancellationToken.None);
|
||
|
||
result.Should().NotBeEmpty("预设列表应包含常见供应商");
|
||
result.Should().Contain(p =>
|
||
p.Provider == (int)ModelProviderEnum.LMStudio && p.Label == "LM Studio");
|
||
}
|
||
}
|