abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-19 09:33:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initiate the GeminiClient with a config

Browse Source 
Also address the open readability improvement comments from #104.
This commit is contained in:
Jaana Dogan
2025-04-22 11:01:09 -07:00
parent 3db2a796ec
commit 8e0fb9ee2f
5 changed files with 43 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
packages/cli/src/gemini.ts
View File

@@ -9,24 +9,9 @@ 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';
import { GeminiClient } from '@gemini-code/server';
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 and initial input
@@ -34,44 +19,35 @@ async function main() {
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);
return;
}
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
const input = await readStdin();
if (!input) {
console.error('No input provided via stdin.');
process.exit(1);
}
// If not a TTY and we have initial input, process it directly
const geminiClient = new GeminiClient(config);
const chat = await geminiClient.startChat();
try {
for await (const event of geminiClient.sendMessageStream(chat, [
{ text: input },
])) {
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);
}
}
// --- Global Unhandled Rejection Handler ---