sparkecoder 0.1.59 → 0.1.61

package/dist/tools/index.js

@@ -10,7 +10,7 @@ var __export = (target, all) => {
 
 // src/config/types.ts
 import { z } from "zod";
-var ToolApprovalConfigSchema, SkillMetadataSchema, SessionConfigSchema, VectorGatewayConfigSchema, RemoteServerConfigSchema, SparkcoderConfigSchema;
+var ToolApprovalConfigSchema, SkillMetadataSchema, TaskConfigSchema, SessionConfigSchema, VectorGatewayConfigSchema, RemoteServerConfigSchema, SparkcoderConfigSchema;
 var init_types = __esm({
   "src/config/types.ts"() {
     "use strict";
@@ -29,11 +29,22 @@ var init_types = __esm({
       // Glob patterns - auto-inject when working with matching files
       globs: z.array(z.string()).optional().default([])
     });
+    TaskConfigSchema = z.object({
+      enabled: z.boolean(),
+      outputSchema: z.record(z.string(), z.unknown()),
+      webhookUrl: z.string().url().optional(),
+      maxIterations: z.number().optional(),
+      status: z.enum(["running", "completed", "failed"]),
+      result: z.unknown().optional(),
+      error: z.string().optional(),
+      iterations: z.number().optional()
+    });
     SessionConfigSchema = z.object({
       toolApprovals: z.record(z.string(), z.boolean()).optional(),
       approvalWebhook: z.string().url().optional(),
       skillsDirectory: z.string().optional(),
-      maxContextChars: z.number().optional().default(2e5)
+      maxContextChars: z.number().optional().default(2e5),
+      task: TaskConfigSchema.optional()
     });
     VectorGatewayConfigSchema = z.object({
       // Redis cluster nodes URL for Vector Gateway (or use REDIS_CLUSTER_NODES env var)
@@ -426,7 +437,7 @@ var init_hasher = __esm({
 });
 
 // src/semantic/chunker.ts
-import { extname as extname5, basename as basename2 } from "path";
+import { extname as extname6, basename as basename2 } from "path";
 var init_chunker = __esm({
   "src/semantic/chunker.ts"() {
     "use strict";
@@ -1296,26 +1307,41 @@ Terminal output is stored in the global SparkECoder data directory. Use the \`ta
 import { tool as tool2 } from "ai";
 import { z as z3 } from "zod";
 import { readFile as readFile2, stat } from "fs/promises";
-import { resolve as resolve2, relative, isAbsolute } from "path";
+import { resolve as resolve2, relative, isAbsolute, extname } from "path";
 import { existsSync as existsSync3 } from "fs";
 var MAX_FILE_SIZE = 5 * 1024 * 1024;
+var MAX_IMAGE_SIZE = 20 * 1024 * 1024;
 var MAX_OUTPUT_CHARS3 = 5e4;
+var IMAGE_EXTENSIONS = {
+  ".png": "image/png",
+  ".jpg": "image/jpeg",
+  ".jpeg": "image/jpeg",
+  ".gif": "image/gif",
+  ".webp": "image/webp"
+};
+function isImageFile(filePath) {
+  return extname(filePath).toLowerCase() in IMAGE_EXTENSIONS;
+}
+function getImageMediaType(filePath) {
+  return IMAGE_EXTENSIONS[extname(filePath).toLowerCase()] || "image/png";
+}
 var readFileInputSchema = z3.object({
-  path: z3.string().describe("The path to the file to read. Can be relative to working directory or absolute."),
-  startLine: z3.number().optional().describe("Optional: Start reading from this line number (1-indexed)"),
-  endLine: z3.number().optional().describe("Optional: Stop reading at this line number (1-indexed, inclusive)")
+  path: z3.string().describe("The path to the file to read. Can be relative to working directory or absolute. Supports text files and images (png, jpg, jpeg, gif, webp)."),
+  startLine: z3.number().optional().describe("Optional: Start reading from this line number (1-indexed). Only for text files."),
+  endLine: z3.number().optional().describe("Optional: Stop reading at this line number (1-indexed, inclusive). Only for text files.")
 });
 function createReadFileTool(options) {
   return tool2({
     description: `Read the contents of a file. Provide a path relative to the working directory (${options.workingDirectory}) or an absolute path.
-Large files will be automatically truncated. Binary files are not supported.
-Use this to understand existing code, check file contents, or gather context.`,
+Supports text files (automatically truncated if large) and image files (png, jpg, jpeg, gif, webp).
+For images, the file contents are returned as visual data you can see and analyze.
+Use this to understand existing code, check file contents, view screenshots, or gather context.`,
     inputSchema: readFileInputSchema,
-    execute: async ({ path, startLine, endLine }) => {
+    execute: async ({ path: filePath, startLine, endLine }) => {
       try {
-        const absolutePath = isAbsolute(path) ? path : resolve2(options.workingDirectory, path);
+        const absolutePath = isAbsolute(filePath) ? filePath : resolve2(options.workingDirectory, filePath);
         const relativePath = relative(options.workingDirectory, absolutePath);
-        if (relativePath.startsWith("..") && !isAbsolute(path)) {
+        if (relativePath.startsWith("..") && !isAbsolute(filePath)) {
           return {
             success: false,
             error: "Path escapes the working directory. Use an absolute path if intentional.",
@@ -1325,22 +1351,43 @@ Use this to understand existing code, check file contents, or gather context.`,
         if (!existsSync3(absolutePath)) {
           return {
             success: false,
-            error: `File not found: ${path}`,
+            error: `File not found: ${filePath}`,
             content: null
           };
         }
         const stats = await stat(absolutePath);
-        if (stats.size > MAX_FILE_SIZE) {
+        if (stats.isDirectory()) {
           return {
             success: false,
-            error: `File is too large (${(stats.size / 1024 / 1024).toFixed(2)}MB). Maximum size is ${MAX_FILE_SIZE / 1024 / 1024}MB.`,
+            error: 'Path is a directory, not a file. Use bash with "ls" to list directory contents.',
             content: null
           };
         }
-        if (stats.isDirectory()) {
+        if (isImageFile(absolutePath)) {
+          if (stats.size > MAX_IMAGE_SIZE) {
+            return {
+              success: false,
+              error: `Image is too large (${(stats.size / 1024 / 1024).toFixed(2)}MB). Maximum size is ${MAX_IMAGE_SIZE / 1024 / 1024}MB.`,
+              content: null
+            };
+          }
+          const buffer = await readFile2(absolutePath);
+          const base64 = buffer.toString("base64");
+          const mediaType = getImageMediaType(absolutePath);
+          return {
+            success: true,
+            path: absolutePath,
+            relativePath: relative(options.workingDirectory, absolutePath),
+            content: `[Image: ${relativePath} (${mediaType}, ${(stats.size / 1024).toFixed(1)}KB)]`,
+            mediaType,
+            imageData: base64,
+            sizeBytes: stats.size
+          };
+        }
+        if (stats.size > MAX_FILE_SIZE) {
           return {
             success: false,
-            error: 'Path is a directory, not a file. Use bash with "ls" to list directory contents.',
+            error: `File is too large (${(stats.size / 1024 / 1024).toFixed(2)}MB). Maximum size is ${MAX_FILE_SIZE / 1024 / 1024}MB.`,
             content: null
           };
         }
@@ -1356,9 +1403,7 @@ Use this to understand existing code, check file contents, or gather context.`,
               content: null
             };
           }
-          content = lines.slice(start, end).join("\n");
-          const lineNumbers = lines.slice(start, end).map((line, idx) => `${(start + idx + 1).toString().padStart(4)}: ${line}`).join("\n");
-          content = lineNumbers;
+          content = lines.slice(start, end).map((line, idx) => `${(start + idx + 1).toString().padStart(4)}: ${line}`).join("\n");
         }
         const truncatedContent = truncateOutput(content, MAX_OUTPUT_CHARS3);
         const wasTruncated = truncatedContent.length < content.length;
@@ -1385,6 +1430,19 @@ Use this to understand existing code, check file contents, or gather context.`,
           content: null
         };
       }
+    },
+    toModelOutput: ({ output }) => {
+      if (output && typeof output === "object" && "imageData" in output && output.imageData) {
+        const result = output;
+        return {
+          type: "content",
+          value: [
+            { type: "text", text: result.content },
+            { type: "image-data", data: result.imageData, mediaType: result.mediaType }
+          ]
+        };
+      }
+      return typeof output === "string" ? { type: "text", value: output } : { type: "json", value: output };
     }
   });
 }
@@ -1449,7 +1507,7 @@ async function backupFile(sessionId, workingDirectory, filePath) {
 }
 
 // src/lsp/index.ts
-import { extname as extname2, dirname as dirname4 } from "path";
+import { extname as extname3, dirname as dirname4 } from "path";
 
 // src/lsp/servers.ts
 import { spawn } from "child_process";
@@ -1572,9 +1630,9 @@ import {
 import { pathToFileURL, fileURLToPath } from "url";
 import { readFile as readFile4 } from "fs/promises";
 import { existsSync as existsSync6 } from "fs";
-import { extname, normalize } from "path";
+import { extname as extname2, normalize } from "path";
 function getLanguageId(filePath) {
-  const ext = extname(filePath).toLowerCase();
+  const ext = extname2(filePath).toLowerCase();
   const map = {
     ".ts": "typescript",
     ".tsx": "typescriptreact",
@@ -1962,7 +2020,7 @@ var state = {
 };
 async function getClientForFile(filePath) {
   const normalized = normalizePath(filePath);
-  const ext = extname2(normalized);
+  const ext = extname3(normalized);
   const serverDef = getServerForExtension(ext);
   if (!serverDef) {
     return null;
@@ -2055,7 +2113,7 @@ async function formatDiagnosticsOutput(filePath, options = {}) {
   return formatDiagnosticsForAgent(filePath, diagnostics, options);
 }
 function isSupported(filePath) {
-  const ext = extname2(filePath);
+  const ext = extname3(filePath);
   return getServerForExtension(ext) !== null;
 }
 
@@ -2402,7 +2460,7 @@ import { z as z6 } from "zod";
 // src/skills/index.ts
 init_types();
 import { readFile as readFile6, readdir } from "fs/promises";
-import { resolve as resolve6, basename, extname as extname3, relative as relative4 } from "path";
+import { resolve as resolve6, basename, extname as extname4, relative as relative4 } from "path";
 import { existsSync as existsSync8 } from "fs";
 import { minimatch } from "minimatch";
 function parseSkillFrontmatter(content) {
@@ -2473,7 +2531,7 @@ function parseSkillFrontmatter(content) {
   }
 }
 function getSkillNameFromPath(filePath) {
-  return basename(filePath, extname3(filePath)).replace(/[-_]/g, " ").replace(/\b\w/g, (c) => c.toUpperCase());
+  return basename(filePath, extname4(filePath)).replace(/[-_]/g, " ").replace(/\b\w/g, (c) => c.toUpperCase());
 }
 async function loadSkillsFromDirectory(directory, options = {}) {
   const {
@@ -2661,7 +2719,7 @@ Once loaded, a skill's content will be available in the conversation context.`,
 // src/tools/linter.ts
 import { tool as tool6 } from "ai";
 import { z as z7 } from "zod";
-import { resolve as resolve7, relative as relative5, isAbsolute as isAbsolute3, extname as extname4 } from "path";
+import { resolve as resolve7, relative as relative5, isAbsolute as isAbsolute3, extname as extname5 } from "path";
 import { existsSync as existsSync9 } from "fs";
 import { readdir as readdir2, stat as stat2 } from "fs/promises";
 var linterInputSchema = z7.object({
@@ -2684,7 +2742,7 @@ async function findSupportedFiles(dir, workingDirectory, maxFiles = 50) {
           }
           await walk(fullPath);
         } else if (entry.isFile()) {
-          const ext = extname4(entry.name);
+          const ext = extname5(entry.name);
           if (supportedExtensions.includes(ext)) {
             files.push(fullPath);
           }
@@ -2864,7 +2922,165 @@ import { nanoid as nanoid2 } from "nanoid";
 
 // src/agent/model.ts
 import { gateway } from "@ai-sdk/gateway";
+
+// src/agent/remote-model.ts
+function serializePrompt(prompt) {
+  return prompt.map((msg) => {
+    if (!Array.isArray(msg.content)) return msg;
+    return {
+      ...msg,
+      content: msg.content.map((part) => {
+        if (part.type === "file" && part.data instanceof Uint8Array) {
+          return {
+            ...part,
+            data: Buffer.from(part.data).toString("base64"),
+            _base64: true
+          };
+        }
+        return part;
+      })
+    };
+  });
+}
+function deserializeValue(value) {
+  if (value && typeof value === "object") {
+    if (value.__uint8array && typeof value.data === "string") {
+      return Buffer.from(value.data, "base64");
+    }
+    if (Array.isArray(value)) {
+      return value.map(deserializeValue);
+    }
+    const result = {};
+    for (const [k, v] of Object.entries(value)) {
+      result[k] = deserializeValue(v);
+    }
+    return result;
+  }
+  return value;
+}
+function prepareOptions(options) {
+  const { abortSignal, ...rest } = options;
+  return {
+    ...rest,
+    prompt: serializePrompt(options.prompt)
+  };
+}
+function createRemoteModel(modelId, config) {
+  const baseUrl = config.url.replace(/\/$/, "");
+  const headers = {
+    "Content-Type": "application/json",
+    "Authorization": `Bearer ${config.authKey}`
+  };
+  return {
+    specificationVersion: "v3",
+    provider: "remote-proxy",
+    modelId,
+    supportedUrls: {},
+    async doGenerate(options) {
+      const res = await fetch(`${baseUrl}/inference/generate`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({
+          modelId,
+          options: prepareOptions(options)
+        }),
+        signal: options.abortSignal
+      });
+      if (!res.ok) {
+        const err = await res.json().catch(() => ({}));
+        throw new Error(
+          `Remote inference failed (${res.status}): ${err.error || res.statusText}`
+        );
+      }
+      const result = await res.json();
+      return deserializeValue(result);
+    },
+    async doStream(options) {
+      const res = await fetch(`${baseUrl}/inference/stream`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({
+          modelId,
+          options: prepareOptions(options)
+        }),
+        signal: options.abortSignal
+      });
+      if (!res.ok) {
+        const err = await res.json().catch(() => ({}));
+        throw new Error(
+          `Remote inference failed (${res.status}): ${err.error || res.statusText}`
+        );
+      }
+      const reader = res.body.getReader();
+      const decoder = new TextDecoder();
+      let buffer = "";
+      const stream = new ReadableStream({
+        async pull(controller) {
+          while (true) {
+            const { done, value } = await reader.read();
+            if (done) {
+              if (buffer.trim()) {
+                try {
+                  const parsed = deserializeValue(JSON.parse(buffer.trim()));
+                  if (parsed.type === "error") {
+                    controller.error(new Error(parsed.error));
+                  } else {
+                    controller.enqueue(parsed);
+                  }
+                } catch {
+                }
+              }
+              controller.close();
+              return;
+            }
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split("\n");
+            buffer = lines.pop() || "";
+            for (const line of lines) {
+              if (!line.trim()) continue;
+              try {
+                const parsed = deserializeValue(JSON.parse(line));
+                if (parsed.type === "error") {
+                  controller.error(new Error(parsed.error));
+                  return;
+                }
+                controller.enqueue(parsed);
+              } catch {
+              }
+            }
+          }
+        },
+        cancel() {
+          reader.cancel();
+        }
+      });
+      const responseHeaders = {};
+      res.headers.forEach((v, k) => {
+        if (k.startsWith("x-upstream-")) {
+          responseHeaders[k.replace("x-upstream-", "")] = v;
+        }
+      });
+      return {
+        stream,
+        response: Object.keys(responseHeaders).length > 0 ? { headers: responseHeaders } : void 0
+      };
+    }
+  };
+}
+
+// src/agent/model.ts
+init_config();
 function resolveModel(modelId) {
+  try {
+    const config = getConfig();
+    if (config.resolvedRemoteServer.isConfigured) {
+      return createRemoteModel(modelId.trim(), {
+        url: config.resolvedRemoteServer.url,
+        authKey: config.resolvedRemoteServer.authKey
+      });
+    }
+  } catch {
+  }
   return gateway(modelId.trim());
 }
 var SUBAGENT_MODELS = {
@@ -4034,6 +4250,59 @@ Context: ${context}` : query;
 
 // src/tools/index.ts
 init_semantic_search();
+
+// src/tools/task.ts
+import { tool as tool11 } from "ai";
+import { z as z12 } from "zod";
+import Ajv from "ajv";
+var ajv = new Ajv({ allErrors: true });
+function createCompleteTaskTool(options) {
+  const validate = ajv.compile(options.outputSchema);
+  return tool11({
+    description: "Call this tool when you have completed the task. Pass the result as a JSON object matching the required output schema. If the result does not match the schema, you will receive validation errors and should fix and retry.",
+    inputSchema: z12.object({
+      result: z12.record(z12.string(), z12.unknown()).describe("The task result as a JSON object matching the output schema")
+    }),
+    execute: async (input) => {
+      const valid = validate(input.result);
+      if (!valid) {
+        const errors = validate.errors?.map((e) => ({
+          path: e.instancePath || "/",
+          message: e.message,
+          params: e.params
+        }));
+        return {
+          status: "validation_error",
+          message: "The result does not match the required output schema. Fix the errors and call complete_task again.",
+          errors,
+          expectedSchema: options.outputSchema
+        };
+      }
+      options.onComplete({ status: "completed", result: input.result });
+      return {
+        status: "completed",
+        message: "Task completed successfully."
+      };
+    }
+  });
+}
+function createTaskFailedTool(options) {
+  return tool11({
+    description: "Call this tool if you are unable to complete the task. Provide a clear reason explaining why the task cannot be completed.",
+    inputSchema: z12.object({
+      reason: z12.string().describe("Explanation of why the task cannot be completed")
+    }),
+    execute: async (input) => {
+      options.onComplete({ status: "failed", error: input.reason });
+      return {
+        status: "failed",
+        message: `Task marked as failed: ${input.reason}`
+      };
+    }
+  });
+}
+
+// src/tools/index.ts
 init_semantic();
 init_semantic_search();
 async function createTools(options) {
@@ -4085,16 +4354,22 @@ async function createTools(options) {
     } catch {
     }
   }
+  if (options.taskTools) {
+    tools.complete_task = createCompleteTaskTool(options.taskTools);
+    tools.task_failed = createTaskFailedTool(options.taskTools);
+  }
   return tools;
 }
 export {
   createBashTool,
   createCodeGraphTool,
+  createCompleteTaskTool,
   createLinterTool,
   createLoadSkillTool,
   createReadFileTool,
   createSearchTool,
   createSemanticSearchTool,
+  createTaskFailedTool,
   createTodoTool,
   createTools,
   createWriteFileTool