getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-19 05:02:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: show (sub) indicator in footer when using OAuth subscription

Browse source
This commit is contained in:
Mario Zechner 2025-12-02 09:32:25 +01:00
parent 8cd3151c2a
commit bc838b021d
4 changed files with 69 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
packages/coding-agent/src/tui/footer.ts
View file

@ -3,6 +3,7 @@ import type { AssistantMessage } from "@mariozechner/pi-ai";
import { type Component, visibleWidth } from "@mariozechner/pi-tui";
import { existsSync, type FSWatcher, readFileSync, watch } from "fs";
import { join } from "path";
import { isModelUsingOAuth } from "../model-config.js";
import { theme } from "../theme/theme.js";
/**
@ -169,7 +170,13 @@ export class FooterComponent implements Component {
if (totalOutput) statsParts.push(`↓${formatTokens(totalOutput)}`);
if (totalCacheRead) statsParts.push(`R${formatTokens(totalCacheRead)}`);
if (totalCacheWrite) statsParts.push(`W${formatTokens(totalCacheWrite)}`);
if (totalCost) statsParts.push(`$${totalCost.toFixed(3)}`);
// Show cost with "(sub)" indicator if using OAuth subscription
const usingSubscription = this.state.model ? isModelUsingOAuth(this.state.model) : false;
if (totalCost || usingSubscription) {
const costStr = `$${totalCost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`;
statsParts.push(costStr);
}
// Colorize context percentage based on usage
let contextPercentStr: string;

7
packages/coding-agent/src/tui/tui-renderer.ts
View file

@ -17,7 +17,7 @@ import {
import { exec } from "child_process";
import { getChangelogPath, parseChangelog } from "../changelog.js";
import { exportSessionToHtml } from "../export-html.js";
import { getApiKeyForModel, getAvailableModels } from "../model-config.js";
import { getApiKeyForModel, getAvailableModels, invalidateOAuthCache } from "../model-config.js";
import { listOAuthProviders, login, logout } from "../oauth/index.js";
import type { SessionManager } from "../session-manager.js";
import type { SettingsManager } from "../settings-manager.js";
@ -1318,7 +1318,8 @@ export class TuiRenderer {
},
);
// Success
// Success - invalidate OAuth cache so footer updates
invalidateOAuthCache();
this.chatContainer.addChild(new Spacer(1));
this.chatContainer.addChild(
new Text(theme.fg("success", `✓ Successfully logged in to ${providerId}`), 1, 0),
@ -1335,6 +1336,8 @@ export class TuiRenderer {
try {
await logout(providerId);
// Invalidate OAuth cache so footer updates
invalidateOAuthCache();
this.chatContainer.addChild(new Spacer(1));
this.chatContainer.addChild(
new Text(theme.fg("success", `✓ Successfully logged out of ${providerId}`), 1, 0),