blob: 360ac987dab6557d247e2165a2889c388f61d8f2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
import { readdir } from "node:fs/promises";
import { join, resolve } from "node:path";
import { z } from "zod";
import type { ToolDefinition } from "../types/index.js";
export function createListFilesTool(workingDirectory: string): ToolDefinition {
return {
name: "list_files",
description: "List files and directories at a path relative to the working directory.",
parameters: z.object({
path: z
.string()
.optional()
.describe("Path to list, relative to the working directory. Defaults to '.'"),
}),
execute: async (args: Record<string, unknown>): Promise<string> => {
const relPath = (args.path as string | undefined) ?? ".";
const absolutePath = resolve(join(workingDirectory, relPath));
const absoluteWorkDir = resolve(workingDirectory);
if (!absolutePath.startsWith(`${absoluteWorkDir}/`) && absolutePath !== absoluteWorkDir) {
return `Error: Path "${relPath}" is outside the working directory.`;
}
try {
const entries = await readdir(absolutePath, { withFileTypes: true });
if (entries.length === 0) {
return "(empty directory)";
}
return entries
.map((entry) => (entry.isDirectory() ? `${entry.name}/` : entry.name))
.join("\n");
} catch (err) {
return `Error listing files: ${err instanceof Error ? err.message : String(err)}`;
}
},
};
}
|
