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 = (