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

Pulled manual commands to seperate function

Browse Source
This commit is contained in:
Seth Troisi
2025-04-25 00:12:20 +00:00
parent 4ce897d19d
commit ed12a2e133
1 changed files with 60 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

104
packages/cli/src/ui/hooks/useGeminiStream.ts
View File

@@ -98,6 +98,61 @@ export const useGeminiStream = (
[setHistory], [setHistory],
); );
// Possibly handle a query manually, return true if handled.
const handleQueryManually = (rawQuery: PartListUnion): boolean => {
if (typeof rawQuery !== 'string') {
return false;
}
const query = rawQuery.trim();
const maybeCommand = query.split(/\s+/)[0];
if (query === 'clear') {
// This just clears the *UI* history, not the model history.
// TODO: add a slash command for that.
setDebugMessage('Clearing terminal.');
setHistory((_) => []);
return true;
}
if (config.getPassthroughCommands().includes(maybeCommand)) {
// Execute and capture output
const targetDir = config.getTargetDir();
setDebugMessage(`Executing shell command in ${targetDir}: ${query}`);
const execOptions = {
cwd: targetDir,
};
_exec(query, execOptions, (error, stdout, stderr) => {
const timestamp = getNextMessageId(Date.now());
if (error) {
addHistoryItem(
setHistory,
{ type: 'error', text: error.message },
timestamp,
);
} else if (stderr) {
addHistoryItem(
setHistory,
{ type: 'error', text: stderr },
timestamp,
);
} else {
// Add stdout as an info message
addHistoryItem(
setHistory,
{ type: 'info', text: stdout || '' },
timestamp,
);
}
// Set state back to Idle *after* command finishes and output is added
setStreamingState(StreamingState.Idle);
});
// Set state to Responding while the command runs
setStreamingState(StreamingState.Responding);
return true; // Prevent Gemini call
}
return true;
}
// Improved submit query function // Improved submit query function
const submitQuery = useCallback( const submitQuery = useCallback(
async (query: PartListUnion) => { async (query: PartListUnion) => {
@@ -105,50 +160,11 @@ export const useGeminiStream = (
if (typeof query === 'string' && query.trim().length === 0) return; if (typeof query === 'string' && query.trim().length === 0) return;
if (typeof query === 'string') { if (typeof query === 'string') {
setDebugMessage(`User query: ${query}`); setDebugMessage(`User query: '${query}'`);
const maybeCommand = query.split(/\s+/)[0]; }
if (query.trim() === 'clear') {
// This just clears the *UI* history, not the model history. if (handleQueryManually(query)) {
// TODO: add a slash command for that. return;
setDebugMessage('Clearing terminal.');
setHistory((_) => []);
return;
} else if (config.getPassthroughCommands().includes(maybeCommand)) {
// Execute and capture output
const targetDir = config.getTargetDir();
setDebugMessage(`Executing shell command in ${targetDir}: ${query}`);
const execOptions = {
cwd: targetDir,
};
_exec(query, execOptions, (error, stdout, stderr) => {
const timestamp = getNextMessageId(Date.now());
if (error) {
addHistoryItem(
setHistory,
{ type: 'error', text: error.message },
timestamp,
);
} else if (stderr) {
addHistoryItem(
setHistory,
{ type: 'error', text: stderr },
timestamp,
);
} else {
// Add stdout as an info message
addHistoryItem(
setHistory,
{ type: 'info', text: stdout || '' },
timestamp,
);
}
// Set state back to Idle *after* command finishes and output is added
setStreamingState(StreamingState.Idle);
});
// Set state to Responding while the command runs
setStreamingState(StreamingState.Responding);
return; // Prevent Gemini call
}
} }
const userMessageTimestamp = Date.now(); const userMessageTimestamp = Date.now();