mirror of
https://github.com/QwenLM/qwen-code.git
synced 2025-12-20 08:47:44 +00:00
shell bones (#160)
* shell bones * Merge remote-tracking branch 'origin/main' into shell_bones * add line break * another line break * drop the log to avoid breaking terminals * rename tool to be consistent with terminal * fix build
This commit is contained in:
@@ -6,7 +6,7 @@
|
||||
"main": "dist/index.js",
|
||||
"scripts": {
|
||||
"start": "node dist/src/index.js",
|
||||
"build": "tsc --build",
|
||||
"build": "../../scripts/build_package.sh",
|
||||
"clean": "rm -rf dist",
|
||||
"lint": "eslint . --ext .ts,.tsx",
|
||||
"format": "prettier --write .",
|
||||
|
||||
@@ -15,9 +15,11 @@ import { GrepTool } from '../tools/grep.js';
|
||||
import { GlobTool } from '../tools/glob.js';
|
||||
import { EditTool } from '../tools/edit.js';
|
||||
import { TerminalTool } from '../tools/terminal.js';
|
||||
import { ShellTool } from '../tools/shell.js';
|
||||
import { WriteFileTool } from '../tools/write-file.js';
|
||||
import { WebFetchTool } from '../tools/web-fetch.js';
|
||||
import { ReadManyFilesTool } from '../tools/read-many-files.js';
|
||||
import { BaseTool, ToolResult } from '../tools/tools.js';
|
||||
|
||||
const DEFAULT_PASSTHROUGH_COMMANDS = ['ls', 'git', 'npm'];
|
||||
|
||||
@@ -132,17 +134,24 @@ function createToolRegistry(config: Config): ToolRegistry {
|
||||
const registry = new ToolRegistry();
|
||||
const targetDir = config.getTargetDir();
|
||||
|
||||
const tools = [
|
||||
const tools: Array<BaseTool<unknown, ToolResult>> = [
|
||||
new LSTool(targetDir),
|
||||
new ReadFileTool(targetDir),
|
||||
new GrepTool(targetDir),
|
||||
new GlobTool(targetDir),
|
||||
new EditTool(targetDir),
|
||||
new TerminalTool(targetDir, config),
|
||||
new WriteFileTool(targetDir),
|
||||
new WebFetchTool(), // Note: WebFetchTool takes no arguments
|
||||
new ReadManyFilesTool(targetDir),
|
||||
];
|
||||
|
||||
// use ShellTool (next-gen TerminalTool) if environment variable is set
|
||||
if (process.env.SHELL_TOOL) {
|
||||
tools.push(new ShellTool(targetDir, config));
|
||||
} else {
|
||||
tools.push(new TerminalTool(targetDir, config));
|
||||
}
|
||||
|
||||
for (const tool of tools) {
|
||||
registry.registerTool(tool);
|
||||
}
|
||||
|
||||
18
packages/server/src/tools/shell.json
Normal file
18
packages/server/src/tools/shell.json
Normal file
@@ -0,0 +1,18 @@
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"command": {
|
||||
"description": "The exact bash command or sequence of commands (using ';' or '&&') to execute. Must adhere to usage guidelines. Example: 'npm install && npm run build'",
|
||||
"type": "string"
|
||||
},
|
||||
"description": {
|
||||
"description": "Optional: A brief, user-centric explanation of what the command does and why it's being run. Used for logging and confirmation prompts. Example: 'Install project dependencies'",
|
||||
"type": "string"
|
||||
},
|
||||
"runInBackground": {
|
||||
"description": "If true, execute the command in the background using '&'. Defaults to false. Use for servers or long tasks.",
|
||||
"type": "boolean"
|
||||
}
|
||||
},
|
||||
"required": ["command"]
|
||||
}
|
||||
1
packages/server/src/tools/shell.md
Normal file
1
packages/server/src/tools/shell.md
Normal file
@@ -0,0 +1 @@
|
||||
This is a minimal shell tool.
|
||||
44
packages/server/src/tools/shell.ts
Normal file
44
packages/server/src/tools/shell.ts
Normal file
@@ -0,0 +1,44 @@
|
||||
/**
|
||||
* @license
|
||||
* Copyright 2025 Google LLC
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*/
|
||||
|
||||
import path from 'path';
|
||||
import fs from 'fs';
|
||||
import { Config } from '../config/config.js';
|
||||
import { BaseTool, ToolResult } from './tools.js';
|
||||
|
||||
export interface ShellToolParams {
|
||||
command: string;
|
||||
description?: string;
|
||||
}
|
||||
|
||||
export class ShellTool extends BaseTool<ShellToolParams, ToolResult> {
|
||||
static Name: string = 'execute_bash_command';
|
||||
private readonly rootDirectory: string;
|
||||
private readonly config: Config;
|
||||
|
||||
constructor(rootDirectory: string, config: Config) {
|
||||
const toolDisplayName = 'Shell';
|
||||
const descriptionUrl = new URL('shell.md', import.meta.url);
|
||||
const toolDescription = fs.readFileSync(descriptionUrl, 'utf-8');
|
||||
const schemaUrl = new URL('shell.json', import.meta.url);
|
||||
const toolParameterSchema = JSON.parse(fs.readFileSync(schemaUrl, 'utf-8'));
|
||||
super(
|
||||
ShellTool.Name,
|
||||
toolDisplayName,
|
||||
toolDescription,
|
||||
toolParameterSchema,
|
||||
);
|
||||
this.config = config;
|
||||
this.rootDirectory = path.resolve(rootDirectory);
|
||||
}
|
||||
|
||||
async execute(_params: ShellToolParams): Promise<ToolResult> {
|
||||
return {
|
||||
llmContent: 'hello',
|
||||
returnDisplay: 'hello',
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user