chuan/claude-code-router
1
0
Fork 0
mirror of https://github.com/musistudio/claude-code-router.git synced 2026-02-02 23:00:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix count_tokens

Browse Source
This commit is contained in:
musistudio
2025-10-06 08:59:23 +08:00
parent 7de9873908
commit 42074cdbac
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/index.ts
View File

@@ -158,7 +158,7 @@ async function run(options: RunOptions = {}) {
}); });
}); });
server.addHook("preHandler", async (req, reply) => { server.addHook("preHandler", async (req, reply) => {
if (req.url.startsWith("/v1/messages")) { if (req.url.startsWith("/v1/messages") && !req.url.startsWith("/v1/messages/count_tokens")) {
const useAgents = [] const useAgents = []
for (const agent of agentsManager.getAllAgents()) { for (const agent of agentsManager.getAllAgents()) {
@@ -198,7 +198,7 @@ async function run(options: RunOptions = {}) {
event.emit('onError', request, reply, error); event.emit('onError', request, reply, error);
}) })
server.addHook("onSend", (req, reply, payload, done) => { server.addHook("onSend", (req, reply, payload, done) => {
if (req.sessionId && req.url.startsWith("/v1/messages")) { if (req.sessionId && req.url.startsWith("/v1/messages") && !req.url.startsWith("/v1/messages/count_tokens")) {
if (payload instanceof ReadableStream) { if (payload instanceof ReadableStream) {
if (req.agents) { if (req.agents) {
const abortController = new AbortController(); const abortController = new AbortController();