mjasin/Nexus.Reader
1
0
Fork 0
Code Issues 9 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat: complete KM-RAG polyglot ingestion pipeline migration

Browse Source
This commit is contained in:
Marek Jaisński
2026-05-20 19:55:42 +02:00
parent 711822f5de
commit 03bc649335
15 changed files with 348 additions and 287 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/NexusReader.Application.Tests/Queries/QueryTests.cs
+18 -35
View File
@@ -8,12 +8,13 @@ using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.AI;
using Moq;
using FluentResults;
using NexusReader.Application.Abstractions.Services;
using NexusReader.Application.DTOs.AI;
using NexusReader.Application.DTOs.User;
using NexusReader.Application.Queries.Library;
using NexusReader.Data.Persistence;
using NexusReader.Domain.Entities;
using Pgvector;
using Xunit;
namespace NexusReader.Application.Tests.Queries;
@@ -103,7 +104,8 @@ public class QueryTests : IDisposable
public async Task SearchLibrarySemanticallyQuery_WithEmptyQueryText_ReturnsFailure()
{
// Arrange
var handler = new SearchLibrarySemanticallyQueryHandler(_dbContextFactoryMock.Object, _embeddingGeneratorMock.Object);
var knowledgeServiceMock = new Mock<IKnowledgeService>();
var handler = new SearchLibrarySemanticallyQueryHandler(knowledgeServiceMock.Object);
var query = new SearchLibrarySemanticallyQuery("", "tenant-123");
// Act
@@ -115,44 +117,25 @@ public class QueryTests : IDisposable
}
[Fact]
public async Task SearchLibrarySemanticallyQuery_WithNoResults_TriggersFallback1536Embedding()
public async Task SearchLibrarySemanticallyQuery_WithValidQuery_CallsKnowledgeService()
{
// Arrange
// Mock 768-dim primary embedding generator response
var embedding768 = new Embedding<float>(new float[768]);
var mockResponse768 = new GeneratedEmbeddings<Embedding<float>>(new List<Embedding<float>> { embedding768 });
_embeddingGeneratorMock.Setup(g => g.GenerateAsync(
It.Is<IEnumerable<string>>(s => s.Contains("test")),
It.Is<EmbeddingGenerationOptions>(o => o.Dimensions == 768),
It.IsAny<CancellationToken>()))
.ReturnsAsync(mockResponse768);
// Mock 1536-dim fallback embedding generator response
var embedding1536 = new Embedding<float>(new float[1536]);
var mockResponse1536 = new GeneratedEmbeddings<Embedding<float>>(new List<Embedding<float>> { embedding1536 });
_embeddingGeneratorMock.Setup(g => g.GenerateAsync(
It.Is<IEnumerable<string>>(s => s.Contains("test")),
It.Is<EmbeddingGenerationOptions>(o => o.Dimensions == 1536),
It.IsAny<CancellationToken>()))
.ReturnsAsync(mockResponse1536);
// Seed one legacy cache entry
using (var context = new AppDbContext(_contextOptions))
var knowledgeServiceMock = new Mock<IKnowledgeService>();
var expectedResults = new List<SemanticSearchResultDto>
{
var cacheEntry = new SemanticKnowledgeCache
new SemanticSearchResultDto
{
TenantId = "tenant-123",
ContentHash = "hash-123",
OriginalText = "Fallback Cache Content Snippet",
Vector = new Vector(new float[1536]),
PromptVersion = "1",
CreatedAt = DateTime.UtcNow
};
context.SemanticKnowledgeCache.Add(cacheEntry);
await context.SaveChangesAsync();
}
Snippet = "Semantic search result content snippet",
UnitType = "Concept",
RelevanceScore = 0.95f
}
};
var handler = new SearchLibrarySemanticallyQueryHandler(_dbContextFactoryMock.Object, _embeddingGeneratorMock.Object);
knowledgeServiceMock.Setup(s => s.SearchLibrarySemanticallyAsync("test", "tenant-123", 5, It.IsAny<CancellationToken>()))
.ReturnsAsync(Result.Ok(expectedResults));
var handler = new SearchLibrarySemanticallyQueryHandler(knowledgeServiceMock.Object);
var query = new SearchLibrarySemanticallyQuery("test", "tenant-123");
// Act
@@ -161,7 +144,7 @@ public class QueryTests : IDisposable
// Assert
result.IsSuccess.Should().BeTrue();
result.Value.Should().HaveCount(1);
result.Value.First().Snippet.Should().Be("Fallback Cache Content Snippet");
result.Value.First().Snippet.Should().Be("Semantic search result content snippet");
result.Value.First().ContentHash.Should().Be("hash-123");
}