mirror of
https://github.com/QwenLM/qwen-code.git
synced 2025-12-19 09:33:53 +00:00
f10aaf7e7e
Introduces a process.on('unhandledRejection') handler in src/gemini.ts
as a workaround for an issue where 429 ClientErrors originating from
the @google/genai library's sendMessageStream during iteration can
cause an unhandled rejection, even when caught within local try/catch
blocks in the application code (e.g., in processGeminiStream).
The handler specifically identifies this known 429 ClientError based on
its type and message content. If matched, it logs a warning indicating
the known issue is being suppressed and prevents process.exit(1).
Any other genuinely unhandled promise rejections will still be logged
as critical errors and will terminate the application, maintaining
default behavior for unexpected issues. This workaround mitigates a
suspected library-internal problem related to error propagation during
asynchronous stream iteration.
91 lines
3.4 KiB
TypeScript
91 lines
3.4 KiB
TypeScript
import React from 'react';
|
|
import { render } from 'ink';
|
|
import App from './ui/App.js';
|
|
import { parseArguments } from './config/args.js';
|
|
import { loadEnvironment } from './config/env.js';
|
|
import { getTargetDirectory } from './utils/paths.js';
|
|
import { toolRegistry } from './tools/tool-registry.js';
|
|
import { LSTool } from './tools/ls.tool.js';
|
|
import { ReadFileTool } from './tools/read-file.tool.js';
|
|
import { GrepTool } from './tools/grep.tool.js';
|
|
import { GlobTool } from './tools/glob.tool.js';
|
|
import { EditTool } from './tools/edit.tool.js';
|
|
import { TerminalTool } from './tools/terminal.tool.js';
|
|
import { WriteFileTool } from './tools/write-file.tool.js';
|
|
|
|
async function main() {
|
|
// 1. Configuration
|
|
loadEnvironment();
|
|
const argv = await parseArguments(); // Ensure args.ts imports printWarning from ui/display
|
|
const targetDir = getTargetDirectory(argv.target_dir);
|
|
|
|
// 2. Configure tools
|
|
registerTools(targetDir);
|
|
|
|
// 3. Render UI
|
|
render(React.createElement(App, { directory: targetDir }));
|
|
}
|
|
|
|
// --- Global Unhandled Rejection Handler ---
|
|
process.on('unhandledRejection', (reason, promise) => {
|
|
// Check if this is the known 429 ClientError that sometimes escapes
|
|
// this is a workaround for a specific issue with the way we are calling gemini
|
|
// where a 429 error is thrown but not caught, causing an unhandled rejection
|
|
// TODO(adh): Remove this when the race condition is fixed
|
|
const isKnownEscaped429 =
|
|
reason instanceof Error &&
|
|
reason.name === 'ClientError' &&
|
|
reason.message.includes('got status: 429');
|
|
|
|
if (isKnownEscaped429) {
|
|
// Log it differently and DON'T exit, as it's likely already handled visually
|
|
console.warn('-----------------------------------------');
|
|
console.warn('WORKAROUND: Suppressed known escaped 429 Unhandled Rejection.');
|
|
console.warn('-----------------------------------------');
|
|
console.warn('Reason:', reason);
|
|
// No process.exit(1);
|
|
} else {
|
|
// Log other unexpected unhandled rejections as critical errors
|
|
console.error('=========================================');
|
|
console.error('CRITICAL: Unhandled Promise Rejection!');
|
|
console.error('=========================================');
|
|
console.error('Reason:', reason);
|
|
console.error('Stack trace may follow:');
|
|
if (!(reason instanceof Error)) {
|
|
console.error(reason);
|
|
}
|
|
// Exit for genuinely unhandled errors
|
|
process.exit(1);
|
|
}
|
|
});
|
|
|
|
// --- Global Entry Point ---
|
|
main().catch((error) => {
|
|
console.error('An unexpected critical error occurred:');
|
|
if (error instanceof Error) {
|
|
console.error(error.message);
|
|
} else {
|
|
console.error(String(error));
|
|
}
|
|
process.exit(1);
|
|
});
|
|
|
|
function registerTools(targetDir: string) {
|
|
const lsTool = new LSTool(targetDir);
|
|
const readFileTool = new ReadFileTool(targetDir);
|
|
const grepTool = new GrepTool(targetDir);
|
|
const globTool = new GlobTool(targetDir);
|
|
const editTool = new EditTool(targetDir);
|
|
const terminalTool = new TerminalTool(targetDir);
|
|
const writeFileTool = new WriteFileTool(targetDir);
|
|
|
|
toolRegistry.registerTool(lsTool);
|
|
toolRegistry.registerTool(readFileTool);
|
|
toolRegistry.registerTool(grepTool);
|
|
toolRegistry.registerTool(globTool);
|
|
toolRegistry.registerTool(editTool);
|
|
toolRegistry.registerTool(terminalTool);
|
|
toolRegistry.registerTool(writeFileTool);
|
|
}
|
|
|