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
commit 8524999284
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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: