jack 3 settimane fa
parent
commit
cf32f20b65
1 ha cambiato i file con 6 aggiunte e 1 eliminazioni
  1. 6 1
      app/core/runner/thread_runner.py

+ 6 - 1
app/core/runner/thread_runner.py

@@ -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(