klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove auto install library for chromadb (#1870)

Browse Source
This commit is contained in:
Dev Khant
2024-09-16 11:22:08 +05:30
committed by GitHub
parent 30edf49aaf
commit 3502344e89
1 changed files with 1 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mem0/configs/vector_stores/chroma.py
View File

@@ -9,17 +9,7 @@ class ChromaDbConfig(BaseModel):
try: try:
from chromadb.api.client import Client from chromadb.api.client import Client
except ImportError: except ImportError:
user_input: Any = input("The 'chromadb' library is required. Install it now? [y/N]: ") raise ImportError("The 'chromadb' library is required. Please install it using 'pip install chromadb'.")
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)
Client: ClassVar[type] = Client Client: ClassVar[type] = Client
collection_name: str = Field("mem0", description="Default name for the collection") collection_name: str = Field("mem0", description="Default name for the collection")