Added support for chromadb cloud (#3436)

mem0/configs/vector_stores/chroma.py

@@ -10,17 +10,37 @@ class ChromaDbConfig(BaseModel):
         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")
+    collection_name: str = Field("mem0", description="Default name for the collection/database")
     client: Optional[Client] = Field(None, description="Existing ChromaDB client instance")
     path: Optional[str] = Field(None, description="Path to the database directory")
     host: Optional[str] = Field(None, description="Database connection remote host")
     port: Optional[int] = Field(None, description="Database connection remote port")
+    # ChromaDB Cloud configuration
+    api_key: Optional[str] = Field(None, description="ChromaDB Cloud API key")
+    tenant: Optional[str] = Field(None, description="ChromaDB Cloud tenant ID")
 
     @model_validator(mode="before")
-    def check_host_port_or_path(cls, values):
+    def check_connection_config(cls, values):
         host, port, path = values.get("host"), values.get("port"), values.get("path")
-        if not path and not (host and port):
-            raise ValueError("Either 'host' and 'port' or 'path' must be provided.")
+        api_key, tenant = values.get("api_key"), values.get("tenant")
+        
+        # Check if cloud configuration is provided
+        cloud_config = bool(api_key and tenant)
+        
+        # If cloud configuration is provided, remove any default path that might have been added
+        if cloud_config and path == "/tmp/chroma":
+            values.pop("path", None)
+            return values
+        
+        # Check if local/server configuration is provided (excluding default tmp path for cloud config)
+        local_config = bool(path and path != "/tmp/chroma") or bool(host and port)
+        
+        if not cloud_config and not local_config:
+            raise ValueError("Either ChromaDB Cloud configuration (api_key, tenant) or local configuration (path or host/port) must be provided.")
+        
+        if cloud_config and local_config:
+            raise ValueError("Cannot specify both cloud configuration and local configuration. Choose one.")
+            
         return values
 
     @model_validator(mode="before")

mem0/vector_stores/chroma.py

@@ -28,6 +28,8 @@ class ChromaDB(VectorStoreBase):
         host: Optional[str] = None,
         port: Optional[int] = None,
         path: Optional[str] = None,
+        api_key: Optional[str] = None,
+        tenant: Optional[str] = None,
     ):
         """
         Initialize the Chromadb vector store.
@@ -38,10 +40,21 @@ class ChromaDB(VectorStoreBase):
             host (str, optional): Host address for chromadb server. Defaults to None.
             port (int, optional): Port for chromadb server. Defaults to None.
             path (str, optional): Path for local chromadb database. Defaults to None.
+            api_key (str, optional): ChromaDB Cloud API key. Defaults to None.
+            tenant (str, optional): ChromaDB Cloud tenant ID. Defaults to None.
         """
         if client:
             self.client = client
+        elif api_key and tenant:
+            # Initialize ChromaDB Cloud client
+            logger.info("Initializing ChromaDB Cloud client")
+            self.client = chromadb.CloudClient(
+                api_key=api_key,
+                tenant=tenant,
+                database="mem0"  # Use fixed database name for cloud
+            )
         else:
+            # Initialize local or server client
             self.settings = Settings(anonymized_telemetry=False)
 
             if host and port: