|
31 | 31 | ModelTypeConst.LLM, win_xin_llm_model_credential, QianfanChatModel), |
32 | 32 | ModelInfo('qianfan-chinese-llama-2-13b', |
33 | 33 | '', |
34 | | - ModelTypeConst.LLM, win_xin_llm_model_credential, QianfanChatModel) |
| 34 | + ModelTypeConst.LLM, win_xin_llm_model_credential, QianfanChatModel), |
| 35 | + ModelInfo('ernie-4.5-turbo-32k', '', ModelTypeConst.LLM, win_xin_llm_model_credential, |
| 36 | + QianfanChatModel), |
| 37 | + ModelInfo('ernie-speed-8k', '', ModelTypeConst.LLM, win_xin_llm_model_credential, QianfanChatModel), |
| 38 | + ModelInfo('ernie-4.5-0.3b', '', ModelTypeConst.LLM, win_xin_llm_model_credential, QianfanChatModel) |
35 | 39 |
|
36 | 40 | ] |
37 | | -embedding_model_info = ModelInfo('Embedding-V1', |
38 | | - _('Embedding-V1 is a text representation model based on Baidu Wenxin large model technology. It can convert text into a vector form represented by numerical values and can be used in text retrieval, information recommendation, knowledge mining and other scenarios. Embedding-V1 provides the Embeddings interface, which can generate corresponding vector representations based on input content. You can call this interface to input text into the model and obtain the corresponding vector representation for subsequent text processing and analysis.'), |
39 | | - ModelTypeConst.EMBEDDING, qianfan_embedding_credential, QianfanEmbeddings) |
| 41 | +embedding_model_info_list = [ModelInfo('Embedding-V1', |
| 42 | + _('Embedding-V1 is a text representation model based on Baidu Wenxin large model technology. It can convert text into a vector form represented by numerical values and can be used in text retrieval, information recommendation, knowledge mining and other scenarios. Embedding-V1 provides the Embeddings interface, which can generate corresponding vector representations based on input content. You can call this interface to input text into the model and obtain the corresponding vector representation for subsequent text processing and analysis.'), |
| 43 | + ModelTypeConst.EMBEDDING, qianfan_embedding_credential, QianfanEmbeddings), |
| 44 | + ModelInfo('tao-8k', '', ModelTypeConst.EMBEDDING, qianfan_embedding_credential, |
| 45 | + QianfanEmbeddings), |
| 46 | + ModelInfo('bge-large-zh', '', ModelTypeConst.EMBEDDING, qianfan_embedding_credential, |
| 47 | + QianfanEmbeddings) |
| 48 | + ] |
40 | 49 | model_info_manage = ModelInfoManage.builder().append_model_info_list(model_info_list).append_default_model_info( |
41 | 50 | ModelInfo('ERNIE-Bot-4', |
42 | 51 | _('ERNIE-Bot-4 is a large language model independently developed by Baidu. It covers massive Chinese data and has stronger capabilities in dialogue Q&A, content creation and generation.'), |
43 | 52 | ModelTypeConst.LLM, |
44 | 53 | win_xin_llm_model_credential, |
45 | | - QianfanChatModel)).append_model_info(embedding_model_info).append_default_model_info( |
46 | | - embedding_model_info).build() |
| 54 | + QianfanChatModel)).append_model_info_list(embedding_model_info_list).append_default_model_info( |
| 55 | + embedding_model_info_list[0]).build() |
47 | 56 |
|
48 | 57 |
|
49 | 58 | class WenxinModelProvider(IModelProvider): |
|
0 commit comments