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

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

@@ -317,7 +317,17 @@ class ThreadRunner:
                 relation_type=RelationType.Assistant,
                 relation_id=assistant_id,
             )
-            token = TokenService.get_token_by_id(self.session, token_id)
+            print(
+                "token_idtoken_idtoken_idtoken_idtoken_idtoken_idtoken_idtoken_idtoken_idtoken_idtoken_idtoken_id"
+            )
+            print(token_id)
+            if not token_id:
+                token = TokenService.get_token_by_id(self.session, token_id)
+            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)
         else:
             # init llm backend with llm settings