ai: add common tool runtime and built-in tools

src/ai/tools/registry.ts

@@ -0,0 +1,116 @@
+import {Environment} from "../../common/environment";
+import {AiTool} from "../tool-types";
+import {braveSearchTool, webSearch} from "./brave-search";
+import {getCurrentDateTime, getCurrentDateTimeTool} from "./datetime";
+import {shellExecute, shellExecuteTool} from "./shell";
+import {ToolHandler} from "./types";
+import {getWeather, getWeatherTool} from "./weather";
+import {getMarketRates, getMarketRatesTool} from "./market-rates";
+import {pythonInterpreterTool, runPythonInterpreter} from "./python-interpretator";
+import {
+    copyPath,
+    copyPathTool,
+    createDirectory,
+    createDirectoryTool,
+    createFile,
+    createFileTool,
+    deletePath,
+    deletePathTool,
+    listDirectory,
+    listDirectoryTool,
+    readFile,
+    readFileTool,
+    renamePath,
+    renamePathTool,
+    updateFile,
+    updateFileTool
+} from "./file-system";
+
+export const getTools = () => {
+    const tools: AiTool[] = [
+        getCurrentDateTimeTool,
+        getMarketRatesTool,
+    ];
+
+    if (Environment.ENABLE_PYTHON_INTERPRETER) {
+        tools.push(pythonInterpreterTool);
+    }
+
+    if (Environment.ENABLE_UNSAFE_EVAL) {
+        tools.push(shellExecuteTool);
+    }
+
+    if (Environment.BRAVE_SEARCH_API_KEY) {
+        tools.push(braveSearchTool);
+    }
+
+    if (Environment.OPEN_WEATHER_MAP_API_KEY) {
+        tools.push(getWeatherTool);
+    }
+
+    if (Environment.FILE_TOOLS_ROOT_DIR && Environment.ENABLE_FS_TOOLS) {
+        tools.push(
+            readFileTool,
+            listDirectoryTool,
+            createFileTool,
+            createDirectoryTool,
+            updateFileTool,
+            renamePathTool,
+            copyPathTool,
+            deletePathTool,
+        );
+    }
+
+    return tools;
+};
+
+export const getToolHandlers = () => {
+    let handlers: Record<string, ToolHandler> = {
+        get_datetime: getCurrentDateTime,
+        get_market_rates: getMarketRates,
+    };
+
+    if (Environment.ENABLE_PYTHON_INTERPRETER) {
+        handlers = {
+            python_interpreter: runPythonInterpreter,
+            ...handlers
+        };
+    }
+
+    if (Environment.ENABLE_UNSAFE_EVAL) {
+        handlers = {
+            shell_execute: shellExecute,
+            ...handlers,
+        };
+    }
+
+    if (Environment.BRAVE_SEARCH_API_KEY) {
+        handlers = {
+            web_search: webSearch,
+            ...handlers,
+        };
+    }
+
+    if (Environment.OPEN_WEATHER_MAP_API_KEY) {
+        handlers = {
+            get_weather: getWeather,
+            ...handlers,
+        };
+    }
+
+    if (Environment.FILE_TOOLS_ROOT_DIR && Environment.ENABLE_FS_TOOLS) {
+        handlers = {
+            read_file: readFile,
+            list_directory: listDirectory,
+            create_file: createFile,
+            create_directory: createDirectory,
+            update_file: updateFile,
+            rename_path: renamePath,
+            copy_path: copyPath,
+            delete_path: deletePath,
+            ...handlers,
+        };
+    }
+
+    return handlers;
+};