Feature/bedrock embedder (#1470)

2024-08-01 13:55:28 -04:00
parent 80945df4ca
commit 563a130141
15 changed files with 390 additions and 26 deletions
--- a/embedchain/embedchain/models/embedding_functions.py
+++ b/embedchain/embedchain/models/embedding_functions.py
@@ -5,5 +5,6 @@ class EmbeddingFunctions(Enum):
    OPENAI = "OPENAI"
    HUGGING_FACE = "HUGGING_FACE"
    VERTEX_AI = "VERTEX_AI"
+    AWS_BEDROCK = "AWS_BEDROCK"
    GPT4ALL = "GPT4ALL"
    OLLAMA = "OLLAMA"
--- a/embedchain/embedchain/models/vector_dimensions.py
+++ b/embedchain/embedchain/models/vector_dimensions.py
@@ -12,3 +12,5 @@ class VectorDimensions(Enum):
    NVIDIA_AI = 1024
    COHERE = 384
    OLLAMA = 384
+    AMAZON_TITAN_V1 = 1536
+    AMAZON_TITAN_V2 = 1024