add Gemini key (#1480)
Browse files### 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 <[email protected]>
- rag/llm/__init__.py +6 -3
rag/llm/__init__.py
CHANGED
|
@@ -33,7 +33,7 @@ EmbeddingModel = {
|
|
| 33 |
"BAAI": DefaultEmbedding,
|
| 34 |
"Mistral": MistralEmbed,
|
| 35 |
"Bedrock": BedrockEmbed,
|
| 36 |
-
"
|
| 37 |
}
|
| 38 |
|
| 39 |
|
|
@@ -44,7 +44,8 @@ CvModel = {
|
|
| 44 |
"Xinference": XinferenceCV,
|
| 45 |
"Tongyi-Qianwen": QWenCV,
|
| 46 |
"ZHIPU-AI": Zhipu4V,
|
| 47 |
-
"Moonshot": LocalCV
|
|
|
|
| 48 |
}
|
| 49 |
|
| 50 |
|
|
@@ -61,7 +62,9 @@ ChatModel = {
|
|
| 61 |
"BaiChuan": BaiChuanChat,
|
| 62 |
"MiniMax": MiniMaxChat,
|
| 63 |
"Mistral": MistralChat,
|
| 64 |
-
|
|
|
|
|
|
|
| 65 |
}
|
| 66 |
|
| 67 |
|
|
|
|
| 33 |
"BAAI": DefaultEmbedding,
|
| 34 |
"Mistral": MistralEmbed,
|
| 35 |
"Bedrock": BedrockEmbed,
|
| 36 |
+
"Gemini":GeminiEmbed,
|
| 37 |
}
|
| 38 |
|
| 39 |
|
|
|
|
| 44 |
"Xinference": XinferenceCV,
|
| 45 |
"Tongyi-Qianwen": QWenCV,
|
| 46 |
"ZHIPU-AI": Zhipu4V,
|
| 47 |
+
"Moonshot": LocalCV,
|
| 48 |
+
'Gemini':GeminiCV
|
| 49 |
}
|
| 50 |
|
| 51 |
|
|
|
|
| 62 |
"BaiChuan": BaiChuanChat,
|
| 63 |
"MiniMax": MiniMaxChat,
|
| 64 |
"Mistral": MistralChat,
|
| 65 |
+
'Gemini' : GeminiChat,
|
| 66 |
+
"Bedrock": BedrockChat,
|
| 67 |
+
"Groq": GroqChat
|
| 68 |
}
|
| 69 |
|
| 70 |
|