minor refactor

This commit is contained in:
Neeraj Gupta 2025-02-03 11:18:01 +05:30
parent 4fd3a269b0
commit e42024c912
2 changed files with 16 additions and 4 deletions

View File

@ -85,6 +85,17 @@ class FileDataEntity {
remoteRawData[_clipKey] as Map<String, dynamic>,
)
: null;
RemoteClipEmbedding? getClipEmbeddingIfCompatible(
int minClipMlVersion,
) {
final clipData = remoteRawData[_clipKey];
if (clipData == null) return null;
final clipEmbedding =
RemoteClipEmbedding.fromJson(clipData as Map<String, dynamic>);
return clipEmbedding.version >= minClipMlVersion ? clipEmbedding : null;
}
}
class RemoteFaceEmbedding {

View File

@ -217,13 +217,14 @@ Stream<List<FileMLInstruction>> fetchEmbeddingsAndInstructions(
faces.addAll(facesFromRemoteEmbedding);
existingInstruction.shouldRunFaces = false;
}
if (fileMl.clipEmbedding != null &&
fileMl.clipEmbedding!.version >= clipMlVersion) {
final remoteClipEmbedding =
fileMl.getClipEmbeddingIfCompatible(clipMlVersion);
if (remoteClipEmbedding != null) {
clipEmbeddings.add(
ClipEmbedding(
fileID: fileMl.fileID,
embedding: fileMl.clipEmbedding!.embedding,
version: fileMl.clipEmbedding!.version,
embedding: remoteClipEmbedding.embedding,
version: remoteClipEmbedding.version,
),
);
existingInstruction.shouldRunClip = false;