From eb81d6d0104ccf2ee8fe5e7967f29c84d0fd4f8b Mon Sep 17 00:00:00 2001 From: Tong Li <31761981+litongjava@users.noreply.github.com> Date: Sat, 7 Sep 2024 19:38:38 -1000 Subject: [PATCH] update openai_model_provider.py (#1129) Co-authored-by: litongjava --- .../openai_model_provider/openai_model_provider.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py b/apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py index cbec2d504..c2cc4a27e 100644 --- a/apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py +++ b/apps/setting/models_provider/impl/openai_model_provider/openai_model_provider.py @@ -31,6 +31,9 @@ model_info_list = [ ModelInfo('gpt-4o', '最新的GPT-4o，比gpt-4-turbo更便宜、更快，随OpenAI调整而更新', ModelTypeConst.LLM, openai_llm_model_credential, OpenAIChatModel), + ModelInfo('gpt-4o-mini', '最新的gpt-4o-mini，比gpt-4o更便宜、更快，随OpenAI调整而更新', + ModelTypeConst.LLM, openai_llm_model_credential, + OpenAIChatModel), ModelInfo('gpt-4-turbo', '最新的gpt-4-turbo，随OpenAI调整而更新', ModelTypeConst.LLM, openai_llm_model_credential, OpenAIChatModel), @@ -74,7 +77,14 @@ open_ai_embedding_credential = OpenAIEmbeddingCredential() model_info_embedding_list = [ ModelInfo('text-embedding-ada-002', '', ModelTypeConst.EMBEDDING, open_ai_embedding_credential, - OpenAIEmbeddingModel)] + OpenAIEmbeddingModel), + ModelInfo('text-embedding-3-small', '', + ModelTypeConst.EMBEDDING, open_ai_embedding_credential, + OpenAIEmbeddingModel), + ModelInfo('text-embedding-3-large', '', + ModelTypeConst.EMBEDDING, open_ai_embedding_credential, + OpenAIEmbeddingModel) +] model_info_manage = ModelInfoManage.builder().append_model_info_list(model_info_list).append_default_model_info( ModelInfo('gpt-3.5-turbo', '最新的gpt-3.5-turbo，随OpenAI调整而更新', ModelTypeConst.LLM,