AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 19:29:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

add Gemini key (#1480)

Browse Source 
### What problem does this PR solve?

#1036

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
This commit is contained in:
黄腾 2024-07-12 13:41:58 +08:00 committed by GitHub
parent d29fd52e14
commit b96ba6f831
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rag/llm/__init__.py
View File

@ -33,7 +33,7 @@ EmbeddingModel = {
"BAAI": DefaultEmbedding,
"Mistral": MistralEmbed,
"Bedrock": BedrockEmbed,
"Groq": GroqChat
"Gemini":GeminiEmbed,
}
@ -44,7 +44,8 @@ CvModel = {
"Xinference": XinferenceCV,
"Tongyi-Qianwen": QWenCV,
"ZHIPU-AI": Zhipu4V,
"Moonshot": LocalCV
"Moonshot": LocalCV,
'Gemini':GeminiCV
}
@ -61,7 +62,9 @@ ChatModel = {
"BaiChuan": BaiChuanChat,
"MiniMax": MiniMaxChat,
"Mistral": MistralChat,
"Bedrock": BedrockChat
'Gemini' : GeminiChat,
"Bedrock": BedrockChat,
"Groq": GroqChat
}