Format fixes

This commit is contained in:
imartinez 2024-02-29 14:50:47 +01:00
parent 8c390812ff
commit 34d48d7b4d
5 changed files with 12 additions and 11 deletions

View file

@ -60,12 +60,14 @@ class VectorStoreComponent:
case "chroma":
try:
from private_gpt.components.vector_store.batched_chroma import \
BatchedChromaVectorStore
import chromadb # type: ignore
from chromadb.config import ( # type: ignore
Settings as ChromaSettings,
)
from private_gpt.components.vector_store.batched_chroma import (
BatchedChromaVectorStore,
)
except ImportError as e:
raise ImportError(
"ChromaDB dependencies not found, install with `poetry install --extras chroma`"