MindWorkAI/AI-Studio
1
0 Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2026-11-05 04:27:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed Semantic Search with external data sources (ERI servers) (#1090)

Browse Source 
Co-authored-by: Thorsten Sommer <SommerEngineering@users.noreply.github.com>
This commit is contained in:
Peer HogeterpandThorsten Sommer authored and GitHub committed 2026-10-20 16:24:55 +02:00
1 parent 0f5b041729
commit f6e48e95f3
30 files changed
+1370 -141
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

app/Tests/Tools/ToolCalling/SemanticSearchToolRequestTests.cs
+9 -2
View File
@@ -83,12 +83,12 @@ public sealed class SemanticSearchToolRequestTests
[Test]
public void APageAfterTheFirstComesThroughForOneDataSource()
{
var named = SemanticSearchTool.ReadRequest(Arguments($$"""{"query":"travel expenses","data_source_ids":["{{INTRANET.Id}}"],"page":2}"""), OFFERED);
var named = SemanticSearchTool.ReadRequest(Arguments($$"""{"query":"travel expenses","data_source_ids":["{{HANDBOOK.Id}}"],"page":2}"""), OFFERED);
var onlyOneOffered = SemanticSearchTool.ReadRequest(Arguments("""{"query":"travel expenses","page":2}"""), [HANDBOOK]);
Assert.Multiple(() =>
{
Assert.That(named.DataSources, Is.EqualTo(new[] { INTRANET }));
Assert.That(named.DataSources, Is.EqualTo(new[] { HANDBOOK }));
Assert.That(named.Page, Is.EqualTo(2));
Assert.That(onlyOneOffered.Page, Is.EqualTo(2), "With a single data source offered, leaving it unnamed still means that one.");
});
@@ -101,6 +101,13 @@ public sealed class SemanticSearchToolRequestTests
Assert.That(message, Does.Contain("at most 9").And.Contain("but was 10").And.Contain("Rephrase the query"));
}
[Test]
public void AnERIDataSourceHasNoFurtherPage()
{
var message = Refusal(() => SemanticSearchTool.ReadRequest(Arguments($$"""{"query":"travel expenses","data_source_ids":["{{INTRANET.Id}}"],"page":2}"""), OFFERED));
Assert.That(message, Does.Contain("at most 1").And.Contain("but was 2").And.Contain("Rephrase the query"), "A server which allows only the configured number of matches would refuse the page, and the data source would count as not searched.");
}
private static JsonElement Arguments(string json) => JsonSerializer.Deserialize<JsonElement>(json);
private static JsonElement Query(string query) => JsonSerializer.SerializeToElement(new Dictionary<string, string> { ["query"] = query });