Refactor _chromadb.py to improve code structure and readability

main
lasseedfast 1 year ago
parent afd61fac25
commit d1261d2d90
  1. 9
      _chromadb.py

@ -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"),

Loading…
Cancel
Save