build-flow.api/tests/BuildFlow.Rag.Tests/GetAiModelProviderPresetsQueryHandlerTests.cs
xiangning 7479521546 feat(rag): 实现RAG模块的Hybrid检索和向量索引优化
- 添加HNSW向量索引提升检索性能
- 实现Hybrid检索模式支持向量和关键词双路召回
- 采用RRF算法融合多路召回结果
- 支持ParentChild分块模式的上下文回填
- 优化向量写入采用批量SQL操作提升性能
- 添加RAG PostgreSQL连接字符串配置
- 更新部署脚本增加PGVector集成测试
- 修正表名引用从document_chunks到b_document_chunks
- 重构RagRetrievalService实现完整的检索流程
- 增加重排策略支持关键词和LLM重排
- 修改服务器部署地址配置
2026-08-11 10:59:09 +08:00

27 lines
897 B
C#
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

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");
}
}