浏览代码

feat: update free plan rules of spark (#1515)

tags/0.3.30
takatost 2 年前
父节点
当前提交
9de67c586f
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 7 次插入3 次删除
  1. 7
    3
      api/core/third_party/spark/spark_llm.py

+ 7
- 3
api/core/third_party/spark/spark_llm.py 查看文件



class SparkLLMClient: class SparkLLMClient:
def __init__(self, model_name: str, app_id: str, api_key: str, api_secret: str, api_domain: Optional[str] = None): def __init__(self, model_name: str, app_id: str, api_key: str, api_secret: str, api_domain: Optional[str] = None):

domain = 'spark-api.xf-yun.com' if not api_domain else api_domain
domain = 'spark-api.xf-yun.com'
endpoint = 'chat'
if api_domain:
domain = api_domain
if model_name == 'spark-v3':
endpoint = 'multimodal'


model_api_configs = { model_api_configs = {
'spark': { 'spark': {
api_version = model_api_configs[model_name]['version'] api_version = model_api_configs[model_name]['version']


self.chat_domain = model_api_configs[model_name]['chat_domain'] self.chat_domain = model_api_configs[model_name]['chat_domain']
self.api_base = f"wss://{domain}/{api_version}/chat"
self.api_base = f"wss://{domain}/{api_version}/{endpoint}"
self.app_id = app_id self.app_id = app_id
self.ws_url = self.create_url( self.ws_url = self.create_url(
urlparse(self.api_base).netloc, urlparse(self.api_base).netloc,

正在加载...
取消
保存