diviner/service/linker/deepseek.py

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

class Deepseek(object):

    def load(self, model = '', streaming = True):
        return ChatOpenAI(
            model_name=model,
            openai_api_key=Demeter.config['dp']['api_key'],
            openai_api_base="https://api.deepseek.com",
            streaming=streaming,
        )