feat: Support gemini embedding model

2024-12-19 18:58:08 +08:00 · 2024-12-19 18:58:08 +08:00 · 2cb2e6d805
commit 2cb2e6d805
parent f533bb5350
1 changed files with 4 additions and 0 deletions
--- 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 = (