move toolUtils to a separate file

Author: ctlai95

packages/core/src/codewhispererChat/tools/toolUtils.ts

@@ -0,0 +1,126 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Writable } from 'stream'
+import { FsRead, FsReadParams } from './fsRead'
+import { FsWrite, FsWriteParams } from './fsWrite'
+import { ExecuteBash, ExecuteBashParams } from './executeBash'
+import { ToolResult, ToolResultContentBlock, ToolResultStatus, ToolUse } from '@amzn/codewhisperer-streaming'
+import { InvokeOutput } from './toolShared'
+
+export enum ToolType {
+    FsRead = 'fsRead',
+    FsWrite = 'fsWrite',
+    ExecuteBash = 'executeBash',
+}
+
+export type Tool =
+    | { type: ToolType.FsRead; tool: FsRead }
+    | { type: ToolType.FsWrite; tool: FsWrite }
+    | { type: ToolType.ExecuteBash; tool: ExecuteBash }
+
+export class ToolUtils {
+    static displayName(tool: Tool): string {
+        switch (tool.type) {
+            case ToolType.FsRead:
+                return 'Read from filesystem'
+            case ToolType.FsWrite:
+                return 'Write to filesystem'
+            case ToolType.ExecuteBash:
+                return 'Execute shell command'
+        }
+    }
+
+    static requiresAcceptance(tool: Tool) {
+        switch (tool.type) {
+            case ToolType.FsRead:
+                return false
+            case ToolType.FsWrite:
+                return true
+            case ToolType.ExecuteBash:
+                return tool.tool.requiresAcceptance()
+        }
+    }
+
+    static async invoke(tool: Tool, updates: Writable): Promise<InvokeOutput> {
+        switch (tool.type) {
+            case ToolType.FsRead:
+                return tool.tool.invoke(updates)
+            case ToolType.FsWrite:
+                return tool.tool.invoke(updates)
+            case ToolType.ExecuteBash:
+                return tool.tool.invoke(updates)
+        }
+    }
+
+    static queueDescription(tool: Tool, updates: Writable): void {
+        switch (tool.type) {
+            case ToolType.FsRead:
+                tool.tool.queueDescription(updates)
+                break
+            case ToolType.FsWrite:
+                tool.tool.queueDescription(updates)
+                break
+            case ToolType.ExecuteBash:
+                tool.tool.queueDescription(updates)
+                break
+        }
+    }
+
+    static async validate(tool: Tool): Promise<void> {
+        switch (tool.type) {
+            case ToolType.FsRead:
+                return tool.tool.validate()
+            case ToolType.FsWrite:
+                return tool.tool.validate()
+            case ToolType.ExecuteBash:
+                return tool.tool.validate()
+        }
+    }
+
+    static tryFromToolUse(value: ToolUse): Tool | ToolResult {
+        const mapErr = (parseError: any): ToolResult => ({
+            toolUseId: value.toolUseId,
+            content: [
+                {
+                    type: 'text',
+                    text: `Failed to validate tool parameters: ${parseError}. The model has either suggested tool parameters which are incompatible with the existing tools, or has suggested one or more tool that does not exist in the list of known tools.`,
+                } as ToolResultContentBlock,
+            ],
+            status: ToolResultStatus.ERROR,
+        })
+
+        try {
+            switch (value.name) {
+                case ToolType.FsRead:
+                    return {
+                        type: ToolType.FsRead,
+                        tool: new FsRead(value.input as unknown as FsReadParams),
+                    }
+                case ToolType.FsWrite:
+                    return {
+                        type: ToolType.FsWrite,
+                        tool: new FsWrite(value.input as unknown as FsWriteParams),
+                    }
+                case ToolType.ExecuteBash:
+                    return {
+                        type: ToolType.ExecuteBash,
+                        tool: new ExecuteBash(value.input as unknown as ExecuteBashParams),
+                    }
+                default:
+                    return {
+                        toolUseId: value.toolUseId,
+                        content: [
+                            {
+                                type: 'text',
+                                text: `The tool, "${value.name}" is not supported by the client`,
+                            } as ToolResultContentBlock,
+                        ],
+                    }
+            }
+        } catch (error) {
+            return mapErr(error)
+        }
+    }
+}