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

Sbazar-GmbH · Sep 6, 2024 · a5f404a · a5f404a
1 parent 8f0aef9
commit a5f404a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/api/core/tools/provider/builtin/stepfun/tools/image.py b/api/core/tools/provider/builtin/stepfun/tools/image.py
@@ -17,11 +17,8 @@ def _invoke(self,
         """
             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'],