AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 23:49:05 +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, "BAAI": DefaultEmbedding,
"Mistral": MistralEmbed, "Mistral": MistralEmbed,
"Bedrock": BedrockEmbed, "Bedrock": BedrockEmbed,
"Groq": GroqChat "Gemini":GeminiEmbed,
} }
@ -44,7 +44,8 @@ CvModel = {
"Xinference": XinferenceCV, "Xinference": XinferenceCV,
"Tongyi-Qianwen": QWenCV, "Tongyi-Qianwen": QWenCV,
"ZHIPU-AI": Zhipu4V, "ZHIPU-AI": Zhipu4V,
"Moonshot": LocalCV "Moonshot": LocalCV,
'Gemini':GeminiCV
} }
@ -61,7 +62,9 @@ ChatModel = {
"BaiChuan": BaiChuanChat, "BaiChuan": BaiChuanChat,
"MiniMax": MiniMaxChat, "MiniMax": MiniMaxChat,
"Mistral": MistralChat, "Mistral": MistralChat,
"Bedrock": BedrockChat 'Gemini' : GeminiChat,
"Bedrock": BedrockChat,
"Groq": GroqChat
} }