Piped input (#104)

* New method for handling stdin. Bypass Ink, and output to stdout. Makes the CLI work like a typical Unix application when called with piped input.

* Fixing a few post-merge errors.

* Format code.

* Clean up lint and format errors.
This commit is contained in:
Allen Hutchison
2025-04-21 17:41:44 -07:00
committed by GitHub
parent cacf0cc0ef
commit 1a167b2ea5
4 changed files with 100 additions and 92 deletions

View File

@@ -8,16 +8,70 @@ import React from 'react';
import { render } from 'ink';
import { App } from './ui/App.js';
import { loadCliConfig } from './config/config.js';
import { readStdin } from './utils/readStdin.js';
import { GeminiClient, ServerTool } from '@gemini-code/server';
import { PartListUnion } from '@google/genai';
async function main() {
let initialInput: string | undefined = undefined;
// Check if input is being piped
if (!process.stdin.isTTY) {
try {
initialInput = await readStdin();
} catch (error) {
console.error('Error reading from stdin:', error);
process.exit(1);
}
}
// Load configuration
const config = loadCliConfig();
// Render UI, passing necessary config values
render(
React.createElement(App, {
config,
}),
);
// Render UI, passing necessary config values and initial input
if (process.stdin.isTTY) {
render(
React.createElement(App, {
config,
initialInput,
}),
);
} else if (initialInput) {
// If not a TTY and we have initial input, process it directly
const geminiClient = new GeminiClient(
config.getApiKey(),
config.getModel(),
);
const toolRegistry = config.getToolRegistry();
const availableTools: ServerTool[] = toolRegistry.getAllTools();
const toolDeclarations = toolRegistry.getFunctionDeclarations();
const chat = await geminiClient.startChat(toolDeclarations);
const request: PartListUnion = [{ text: initialInput }];
try {
for await (const event of geminiClient.sendMessageStream(
chat,
request,
availableTools,
)) {
if (event.type === 'content') {
process.stdout.write(event.value);
}
// We might need to handle other event types later, but for now, just content.
}
process.stdout.write('\n'); // Add a newline at the end
process.exit(0);
} catch (error) {
console.error('Error processing piped input:', error);
process.exit(1);
}
} else {
// If not a TTY and no initial input, exit with an error
console.error('No input provided via stdin.');
process.exit(1);
}
}
// --- Global Unhandled Rejection Handler ---