feat: create tool factory

helpers/typescript.ts

@@ -166,7 +166,17 @@ export const installTSTemplate = async ({
         cwd: path.join(compPath, "engines", "typescript", "agent"),
       });
 
-      // TODO: action to install tools
+      // Write tools_config.json
+      const configContent: Record<string, any> = {};
+      tools.forEach((tool) => {
+        configContent[tool.name] = tool.config ?? {};
+      });
+      const configFilePath = path.join(enginePath, "tools_config.json");
+      console.log(configFilePath)
+      await fs.writeFile(
+        configFilePath,
+        JSON.stringify(configContent, null, 2),
+      );
     } else {
       await copy("**", enginePath, {
         parents: true,

templates/components/engines/typescript/agent/chat.ts

@@ -5,6 +5,7 @@ import {
 } from "llamaindex";
 import { getDataSource } from "./index";
 import { STORAGE_CACHE_DIR } from "./constants.mjs";
+import ToolFactory from "./tools";
 
 export async function createChatEngine(llm: OpenAI) {
   const index = await getDataSource(llm);
@@ -17,8 +18,10 @@ export async function createChatEngine(llm: OpenAI) {
     },
   });
 
+  const externalTools = await ToolFactory.list();
+
   const agent = new OpenAIAgent({
-    tools: [queryEngineTool],
+    tools: [queryEngineTool, ...externalTools],
     verbose: true,
     llm,
   });

templates/components/engines/typescript/agent/tools.ts

@@ -1 +1,35 @@
-// Load tools here
+import { BaseTool } from "llamaindex";
+import config from "./tool_config.json";
+
+enum ExternalTool {
+	GoogleSearch = "google.GoogleSearchToolSpec",
+	Wikipedia = "wikipedia.WikipediaToolSpec",
+}
+
+type ToolConfig = { [key in ExternalTool]: Record<string, any> };
+
+export default class ToolFactory {
+	private static async createTool(
+		key: ExternalTool,
+		options: Record<string, any>
+	): Promise<BaseTool> {
+		if (key === ExternalTool.Wikipedia) {
+			const WikipediaTool = (await import("llamaindex")).WikipediaTool;
+			const tool = new WikipediaTool();
+			return tool;
+		}
+
+    // TODO: Implement GoogleSearchToolSpec
+
+		throw new Error(`Sorry! Tool ${key} is not supported yet. Options: ${options}`);
+	}
+
+	public static async list(): Promise<BaseTool[]> {
+		const tools: BaseTool[] = [];
+    for (const [key, value] of Object.entries(config as ToolConfig)) {
+      const tool = await ToolFactory.createTool(key as ExternalTool, value);
+      tools.push(tool);
+    }
+		return tools;
+	}
+}