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

1 year ago · 596f160a1e
parent d8b6c053a2
commit 596f160a1e
1 changed files with 2 additions and 5 deletions
--- a/api/core/tools/provider/builtin/stepfun/tools/image.py
+++ b/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'],