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

refactor: Remove duplicate embedding function from RAGStorage

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-05 10:44:40 +00:00
parent 509fb375ca
commit 1c7c4cb828
1 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -38,9 +38,10 @@ class RAGStorage(BaseRAGStorage):
""" """
app: ClientAPI | None = None app: ClientAPI | None = None
collection: Any = None
def __init__( def __init__(
self, type, allow_reset=True, embedder_config=None, crew=None, path=None self, type: str, allow_reset: bool = True, embedder_config: Dict[str, Any] | None = None, crew: Any = None, path: str | None = None
): ):
super().__init__(type, allow_reset, embedder_config, crew) super().__init__(type, allow_reset, embedder_config, crew)
agents = crew.agents if crew else [] agents = crew.agents if crew else []
@@ -50,7 +51,6 @@ class RAGStorage(BaseRAGStorage):
self.storage_file_name = self._build_storage_file_name(type, agents) self.storage_file_name = self._build_storage_file_name(type, agents)
self.type = type self.type = type
self.allow_reset = allow_reset self.allow_reset = allow_reset
self.path = path self.path = path
self._initialize_app() self._initialize_app()
@@ -163,12 +163,3 @@ class RAGStorage(BaseRAGStorage):
raise Exception( raise Exception(
f"An error occurred while resetting the {self.type} memory: {e}" f"An error occurred while resetting the {self.type} memory: {e}"
) )
def _create_default_embedding_function(self):
from chromadb.utils.embedding_functions.openai_embedding_function import (
OpenAIEmbeddingFunction,
)
return OpenAIEmbeddingFunction(
api_key=os.getenv("OPENAI_API_KEY"), model_name="text-embedding-3-small"
)