improvements all around Knowledge class

src/crewai/knowledge/storage/knowledge_storage.py

@@ -1,14 +1,12 @@
-from crewai.memory.storage.base_rag_storage import BaseRAGStorage
-from crewai.utilities.paths import db_storage_path
-from typing import Optional, List
-
-import chromadb
-import numpy as np
-from typing import Dict, Any
 import uuid
 import contextlib
 import io
 import logging
+import chromadb
+
+from crewai.utilities.paths import db_storage_path
+from typing import Optional, List
+from typing import Dict, Any
 
 
 @contextlib.contextmanager
@@ -28,7 +26,7 @@ def suppress_logging(
     logger.setLevel(original_level)
 
 
-class KnowledgeStorage(BaseRAGStorage):
+class KnowledgeStorage:
     """
     Extends Storage to handle embeddings for memory entries, improving
     search efficiency.
@@ -42,9 +40,6 @@ class KnowledgeStorage(BaseRAGStorage):
         )
         self._initialize_app()
 
-    def _sanitize_role(self, role: str) -> str:
-        return role.replace(" ", "_")
-
     def search(
         self,
         query: List[str],
@@ -94,12 +89,16 @@ class KnowledgeStorage(BaseRAGStorage):
         if self.app:
             self.app.reset()
 
-    def save(self, documents: List[str], metadata: Dict[str, Any]):
+    def save(
+        self, documents: List[str], metadata: Dict[str, Any] | List[Dict[str, Any]]
+    ):
         if self.collection:
+            metadatas = [metadata] if isinstance(metadata, dict) else metadata
+
             self.collection.add(
                 documents=documents,
-                metadatas=metadata,
-                ids=[str(uuid.uuid4())],
+                metadatas=metadatas,
+                ids=[str(uuid.uuid4()) for _ in range(len(documents))],
             )
         else:
             raise Exception("Collection not initialized")