From 3502344e89391f1a62bb98782bdba184254e9737 Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Mon, 16 Sep 2024 11:22:08 +0530 Subject: [PATCH] Remove auto install library for chromadb (#1870) --- mem0/configs/vector_stores/chroma.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/mem0/configs/vector_stores/chroma.py b/mem0/configs/vector_stores/chroma.py index ba58b3ac..4a5ecbf5 100644 --- a/mem0/configs/vector_stores/chroma.py +++ b/mem0/configs/vector_stores/chroma.py @@ -9,17 +9,7 @@ class ChromaDbConfig(BaseModel): try: from chromadb.api.client import Client except ImportError: - user_input: Any = input("The 'chromadb' library is required. Install it now? [y/N]: ") - if user_input.lower() == 'y': - try: - subprocess.check_call([sys.executable, "-m", "pip", "install", "chromadb"]) - from chromadb.api.client import Client - except subprocess.CalledProcessError: - print("Failed to install 'chromadb'. Please install it manually using 'pip install chromadb'.") - sys.exit(1) - else: - print("The required 'chromadb' library is not installed.") - sys.exit(1) + raise ImportError("The 'chromadb' library is required. Please install it using 'pip install chromadb'.") Client: ClassVar[type] = Client collection_name: str = Field("mem0", description="Default name for the collection")