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

class Spark(object):

    def load(self, model = 'qwen-turbo', streaming = True):
        return SparkLLM(  # 科大讯飞星火（豆包）适配
            app_id=Demeter.config['spark']['app_id'],
            api_key=Demeter.config['spark']['api_key'],
            api_secret=Demeter.config['spark']['api_secret'],
            domain=Demeter.config['spark']['domain'],
            spark_url=Demeter.config['spark']['spark_url'],
            streaming=streaming,
        )