[Misc] Lint code and fix code smells (#1871)

mem0/vector_stores/chroma.py

@@ -80,24 +80,14 @@ class ChromaDB(VectorStoreBase):
             values.append(value)
 
         ids, distances, metadatas = values
-        max_length = max(
-            len(v) for v in values if isinstance(v, list) and v is not None
-        )
+        max_length = max(len(v) for v in values if isinstance(v, list) and v is not None)
 
         result = []
         for i in range(max_length):
             entry = OutputData(
                 id=ids[i] if isinstance(ids, list) and ids and i < len(ids) else None,
-                score=(
-                    distances[i]
-                    if isinstance(distances, list) and distances and i < len(distances)
-                    else None
-                ),
-                payload=(
-                    metadatas[i]
-                    if isinstance(metadatas, list) and metadatas and i < len(metadatas)
-                    else None
-                ),
+                score=(distances[i] if isinstance(distances, list) and distances and i < len(distances) else None),
+                payload=(metadatas[i] if isinstance(metadatas, list) and metadatas and i < len(metadatas) else None),
             )
             result.append(entry)
 
@@ -143,9 +133,7 @@ class ChromaDB(VectorStoreBase):
         logger.info(f"Inserting {len(vectors)} vectors into collection {self.collection_name}")
         self.collection.add(ids=ids, embeddings=vectors, metadatas=payloads)
 
-    def search(
-        self, query: List[list], limit: int = 5, filters: Optional[Dict] = None
-    ) -> List[OutputData]:
+    def search(self, query: List[list], limit: int = 5, filters: Optional[Dict] = None) -> List[OutputData]:
         """
         Search for similar vectors.
 
@@ -157,9 +145,7 @@ class ChromaDB(VectorStoreBase):
         Returns:
             List[OutputData]: Search results.
         """
-        results = self.collection.query(
-            query_embeddings=query, where=filters, n_results=limit
-        )
+        results = self.collection.query(query_embeddings=query, where=filters, n_results=limit)
         final_results = self._parse_output(results)
         return final_results
 
@@ -225,9 +211,7 @@ class ChromaDB(VectorStoreBase):
         """
         return self.client.get_collection(name=self.collection_name)
 
-    def list(
-        self, filters: Optional[Dict] = None, limit: int = 100
-    ) -> List[OutputData]:
+    def list(self, filters: Optional[Dict] = None, limit: int = 100) -> List[OutputData]:
         """
         List all vectors in a collection.