more fixes

This commit is contained in:
Brandon Hancock
2024-12-12 14:12:35 -05:00
parent 8ad6bf5cd1
commit 476d7433f9
2 changed files with 37 additions and 24 deletions

View File

@@ -177,6 +177,19 @@ class KnowledgeStorage(BaseKnowledgeStorage):
except Exception as e:
Logger(verbose=True).log("error", f"Failed to upsert documents: {e}", "red")
raise
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"
)
def _set_embedder_config(
self, embedder_config: Optional[Dict[str, Any]] = None
) -> None:
"""Set the embedding configuration for the knowledge storage.
Args: