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

feature: Introducing VoyageAI (#1871)

Browse Source 
* Introducing VoyageAI's embedding models

* Adding back the whitespaces

* Adding the whitespaces back
This commit is contained in:
fzowl
2025-01-16 19:49:46 +01:00
committed by GitHub
parent cc129a0bce
commit 3fecde49b6
4 changed files with 34 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/crewai/utilities/embedding_configurator.py
View File

@@ -14,6 +14,7 @@ class EmbeddingConfigurator:
"vertexai": self._configure_vertexai,
"google": self._configure_google,
"cohere": self._configure_cohere,
"voyageai": self._configure_voyageai,
"bedrock": self._configure_bedrock,
"huggingface": self._configure_huggingface,
"watson": self._configure_watson,
@@ -124,6 +125,17 @@ class EmbeddingConfigurator:
api_key=config.get("api_key"),
)
@staticmethod
def _configure_voyageai(config, model_name):
from chromadb.utils.embedding_functions.voyageai_embedding_function import (
VoyageAIEmbeddingFunction,
)
return VoyageAIEmbeddingFunction(
model_name=model_name,
api_key=config.get("api_key"),
)
@staticmethod
def _configure_bedrock(config, model_name):
from chromadb.utils.embedding_functions.amazon_bedrock_embedding_function import (