Add tool configuration management and integrate with session handling

Author: PrashamTrivedi

deno.json

@@ -7,6 +7,7 @@
   "imports": {
     "@std/assert": "jsr:@std/assert@1",
     "@db/sqlite": "jsr:@db/sqlite",
-    "table": "jsr:@sauber/table"
+    "table": "jsr:@sauber/table",
+    "anthropic": "npm:@anthropic-ai/sdk"
   }
 }

deno.lock

@@ -4,7 +4,7 @@ import {parseFlagForHelp, getCommandHelp} from "../utils/functions.ts"
 
 export async function handleSettings(args: string[]): Promise<void> {
     const settingsFlags = {
-        string: ["set-name", "add-command", "remove-command", "set-jina-key"],
+        string: ["set-name", "add-command", "remove-command", "set-jina-key", "set-config"],
         boolean: ["list"],
     }
     const flags = parseArgs(args, settingsFlags)
@@ -15,7 +15,11 @@ export async function handleSettings(args: string[]): Promise<void> {
     }
     const settings = await loadUserSettings()
 
-    if (flags["set-name"]) {
+    if (flags["set-config"]) {
+        settings.toolConfigPath = flags["set-config"]
+        console.log(`Tool configuration path set to: ${flags["set-config"]}`)
+    }
+    else if (flags["set-name"]) {
         settings.userName = flags["set-name"]
     }
     else if (flags["set-jina-key"]) {

src/commands/settings.ts

@@ -1,11 +1,12 @@
 import {join} from "jsr:@std/path"
-import Anthropic from "npm:@anthropic-ai/sdk"
+import Anthropic from "anthropic"
 import {homedir} from "node:os"
 import {isJinaAvailable, loadUserSettings} from "./settings.ts"
 
 export const EDITOR_DIR = join(homedir(), ".ComputerUseAgent", "editor_dir")
 export const SESSIONS_DIR = join(homedir(), ".ComputerUseAgent", "sessions")
 export const LOGS_DIR = join(homedir(), ".ComputerUseAgent", "logs")
+export const DEFAULT_TOOLS_CONFIG_PATH = join(homedir(), ".ComputerUseAgent", "settings.json")
 
 export const EDITOR_SYSTEM_PROMPT = Deno.env.get("EDITOR_SYSTEM_PROMPT") ??
     `You are a helpful assistant that helps users edit and work with text files.
@@ -216,11 +217,7 @@ export const JINA_TOOLS: Anthropic.Beta.BetaTool[] = [
     },
 ]
 
-export const ALL_TOOLS = [
-    ...MEMORY_TOOLS,
-    ...CLIPBOARD_TOOLS,
-    ...(isJinaAvailable() ? JINA_TOOLS : []),
-]
+// Remove ALL_TOOLS export and definition
 
 export function getSystemContext(basePrompt: string): string {
     const settings = loadUserSettings()

src/config/constants.ts

@@ -5,7 +5,8 @@ import {UserSettings} from "../types/interfaces.ts"
 const DEFAULT_SETTINGS: UserSettings = {
     userName: "User",
     customCommands: [],
-    jinaApiKey: undefined
+    jinaApiKey: undefined,
+    toolConfigPath: join(homedir(), ".ComputerUseAgent", "tools.json")
 }
 
 const SETTINGS_PATH = join(homedir(), ".ComputerUseAgent", "settings.json")
@@ -24,11 +25,16 @@ export function saveUserSettings(settings: UserSettings) {
 }
 
 export function isJinaAvailable(): boolean {
-    const settings = loadUserSettings();
-    return Boolean(settings.jinaApiKey);
+    const settings = loadUserSettings()
+    return Boolean(settings.jinaApiKey)
 }
 
 export function getJinaApiKey(): string {
-    const settings = loadUserSettings();
-    return settings.jinaApiKey || "";
+    const settings = loadUserSettings()
+    return settings.jinaApiKey || ""
+}
+
+export function getConfigFileLocation(): string {
+    const settings = loadUserSettings()
+    return settings.toolConfigPath
 }

src/config/settings.ts

@@ -0,0 +1,14 @@
+import {ToolConfig} from "../types/interfaces.ts"
+
+export class ToolConfigManager {
+    loadConfig(configPath: string): ToolConfig[] {
+        try {
+            const configText = Deno.readTextFileSync(configPath)
+            const config = JSON.parse(configText)
+            return Array.isArray(config) ? config : config.tools || []
+        } catch (error) {
+            console.warn(`Could not load config from ${configPath}: ${error}`)
+            return []
+        }
+    }
+}

src/config/tool_config.ts

@@ -49,7 +49,6 @@ async function main() {
     const session = new BashSession(sessionId, flags["no-agi"])
     await session.processBashCommand(prompt)
   } else {
-    // Default to hybrid mode for any other mode value
     const session = new HybridSession(sessionId, flags["no-agi"])
     await session.process(prompt)
   }

src/main.ts

@@ -1,16 +1,18 @@
 import {BaseSession} from "../../utils/session.ts"
-import {COMBINED_SYSTEM_PROMPT, API_CONFIG, MEMORY_TOOLS, CLIPBOARD_TOOLS, JINA_TOOLS} from "../../config/constants.ts"
+import {COMBINED_SYSTEM_PROMPT, API_CONFIG} from "../../config/constants.ts"
 import {log} from "../../config/logging.ts"
 import {ToolHandler} from "../../utils/tool_handler.ts"
-
-
+import {getConfigFileLocation} from "../../config/settings.ts"
+import {ToolConfigManager} from "../../config/tool_config.ts"
 
 export class HybridSession extends BaseSession {
     private toolHandler: ToolHandler
 
     constructor(sessionId: string, noAgi = false) {
         super(sessionId)
-        this.toolHandler = new ToolHandler(noAgi)
+        const configPath = getConfigFileLocation()
+        const toolConfig = new ToolConfigManager().loadConfig(configPath)
+        this.toolHandler = new ToolHandler(noAgi, toolConfig)
     }
 
     async process(prompt: string): Promise<void> {
@@ -39,9 +41,7 @@ export class HybridSession extends BaseSession {
                     tools: [
                         {type: "bash_20241022", name: "bash"},
                         {type: "text_editor_20241022", name: "str_replace_editor"},
-                        ...MEMORY_TOOLS,
-                        ...CLIPBOARD_TOOLS,
-                        ...JINA_TOOLS
+                        ...this.toolHandler.getAllTools()
                     ],
                     system: this.getSystemPrompt(`${COMBINED_SYSTEM_PROMPT}\nSystem Context: ${JSON.stringify(systemInfo)}`),
                     betas: ["computer-use-2024-10-22"],