[mob] Keep fetching embeddings till no embedding is returned

This commit is contained in:
Neeraj Gupta 2024-05-13 15:50:27 +05:30
parent 50613b8c57
commit 883e246f30

View file

@ -145,9 +145,12 @@ class EmbeddingStore {
}
_logger.info("${remoteEmbeddings.length} embeddings fetched");
return RemoteEmbeddings(
remoteEmbeddings,
remoteEmbeddings.length == limit,
// keep fetching until we get all embeddings. Avoid limit check as
// some embedding fetch might fail on server
remoteEmbeddings.isNotEmpty,
);
}