extracting token count from uid marker

Author: spoons-and-mirrors

lib/messages/utils.ts

@@ -294,7 +294,9 @@ export function annotateContext(messages: WithParts[]): void {
             if (part.type !== "tool") continue
             uid++
             const tokens = countToolTokens(part)
-            const tag = tokens > 0 ? `[uid_${uid}, ${formatTokenCount(tokens)}]` : `[uid_${uid}]`
+            const uidTag = `[uid_${uid}]`
+            const tokenTag = tokens > 0 ? ` [${formatTokenCount(tokens)}]` : ""
+            const tag = `${uidTag}${tokenTag}`
             if (part.state?.status === "completed" && typeof part.state.output === "string") {
                 part.state.output = `${tag}\n${part.state.output}`
             } else if (part.state?.status === "error" && typeof part.state.error === "string") {