Browse Source

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

crazywoola 7 tháng trước cách đây
mục cha
commit
596f160a1e
1 tập tin đã thay đổi với 2 bổ sung5 xóa
  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'],