From 619df1429122517565350f267d4fb6ae7c4da3ba Mon Sep 17 00:00:00 2001
From: Hureru <3507039083@qq.com>
Date: Thu, 19 Mar 2026 01:30:37 +0800
Subject: [PATCH] fix(proxy): apply account-level proxy to count_tokens and
 completions routes

---
 src/server/routes/proxy/chat.ts        | 4 ++--
 src/server/routes/proxy/completions.ts | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/server/routes/proxy/chat.ts b/src/server/routes/proxy/chat.ts
index 51531bb..d4c758f 100644
--- a/src/server/routes/proxy/chat.ts
+++ b/src/server/routes/proxy/chat.ts
@@ -839,7 +839,7 @@ async function handleClaudeCountTokensRequest(
           method: 'POST',
           headers: requestForFetch.headers,
           body: JSON.stringify(requestForFetch.body),
-        }),
+        }, resolveChannelProxyUrl(selected.site, selected.account.extraConfig)),
       });
 
       if (upstream.status === 401 && oauth) {
@@ -859,7 +859,7 @@ async function handleClaudeCountTokensRequest(
               method: 'POST',
               headers: requestForFetch.headers,
               body: JSON.stringify(requestForFetch.body),
-            }),
+            }, resolveChannelProxyUrl(selected.site, selected.account.extraConfig)),
           });
         } catch {
           // Fall through to the regular upstream error handling below.
diff --git a/src/server/routes/proxy/completions.ts b/src/server/routes/proxy/completions.ts
index 1a2bdcd..7e7d1b5 100644
--- a/src/server/routes/proxy/completions.ts
+++ b/src/server/routes/proxy/completions.ts
@@ -71,7 +71,7 @@ export async function completionsProxyRoute(app: FastifyInstance) {
             'Authorization': `Bearer ${selected.tokenValue}`,
           },
           body: JSON.stringify(forwardBody),
-        }));
+        }, getProxyUrlFromExtraConfig(selected.account.extraConfig)));
 
         if (!upstream.ok) {
           const errText = await upstream.text().catch(() => 'unknown error');