[Mem0] Update dependencies and make the package lighter (#1708)

Co-authored-by: Dev-Khant <devkhant24@gmail.com>

mem0/configs/vector_stores/chroma.py

@@ -2,15 +2,20 @@ from typing import Optional, ClassVar, Dict, Any
 
 from pydantic import BaseModel, Field, model_validator
 
+
 class ChromaDbConfig(BaseModel):
     try:
         from chromadb.api.client import Client
     except ImportError:
-        raise ImportError("Chromadb requires extra dependencies. Install with `pip install chromadb`") from None
+        raise ImportError(
+            "Chromadb requires extra dependencies. Install with `pip install chromadb`"
+        ) from None
     Client: ClassVar[type] = Client
 
     collection_name: str = Field("mem0", description="Default name for the collection")
-    client: Optional[Client] = Field(None, description="Existing ChromaDB client instance")
+    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")
@@ -29,9 +34,11 @@ class ChromaDbConfig(BaseModel):
         input_fields = set(values.keys())
         extra_fields = input_fields - allowed_fields
         if extra_fields:
-            raise ValueError(f"Extra fields not allowed: {', '.join(extra_fields)}. Please input only the following fields: {', '.join(allowed_fields)}")
+            raise ValueError(
+                f"Extra fields not allowed: {', '.join(extra_fields)}. Please input only the following fields: {', '.join(allowed_fields)}"
+            )
         return values
-    
+
     model_config = {
         "arbitrary_types_allowed": True,
-    }
+    }

mem0/configs/vector_stores/pgvector.py

@@ -2,11 +2,14 @@ from typing import Optional, Dict, Any
 
 from pydantic import BaseModel, Field, model_validator
 
+
 class PGVectorConfig(BaseModel):
 
     dbname: str = Field("postgres", description="Default name for the database")
     collection_name: str = Field("mem0", description="Default name for the collection")
-    embedding_model_dims: Optional[int] = Field(1536, description="Dimensions of the embedding model")
+    embedding_model_dims: Optional[int] = Field(
+        1536, description="Dimensions of the embedding model"
+    )
     user: Optional[str] = Field(None, description="Database user")
     password: Optional[str] = Field(None, description="Database password")
     host: Optional[str] = Field(None, description="Database host. Default is localhost")
@@ -21,7 +24,7 @@ class PGVectorConfig(BaseModel):
         if not host and not port:
             raise ValueError("Both 'host' and 'port' must be provided.")
         return values
-    
+
     @model_validator(mode="before")
     @classmethod
     def validate_extra_fields(cls, values: Dict[str, Any]) -> Dict[str, Any]:
@@ -29,6 +32,7 @@ class PGVectorConfig(BaseModel):
         input_fields = set(values.keys())
         extra_fields = input_fields - allowed_fields
         if extra_fields:
-            raise ValueError(f"Extra fields not allowed: {', '.join(extra_fields)}. Please input only the following fields: {', '.join(allowed_fields)}")
+            raise ValueError(
+                f"Extra fields not allowed: {', '.join(extra_fields)}. Please input only the following fields: {', '.join(allowed_fields)}"
+            )
         return values
-    

mem0/configs/vector_stores/qdrant.py

@@ -1,16 +1,24 @@
 from pydantic import BaseModel, Field, model_validator
 from typing import Optional, ClassVar, Dict, Any
 
+
 class QdrantConfig(BaseModel):
-    from qdrant_client import QdrantClient 
+    from qdrant_client import QdrantClient
+
     QdrantClient: ClassVar[type] = QdrantClient
 
     collection_name: str = Field("mem0", description="Name of the collection")
-    embedding_model_dims: Optional[int] = Field(1536, description="Dimensions of the embedding model")
-    client: Optional[QdrantClient] = Field(None, description="Existing Qdrant client instance")
+    embedding_model_dims: Optional[int] = Field(
+        1536, description="Dimensions of the embedding model"
+    )
+    client: Optional[QdrantClient] = Field(
+        None, description="Existing Qdrant client instance"
+    )
     host: Optional[str] = Field(None, description="Host address for Qdrant server")
     port: Optional[int] = Field(None, description="Port for Qdrant server")
-    path: Optional[str] = Field("/tmp/qdrant", description="Path for local Qdrant database")
+    path: Optional[str] = Field(
+        "/tmp/qdrant", description="Path for local Qdrant database"
+    )
     url: Optional[str] = Field(None, description="Full URL for Qdrant server")
     api_key: Optional[str] = Field(None, description="API key for Qdrant server")
     on_disk: Optional[bool] = Field(False, description="Enables persistent storage")
@@ -38,9 +46,11 @@ class QdrantConfig(BaseModel):
         input_fields = set(values.keys())
         extra_fields = input_fields - allowed_fields
         if extra_fields:
-            raise ValueError(f"Extra fields not allowed: {', '.join(extra_fields)}. Please input only the following fields: {', '.join(allowed_fields)}")
+            raise ValueError(
+                f"Extra fields not allowed: {', '.join(extra_fields)}. Please input only the following fields: {', '.join(allowed_fields)}"
+            )
         return values
 
     model_config = {
         "arbitrary_types_allowed": True,
-    }
+    }