0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add retrials for model downloads

Browse source
This commit is contained in:
vishnukvmd 2024-01-05 15:59:58 +05:30
parent f24faa0404
commit 39affa0852
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lib/services/semantic_search/frameworks/ml_framework.dart
View file

@ -8,6 +8,7 @@ import "package:photos/core/network/network.dart";
abstract class MLFramework {
static const kImageEncoderEnabled = true;
static const kMaximumRetrials = 3;
final _logger = Logger("MLFramework");
@ -97,13 +98,26 @@ abstract class MLFramework {
basename(getTextModelRemotePath());
}
Future<void> _downloadFile(String url, String savePath) async {
Future<void> _downloadFile(
String url,
String savePath, {
int trialCount = 1,
}) async {
_logger.info("Downloading " + url);
final existingFile = File(savePath);
if (await existingFile.exists()) {
await existingFile.delete();
}
await NetworkClient.instance.getDio().download(url, savePath);
try {
await NetworkClient.instance.getDio().download(url, savePath);
} catch (e, s) {
_logger.severe(e, s);
if (trialCount < kMaximumRetrials) {
return _downloadFile(url, savePath, trialCount: trialCount + 1);
} else {
rethrow;
}
}
}
Future<String> getAccessiblePathForAsset(