zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Support gemini embedding model

Browse Source
This commit is contained in:
CaptainB 2024-12-19 18:58:08 +08:00
parent f533bb5350
commit 2cb2e6d805
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/setting/models_provider/impl/gemini_model_provider/gemini_model_provider.py
View File

@ -64,6 +64,10 @@ model_embedding_info_list = [
ModelTypeConst.EMBEDDING, ModelTypeConst.EMBEDDING,
gemini_embedding_model_credential, gemini_embedding_model_credential,
GeminiEmbeddingModel), GeminiEmbeddingModel),
ModelInfo('models/text-embedding-004', '',
ModelTypeConst.EMBEDDING,
gemini_embedding_model_credential,
GeminiEmbeddingModel),
] ]
model_info_manage = ( model_info_manage = (