andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: Update _generate_embedding signature to match BaseRAGStorage

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-09 19:43:24 +00:00
parent b442fe20a2
commit 86844ff3df
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai/memory/storage/rag_storage.py
View File

@@ -181,12 +181,15 @@ class RAGStorage(BaseRAGStorage):
logging.error(f"Error during {self.type} search: {str(e)}") logging.error(f"Error during {self.type} search: {str(e)}")
return [] return []
def _generate_embedding(self, text: str, metadata: Dict[str, Any]) -> None: def _generate_embedding(self, text: str, metadata: Optional[Dict[str, Any]] = None) -> Any:
"""Generate and store embeddings for the given text. """Generate and store embeddings for the given text.
Args: Args:
text: The text to generate embeddings for text: The text to generate embeddings for
metadata: Additional metadata to store with the embeddings metadata: Optional additional metadata to store with the embeddings
Returns:
Any: The generated embedding or None if only storing
""" """
if not hasattr(self, "app") or not hasattr(self, "collection"): if not hasattr(self, "app") or not hasattr(self, "collection"):
self._initialize_app() self._initialize_app()
@@ -197,6 +200,7 @@ class RAGStorage(BaseRAGStorage):
metadatas=[metadata or {}], metadatas=[metadata or {}],
ids=[str(uuid.uuid4())], ids=[str(uuid.uuid4())],
) )
return None
except Exception as e: except Exception as e:
raise EmbeddingInitializationError(self.type, f"Failed to generate embedding: {str(e)}") raise EmbeddingInitializationError(self.type, f"Failed to generate embedding: {str(e)}")