mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-01 07:13:00 +00:00
fix: Update embedding configuration and fix type errors
- Add configurable embedding provider support - Remove OpenAI dependency for memory reset - Add tests for different embedding providers - Fix type hints and improve docstrings Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -163,12 +163,3 @@ class RAGStorage(BaseRAGStorage):
|
||||
raise Exception(
|
||||
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"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user