add Gemini key (#1480)

### 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]>
infiniflow · Jul 12, 2024 · b96ba6f · b96ba6f
1 parent d29fd52
commit b96ba6f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py
@@ -33,7 +33,7 @@
     "BAAI": DefaultEmbedding,
     "Mistral": MistralEmbed,
     "Bedrock": BedrockEmbed,
-    "Groq": GroqChat
+    "Gemini":GeminiEmbed,
 }
 
 
@@ -44,7 +44,8 @@
     "Xinference": XinferenceCV,
     "Tongyi-Qianwen": QWenCV,
     "ZHIPU-AI": Zhipu4V,
-    "Moonshot": LocalCV
+    "Moonshot": LocalCV,
+    'Gemini':GeminiCV
 }
 
 
@@ -61,7 +62,9 @@
     "BaiChuan": BaiChuanChat,
     "MiniMax": MiniMaxChat,
     "Mistral": MistralChat,
-    "Bedrock": BedrockChat
+    'Gemini' : GeminiChat,
+    "Bedrock": BedrockChat,
+    "Groq": GroqChat
 }