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

This commit is contained in:
crazywoola 2024-09-04 08:32:22 +08:00 committed by GitHub
parent d8b6c053a2
commit 596f160a1e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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