diff --git a/src/browser/App.stories.tsx b/src/browser/App.stories.tsx index ab7e4443dc..aa3e5f5725 100644 --- a/src/browser/App.stories.tsx +++ b/src/browser/App.stories.tsx @@ -541,6 +541,18 @@ export const ActiveWorkspaceWithChat: Story = { projects, workspaces, apiOverrides: { + tokenizer: { + countTokens: () => Promise.resolve(42), + countTokensBatch: (_model, texts) => Promise.resolve(texts.map(() => 42)), + calculateStats: () => + Promise.resolve({ + consumers: [], + totalTokens: 0, + model: "mock-model", + tokenizerName: "mock-tokenizer", + usageHistory: [], + }), + }, providers: { setProviderConfig: () => Promise.resolve({ success: true, data: undefined }), list: () => Promise.resolve(["anthropic", "openai", "xai"]), @@ -1246,6 +1258,13 @@ main }) ); + // Pre-fill input with text so token count is visible + localStorage.setItem( + `input:${workspaceId}`, + "Add OAuth2 support with Google and GitHub providers" + ); + localStorage.setItem(`model:${workspaceId}`, "anthropic:claude-sonnet-4-5"); + initialized.current = true; } diff --git a/src/browser/components/ChatInput/index.tsx b/src/browser/components/ChatInput/index.tsx index 0ad550b505..a6c80316ca 100644 --- a/src/browser/components/ChatInput/index.tsx +++ b/src/browser/components/ChatInput/index.tsx @@ -879,7 +879,7 @@ export const ChatInput: React.FC = (props) => { Calculating tokens… @@ -940,7 +940,7 @@ const TokenCountDisplay: React.FC<{ reader: TokenCountReader }> = ({ reader }) = return null; } return ( -
+
{tokens.toLocaleString()} tokens
);