刘旭东

描述

增加了对百度文心一言模型的支持，主要支持百度三个主要模型ERNIE-Bot、ERNIE-Bot-4、ERNIE-Bot-Tubor

相关问题

[功能请求]: 请问什么时候能够支持文心一言呢 #928

补充信息

无

Copilot4PR

🤖 Generated by Copilot at 5a27151

Summary

🗣️🌐🐼

This pull request adds support for using ERNIE, a pre-trained dialogue model from Baidu Cloud, as an online service for generating chat responses. It updates the base model class, the presets module, the config files, and the models module to handle the ERNIE model type and its parameters. It also adds a new module ERNIE.py that defines a class to interact with the ERNIE model service.

To chat with ERNIE online
We need some keys to access its mind
We read them from config
Or env vars if we pick
And use ERNIE_Client of a kind

Walkthrough

• Add support for ERNIE dialogue model from Baidu Cloud as an online service (link, link, link, link, link, link, link, link)

• Add two new fields to config_example.json for ERNIE API key and secret key (link)
• Read ERNIE keys from config file or environment variables in modules/config.py and set them as global variables (link)
• Add a new enum value ERNIE to the ModelType class in modules/models/base_model.py to represent the ERNIE model type (link)
• Add a new condition to the get_type method of the ModelType class to check if the model name contains "ernie" and return the ERNIE model type (link)
• Add a new module modules/models/ERNIE.py that defines a class ERNIE_Client that inherits from the BaseLLMModel class and implements the methods to interact with the ERNIE model service (link)
• Add a new branch to the get_model function in modules/models/models.py that imports the ERNIE_Client class and creates an instance of it with the given model name and the ERNIE keys (link)
• Add the three ERNIE model names to the ONLINE_MODELS list in modules/presets.py to check if a model is an online service or a local file (link)
• Add the metadata for the three ERNIE models to the MODEL_METADATA dictionary in modules/presets.py to store information such as the model name and the token limit for each model (link)