pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add local llm implementation (#119)

Browse Source
This commit is contained in:
KevinHuSh
2024-03-12 11:57:08 +08:00
committed by GitHub
parent 0452a6db73
commit f1f09df901
17 changed files with 196 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rag/llm/__init__.py
View File

@ -19,22 +19,25 @@ from .cv_model import *
EmbeddingModel = {
"Infiniflow": HuEmbedding,
"local": HuEmbedding,
"OpenAI": OpenAIEmbed,
"通义千问": HuEmbedding, #QWenEmbed,
"智谱AI": ZhipuEmbed
}
CvModel = {
"OpenAI": GptV4,
"Infiniflow": GptV4,
"local": LocalCV,
"通义千问": QWenCV,
"智谱AI": Zhipu4V
}
ChatModel = {
"OpenAI": GptTurbo,
"Infiniflow": GptTurbo,
"智谱AI": ZhipuChat,
"通义千问": QWenChat,
"local": LocalLLM
}