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/configs/ollama.yaml
+++ b/configs/ollama.yaml
@@ -8,6 +8,7 @@ llm:
    base_url: http://localhost:11434

 embedder:
-  provider: huggingface
+  provider: ollama
  config:
-    model: 'BAAI/bge-small-en-v1.5'
+    model: 'mxbai-embed-large:latest'
+    base_url: http://localhost:11434