Merge branch 'agent-paper-main' of https://github.com/brown-dash/Dash-Web into joanne-tutorialagent

1 files changed, 34 insertions, 0 deletions

diff --git a/src/client/views/nodes/chatbot/tools/FileNamesTool.ts b/src/client/views/nodes/chatbot/tools/FileNamesTool.ts
new file mode 100644
index 000000000..b69874afa
--- /dev/null
+++ b/src/client/views/nodes/chatbot/tools/FileNamesTool.ts
@@ -0,0 +1,34 @@
+import { Observation } from '../types/types';
+import { ParametersType, ToolInfo } from '../types/tool_types';
+import { Vectorstore } from '../vectorstore/Vectorstore';
+import { BaseTool } from './BaseTool';
+
+const fileNamesToolParams = [] as const;
+
+type FileNamesToolParamsType = typeof fileNamesToolParams;
+
+const fileNamesToolInfo: ToolInfo<FileNamesToolParamsType> = {
+    name: 'fileNames',
+    description: 'Retrieves the names of all files in the Dash codebase to help understand the codebase structure.',
+    citationRules: `No citation needed.`,
+    parameterRules: fileNamesToolParams,
+};
+
+export class FileNamesTool extends BaseTool<FileNamesToolParamsType> {
+    constructor(private vectorstore: Vectorstore) {
+        super(fileNamesToolInfo);
+    }
+
+    async execute(args: ParametersType<FileNamesToolParamsType>): Promise<Observation[]> {
+        console.log(`Executing file names retrieval`);
+
+        const filepaths = await this.vectorstore.getFileNames();
+
+        return [
+            {
+                type: 'text',
+                text: `The file names in the codebase are: ${filepaths}`,
+            },
+        ];
+    }
+}