# -*- coding: utf-8 -*-
from .__load__ import *
from langchain_community.chat_models import QianfanChatEndpoint

class Baidu(object):

    def load(self, model = 'ERNIE-Bot-turbo', streaming = True):
        for key,value in Demeter.config['baidu'].items():
            os.environ[key.upper()] = value
        return QianfanChatEndpoint(
            streaming=streaming,
            model=model,
        )