Merge pull request #184 from SecretiveShell/Infinity-Embed-TODO

Complete conditional infinity import TODO
This commit is contained in:
Brian Dashore
2024-09-04 21:47:49 -04:00
committed by GitHub

View File

@@ -7,13 +7,13 @@ from typing import List, Optional
from common.utils import unwrap
# Conditionally import infinity to sidestep its logger
# TODO: Make this prettier
has_infinity_emb: bool = False
try:
from infinity_emb import EngineArgs, AsyncEmbeddingEngine
has_infinity_emb = True
except ImportError:
has_infinity_emb = False
pass
class InfinityContainer: