Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: expose Cohere and HF embedding support on UI (#236)

Browse Source
This commit is contained in:
Tuan Anh Nguyen Dang (Tadashi_Cin)
2024-09-06 18:18:19 +07:00
committed by GitHub
parent 4d7f16475f
commit 069f0f3c83
2 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
flowsettings.py
View File

@@ -167,7 +167,7 @@ if config("LOCAL_MODEL", default=""):
"default": False,
}
KH_EMBEDDINGS["local-bge-en"] = {
KH_EMBEDDINGS["fast_embed"] = {
"spec": {
"__type__": "kotaemon.embeddings.FastEmbedEmbeddings",
"model_name": "BAAI/bge-base-en-v1.5",
@@ -202,6 +202,23 @@ KH_LLMS["groq"] = {
"default": False,
}
# additional embeddings configurations
KH_EMBEDDINGS["cohere"] = {
"spec": {
"__type__": "kotaemon.embeddings.LCCohereEmbeddings",
"model": "embed-multilingual-v2.0",
"cohere_api_key": "your-key",
},
"default": False,
}
# KH_EMBEDDINGS["huggingface"] = {
# "spec": {
# "__type__": "kotaemon.embeddings.LCHuggingFaceEmbeddings",
# "model_name": "sentence-transformers/all-mpnet-base-v2",
# },
# "default": False,
# }
KH_REASONINGS = [
"ktem.reasoning.simple.FullQAPipeline",
"ktem.reasoning.simple.FullDecomposeQAPipeline",