Add Ollama as a supported embedding provider (#1344)

2024-05-02 01:08:47 -04:00
parent 1a66f961f4
commit a795798156
9 changed files with 46 additions and 2 deletions
--- a/embedchain/models/embedding_functions.py
+++ b/embedchain/models/embedding_functions.py
@@ -6,3 +6,4 @@ class EmbeddingFunctions(Enum):
    HUGGING_FACE = "HUGGING_FACE"
    VERTEX_AI = "VERTEX_AI"
    GPT4ALL = "GPT4ALL"
+    OLLAMA = "OLLAMA"
--- a/embedchain/models/vector_dimensions.py
+++ b/embedchain/models/vector_dimensions.py
@@ -11,3 +11,4 @@ class VectorDimensions(Enum):
    MISTRAL_AI = 1024
    NVIDIA_AI = 1024
    COHERE = 384
+    OLLAMA = 384