def test_should_generate_lmdb_cache(
self, embedding_manager: EmbeddingManager):
embedding_manager.add_embedding_config(EMBEDDING_1)
assert embedding_manager.ensure_available(
EMBEDDING_NAME_1) == EMBEDDING_NAME_1
assert not embedding_manager.has_lmdb_cache(EMBEDDING_NAME_1)
embedding_manager.ensure_lmdb_cache_if_enabled(EMBEDDING_NAME_1)
assert embedding_manager.has_lmdb_cache(EMBEDDING_NAME_1)
def test_should_download_registered_embedding(
self, download_manager: MagicMock,
embedding_manager: EmbeddingManager, download_path_1: Path):
embedding_manager.add_embedding_config({
'name': EMBEDDING_NAME_1,
'path': str(download_path_1),
'url': EXTERNAL_TXT_URL_1
})
assert embedding_manager.ensure_available(
EMBEDDING_NAME_1) == EMBEDDING_NAME_1
download_manager.download.assert_called_with(
EXTERNAL_TXT_URL_1, local_file=str(download_path_1))
def test_should_download_registered_mdb_embedding(
self, download_manager: MagicMock,
embedding_manager: EmbeddingManager):
embedding_manager.add_embedding_config({
'name': EMBEDDING_NAME_1,
'url': EXTERNAL_MDB_GZ_URL_1
})
assert embedding_manager.ensure_available(
EMBEDDING_NAME_1) == EMBEDDING_NAME_1
download_manager.download.assert_called_with(
EXTERNAL_MDB_GZ_URL_1,
local_file=str(
embedding_manager.get_embedding_lmdb_cache_data_path(
EMBEDDING_NAME_1)))
