lucretia/embeddingsearch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added filtering query output to top n elements

Browse Source
This commit is contained in:
LD-Reborn
2025-12-20 20:13:28 +01:00
parent 1375e45f59
commit e4a711fcbd
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Server/Controllers/EntityController.cs
View File

@@ -25,7 +25,7 @@ public class EntityController : ControllerBase
}
[HttpGet("Query")]
public ActionResult<EntityQueryResults> Query(string searchdomain, string query)
public ActionResult<EntityQueryResults> Query(string searchdomain, string query, int? topN)
{
Searchdomain searchdomain_;
try
@@ -40,7 +40,7 @@ public class EntityController : ControllerBase
_logger.LogError("Unable to retrieve the searchdomain {searchdomain} - {ex.Message} - {ex.StackTrace}", [searchdomain, ex.Message, ex.StackTrace]);
return Ok(new EntityQueryResults() {Results = [], Success = false, Message = "Unable to retrieve the searchdomain - it likely exists, but some other error happened." });
}
var results = searchdomain_.Search(query);
List<(float, string)> results = searchdomain_.Search(query, topN);
List<EntityQueryResult> queryResults = [.. results.Select(r => new EntityQueryResult
{
Name = r.Item2,