spark.py 577 B

123456789101112131415
  1. # -*- coding: utf-8 -*-
  2. from .__load__ import *
  3. from langchain_community.chat_models import SparkLLM
  4. class Spark(object):
  5. def load(self, model = 'qwen-turbo', streaming = True):
  6. return SparkLLM( # 科大讯飞星火(豆包)适配
  7. app_id=Demeter.config['spark']['app_id'],
  8. api_key=Demeter.config['spark']['api_key'],
  9. api_secret=Demeter.config['spark']['api_secret'],
  10. domain=Demeter.config['spark']['domain'],
  11. spark_url=Demeter.config['spark']['spark_url'],
  12. streaming=streaming,
  13. )