Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 03:32:23 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
takatost 2023-11-13 17:00:36 +08:00 committed by GitHub
parent 92f594f5e7
commit 9de67c586f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/third_party/spark/spark_llm.py vendored
View File

@ -17,8 +17,12 @@ import websocket
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'
domain = 'spark-api.xf-yun.com' if not api_domain else api_domain endpoint = 'chat'
if api_domain:
domain = api_domain
if model_name == 'spark-v3':
endpoint = 'multimodal'
model_api_configs = { model_api_configs = {
'spark': { 'spark': {
@ -38,7 +42,7 @@ class SparkLLMClient:
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,