feat: add type to tool maps

mlc-ai · Oct 22, 2024 · f90a54f · f90a54f
1 parent a2839e4
commit f90a54f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/src/action.ts b/src/action.ts
@@ -126,6 +126,7 @@ export const actions: Record<string, Tool> = {
         },
       },
     },
+    type: "action",
     implementation: replaceSelectedText,
   },
   appendTextToDocument: {
@@ -147,6 +148,7 @@ export const actions: Record<string, Tool> = {
         },
       },
     },
+    type: "action",
     implementation: appendTextToDocument,
   },
   createCalendarEvent: {
@@ -173,6 +175,7 @@ export const actions: Record<string, Tool> = {
         },
       },
     },
+    type: "action",
     implementation: createCalendarEvent,
   },
 };
diff --git a/src/retriever.ts b/src/retriever.ts
@@ -63,6 +63,7 @@ export const retrievers: Record<string, Tool> = {
         parameters: { type: "object", properties: {}, required: [] },
       },
     },
+    type: "retriever",
     implementation: getSelectedText,
   },
   getCalendarEvents: {
@@ -83,6 +84,7 @@ export const retrievers: Record<string, Tool> = {
         },
       },
     },
+    type: "retriever",
     implementation: getCalendarEvents,
   },
 };
diff --git a/src/tool.ts b/src/tool.ts
@@ -28,5 +28,6 @@ export interface Tool {
       };
     };
   };
+  type: "retriever" | "action";
   implementation: (state: State, parameters: any) => void;
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -28,5 +28,6 @@ export interface Tool { @@
           };
         };
       };
+      type: "retriever" | "action";
       implementation: (state: State, parameters: any) => void;
     }