diviner/service/linker/moonshot.py

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

class Moonshot(object):

    def load(self, model = 'moonshot-v1-32k', streaming = True):
        return MoonshotChat(
            model=model,
            moonshot_api_key=Demeter.config['moonshot']['api_key'],
            streaming=streaming,
        )