From 2cb2e6d805601f7292292e01092f319546fbccdf Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Thu, 19 Dec 2024 18:58:08 +0800
Subject: [PATCH] feat: Support gemini embedding model

---
 .../impl/gemini_model_provider/gemini_model_provider.py       | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/apps/setting/models_provider/impl/gemini_model_provider/gemini_model_provider.py b/apps/setting/models_provider/impl/gemini_model_provider/gemini_model_provider.py
index e8adb8082..556c87ccb 100644
--- a/apps/setting/models_provider/impl/gemini_model_provider/gemini_model_provider.py
+++ b/apps/setting/models_provider/impl/gemini_model_provider/gemini_model_provider.py
@@ -64,6 +64,10 @@ model_embedding_info_list = [
               ModelTypeConst.EMBEDDING,
               gemini_embedding_model_credential,
               GeminiEmbeddingModel),
+    ModelInfo('models/text-embedding-004', '',
+              ModelTypeConst.EMBEDDING,
+              gemini_embedding_model_credential,
+              GeminiEmbeddingModel),
 ]
 
 model_info_manage = (