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

11
src/Server/Searchdomain.cs
View File

@@ -154,7 +154,7 @@ public class Searchdomain
embeddingCache = []; // TODO remove this and implement proper remediation to improve performance
}
public List<(float, string)> Search(string query)
public List<(float, string)> Search(string query, int? topN = null)
{
if (searchCache.TryGetValue(query, out DateTimedSearchResult cachedResult))
{
@@ -190,9 +190,14 @@ public class Searchdomain
}
result.Add((entity.probMethod(datapointProbs), entity.name));
}
List<(float, string)> results = [.. result.OrderByDescending(s => s.Item1)];
IEnumerable<(float, string)> sortedResults = result.OrderByDescending(s => s.Item1);
if (topN is not null)
{
sortedResults = sortedResults.Take(topN ?? 0);
}
List<(float, string)> results = [.. sortedResults];
List<ResultItem> searchResult = new(
[.. results.Select(r =>
[.. sortedResults.Select(r =>
new ResultItem(r.Item1, r.Item2 ))]
);
searchCache[query] = new DateTimedSearchResult(DateTime.Now, searchResult);