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

View File

@@ -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'],