refine db connection (#1494)

### What problem does this PR solve? ### Type of change - [x] Refactoring
2024-07-12 18:11:34 +08:00
parent 0c9a7caa9d
commit 26de9adb41
3 changed files with 3 additions and 3 deletions
--- a/rag/llm/init.py
+++ b/rag/llm/init.py
@@ -33,7 +33,7 @@ EmbeddingModel = {
    "BAAI": DefaultEmbedding,
    "Mistral": MistralEmbed,
    "Bedrock": BedrockEmbed,
-    "Gemini":GeminiEmbed,
+    "Gemini":GeminiEmbed
 }