From 42074cdbacd5bfdc876e117efaf23f554680e97b Mon Sep 17 00:00:00 2001
From: musistudio <m@musiiot.top>
Date: Mon, 6 Oct 2025 08:59:23 +0800
Subject: [PATCH] fix count_tokens

---
 src/index.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index d103290..9f19b67 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -158,7 +158,7 @@ async function run(options: RunOptions = {}) {
     });
   });
   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 = []
 
       for (const agent of agentsManager.getAllAgents()) {
@@ -198,7 +198,7 @@ async function run(options: RunOptions = {}) {
     event.emit('onError', request, reply, error);
   })
   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 (req.agents) {
           const abortController = new AbortController();