|
|
|
|
@ -7,12 +7,15 @@ from chromadb.config import Settings |
|
|
|
|
from dotenv import load_dotenv |
|
|
|
|
from chromadb.utils import embedding_functions |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
load_dotenv('.chroma_env') |
|
|
|
|
|
|
|
|
|
class ChromaDB: |
|
|
|
|
def __init__(self): |
|
|
|
|
def __init__(self, local_deployment: bool = False, db='sci_articles'): |
|
|
|
|
if local_deployment: |
|
|
|
|
self.db = chromadb.PersistentClient(f'chroma_{db}') |
|
|
|
|
else: |
|
|
|
|
self.db = chromadb.HttpClient( |
|
|
|
|
host="https://lasseedfast.se/chroma_ev_cars", |
|
|
|
|
host=os.getenv('CHROMA_HOST'), |
|
|
|
|
settings=Settings( |
|
|
|
|
chroma_client_auth_provider="chromadb.auth.token_authn.TokenAuthClientProvider", |
|
|
|
|
chroma_client_auth_credentials=os.getenv("CHROMA_CLIENT_AUTH_CREDENTIALS"), |
|
|
|
|
|