|
@@ -323,12 +323,17 @@ class ThreadRunner:
|
|
|
print(token_id)
|
|
|
if not token_id:
|
|
|
token = TokenService.get_token_by_id(self.session, token_id)
|
|
|
+ return LLMBackend(
|
|
|
+ base_url=token.llm_base_url, api_key=token.llm_api_key
|
|
|
+ )
|
|
|
else:
|
|
|
token = {
|
|
|
"llm_base_url": "https://onehub.cocorobo.cn/v1",
|
|
|
"llm_api_key": "sk-vTqeBKDC2j6osbGt89A2202dAd1c4fE8B1D294388b569e54",
|
|
|
}
|
|
|
- return LLMBackend(base_url=token.llm_base_url, api_key=token.llm_api_key)
|
|
|
+ return LLMBackend(
|
|
|
+ base_url=token.get("llm_base_url"), api_key=token.get("llm_api_key")
|
|
|
+ )
|
|
|
else:
|
|
|
# init llm backend with llm settings
|
|
|
return LLMBackend(
|