mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-10 22:21:22 -07:00
Add session subtask in /stats command (#13750)
This commit is contained in:
@@ -13,34 +13,46 @@ import {
|
||||
CommandKind,
|
||||
} from './types.js';
|
||||
|
||||
function defaultSessionView(context: CommandContext) {
|
||||
const now = new Date();
|
||||
const { sessionStartTime } = context.session.stats;
|
||||
if (!sessionStartTime) {
|
||||
context.ui.addItem(
|
||||
{
|
||||
type: MessageType.ERROR,
|
||||
text: 'Session start time is unavailable, cannot calculate stats.',
|
||||
},
|
||||
Date.now(),
|
||||
);
|
||||
return;
|
||||
}
|
||||
const wallDuration = now.getTime() - sessionStartTime.getTime();
|
||||
|
||||
const statsItem: HistoryItemStats = {
|
||||
type: MessageType.STATS,
|
||||
duration: formatDuration(wallDuration),
|
||||
};
|
||||
|
||||
context.ui.addItem(statsItem, Date.now());
|
||||
}
|
||||
|
||||
export const statsCommand: SlashCommand = {
|
||||
name: 'stats',
|
||||
altNames: ['usage'],
|
||||
description: 'Check session stats. Usage: /stats [model|tools]',
|
||||
description: 'Check session stats. Usage: /stats [session|model|tools]',
|
||||
kind: CommandKind.BUILT_IN,
|
||||
action: (context: CommandContext) => {
|
||||
const now = new Date();
|
||||
const { sessionStartTime } = context.session.stats;
|
||||
if (!sessionStartTime) {
|
||||
context.ui.addItem(
|
||||
{
|
||||
type: MessageType.ERROR,
|
||||
text: 'Session start time is unavailable, cannot calculate stats.',
|
||||
},
|
||||
Date.now(),
|
||||
);
|
||||
return;
|
||||
}
|
||||
const wallDuration = now.getTime() - sessionStartTime.getTime();
|
||||
|
||||
const statsItem: HistoryItemStats = {
|
||||
type: MessageType.STATS,
|
||||
duration: formatDuration(wallDuration),
|
||||
};
|
||||
|
||||
context.ui.addItem(statsItem, Date.now());
|
||||
defaultSessionView(context);
|
||||
},
|
||||
subCommands: [
|
||||
{
|
||||
name: 'session',
|
||||
description: 'Show session-specific usage statistics',
|
||||
kind: CommandKind.BUILT_IN,
|
||||
action: (context: CommandContext) => {
|
||||
defaultSessionView(context);
|
||||
},
|
||||
},
|
||||
{
|
||||
name: 'model',
|
||||
description: 'Show model-specific usage statistics',
|
||||
|
||||
@@ -183,7 +183,8 @@ describe('useSlashCompletion', () => {
|
||||
createTestCommand({
|
||||
name: 'stats',
|
||||
altNames: ['usage'],
|
||||
description: 'check session stats. Usage: /stats [model|tools]',
|
||||
description:
|
||||
'check session stats. Usage: /stats [session|model|tools]',
|
||||
}),
|
||||
createTestCommand({ name: 'clear', description: 'Clear the screen' }),
|
||||
createTestCommand({
|
||||
@@ -282,7 +283,8 @@ describe('useSlashCompletion', () => {
|
||||
createTestCommand({
|
||||
name: 'stats',
|
||||
altNames: ['usage'],
|
||||
description: 'check session stats. Usage: /stats [model|tools]',
|
||||
description:
|
||||
'check session stats. Usage: /stats [session|model|tools]',
|
||||
}),
|
||||
];
|
||||
let result: {
|
||||
@@ -307,7 +309,8 @@ describe('useSlashCompletion', () => {
|
||||
{
|
||||
label: 'stats',
|
||||
value: 'stats',
|
||||
description: 'check session stats. Usage: /stats [model|tools]',
|
||||
description:
|
||||
'check session stats. Usage: /stats [session|model|tools]',
|
||||
commandKind: CommandKind.BUILT_IN,
|
||||
},
|
||||
]);
|
||||
@@ -360,7 +363,8 @@ describe('useSlashCompletion', () => {
|
||||
createTestCommand({
|
||||
name: 'stats',
|
||||
altNames: ['usage'],
|
||||
description: 'check session stats. Usage: /stats [model|tools]',
|
||||
description:
|
||||
'check session stats. Usage: /stats [session|model|tools]',
|
||||
action: vi.fn(),
|
||||
}),
|
||||
];
|
||||
|
||||
Reference in New Issue
Block a user