diviner/service/linker/zhipu.py

# -*- coding: utf-8 -*-
from .__load__ import *
from langchain.chat_models import ChatOpenAI

class Zhipu(object):

    def load(self, model = 'glm-4', streaming = True):
        return ZhipuAI(
            model=model,
            api_key=Demeter.config['zhipu']['api_key'],
            streaming=streaming,
        )