MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-04-19 01:30:42 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): migrate chat recording to JSONL streaming (#23749)

Browse Source
This commit is contained in:
Spencer
2026-04-09 17:13:55 -04:00
committed by GitHub
parent 45100f7c0e
commit f744913584
15 changed files with 906 additions and 665 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
packages/cli/src/utils/sessionUtils.ts
View File

@@ -12,6 +12,7 @@ import {
type Storage,
type ConversationRecord,
type MessageRecord,
loadConversationRecord,
} from '@google/gemini-cli-core';
import * as fs from 'node:fs/promises';
import path from 'node:path';
@@ -250,23 +251,27 @@ export const getAllSessionFiles = async (
try {
const files = await fs.readdir(chatsDir);
const sessionFiles = files
.filter((f) => f.startsWith(SESSION_FILE_PREFIX) && f.endsWith('.json'))
.filter(
(f) =>
f.startsWith(SESSION_FILE_PREFIX) &&
(f.endsWith('.json') || f.endsWith('.jsonl')),
)
.sort(); // Sort by filename, which includes timestamp
const sessionPromises = sessionFiles.map(
async (file): Promise<SessionFileEntry> => {
const filePath = path.join(chatsDir, file);
try {
// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
const content: ConversationRecord = JSON.parse(
await fs.readFile(filePath, 'utf8'),
);
const content = await loadConversationRecord(filePath, {
metadataOnly: !options.includeFullContent,
});
if (!content) {
return { fileName: file, sessionInfo: null };
}
// Validate required fields
if (
!content.sessionId ||
!content.messages ||
!Array.isArray(content.messages) ||
!content.startTime ||
!content.lastUpdated
) {
@@ -275,7 +280,7 @@ export const getAllSessionFiles = async (
}
// Skip sessions that only contain system messages (info, error, warning)
if (!hasUserOrAssistantMessage(content.messages)) {
if (!content.hasUserOrAssistantMessage) {
return { fileName: file, sessionInfo: null };
}
@@ -285,7 +290,9 @@ export const getAllSessionFiles = async (
return { fileName: file, sessionInfo: null };
}
const firstUserMessage = extractFirstUserMessage(content.messages);
const firstUserMessage = content.firstUserMessage
? cleanMessage(content.firstUserMessage)
: extractFirstUserMessage(content.messages);
const isCurrentSession = currentSessionId
? file.includes(currentSessionId.slice(0, 8))
: false;
@@ -310,11 +317,11 @@ export const getAllSessionFiles = async (
const sessionInfo: SessionInfo = {
id: content.sessionId,
file: file.replace('.json', ''),
file: file.replace(/\.jsonl?$/, ''),
fileName: file,
startTime: content.startTime,
lastUpdated: content.lastUpdated,
messageCount: content.messages.length,
messageCount: content.messageCount ?? content.messages.length,
displayName: content.summary
? stripUnsafeCharacters(content.summary)
: firstUserMessage,
@@ -505,10 +512,10 @@ export class SessionSelector {
const sessionPath = path.join(chatsDir, sessionInfo.fileName);
try {
// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
const sessionData: ConversationRecord = JSON.parse(
await fs.readFile(sessionPath, 'utf8'),
);
const sessionData = await loadConversationRecord(sessionPath);
if (!sessionData) {
throw new Error('Failed to load session data');
}
const displayInfo = `Session ${sessionInfo.index}: ${sessionInfo.firstUserMessage} (${sessionInfo.messageCount} messages, ${formatRelativeTime(sessionInfo.lastUpdated)})`;