changin token count wording

Author: spoons-and-mirrors

lib/messages/inject.ts

@@ -141,7 +141,7 @@ const buildContextPressureTools = (state: SessionState, config: PluginConfig): s
         const paramKey = extractParameterKey(entry.tool, entry.parameters)
         const description = paramKey ? `${entry.tool}, ${paramKey}` : entry.tool
         const tokens = entry.tokenCount ?? 0
-        const tokenSuffix = entry.tokenCount !== undefined ? ` (~${entry.tokenCount} tokens)` : ""
+        const tokenSuffix = entry.tokenCount !== undefined ? ` (${entry.tokenCount} tokens)` : ""
         lines.push({ tokens, text: `- ${description}${tokenSuffix}` })
     })
 

lib/messages/utils.ts

@@ -269,24 +269,19 @@ export const findMessageIndex = (messages: WithParts[], messageId: string): numb
     return messages.findIndex((msg) => msg.info.id === messageId)
 }
 
-function formatTokenCount(n: number): string {
-    if (n >= 1000) return `${(n / 1000).toFixed(1)}k tokens`
-    return `${n} tokens`
-}
-
 export function annotateContext(messages: WithParts[]): void {
     for (const msg of messages) {
         const parts = Array.isArray(msg.parts) ? msg.parts : []
         for (const part of parts) {
             if (part.type !== "tool") continue
             const tokens = countToolTokens(part)
             if (tokens <= 0) continue
-            const tag = `[${formatTokenCount(tokens)}]`
+            const tag = `[This tool added ${tokens} tokens to the context]\n---\n`
             if (part.state?.status === "completed" && typeof part.state.output === "string") {
-                part.state.output = `${tag}\n${part.state.output}`
+                part.state.output = `${tag}${part.state.output}`
             }
             if (part.state?.status === "error" && typeof part.state.error === "string") {
-                part.state.error = `${tag}\n${part.state.error}`
+                part.state.error = `${tag}${part.state.error}`
             }
         }
     }

lib/state/tool-cache.ts

@@ -76,7 +76,7 @@ export function syncToolCache(
                     tokenCount,
                 })
                 logger.info(
-                    `Cached tool id: ${part.callID} (turn ${turnCounter}${tokenCount !== undefined ? `, ~${tokenCount} tokens` : ""})`,
+                    `Cached tool id: ${part.callID} (turn ${turnCounter}${tokenCount !== undefined ? `, ${tokenCount} tokens` : ""})`,
                 )
             }
         }