81f0f618f7
- The tl;dr; is that GC couldn't see what the user was saying when tool call events happened in response. The rason why this was happening was because we were instantly invoking tools that the model told us to invoke and then instantly re-requesting. This resulted in the bug because the genai APIs can't update the chat history before a full response has been completed (doesn't know how to update if it's incomplete). - To address the above issue I had to do quite the large refactor. The gist is that now turns truly drive everything on the server (vs. a server client split). This ensured that when we got tool invocations we could control when/how re-requesting would happen and then also ensure that history was updated. This change also meant that the server would act as an event publisher to enable the client to react to events rather than try and weave in complex logic between the events. - A BIG change that this changeset incudes is the removal of all of the CLI tools in favor of the server tools. - Removed some dead code as part of this - **NOTE: Confirmations are still broken (they were broken prior to this); however, I've set them up to be able to work in the future, I'll dot hat in a follow up to be less breaking to others.** Fixes https://b.corp.google.com/issues/412320087
103 lines
3.1 KiB
TypeScript
103 lines
3.1 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import React from 'react';
|
|
import { render } from 'ink';
|
|
import { App } from './ui/App.js';
|
|
import { toolRegistry } from './tools/tool-registry.js';
|
|
import { loadCliConfig } from './config/config.js';
|
|
import {
|
|
LSTool,
|
|
ReadFileTool,
|
|
GrepTool,
|
|
GlobTool,
|
|
EditTool,
|
|
TerminalTool,
|
|
WriteFileTool,
|
|
WebFetchTool,
|
|
} from '@gemini-code/server';
|
|
|
|
async function main() {
|
|
// Load configuration
|
|
const config = loadCliConfig();
|
|
|
|
// Configure tools using the loaded config
|
|
registerTools(config.getTargetDir());
|
|
|
|
// Render UI, passing necessary config values
|
|
render(
|
|
React.createElement(App, {
|
|
config,
|
|
}),
|
|
);
|
|
}
|
|
|
|
// --- 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 config = loadCliConfig();
|
|
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, config);
|
|
const writeFileTool = new WriteFileTool(targetDir);
|
|
const webFetchTool = new WebFetchTool();
|
|
|
|
toolRegistry.registerTool(lsTool);
|
|
toolRegistry.registerTool(readFileTool);
|
|
toolRegistry.registerTool(grepTool);
|
|
toolRegistry.registerTool(globTool);
|
|
toolRegistry.registerTool(editTool);
|
|
toolRegistry.registerTool(terminalTool);
|
|
toolRegistry.registerTool(writeFileTool);
|
|
toolRegistry.registerTool(webFetchTool);
|
|
}
|