@nestbox-ai/cli 1.0.13 → 1.0.16

package/dist/commands/agent.js

@@ -1,4 +1,37 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -13,6 +46,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerAgentCommands = registerAgentCommands;
+const error_1 = require("../utils/error");
 const chalk_1 = __importDefault(require("chalk"));
 const ora_1 = __importDefault(require("ora"));
 const cli_table3_1 = __importDefault(require("cli-table3"));
@@ -20,22 +54,109 @@ const auth_1 = require("../utils/auth");
 const admin_1 = require("@nestbox-ai/admin");
 const project_1 = require("../utils/project");
 const fs_1 = __importDefault(require("fs"));
+const js_yaml_1 = __importDefault(require("js-yaml"));
 const agent_1 = require("../utils/agent");
 const axios_1 = __importDefault(require("axios"));
 const agentType_1 = require("../types/agentType");
+const inquirer_1 = __importDefault(require("inquirer"));
+const path_1 = __importDefault(require("path"));
+const user_1 = require("../utils/user");
+/**
+ * Create a new agent in the Nestbox platform
+ *
+ * @param agentName The name of the agent
+ * @param options Options including lang, template, and project
+ * @param agentsApi The MachineAgentApi instance
+ * @param projectsApi The ProjectsApi instance
+ */
+function createAgent(agentName, options, agentsApi, projectsApi) {
+    return __awaiter(this, void 0, void 0, function* () {
+        var _a, _b, _c;
+        const authToken = (0, auth_1.getAuthToken)();
+        if (!authToken) {
+            throw new Error("No authentication token found. Please login first.");
+        }
+        // Create API instances if not provided
+        if (!agentsApi || !projectsApi) {
+            const configuration = new admin_1.Configuration({
+                basePath: authToken === null || authToken === void 0 ? void 0 : authToken.serverUrl,
+                baseOptions: {
+                    headers: {
+                        Authorization: authToken === null || authToken === void 0 ? void 0 : authToken.token,
+                    },
+                },
+            });
+            agentsApi = agentsApi || new admin_1.MachineAgentApi(configuration);
+            projectsApi = projectsApi || new admin_1.ProjectsApi(configuration);
+        }
+        // Resolve project - convert options to match CommandOptions interface
+        const projectData = yield (0, project_1.resolveProject)(projectsApi, Object.assign({ project: options.project, instance: options.instanceName || '' }, options));
+        // Prepare agent creation payload
+        // Determine the correct type value based on options.type
+        const agentTypeValue = ((_a = options.type) === null || _a === void 0 ? void 0 : _a.includes("AGENT")) ? "REGULAR" : options.type || "CHAT";
+        const payload = {
+            agentName,
+            goal: options.goal || `AI agent for ${agentName}`,
+            modelBaseId: options.modelBaseId || "",
+            machineName: options.machineName,
+            machineInstanceId: options.machineInstanceId,
+            instanceIP: options.instanceIP,
+            machineManifestId: options.machineManifestId,
+            parameters: ((_b = options.parameters) === null || _b === void 0 ? void 0 : _b.map((param) => ({
+                name: param.name,
+                description: param.description,
+                default_value: param.default || "",
+                isUserParam: param.isUserParam !== undefined ? param.isUserParam : true
+            }))) || [],
+            projectId: projectData.id,
+            type: agentTypeValue,
+            userId: options.userId || 0,
+        };
+        // Determine the type of resource (Agent or Chat)
+        const agentType = options.type || "CHAT";
+        const resourceType = agentType === "AGENT" || agentType === "REGULAR" ? "Agent" : "Chatbot";
+        // Create the agent
+        const spinner = (0, ora_1.default)(`Creating ${resourceType.toLowerCase()} ${agentName}...`).start();
+        try {
+            const response = yield agentsApi.machineAgentControllerCreateMachineAgent(projectData.id, payload);
+            spinner.succeed(`${resourceType} '${agentName}' created successfully`);
+            return response.data;
+        }
+        catch (error) {
+            spinner.fail(`Failed to create ${resourceType.toLowerCase()} ${agentName}`);
+            if (error.response && error.response.status === 401) {
+                throw new Error('Authentication token has expired. Please login again using "nestbox login <domain>".');
+            }
+            else if (error.response) {
+                throw new Error(`API Error (${error.response.status}): ${((_c = error.response.data) === null || _c === void 0 ? void 0 : _c.message) || "Unknown error"}`);
+            }
+            else {
+                throw new Error(error.message || "Unknown error");
+            }
+        }
+    });
+}
 function registerAgentCommands(program) {
-    // Get authentication token and create API configuration
-    const authToken = (0, auth_1.getAuthToken)();
-    const configuration = new admin_1.Configuration({
-        basePath: authToken === null || authToken === void 0 ? void 0 : authToken.serverUrl,
-        baseOptions: {
-            headers: {
-                Authorization: authToken === null || authToken === void 0 ? void 0 : authToken.token,
+    // Function to create/recreate API instances
+    const createApis = () => {
+        const authToken = (0, auth_1.getAuthToken)();
+        if (!authToken) {
+            throw new Error('No authentication token found. Please log in first.');
+        }
+        const configuration = new admin_1.Configuration({
+            basePath: authToken.serverUrl,
+            baseOptions: {
+                headers: {
+                    Authorization: authToken.token,
+                },
             },
-        },
-    });
-    const agentsApi = new admin_1.MachineAgentApi(configuration);
-    const projectsApi = new admin_1.ProjectsApi(configuration);
+        });
+        return {
+            agentsApi: new admin_1.MachineAgentApi(configuration),
+            projectsApi: new admin_1.ProjectsApi(configuration),
+            instanceApi: new admin_1.MachineInstancesApi(configuration)
+        };
+    };
     // Create the main agent command
     const agentCommand = program
         .command("agent")
@@ -48,17 +169,16 @@ function registerAgentCommands(program) {
         .action((options) => __awaiter(this, void 0, void 0, function* () {
         var _a, _b;
         try {
-            if (!authToken) {
-                console.error(chalk_1.default.red("No authentication token found. Please login first."));
-                return;
-            }
-            try {
-                // Use the resolveProject helper to get project information
-                const projectData = yield (0, project_1.resolveProject)(projectsApi, options);
+            let apis = createApis();
+            // Execute with token refresh support
+            yield (0, error_1.withTokenRefresh)(() => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                // Resolve project
+                const projectData = yield (0, project_1.resolveProject)(apis.projectsApi, options);
                 const spinner = (0, ora_1.default)(`Listing agents in project ${projectData.name}...`).start();
                 try {
-                    // Now get the agents for the specific project
-                    const agentsResponse = yield agentsApi.machineAgentControllerGetMachineAgentByProjectId(projectData.id, 0, 10, agentType_1.AgentType.REGULAR);
+                    // Get the agents for the specific project
+                    const agentsResponse = yield apis.agentsApi.machineAgentControllerGetMachineAgentByProjectId(projectData.id, 0, 10, agentType_1.AgentType.REGULAR);
                     spinner.succeed("Successfully retrieved agents");
                     // Display the results
                     const agents = ((_a = agentsResponse.data) === null || _a === void 0 ? void 0 : _a.machineAgents) || [];
@@ -67,7 +187,7 @@ function registerAgentCommands(program) {
                         return;
                     }
                     console.log(chalk_1.default.blue(`\nAgents in project ${projectData.name}:\n`));
-                    // Create a formatted table focusing on id, name, and URL
+                    // Create a formatted table
                     const table = new cli_table3_1.default({
                         head: [
                             chalk_1.default.white.bold("ID"),
@@ -75,46 +195,40 @@ function registerAgentCommands(program) {
                             chalk_1.default.white.bold("URL"),
                         ],
                         style: {
-                            head: [], // Disable the default styling
+                            head: [],
                             border: [],
                         },
                     });
-                    // Add agents to the table with the requested info
+                    // Add agents to the table
                     agents.forEach((agent) => {
-                        // Format the agent URL
                         let url = "N/A";
                         if (agent.instanceIP) {
-                            // Construct an agent-specific URL if possible
                             url = `${agent.instanceIP}/v1/agents/${agent.modelBaseId}/query`;
                         }
-                        // Format date for readability
-                        let createdAt = agent.createdAt || "N/A";
-                        if (createdAt !== "N/A") {
-                            createdAt = new Date(createdAt).toLocaleString();
-                        }
                         table.push([agent.id || "N/A", agent.agentName || "N/A", url]);
                     });
-                    // Display the table
                     console.log(table.toString());
-                    // Display totals
                     console.log(`\nTotal agents: ${agents.length}`);
                 }
                 catch (error) {
                     spinner.fail("Failed to retrieve agents");
-                    if (error.response) {
-                        console.error(chalk_1.default.red("API Error:"), ((_b = error.response.data) === null || _b === void 0 ? void 0 : _b.message) || "Unknown error");
-                    }
-                    else {
-                        console.error(chalk_1.default.red("Error:"), error.message || "Unknown error");
-                    }
+                    throw error;
                 }
-            }
-            catch (error) {
-                console.error(chalk_1.default.red("Error:"), error instanceof Error ? error.message : "Unknown error");
-            }
+            }), () => {
+                // Recreate APIs after token refresh
+                apis = createApis();
+            });
         }
         catch (error) {
-            console.error(chalk_1.default.red("Error:"), error instanceof Error ? error.message : "Unknown error");
+            if (error.message && error.message.includes('Authentication')) {
+                console.error(chalk_1.default.red(error.message));
+            }
+            else if ((_b = (_a = error.response) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.message) {
+                console.error(chalk_1.default.red("API Error:"), error.response.data.message);
+            }
+            else {
+                console.error(chalk_1.default.red("Error:"), error.message || "Unknown error");
+            }
         }
     }));
     // Remove agent
@@ -126,168 +240,468 @@ function registerAgentCommands(program) {
         .action((options) => __awaiter(this, void 0, void 0, function* () {
         var _a, _b;
         try {
-            if (!authToken) {
-                console.error(chalk_1.default.red("No authentication token found. Please login first."));
-                return;
-            }
+            let apis = createApis();
             const { agent } = options;
-            // Use the resolveProject helper to get project information
-            const projectData = yield (0, project_1.resolveProject)(projectsApi, options);
-            const spinner = (0, ora_1.default)(`Finding agent ${agent} in project ${projectData.name}...`).start();
-            try {
-                // First, get the list of agents to find the correct modelbaseId
-                const agentsResponse = yield agentsApi.machineAgentControllerGetMachineAgentByProjectId(projectData.id, 0, 100, agentType_1.AgentType.REGULAR);
-                // Get the agents array
-                const agents = ((_a = agentsResponse.data) === null || _a === void 0 ? void 0 : _a.machineAgents) || [];
-                // Find the specific agent by ID
-                const targetAgent = agents.find((a) => a.id.toString() === agent.toString());
-                if (!targetAgent) {
-                    spinner.fail(`Agent with ID ${agent} not found in project ${projectData.name}`);
-                    return;
-                }
-                // Extract the modelbaseId from the found agent
-                const modelbaseId = targetAgent.modelBaseId;
-                if (!modelbaseId) {
-                    spinner.fail(`Could not find modelbaseId for agent ${agent}. Please try again.`);
-                    return;
-                }
-                spinner.text = `Removing agent ${agent} from project ${projectData.name}...`;
-                // Now remove the agent with the dynamically retrieved modelbaseId
-                const payload = [
-                    {
-                        id: parseInt(agent, 10),
-                        modelbaseId: modelbaseId,
-                    },
-                ];
-                const removeResponse = yield agentsApi.machineAgentControllerDeleteMachineAgents(projectData.id, agent, payload);
-                spinner.succeed("Successfully removed agent");
-                // Display the results
-                console.log(chalk_1.default.green(`Agent ${agent} removed successfully from project ${projectData.name}`));
-            }
-            catch (error) {
-                spinner.fail("Failed to remove agent");
-                if (error.response) {
-                    console.error(chalk_1.default.red("API Error:"), ((_b = error.response.data) === null || _b === void 0 ? void 0 : _b.message) || "Unknown error");
+            yield (0, error_1.withTokenRefresh)(() => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                // Resolve project
+                const projectData = yield (0, project_1.resolveProject)(apis.projectsApi, options);
+                const spinner = (0, ora_1.default)(`Finding agent ${agent} in project ${projectData.name}...`).start();
+                try {
+                    // Get the list of agents to find the correct modelbaseId
+                    const agentsResponse = yield apis.agentsApi.machineAgentControllerGetMachineAgentByProjectId(projectData.id, 0, 100, agentType_1.AgentType.REGULAR);
+                    const agents = ((_a = agentsResponse.data) === null || _a === void 0 ? void 0 : _a.machineAgents) || [];
+                    const targetAgent = agents.find((a) => a.id.toString() === agent.toString());
+                    if (!targetAgent) {
+                        spinner.fail(`Agent with ID ${agent} not found in project ${projectData.name}`);
+                        return;
+                    }
+                    const modelbaseId = targetAgent.modelBaseId;
+                    if (!modelbaseId) {
+                        spinner.fail(`Could not find modelbaseId for agent ${agent}. Please try again.`);
+                        return;
+                    }
+                    spinner.text = `Removing agent ${agent} from project ${projectData.name}...`;
+                    // Remove the agent
+                    const payload = [
+                        {
+                            id: parseInt(agent, 10),
+                            modelbaseId: modelbaseId,
+                        },
+                    ];
+                    yield apis.agentsApi.machineAgentControllerDeleteMachineAgents(projectData.id, agent, payload);
+                    spinner.succeed("Successfully removed agent");
+                    console.log(chalk_1.default.green(`Agent ${agent} removed successfully from project ${projectData.name}`));
                 }
-                else {
-                    console.error(chalk_1.default.red("Error:"), error.message || "Unknown error");
+                catch (error) {
+                    spinner.fail("Failed to remove agent");
+                    throw error;
                 }
-            }
+            }), () => {
+                apis = createApis();
+            });
         }
         catch (error) {
-            console.error(chalk_1.default.red("Error:"), error instanceof Error ? error.message : "Unknown error");
+            if (error.message && error.message.includes('Authentication')) {
+                console.error(chalk_1.default.red(error.message));
+            }
+            else if ((_b = (_a = error.response) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.message) {
+                console.error(chalk_1.default.red("API Error:"), error.response.data.message);
+            }
+            else {
+                console.error(chalk_1.default.red("Error:"), error.message || "Unknown error");
+            }
         }
     }));
     agentCommand
         .command("deploy")
         .description("Deploy an AI agent to the Nestbox platform")
-        .requiredOption("--agent <agentId>", "Agent ID to deploy")
-        .requiredOption("--instance <instanceId>", "Instance ID")
+        .option("--agent <agentName>", "Agent name to deploy")
+        .option("--chatbot <chatbotName>", "Chatbot name to deploy")
+        .requiredOption("--instance <instanceName>", "Instance name")
         .option("--zip <zipFileOrDirPath>", "Path to the zip file or directory to upload")
         .option("--project <projectName>", "Project name (defaults to the current project)")
         .option("--entry <entryFunction>", "Entry function name", "main")
         .action((options) => __awaiter(this, void 0, void 0, function* () {
-        var _a, _b, _c, _d, _e, _f;
+        var _a;
         try {
-            if (!authToken) {
-                console.error(chalk_1.default.red("No authentication token found. Please login first."));
+            const { agent: agentName, chatbot: chatbotName, instance: instanceName, zip: customZipPath, entry, } = options;
+            // Ensure either agent or chatbot is provided, but not both
+            if ((!agentName && !chatbotName) || (agentName && chatbotName)) {
+                console.error(chalk_1.default.red("Please provide either --agent OR --chatbot option, but not both."));
                 return;
             }
-            const { agent: agentId, instance: instanceId, zip: customZipPath, entry, } = options;
-            // Find project root (CLI tools directory)
+            let apis = createApis();
+            // Find project root
             const projectRoot = yield (0, agent_1.findProjectRoot)();
             console.log(chalk_1.default.blue(`Project root detected at: ${projectRoot}`));
-            // Use the resolveProject helper to get project information
-            const projectData = yield (0, project_1.resolveProject)(projectsApi, options);
-            // Load nestbox.config.json from CLI tools directory
-            const config = (0, agent_1.loadNestboxConfig)(projectRoot);
-            // Start the deployment process
-            const spinner = (0, ora_1.default)(`Preparing to deploy agent ${agentId} to instance ${instanceId}...`).start();
-            try {
-                // Determine the source path (custom path or project root)
-                const sourcePath = customZipPath || projectRoot;
-                let zipFilePath;
-                // Check if the specified path exists
-                if (!fs_1.default.existsSync(sourcePath)) {
-                    spinner.fail(`Path not found: ${sourcePath}`);
+            // Main deployment logic with token refresh
+            yield (0, error_1.withTokenRefresh)(() => __awaiter(this, void 0, void 0, function* () {
+                var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+                // Resolve project
+                const projectData = yield (0, project_1.resolveProject)(apis.projectsApi, options);
+                // Determine if we're deploying an agent or chatbot
+                const isAgent = !!agentName;
+                const resourceName = isAgent ? agentName : chatbotName;
+                const resourceType = isAgent ? "Agent" : "Chatbot";
+                const agentType = isAgent ? agentType_1.AgentType.REGULAR : "CHAT";
+                // Get agents data and find agent/chatbot by name
+                const agentsData = yield apis.agentsApi.machineAgentControllerGetMachineAgentByProjectId(projectData.id, 0, 10, agentType);
+                const targetAgent = agentsData.data.machineAgents.find((agent) => agent.agentName === resourceName);
+                if (!targetAgent) {
+                    console.error(chalk_1.default.red(`${resourceType} with name "${resourceName}" not found in project "${projectData.name}".`));
+                    console.log(chalk_1.default.yellow(`Available ${resourceType.toLowerCase()}s:`));
+                    agentsData.data.machineAgents.forEach((agent) => {
+                        console.log(chalk_1.default.yellow(`  - ${agent.agentName} (ID: ${agent.id})`));
+                    });
                     return;
                 }
-                // Check if the path is a zip file or directory
-                const stats = fs_1.default.statSync(sourcePath);
-                if (stats.isFile()) {
-                    // Case 1: It's a file - verify it's a zip and use directly
-                    if (!sourcePath.toLowerCase().endsWith(".zip")) {
-                        spinner.fail(`File is not a zip archive: ${sourcePath}`);
-                        return;
-                    }
-                    // Use the zip file directly
-                    spinner.text = `Using provided zip file: ${sourcePath}`;
-                    zipFilePath = sourcePath;
+                // Get instance data and find instance by name
+                const instanceData = yield apis.instanceApi.machineInstancesControllerGetMachineInstanceByUserId(projectData.id, 0, 10);
+                const targetInstance = instanceData.data.machineInstances.find((instance) => instance.instanceName === instanceName);
+                if (!targetInstance) {
+                    console.error(chalk_1.default.red(`Instance with name "${instanceName}" not found in project "${projectData.name}".`));
+                    console.log(chalk_1.default.yellow("Available instances:"));
+                    instanceData.data.machineInstances.forEach((instance) => {
+                        console.log(chalk_1.default.yellow(`  - ${instance.instanceName} (ID: ${instance.id})`));
+                    });
+                    return;
                 }
-                else if (stats.isDirectory()) {
-                    // Case 2: It's a directory - check for predeploy scripts in CLI config
-                    // Determine if it's a TypeScript project
-                    const isTypeScript = (0, agent_1.isTypeScriptProject)(sourcePath);
-                    if (isTypeScript) {
-                        spinner.text = `TypeScript project detected. Checking for predeploy scripts...`;
-                        // Run predeploy scripts if defined in CLI tools config
-                        if (((_a = config === null || config === void 0 ? void 0 : config.agent) === null || _a === void 0 ? void 0 : _a.predeploy) || ((_b = config === null || config === void 0 ? void 0 : config.agents) === null || _b === void 0 ? void 0 : _b.predeploy)) {
-                            const predeployScripts = ((_c = config === null || config === void 0 ? void 0 : config.agent) === null || _c === void 0 ? void 0 : _c.predeploy) || ((_d = config === null || config === void 0 ? void 0 : config.agents) === null || _d === void 0 ? void 0 : _d.predeploy);
-                            spinner.text = `Running predeploy scripts on target directory...`;
-                            yield (0, agent_1.runPredeployScripts)(predeployScripts, sourcePath);
+                // Extract IDs
+                const agentId = targetAgent.id;
+                const instanceId = targetInstance.id;
+                // Load nestbox.config.json
+                const config = (0, agent_1.loadNestboxConfig)(projectRoot);
+                // Start the deployment process
+                const spinner = (0, ora_1.default)(`Preparing to deploy ${resourceType.toLowerCase()} ${agentId} to instance ${instanceId}...`).start();
+                try {
+                    let zipFilePath;
+                    if (customZipPath) {
+                        // Process custom zip path
+                        if (!fs_1.default.existsSync(customZipPath)) {
+                            spinner.fail(`Path not found: ${customZipPath}`);
+                            return;
                         }
-                        else {
-                            spinner.info("No predeploy scripts found in CLI tools nestbox.config.json");
+                        const stats = fs_1.default.statSync(customZipPath);
+                        if (stats.isFile()) {
+                            if (!customZipPath.toLowerCase().endsWith(".zip")) {
+                                spinner.fail(`File is not a zip archive: ${customZipPath}`);
+                                return;
+                            }
+                            spinner.text = `Using provided zip file: ${customZipPath}`;
+                            zipFilePath = customZipPath;
+                        }
+                        else if (stats.isDirectory()) {
+                            // Process directory
+                            spinner.text = `Processing directory: ${customZipPath}`;
+                            const isTypeScript = (0, agent_1.isTypeScriptProject)(customZipPath);
+                            if (isTypeScript && (((_a = config === null || config === void 0 ? void 0 : config.agent) === null || _a === void 0 ? void 0 : _a.predeploy) || ((_b = config === null || config === void 0 ? void 0 : config.agents) === null || _b === void 0 ? void 0 : _b.predeploy))) {
+                                const predeployScripts = ((_c = config === null || config === void 0 ? void 0 : config.agent) === null || _c === void 0 ? void 0 : _c.predeploy) || ((_d = config === null || config === void 0 ? void 0 : config.agents) === null || _d === void 0 ? void 0 : _d.predeploy);
+                                spinner.text = `Running predeploy scripts on target directory...`;
+                                yield (0, agent_1.runPredeployScripts)(predeployScripts, customZipPath);
+                            }
+                            spinner.text = `Creating zip archive from directory ${customZipPath}...`;
+                            zipFilePath = (0, agent_1.createZipFromDirectory)(customZipPath);
+                            spinner.text = `Directory zipped successfully to ${zipFilePath}`;
                         }
                     }
                     else {
-                        // JavaScript directory - just zip it
-                        spinner.text = `JavaScript project detected. Skipping predeploy scripts.`;
+                        // Use project root
+                        spinner.text = `Using project root: ${projectRoot}`;
+                        const isTypeScript = (0, agent_1.isTypeScriptProject)(projectRoot);
+                        if (isTypeScript && (((_e = config === null || config === void 0 ? void 0 : config.agent) === null || _e === void 0 ? void 0 : _e.predeploy) || ((_f = config === null || config === void 0 ? void 0 : config.agents) === null || _f === void 0 ? void 0 : _f.predeploy))) {
+                            const predeployScripts = ((_g = config === null || config === void 0 ? void 0 : config.agent) === null || _g === void 0 ? void 0 : _g.predeploy) || ((_h = config === null || config === void 0 ? void 0 : config.agents) === null || _h === void 0 ? void 0 : _h.predeploy);
+                            spinner.text = `Running predeploy scripts on project root...`;
+                            yield (0, agent_1.runPredeployScripts)(predeployScripts, projectRoot);
+                        }
+                        spinner.text = `Creating zip archive from project root ${projectRoot}...`;
+                        zipFilePath = (0, agent_1.createZipFromDirectory)(projectRoot);
+                        spinner.text = `Directory zipped successfully to ${zipFilePath}`;
                     }
-                    // Create zip archive with node_modules excluded
-                    spinner.text = `Creating zip archive from directory ${sourcePath}...`;
-                    zipFilePath = (0, agent_1.createZipFromDirectory)(sourcePath);
-                    spinner.text = `Directory zipped successfully to ${zipFilePath}`;
+                    spinner.text = `Deploying ${resourceType.toLowerCase()} ${agentId} to instance ${instanceId}...`;
+                    // Prepare deployment
+                    const authToken = (0, auth_1.getAuthToken)();
+                    const baseUrl = ((_j = authToken === null || authToken === void 0 ? void 0 : authToken.serverUrl) === null || _j === void 0 ? void 0 : _j.endsWith("/"))
+                        ? authToken.serverUrl.slice(0, -1)
+                        : authToken === null || authToken === void 0 ? void 0 : authToken.serverUrl;
+                    const { default: FormData } = yield Promise.resolve().then(() => __importStar(require("form-data")));
+                    const form = new FormData();
+                    form.append("file", fs_1.default.createReadStream(zipFilePath));
+                    form.append("machineAgentId", agentId.toString());
+                    form.append("instanceId", instanceId.toString());
+                    form.append("entryFunctionName", entry);
+                    form.append("isSourceCodeUpdate", "true");
+                    form.append("projectId", projectData.id);
+                    const axiosInstance = axios_1.default.create({
+                        baseURL: baseUrl,
+                        headers: Object.assign(Object.assign({}, form.getHeaders()), { Authorization: authToken === null || authToken === void 0 ? void 0 : authToken.token }),
+                    });
+                    const endpoint = `/projects/${projectData.id}/agents/${agentId}`;
+                    console.log(chalk_1.default.blue("\nMaking API request:"));
+                    console.log(chalk_1.default.blue(`  URL: ${baseUrl}${endpoint}`));
+                    console.log(chalk_1.default.blue(`  Method: PATCH`));
+                    console.log(chalk_1.default.blue(`  File: ${path_1.default.basename(zipFilePath)}`));
+                    spinner.text = `Sending API request to deploy ${resourceType.toLowerCase()}...`;
+                    const res = yield axiosInstance.patch(endpoint, form);
+                    console.log(chalk_1.default.green("\nAPI Response received:"));
+                    console.log(chalk_1.default.green(`  Status: ${res.status} ${res.statusText}`));
+                    if (!customZipPath && zipFilePath && fs_1.default.existsSync(zipFilePath)) {
+                        fs_1.default.unlinkSync(zipFilePath);
+                    }
+                    spinner.succeed(`Successfully deployed ${resourceType.toLowerCase()} ${agentId} to instance ${instanceId}`);
+                    console.log(chalk_1.default.green(`${resourceType} deployed successfully`));
                 }
-                else {
-                    spinner.fail(`Unsupported file type: ${sourcePath}`);
+                catch (error) {
+                    spinner.fail(`Failed to deploy ${resourceType.toLowerCase()}`);
+                    throw error;
+                }
+            }), () => {
+                apis = createApis();
+            });
+        }
+        catch (error) {
+            if (error.message && error.message.includes('Authentication')) {
+                console.error(chalk_1.default.red(error.message));
+            }
+            else if (error.response) {
+                console.error(chalk_1.default.red(`API Error (${error.response.status}): ${((_a = error.response.data) === null || _a === void 0 ? void 0 : _a.message) || "Unknown error"}`));
+                if (error.response.data) {
+                    console.error(chalk_1.default.red(`Error Data: ${JSON.stringify(error.response.data, null, 2)}`));
+                }
+            }
+            else {
+                console.error(chalk_1.default.red("Error:"), error.message || "Unknown error");
+            }
+        }
+    }));
+    agentCommand
+        .command("generate <folder>")
+        .description("Generate a new project from templates")
+        .option("--lang <language>", "Project language (ts|js)")
+        .option("--template <type>", "Template type (agent|chatbot)")
+        .option("--project <projectId>", "Project ID")
+        .action((folder, options) => __awaiter(this, void 0, void 0, function* () {
+        try {
+            const spinner = (0, ora_1.default)("Initializing project generation...").start();
+            // Ensure target folder doesn't exist
+            if (fs_1.default.existsSync(folder)) {
+                spinner.fail(`Folder ${folder} already exists`);
+                return;
+            }
+            let selectedLang = options.lang;
+            let selectedTemplate = options.template;
+            // Interactive selection if not provided
+            if (!selectedLang || !selectedTemplate) {
+                spinner.stop();
+                const answers = yield inquirer_1.default.prompt([
+                    {
+                        type: 'list',
+                        name: 'lang',
+                        message: 'Select project language:',
+                        choices: [
+                            { name: 'TypeScript', value: 'ts' },
+                            { name: 'JavaScript', value: 'js' }
+                        ],
+                        when: () => !selectedLang
+                    },
+                    {
+                        type: 'list',
+                        name: 'template',
+                        message: 'Select template type:',
+                        choices: [
+                            { name: 'Agent', value: 'agent' },
+                            { name: 'Chatbot', value: 'chatbot' }
+                        ],
+                        when: () => !selectedTemplate
+                    }
+                ]);
+                selectedLang = selectedLang || answers.lang;
+                selectedTemplate = selectedTemplate || answers.template;
+                spinner.start("Generating project...");
+            }
+            // Find matching template in local templates folder
+            const templateMapping = {
+                'agent': 'base',
+                'chatbot': 'chatbot'
+            };
+            const mappedTemplateType = templateMapping[selectedTemplate] || selectedTemplate;
+            const templateKey = `template-${mappedTemplateType}-${selectedLang}.zip`;
+            // Try process.cwd() first, then __dirname fallback
+            let templatePath = path_1.default.resolve(process.cwd(), 'templates', templateKey);
+            if (!fs_1.default.existsSync(templatePath)) {
+                // fallback to __dirname
+                templatePath = path_1.default.resolve(__dirname, '../../templates', templateKey);
+            }
+            if (!fs_1.default.existsSync(templatePath)) {
+                spinner.fail(`Template not found: ${templatePath}`);
+                // Show available templates in both locations
+                const cwdTemplates = path_1.default.resolve(process.cwd(), 'templates');
+                const dirTemplates = path_1.default.resolve(__dirname, '../../templates');
+                let shown = false;
+                if (fs_1.default.existsSync(cwdTemplates)) {
+                    console.log(chalk_1.default.yellow('Available templates in ./templates:'));
+                    fs_1.default.readdirSync(cwdTemplates).forEach(file => {
+                        console.log(chalk_1.default.yellow(`  - ${file}`));
+                    });
+                    shown = true;
+                }
+                if (fs_1.default.existsSync(dirTemplates)) {
+                    console.log(chalk_1.default.yellow('Available templates in src/commands/../../templates:'));
+                    fs_1.default.readdirSync(dirTemplates).forEach(file => {
+                        console.log(chalk_1.default.yellow(`  - ${file}`));
+                    });
+                    shown = true;
+                }
+                if (!shown) {
+                    console.log(chalk_1.default.red('No templates directory found. Please add your templates.'));
+                }
+                return;
+            }
+            spinner.text = `Extracting template to ${folder}...`;
+            try {
+                // Extract template to target folder
+                (0, agent_1.extractZip)(templatePath, folder);
+                // Create nestbox.config.json for TypeScript projects
+                (0, agent_1.createNestboxConfig)(folder, selectedLang === 'ts');
+                // Update package.json with project name if it exists
+                const packageJsonPath = path_1.default.join(folder, 'package.json');
+                if (fs_1.default.existsSync(packageJsonPath)) {
+                    const packageJson = JSON.parse(fs_1.default.readFileSync(packageJsonPath, 'utf8'));
+                    packageJson.name = path_1.default.basename(folder);
+                    fs_1.default.writeFileSync(packageJsonPath, JSON.stringify(packageJson, null, 2));
+                }
+                spinner.succeed(`Successfully generated ${mappedTemplateType} project in ${folder}`);
+                console.log(chalk_1.default.green("\nNext steps:"));
+                console.log(chalk_1.default.yellow(`  cd ${folder}`));
+                console.log(chalk_1.default.yellow("  npm install"));
+                if (selectedLang === 'ts') {
+                    console.log(chalk_1.default.yellow("  npm run build"));
+                }
+                console.log(chalk_1.default.yellow("  nestbox agent deploy --agent <agent-name> --instance <instance-name>"));
+            }
+            catch (error) {
+                // Clean up on error
+                if (fs_1.default.existsSync(folder)) {
+                    fs_1.default.rmSync(folder, { recursive: true, force: true });
+                }
+                throw error;
+            }
+        }
+        catch (error) {
+            console.error(chalk_1.default.red("Error:"), error.message || "Failed to generate project");
+        }
+    }));
+    // Command for creating agents from YAML files
+    agentCommand
+        .command("create [firstArg] [secondArg]")
+        .description("Create multiple agents from a YAML configuration file")
+        .option("--project <projectId>", "Project ID (defaults to the current project)")
+        .action((firstArg, secondArg, options) => __awaiter(this, void 0, void 0, function* () {
+        try {
+            let apis = createApis();
+            // Determine which argument is the YAML file path
+            let yamlFilePath;
+            if (firstArg === 'file' && secondArg) {
+                yamlFilePath = secondArg;
+            }
+            else if (firstArg) {
+                yamlFilePath = firstArg;
+                if (typeof secondArg === 'object' && !options) {
+                    options = secondArg;
+                }
+            }
+            else {
+                console.error(chalk_1.default.red("Missing YAML file path. Usage: nestbox agent create <yamlFile> OR nestbox agent create file <yamlFile>"));
+                return;
+            }
+            // Check if file exists
+            if (!fs_1.default.existsSync(yamlFilePath)) {
+                console.error(chalk_1.default.red(`YAML file not found: ${yamlFilePath}`));
+                return;
+            }
+            // Read and parse the YAML file
+            const spinner = (0, ora_1.default)(`Reading agents configuration from ${yamlFilePath}...`).start();
+            try {
+                const fileContents = fs_1.default.readFileSync(yamlFilePath, 'utf8');
+                const config = js_yaml_1.default.load(fileContents);
+                if (!config || !config.agents || !Array.isArray(config.agents)) {
+                    spinner.fail("Invalid YAML configuration: Missing 'agents' array");
+                    console.error(chalk_1.default.red("The YAML file should contain an 'agents' array with agent configurations"));
                     return;
                 }
-                spinner.text = `Deploying agent ${agentId} to instance ${instanceId}...`;
-                // Clean the base URL to avoid path duplication
-                const baseUrl = ((_e = authToken === null || authToken === void 0 ? void 0 : authToken.serverUrl) === null || _e === void 0 ? void 0 : _e.endsWith("/"))
-                    ? authToken.serverUrl.slice(0, -1)
-                    : authToken === null || authToken === void 0 ? void 0 : authToken.serverUrl;
-                const FormData = require("form-data");
-                const form = new FormData();
-                // Add file as a readable stream
-                form.append("file", fs_1.default.createReadStream(zipFilePath));
-                // Add all the required fields
-                form.append("machineAgentId", agentId.toString());
-                form.append("instanceId", instanceId.toString());
-                form.append("entryFunctionName", entry);
-                form.append("isSourceCodeUpdate", "true");
-                form.append("projectId", projectData.id);
-                // Create a custom axios instance with form-data headers
-                const axiosInstance = axios_1.default.create({
-                    baseURL: baseUrl,
-                    headers: Object.assign(Object.assign({}, form.getHeaders()), { Authorization: authToken === null || authToken === void 0 ? void 0 : authToken.token }),
-                });
-                // Construct the endpoint URL
-                const endpoint = `/projects/${projectData.id}/agents/${agentId}`;
-                // Make direct axios request
-                yield axiosInstance.patch(endpoint, form);
-                // Clean up temporary zip file if we created one
-                if (zipFilePath !== sourcePath && fs_1.default.existsSync(zipFilePath)) {
-                    fs_1.default.unlinkSync(zipFilePath);
+                spinner.succeed(`Found ${config.agents.length} agents in configuration file`);
+                // Process each agent with token refresh support
+                const results = {
+                    success: 0,
+                    failed: 0,
+                    agents: []
+                };
+                // Get user data once
+                const user = yield (0, user_1.userData)();
+                for (const agent of config.agents) {
+                    if (!agent.name) {
+                        console.log(chalk_1.default.yellow("Skipping agent with no name defined"));
+                        results.failed++;
+                        results.agents.push({
+                            name: "unnamed",
+                            success: false,
+                            message: "Name is required"
+                        });
+                        continue;
+                    }
+                    let agentType = agent.type || "CHAT";
+                    const resourceType = agentType === "AGENT" ? "Agent" : "Chatbot";
+                    const agentSpinner = (0, ora_1.default)(`Creating ${resourceType.toLowerCase()} '${agent.name}'...`).start();
+                    try {
+                        // Create agent with token refresh support
+                        yield (0, error_1.withTokenRefresh)(() => __awaiter(this, void 0, void 0, function* () {
+                            // Map YAML config to createAgent options
+                            const createOptions = Object.assign(Object.assign({}, options), { goal: agent.goal || "No goal specified", modelBaseId: agent.modelBaseId || "", instanceIP: agent.instanceIP || "localhost", machineInstanceId: agent.machineInstanceId || 1, machineManifestId: agent.machineManifestId || "default", machineName: agent.machineName || `agent-${agent.name.toLowerCase()}`, type: agentType, userId: user.id, parameters: agent.parameters ? agent.parameters.map((p) => {
+                                    return {
+                                        name: p.name || "unnamed",
+                                        description: p.description || "",
+                                        default: p.default || "",
+                                        isUserParam: p.isUserParam !== undefined ? p.isUserParam : true
+                                    };
+                                }) : [] });
+                            yield createAgent(agent.name, createOptions, apis.agentsApi, apis.projectsApi);
+                        }), () => {
+                            apis = createApis();
+                        });
+                        agentSpinner.stop();
+                        results.success++;
+                        results.agents.push({
+                            name: agent.name,
+                            success: true,
+                            message: `Created successfully`
+                        });
+                    }
+                    catch (error) {
+                        agentSpinner.fail(`Failed to create ${resourceType.toLowerCase()} '${agent.name}'`);
+                        console.error(chalk_1.default.red(`Error: ${error.message}`));
+                        results.failed++;
+                        results.agents.push({
+                            name: agent.name,
+                            success: false,
+                            message: error.message
+                        });
+                    }
                 }
-                spinner.succeed(`Successfully deployed agent ${agentId} to instance ${instanceId}`);
-                console.log(chalk_1.default.green("Agent deployed successfully"));
+                // Final summary
+                console.log(chalk_1.default.blue("\nResource creation summary:"));
+                const table = new cli_table3_1.default({
+                    head: [
+                        chalk_1.default.white.bold("Name"),
+                        chalk_1.default.white.bold("Type"),
+                        chalk_1.default.white.bold("Status"),
+                        chalk_1.default.white.bold("Message"),
+                    ],
+                    style: {
+                        head: [],
+                        border: [],
+                    },
+                });
+                results.agents.forEach((agent, index) => {
+                    const agentConfig = config.agents.find(a => a.name === agent.name) || config.agents[index];
+                    const agentType = (agentConfig === null || agentConfig === void 0 ? void 0 : agentConfig.type) || "CHAT";
+                    const resourceType = agentType === "AGENT" ? "Agent" : "Chatbot";
+                    table.push([
+                        agent.name,
+                        resourceType,
+                        agent.success ? chalk_1.default.green("Success") : chalk_1.default.red("Failed"),
+                        agent.message
+                    ]);
+                });
+                console.log(table.toString());
+                console.log(`\nTotal: ${results.success + results.failed}, Successful: ${results.success}, Failed: ${results.failed}`);
             }
             catch (error) {
-                spinner.fail("Failed to deploy agent");
-                if (error.response) {
-                    console.error(chalk_1.default.red(`API Error (${error.response.status}): ${((_f = error.response.data) === null || _f === void 0 ? void 0 : _f.message) || "Unknown error"}`));
+                spinner.fail("Failed to process YAML file");
+                if (error.code === 'ENOENT') {
+                    console.error(chalk_1.default.red(`File not found: ${yamlFilePath}`));
+                }
+                else if (error.name === 'YAMLException') {
+                    console.error(chalk_1.default.red(`Invalid YAML format: ${error.message}`));
                 }
                 else {
                     console.error(chalk_1.default.red("Error:"), error.message || "Unknown error");
@@ -295,7 +709,12 @@ function registerAgentCommands(program) {
             }
         }
         catch (error) {
-            console.error(chalk_1.default.red("Error:"), error instanceof Error ? error.message : "Unknown error");
+            if (error.message && error.message.includes('Authentication')) {
+                console.error(chalk_1.default.red(error.message));
+            }
+            else {
+                console.error(chalk_1.default.red("Error:"), error instanceof Error ? error.message : "Unknown error");
+            }
         }
     }));
 }