feat: add readFile tool

Author: bajrangCoder

package-lock.json

@@ -120,7 +120,8 @@
     "mustache": "^4.2.0",
     "url-parse": "^1.5.10",
     "vanilla-picker": "^2.12.3",
-    "yargs": "^17.7.2"
+    "yargs": "^17.7.2",
+    "zod": "^3.25.47"
   },
   "browserslist": "cover 100%,not android < 5"
-}
+}

package.json

@@ -19,6 +19,7 @@ import {
 	updateConversation,
 } from "./db";
 import { SYSTEM_PROMPT } from "./system_prompt";
+import { allTools } from "./tools";
 
 export default function openAIAssistantPage() {
 	// References
@@ -38,17 +39,18 @@ export default function openAIAssistantPage() {
 
 	const GEMINI_API_KEY = ""; // Replace
 
-	const searchTool = {
-		googleSearch: {},
-	};
 	const agentCheckpointer = new MemorySaver();
 	const model = new ChatGoogleGenerativeAI({
 		model: "gemini-2.0-flash",
 		apiKey: GEMINI_API_KEY,
 	});
+
+	// Get all tools as an array for the agent including search
+	const toolsArray = Object.values(allTools);
+
 	const agent = createReactAgent({
 		llm: model,
-		tools: [searchTool],
+		tools: toolsArray,
 		checkpointSaver: agentCheckpointer,
 		stateModifier: SYSTEM_PROMPT,
 	});

src/pages/aiAssistant/assistant.js

@@ -0,0 +1,6 @@
+import { readFile } from "./readFile";
+
+// Export all tools as a single object
+export const allTools = {
+	readFile,
+};

src/pages/aiAssistant/tools/index.js

@@ -0,0 +1,66 @@
+import { StructuredTool } from "@langchain/core/tools";
+import fsOperation from "fileSystem";
+import { addedFolder } from "lib/openFolder";
+import { z } from "zod";
+
+/**
+ * Tool for reading file contents in Acode
+ */
+class ReadFileTool extends StructuredTool {
+	name = "readFile";
+	description = "Reads the content of the given file in the project.";
+	schema = z.object({
+		path: z
+			.string()
+			.describe(
+				"The relative path of the file to read. This path should never be absolute, and the first component of the path should always be a root directory in a project (opened in sidebar). For example, if root directories are 'directory1' and 'directory2', to access 'file.txt' in 'directory1', use 'directory1/file.txt'. To access 'file.txt' in 'directory2', use 'directory2/file.txt'.",
+			),
+		startLine: z
+			.number()
+			.min(1)
+			.optional()
+			.describe("line number to start reading on (1-based index)"),
+		endLine: z
+			.number()
+			.min(1)
+			.optional()
+			.describe("line number to end reading on (1-based index, inclusive)"),
+	});
+
+	async _call({ path, startLine, endLine }) {
+		try {
+			// Split the path to get project name and file path
+			const pathParts = path.split("/");
+			const projectName = pathParts[0];
+			const filePath = pathParts.slice(1).join("/");
+
+			// Find the project in addedFolder array
+			const project = addedFolder.find(
+				(folder) => folder.title === projectName,
+			);
+			if (!project) {
+				return `Error: Project '${projectName}' not found in opened projects`;
+			}
+
+			// Construct the full file URL
+			const fileUrl = project.url + "/" + filePath;
+
+			// Read the file content
+			const content = await fsOperation(fileUrl).readFile("utf8");
+
+			// If startLine or endLine are specified, filter the content
+			if (startLine !== undefined || endLine !== undefined) {
+				const lines = content.split("\n");
+				const start = startLine ? startLine - 1 : 0;
+				const end = endLine ? endLine : lines.length;
+				return lines.slice(start, end).join("\n");
+			}
+
+			return content;
+		} catch (error) {
+			return `Error reading file: ${error.message}`;
+		}
+	}
+}
+
+export const readFile = new ReadFileTool();