@zds-ai/cli 0.1.0

package/dist/tools/image-tool.js

@@ -0,0 +1,268 @@
+import { exec } from "child_process";
+import { promisify } from "util";
+import { getHandledToolNames } from "./tool-discovery.js";
+const execAsync = promisify(exec);
+export class ImageTool {
+    agent; // Reference to the GrokAgent
+    setAgent(agent) {
+        this.agent = agent;
+    }
+    getHandledToolNames() {
+        return getHandledToolNames(this);
+    }
+    /**
+     * Generate an image using AI image generation.
+     * Uses generate_image_asl.sh script with Stable Diffusion.
+     */
+    async generateImage(prompt, negativePrompt, width, height, model, sampler, configScale, numSteps, nsfw, name, move, seed) {
+        try {
+            if (!prompt) {
+                return {
+                    success: false,
+                    error: "Prompt is required",
+                    output: "Prompt is required"
+                };
+            }
+            if (!this.agent) {
+                return {
+                    success: false,
+                    error: "Agent not available",
+                    output: "Agent not available"
+                };
+            }
+            // Get the agent's home directory and create Images path
+            const agentHome = process.env.ZDS_AI_AGENT_HOME_DIR;
+            if (!agentHome) {
+                return {
+                    success: false,
+                    error: "Agent home directory not found",
+                    output: "Agent home directory not found"
+                };
+            }
+            const imagesDir = `${agentHome}/Images`;
+            // Ensure Images directory exists
+            try {
+                await execAsync(`mkdir -p "${imagesDir}"`);
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: `Failed to create Images directory: ${error}`,
+                    output: `Failed to create Images directory: ${error}`
+                };
+            }
+            // Set defaults
+            const defaultNegativePrompt = "score_6, score_5, score_4, (worst quality:1.2), (low quality:1.2), (normal quality:1.2), lowres, bad anatomy, bad hands, signature, watermarks, ugly, imperfect eyes, skewed eyes, unnatural face, unnatural body, error, extra limb, missing limbs";
+            const finalModel = model || "cyberrealisticPony_v130";
+            const finalWidth = Math.floor(width || 480);
+            const finalHeight = Math.floor(height || 720);
+            const finalSampler = sampler || "DPM++ 2M Karras";
+            const finalConfigScale = configScale || 5.0;
+            const finalNumSteps = Math.floor(numSteps || 30);
+            const finalNsfw = nsfw !== undefined ? nsfw : false;
+            const finalMove = move !== undefined ? move : false;
+            const finalNegativePrompt = negativePrompt || defaultNegativePrompt;
+            // Handle NSFW flag: when nsfw=true, allow NSFW content (prepend to prompt)
+            // when nsfw=false (default), avoid NSFW content (prepend to negative prompt)
+            const finalPrompt = finalNsfw ? `NSFW, ${prompt}` : prompt;
+            // If nsfw is on, strip out any "NSFW" from negative prompt (case-insensitive)
+            let finalNegativePromptWithNsfw;
+            if (finalNsfw) {
+                // Remove all "NSFW" variants - keep replacing until none left
+                finalNegativePromptWithNsfw = finalNegativePrompt;
+                let prev = '';
+                while (prev !== finalNegativePromptWithNsfw) {
+                    prev = finalNegativePromptWithNsfw;
+                    finalNegativePromptWithNsfw = finalNegativePromptWithNsfw
+                        .replace(/NSFW,?\s*/gi, '') // Remove NSFW with optional comma and spaces
+                        .replace(/,\s*,/g, ',') // Clean up double commas
+                        .replace(/^\s*,\s*/, '') // Remove leading comma
+                        .trim();
+                }
+            }
+            else {
+                // nsfw is off, prepend "NSFW, " to block NSFW content
+                finalNegativePromptWithNsfw = `NSFW, ${finalNegativePrompt}`;
+            }
+            // Escape prompts for shell
+            const escapedPrompt = finalPrompt.replace(/'/g, "'\\''");
+            const escapedNegativePrompt = finalNegativePromptWithNsfw.replace(/'/g, "'\\''");
+            // Build command using generate_image_asl.sh
+            let command = `generate_image_asl.sh '${escapedPrompt}' '${escapedNegativePrompt}'`;
+            if (finalMove) {
+                command += ' --move';
+            }
+            command += ` --width ${finalWidth} --height ${finalHeight} --cfg-scale ${finalConfigScale} --steps ${finalNumSteps} --sampler '${finalSampler}' --model '${finalModel}'`;
+            if (seed !== undefined) {
+                command += ` --seed ${Math.floor(seed)}`;
+            }
+            if (name) {
+                const escapedName = name.replace(/'/g, "'\\''");
+                command += ` --name '${escapedName}'`;
+            }
+            try {
+                const { stdout, stderr } = await execAsync(command, {
+                    timeout: 300000 // 5 minute timeout
+                });
+                // Parse output to find the generated file path
+                // The script must output the absolute path to the generated image
+                const lines = stdout.trim().split('\n');
+                const lastLine = lines[lines.length - 1];
+                // Verify we got an absolute path
+                if (!lastLine.startsWith('/')) {
+                    return {
+                        success: false,
+                        error: `Image generation script did not return a valid file path. Output: ${stdout}`,
+                        output: `Image generation failed - no file path returned.\n\nScript output:\n${stdout}`
+                    };
+                }
+                const filepath = lastLine;
+                // Get file size if file exists
+                let fileSize = "unknown";
+                try {
+                    const { stdout: sizeOutput } = await execAsync(`ls -lh "${filepath}" 2>/dev/null | awk '{print $5}'`);
+                    fileSize = sizeOutput.trim() || "unknown";
+                }
+                catch {
+                    fileSize = "unknown";
+                }
+                return {
+                    success: true,
+                    output: `Image generated successfully\nPrompt: ${prompt}\nModel: ${finalModel}\nSize: ${finalWidth}x${finalHeight}\nSteps: ${finalNumSteps}\nGuidance: ${finalConfigScale}\nPath: ${filepath}\nFile Size: ${fileSize}`,
+                    displayOutput: `Generated image (${finalWidth}x${finalHeight})`
+                };
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: `Image generation failed: ${error.message}`,
+                    output: `Image generation failed: ${error.message}`
+                };
+            }
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : "Unknown error generating image",
+                output: error instanceof Error ? error.message : "Unknown error generating image"
+            };
+        }
+    }
+    /**
+     * Caption an image using AI image captioning.
+     * Uses joycaption script.
+     */
+    async captionImage(filename, prompt) {
+        try {
+            if (!filename) {
+                return {
+                    success: false,
+                    error: "Filename is required",
+                    output: "Filename is required"
+                };
+            }
+            // Build command using joycaption
+            let command = `joycaption '${filename.replace(/'/g, "'\\''")}'`;
+            if (prompt) {
+                command += ` --prompt "${prompt.replace(/"/g, '\\"')}"`;
+            }
+            try {
+                const { stdout, stderr } = await execAsync(command, {
+                    timeout: 90000, // 90 second timeout
+                    shell: '/bin/zsh'
+                });
+                // If there's stderr output, include it but still return success if we got stdout
+                if (stderr && !stdout) {
+                    return {
+                        success: false,
+                        error: `Image captioning failed: ${stderr}`,
+                        output: stderr
+                    };
+                }
+                const caption = stdout.trim();
+                return {
+                    success: true,
+                    output: caption,
+                    displayOutput: `Caption: ${caption}`
+                };
+            }
+            catch (error) {
+                // Extract error details
+                const errorMessage = error.message || "Unknown error";
+                const stderr = error.stderr || "";
+                const stdout = error.stdout || "";
+                return {
+                    success: false,
+                    error: `Image captioning failed (code ${error.code || 'unknown'}): ${errorMessage}${stderr ? '\nstderr: ' + stderr : ''}${stdout ? '\nstdout: ' + stdout : ''}`,
+                    output: `Error code: ${error.code || 'unknown'}\n${errorMessage}${stderr ? '\nstderr: ' + stderr : ''}${stdout ? '\nstdout: ' + stdout : ''}`
+                };
+            }
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : "Unknown error captioning image",
+                output: error instanceof Error ? error.message : "Unknown error captioning image"
+            };
+        }
+    }
+    /**
+     * Extract PNG metadata (generation settings) from a PNG file.
+     * Uses exiftool to read embedded generation parameters.
+     */
+    async pngInfo(filename) {
+        try {
+            if (!filename) {
+                return {
+                    success: false,
+                    error: "Filename is required",
+                    output: "Filename is required"
+                };
+            }
+            const escapedFilename = filename.replace(/'/g, "'\\''");
+            const command = `exiftool -Parameters '${escapedFilename}'`;
+            try {
+                const { stdout, stderr } = await execAsync(command, {
+                    timeout: 10000 // 10 second timeout
+                });
+                if (stderr && !stdout) {
+                    return {
+                        success: false,
+                        error: `Failed to read PNG metadata: ${stderr}`,
+                        output: stderr
+                    };
+                }
+                const metadata = stdout.trim();
+                if (!metadata) {
+                    return {
+                        success: false,
+                        error: "No generation parameters found in file",
+                        output: "No generation parameters found in file"
+                    };
+                }
+                return {
+                    success: true,
+                    output: metadata,
+                    displayOutput: "PNG generation parameters extracted"
+                };
+            }
+            catch (error) {
+                const errorMessage = error.message || "Unknown error";
+                const stderr = error.stderr || "";
+                return {
+                    success: false,
+                    error: `Failed to read PNG metadata: ${errorMessage}${stderr ? '\n' + stderr : ''}`,
+                    output: `Failed to read PNG metadata: ${errorMessage}${stderr ? '\n' + stderr : ''}`
+                };
+            }
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : "Unknown error reading PNG metadata",
+                output: error instanceof Error ? error.message : "Unknown error reading PNG metadata"
+            };
+        }
+    }
+}
+//# sourceMappingURL=image-tool.js.map

package/dist/tools/image-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-tool.js","sourceRoot":"","sources":["../../src/tools/image-tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AAEjC,OAAO,EAAiB,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAEzE,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;AAElC,MAAM,OAAO,SAAS;IACZ,KAAK,CAAM,CAAC,6BAA6B;IAEjD,QAAQ,CAAC,KAAU;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,mBAAmB;QACjB,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,aAAa,CACjB,MAAc,EACd,cAAuB,EACvB,KAAc,EACd,MAAe,EACf,KAAc,EACd,OAAgB,EAChB,WAAoB,EACpB,QAAiB,EACjB,IAAc,EACd,IAAa,EACb,IAAc,EACd,IAAa;QAEb,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,oBAAoB;oBAC3B,MAAM,EAAE,oBAAoB;iBAC7B,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBAChB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,qBAAqB;oBAC5B,MAAM,EAAE,qBAAqB;iBAC9B,CAAC;YACJ,CAAC;YAED,wDAAwD;YACxD,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;YACpD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,gCAAgC;oBACvC,MAAM,EAAE,gCAAgC;iBACzC,CAAC;YACJ,CAAC;YAED,MAAM,SAAS,GAAG,GAAG,SAAS,SAAS,CAAC;YAExC,iCAAiC;YACjC,IAAI,CAAC;gBACH,MAAM,SAAS,CAAC,aAAa,SAAS,GAAG,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,sCAAsC,KAAK,EAAE;oBACpD,MAAM,EAAE,sCAAsC,KAAK,EAAE;iBACtD,CAAC;YACJ,CAAC;YAED,eAAe;YACf,MAAM,qBAAqB,GAAG,qPAAqP,CAAC;YACpR,MAAM,UAAU,GAAG,KAAK,IAAI,yBAAyB,CAAC;YACtD,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC,CAAC;YAC5C,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;YAC9C,MAAM,YAAY,GAAG,OAAO,IAAI,iBAAiB,CAAC;YAClD,MAAM,gBAAgB,GAAG,WAAW,IAAI,GAAG,CAAC;YAC5C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;YACpD,MAAM,SAAS,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;YACpD,MAAM,mBAAmB,GAAG,cAAc,IAAI,qBAAqB,CAAC;YAEpE,2EAA2E;YAC3E,6EAA6E;YAC7E,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;YAE3D,8EAA8E;YAC9E,IAAI,2BAAmC,CAAC;YACxC,IAAI,SAAS,EAAE,CAAC;gBACd,8DAA8D;gBAC9D,2BAA2B,GAAG,mBAAmB,CAAC;gBAClD,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,OAAO,IAAI,KAAK,2BAA2B,EAAE,CAAC;oBAC5C,IAAI,GAAG,2BAA2B,CAAC;oBACnC,2BAA2B,GAAG,2BAA2B;yBACtD,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAE,6CAA6C;yBACzE,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAE,yBAAyB;yBACjD,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAE,uBAAuB;yBAChD,IAAI,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,sDAAsD;gBACtD,2BAA2B,GAAG,SAAS,mBAAmB,EAAE,CAAC;YAC/D,CAAC;YAED,2BAA2B;YAC3B,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACzD,MAAM,qBAAqB,GAAG,2BAA2B,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAEjF,4CAA4C;YAC5C,IAAI,OAAO,GAAG,0BAA0B,aAAa,MAAM,qBAAqB,GAAG,CAAC;YAEpF,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO,IAAI,SAAS,CAAC;YACvB,CAAC;YAED,OAAO,IAAI,YAAY,UAAU,aAAa,WAAW,gBAAgB,gBAAgB,YAAY,aAAa,eAAe,YAAY,cAAc,UAAU,GAAG,CAAC;YAEzK,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,OAAO,IAAI,WAAW,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3C,CAAC;YAED,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAChD,OAAO,IAAI,YAAY,WAAW,GAAG,CAAC;YACxC,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE;oBAClD,OAAO,EAAE,MAAM,CAAC,mBAAmB;iBACpC,CAAC,CAAC;gBAEH,+CAA+C;gBAC/C,kEAAkE;gBAClE,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACxC,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEzC,iCAAiC;gBACjC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC9B,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,qEAAqE,MAAM,EAAE;wBACpF,MAAM,EAAE,uEAAuE,MAAM,EAAE;qBACxF,CAAC;gBACJ,CAAC;gBAED,MAAM,QAAQ,GAAG,QAAQ,CAAC;gBAE1B,+BAA+B;gBAC/B,IAAI,QAAQ,GAAG,SAAS,CAAC;gBACzB,IAAI,CAAC;oBACH,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,SAAS,CAAC,WAAW,QAAQ,kCAAkC,CAAC,CAAC;oBACtG,QAAQ,GAAG,UAAU,CAAC,IAAI,EAAE,IAAI,SAAS,CAAC;gBAC5C,CAAC;gBAAC,MAAM,CAAC;oBACP,QAAQ,GAAG,SAAS,CAAC;gBACvB,CAAC;gBAED,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,yCAAyC,MAAM,YAAY,UAAU,WAAW,UAAU,IAAI,WAAW,YAAY,aAAa,eAAe,gBAAgB,WAAW,QAAQ,gBAAgB,QAAQ,EAAE;oBACtN,aAAa,EAAE,oBAAoB,UAAU,IAAI,WAAW,GAAG;iBAChE,CAAC;YACJ,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,4BAA4B,KAAK,CAAC,OAAO,EAAE;oBAClD,MAAM,EAAE,4BAA4B,KAAK,CAAC,OAAO,EAAE;iBACpD,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gCAAgC;gBAChF,MAAM,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gCAAgC;aAClF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,YAAY,CAChB,QAAgB,EAChB,MAAe;QAEf,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,sBAAsB;oBAC7B,MAAM,EAAE,sBAAsB;iBAC/B,CAAC;YACJ,CAAC;YAED,iCAAiC;YACjC,IAAI,OAAO,GAAG,eAAe,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC;YAEhE,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,IAAI,cAAc,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;YAC1D,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE;oBAClD,OAAO,EAAE,KAAK,EAAE,oBAAoB;oBACpC,KAAK,EAAE,UAAU;iBAClB,CAAC,CAAC;gBAEH,iFAAiF;gBACjF,IAAI,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;oBACtB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,4BAA4B,MAAM,EAAE;wBAC3C,MAAM,EAAE,MAAM;qBACf,CAAC;gBACJ,CAAC;gBAED,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC9B,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,OAAO;oBACf,aAAa,EAAE,YAAY,OAAO,EAAE;iBACrC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,wBAAwB;gBACxB,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,eAAe,CAAC;gBACtD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC;gBAElC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,iCAAiC,KAAK,CAAC,IAAI,IAAI,SAAS,MAAM,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC/J,MAAM,EAAE,eAAe,KAAK,CAAC,IAAI,IAAI,SAAS,KAAK,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;iBAC9I,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gCAAgC;gBAChF,MAAM,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gCAAgC;aAClF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO,CACX,QAAgB;QAEhB,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,sBAAsB;oBAC7B,MAAM,EAAE,sBAAsB;iBAC/B,CAAC;YACJ,CAAC;YAED,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,yBAAyB,eAAe,GAAG,CAAC;YAE5D,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE;oBAClD,OAAO,EAAE,KAAK,CAAC,oBAAoB;iBACpC,CAAC,CAAC;gBAEH,IAAI,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;oBACtB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,gCAAgC,MAAM,EAAE;wBAC/C,MAAM,EAAE,MAAM;qBACf,CAAC;gBACJ,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC/B,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,wCAAwC;wBAC/C,MAAM,EAAE,wCAAwC;qBACjD,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,QAAQ;oBAChB,aAAa,EAAE,qCAAqC;iBACrD,CAAC;YACJ,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,eAAe,CAAC;gBACtD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC;gBAElC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,gCAAgC,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBACnF,MAAM,EAAE,gCAAgC,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;iBACrF,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,oCAAoC;gBACpF,MAAM,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,oCAAoC;aACtF,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/tools/index.d.ts

@@ -0,0 +1,14 @@
+export { ZshTool } from "./zsh.js";
+export { TextEditorTool } from "./text-editor.js";
+export { MorphEditorTool } from "./morph-editor.js";
+export { ConfirmationTool } from "./confirmation-tool.js";
+export { SearchTool } from "./search.js";
+export { EnvTool } from "./env-tool.js";
+export { IntrospectTool } from "./introspect-tool.js";
+export { ClearCacheTool } from "./clear-cache-tool.js";
+export { CharacterTool } from "./character-tool.js";
+export { TaskTool } from "./task-tool.js";
+export { InternetTool } from "./internet-tool.js";
+export { ImageTool } from "./image-tool.js";
+export { FileConversionTool } from "./file-conversion-tool.js";
+export { RestartTool } from "./restart-tool.js";

package/dist/tools/index.js

@@ -0,0 +1,15 @@
+export { ZshTool } from "./zsh.js";
+export { TextEditorTool } from "./text-editor.js";
+export { MorphEditorTool } from "./morph-editor.js";
+export { ConfirmationTool } from "./confirmation-tool.js";
+export { SearchTool } from "./search.js";
+export { EnvTool } from "./env-tool.js";
+export { IntrospectTool } from "./introspect-tool.js";
+export { ClearCacheTool } from "./clear-cache-tool.js";
+export { CharacterTool } from "./character-tool.js";
+export { TaskTool } from "./task-tool.js";
+export { InternetTool } from "./internet-tool.js";
+export { ImageTool } from "./image-tool.js";
+export { FileConversionTool } from "./file-conversion-tool.js";
+export { RestartTool } from "./restart-tool.js";
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AACnC,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/tools/internet-tool.d.ts

@@ -0,0 +1,11 @@
+import { ToolResult } from "../types/index.js";
+import { ToolDiscovery } from "./tool-discovery.js";
+export declare class InternetTool implements ToolDiscovery {
+    private agent;
+    setAgent(agent: any): void;
+    getHandledToolNames(): string[];
+    /**
+     * Download a file < 10MB from the Internet.
+     */
+    downloadFile(url: string): Promise<ToolResult>;
+}

package/dist/tools/internet-tool.js

@@ -0,0 +1,108 @@
+import { exec } from "child_process";
+import { promisify } from "util";
+const execAsync = promisify(exec);
+export class InternetTool {
+    agent; // Reference to the GrokAgent
+    setAgent(agent) {
+        this.agent = agent;
+    }
+    getHandledToolNames() {
+        return ["downloadFile"];
+    }
+    /**
+     * Download a file < 10MB from the Internet.
+     */
+    async downloadFile(url) {
+        try {
+            if (!this.agent) {
+                return {
+                    success: false,
+                    error: "Agent not available",
+                    output: "Agent not available"
+                };
+            }
+            // Get the agent's home directory and create Downloads path
+            const agentHome = process.env.ZDS_AI_AGENT_HOME_DIR;
+            if (!agentHome) {
+                return {
+                    success: false,
+                    error: "Agent home directory not found",
+                    output: "Agent home directory not found"
+                };
+            }
+            const downloadsDir = `${agentHome}/Downloads`;
+            // Ensure Downloads directory exists
+            try {
+                await execAsync(`mkdir -p "${downloadsDir}"`);
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: `Failed to create Downloads directory: ${error}`,
+                    output: `Failed to create Downloads directory: ${error}`
+                };
+            }
+            // Check file size first using curl -I (HEAD request)
+            let contentLength;
+            try {
+                const { stdout } = await execAsync(`curl -sI "${url}" | grep -i "content-length" | awk '{print $2}' | tr -d '\r'`);
+                contentLength = parseInt(stdout.trim() || "0");
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: `Failed to check file size: ${error}`,
+                    output: `Failed to check file size: ${error}`
+                };
+            }
+            const maxSize = 10 * 1024 * 1024; // 10MB
+            if (contentLength > maxSize) {
+                return {
+                    success: false,
+                    error: `File too large: ${contentLength} bytes (max: ${maxSize} bytes)`,
+                    output: `File too large: ${contentLength} bytes (max: ${maxSize} bytes)`
+                };
+            }
+            // Extract filename from URL
+            const filename = url.split('/').pop()?.split('?')[0] || 'downloaded_file';
+            const filepath = `${downloadsDir}/${filename}`;
+            // Download the file
+            try {
+                await execAsync(`curl -L -o "${filepath}" "${url}"`);
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: `Download failed: ${error}`,
+                    output: `Download failed: ${error}`
+                };
+            }
+            // Get MIME info using file command
+            let fileInfo;
+            try {
+                const { stdout } = await execAsync(`file "${filepath}"`);
+                fileInfo = stdout.trim() || "Unknown file type";
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: `Failed to get file info: ${error}`,
+                    output: `Download completed but failed to get file info: ${error}`
+                };
+            }
+            return {
+                success: true,
+                output: `File downloaded successfully\nPath: ${filepath}\nSize: ${contentLength} bytes\nInfo: ${fileInfo}`,
+                displayOutput: `Downloaded ${filename} (${contentLength} bytes)`
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : "Unknown error downloading file",
+                output: error instanceof Error ? error.message : "Unknown error downloading file"
+            };
+        }
+    }
+}
+//# sourceMappingURL=internet-tool.js.map

package/dist/tools/internet-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"internet-tool.js","sourceRoot":"","sources":["../../src/tools/internet-tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AAIjC,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;AAElC,MAAM,OAAO,YAAY;IACf,KAAK,CAAM,CAAC,6BAA6B;IAEjD,QAAQ,CAAC,KAAU;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,mBAAmB;QACjB,OAAO,CAAC,cAAc,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,GAAW;QAC5B,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBAChB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,qBAAqB;oBAC5B,MAAM,EAAE,qBAAqB;iBAC9B,CAAC;YACJ,CAAC;YAED,2DAA2D;YAC3D,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;YACpD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,gCAAgC;oBACvC,MAAM,EAAE,gCAAgC;iBACzC,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,GAAG,GAAG,SAAS,YAAY,CAAC;YAE9C,oCAAoC;YACpC,IAAI,CAAC;gBACH,MAAM,SAAS,CAAC,aAAa,YAAY,GAAG,CAAC,CAAC;YAChD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,yCAAyC,KAAK,EAAE;oBACvD,MAAM,EAAE,yCAAyC,KAAK,EAAE;iBACzD,CAAC;YACJ,CAAC;YAED,qDAAqD;YACrD,IAAI,aAAqB,CAAC;YAC1B,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,aAAa,GAAG,8DAA8D,CAAC,CAAC;gBACnH,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,GAAG,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,8BAA8B,KAAK,EAAE;oBAC5C,MAAM,EAAE,8BAA8B,KAAK,EAAE;iBAC9C,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO;YAEzC,IAAI,aAAa,GAAG,OAAO,EAAE,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,mBAAmB,aAAa,gBAAgB,OAAO,SAAS;oBACvE,MAAM,EAAE,mBAAmB,aAAa,gBAAgB,OAAO,SAAS;iBACzE,CAAC;YACJ,CAAC;YAED,4BAA4B;YAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC;YAC1E,MAAM,QAAQ,GAAG,GAAG,YAAY,IAAI,QAAQ,EAAE,CAAC;YAE/C,oBAAoB;YACpB,IAAI,CAAC;gBACH,MAAM,SAAS,CAAC,eAAe,QAAQ,MAAM,GAAG,GAAG,CAAC,CAAC;YACvD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,oBAAoB,KAAK,EAAE;oBAClC,MAAM,EAAE,oBAAoB,KAAK,EAAE;iBACpC,CAAC;YACJ,CAAC;YAED,mCAAmC;YACnC,IAAI,QAAgB,CAAC;YACrB,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,SAAS,CAAC,SAAS,QAAQ,GAAG,CAAC,CAAC;gBACzD,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,IAAI,mBAAmB,CAAC;YAClD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,4BAA4B,KAAK,EAAE;oBAC1C,MAAM,EAAE,mDAAmD,KAAK,EAAE;iBACnE,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,uCAAuC,QAAQ,WAAW,aAAa,iBAAiB,QAAQ,EAAE;gBAC1G,aAAa,EAAE,cAAc,QAAQ,KAAK,aAAa,SAAS;aACjE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gCAAgC;gBAChF,MAAM,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gCAAgC;aAClF,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/tools/introspect-tool.d.ts

@@ -0,0 +1,11 @@
+import { ToolResult } from "../types/index.js";
+import { ToolDiscovery } from "./tool-discovery.js";
+export declare class IntrospectTool implements ToolDiscovery {
+    private agent;
+    setAgent(agent: any): void;
+    /**
+     * Introspect available tools and system information
+     */
+    introspect(target: string): Promise<ToolResult>;
+    getHandledToolNames(): string[];
+}