kenmirrors/ente
1
0
Fork 0
mirror of https://github.com/ente-io/ente.git synced 2025-08-08 07:28:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

[mob] Allow injecting score threshold in query

Browse Source
This commit is contained in:
Neeraj Gupta 2024-08-14 15:37:29 +05:30
parent b7acd3033d
commit 0c20be98d7
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart
View File

@ -144,6 +144,14 @@ class SemanticSearchService {
String query, {
double? scoreThreshold,
}) async {
// if the query starts with 0.xxx, the split the query to get score threshold and actual query
if (query.startsWith(RegExp(r"0\.\d+"))) {
final parts = query.split(" ");
if (parts.length > 1) {
scoreThreshold = double.parse(parts[0]);
query = parts.sublist(1).join(" ");
}
}
final textEmbedding = await _getTextEmbedding(query);
final queryResults = await _getSimilarities(