klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Ollama as a supported embedding provider (#1344)

Browse Source
This commit is contained in:
Colin O'Brien
2024-05-02 01:08:47 -04:00
committed by GitHub
parent 1a66f961f4
commit a795798156
9 changed files with 46 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
embedchain/utils/misc.py
View File

@@ -449,6 +449,7 @@ def validate_config(config_data):
"google",
"mistralai",
"nvidia",
"ollama",
),
Optional("config"): {
Optional("model"): Optional(str),
@@ -458,6 +459,7 @@ def validate_config(config_data):
Optional("title"): str,
Optional("task_type"): str,
Optional("vector_dimension"): int,
Optional("base_url"): str,
},
},
Optional("embedding_model"): {
@@ -470,6 +472,7 @@ def validate_config(config_data):
"google",
"mistralai",
"nvidia",
"ollama",
),
Optional("config"): {
Optional("model"): str,
@@ -478,6 +481,7 @@ def validate_config(config_data):
Optional("title"): str,
Optional("task_type"): str,
Optional("vector_dimension"): int,
Optional("base_url"): str,
},
},
Optional("chunker"): {