ソースを参照

Chore/add default step 1x url (#7933)

crazywoola 7 ヶ月 前
コミット
596f160a1e
1 ファイル変更2 行追加5 行削除
  1. 2 5
      api/core/tools/provider/builtin/stepfun/tools/image.py

+ 2 - 5
api/core/tools/provider/builtin/stepfun/tools/image.py

@@ -17,11 +17,8 @@ class StepfunTool(BuiltinTool):
         """
             invoke tools
         """
-        base_url = self.runtime.credentials.get('stepfun_base_url', None)
-        if not base_url:
-            base_url = None
-        else:
-            base_url = str(URL(base_url) / 'v1')
+        base_url = self.runtime.credentials.get('stepfun_base_url', 'https://api.stepfun.com')
+        base_url = str(URL(base_url) / 'v1')
 
         client = OpenAI(
             api_key=self.runtime.credentials['stepfun_api_key'],