@polka-codes/cli 0.1.5 → 0.1.6

package/dist/handlers.js

@@ -1,9 +1,9 @@
-import { spawn } from "node:child_process";
-import { readFile, writeFile } from "node:fs/promises";
-import { ToolResponseType, allTools, createServiceLogger } from "@polka-codes/core";
-import { listFiles } from "./utils/listFiles.js";
-import { replaceInFile } from "./utils/replaceInFile.js";
-const logger = createServiceLogger("cli/handlers");
+import { spawn } from 'node:child_process';
+import { readFile, writeFile } from 'node:fs/promises';
+import { ToolResponseType, allTools, createServiceLogger } from '@polka-codes/core';
+import { listFiles } from './utils/listFiles';
+import { replaceInFile } from './utils/replaceInFile';
+const logger = createServiceLogger('cli/handlers');
 const handleError = async (name, fn) => {
     try {
         return await fn();
@@ -33,7 +33,7 @@ const getStringArray = (args, name, defaultValue) => {
         }
         return defaultValue;
     }
-    return ret.split(",");
+    return ret.split(',');
 };
 const getBoolean = (args, name, defaultValue) => {
     const ret = args[name];
@@ -44,9 +44,9 @@ const getBoolean = (args, name, defaultValue) => {
         return defaultValue;
     }
     switch (ret.toLowerCase()) {
-        case "true":
+        case 'true':
             return true;
-        case "false":
+        case 'false':
             return false;
         default:
             throw new Error(`Invalid argument value: ${name}`);
@@ -68,7 +68,7 @@ const getInt = (args, name, defaultValue) => {
 };
 const handlers = {
     executeCommand: async (options, args) => {
-        const command = getString(args, "command");
+        const command = getString(args, 'command');
         const shouldExecute = (await options.executeCommand?.shouldExecute?.(command)) ?? true;
         if (!shouldExecute) {
             return {
@@ -76,7 +76,7 @@ const handlers = {
                 message: `<command>${command}</command><command_rejected>Rejected by policy</command_rejected>`,
             };
         }
-        const requiresApproval = getBoolean(args, "requires_approval", false);
+        const requiresApproval = getBoolean(args, 'requires_approval', false);
         if (requiresApproval) {
             const shouldApprove = await options.executeCommand?.shouldApprove?.(command);
             if (!shouldApprove) {
@@ -92,18 +92,18 @@ const handlers = {
             logger.trace(`Executing command: ${command}`);
             const child = spawn(command, [], {
                 shell: true,
-                stdio: "pipe",
+                stdio: 'pipe',
             });
             options.executeCommand?.commandStarted?.(command, child);
-            let stdoutText = "";
-            let stderrText = "";
-            child.stdout.on("data", (data) => {
+            let stdoutText = '';
+            let stderrText = '';
+            child.stdout.on('data', (data) => {
                 stdoutText += data.toString();
             });
-            child.stderr.on("data", (data) => {
+            child.stderr.on('data', (data) => {
                 stderrText += data.toString();
             });
-            child.on("close", (code) => {
+            child.on('close', (code) => {
                 const message = `<command>${command}</command>
 <command_exit_code>${code}</command_exit_code>
 <command_stdout>
@@ -126,13 +126,13 @@ ${stderrText}
                     });
                 }
             });
-            child.on("error", (err) => {
+            child.on('error', (err) => {
                 reject(err);
             });
         });
     },
     readFile: async (options, args) => {
-        const paths = getStringArray(args, "path");
+        const paths = getStringArray(args, 'path');
         for (const path of paths) {
             const isPathAllowed = (await options.files?.isPathAllowed?.(path)) ?? true;
             if (!isPathAllowed) {
@@ -144,17 +144,17 @@ ${stderrText}
         }
         const resp = [];
         for (const path of paths) {
-            const fileContent = await readFile(path, "utf8");
+            const fileContent = await readFile(path, 'utf8');
             resp.push(`<read_file_file_conten path="${path}">${fileContent}</read_file_file_content>`);
         }
         return {
             type: ToolResponseType.Reply,
-            message: resp.join("\n"),
+            message: resp.join('\n'),
         };
     },
     writeToFile: async (options, args) => {
-        const path = getString(args, "path");
-        const content = getString(args, "content");
+        const path = getString(args, 'path');
+        const content = getString(args, 'content');
         const isPathAllowed = (await options.files?.isPathAllowed?.(path)) ?? true;
         if (!isPathAllowed) {
             return {
@@ -162,15 +162,15 @@ ${stderrText}
                 message: `<path>${path}</path><path_rejected>Rejected by policy</path_rejected>`,
             };
         }
-        await writeFile(path, content, "utf8");
+        await writeFile(path, content, 'utf8');
         return {
             type: ToolResponseType.Reply,
             message: `<write_file_path>${path}</write_file_path><status>Success</status>`,
         };
     },
     replaceInFile: async (options, args) => {
-        const path = getString(args, "path");
-        const diff = getString(args, "diff");
+        const path = getString(args, 'path');
+        const diff = getString(args, 'diff');
         const isPathAllowed = (await options.files?.isPathAllowed?.(path)) ?? true;
         if (!isPathAllowed) {
             return {
@@ -178,9 +178,9 @@ ${stderrText}
                 message: `<path>${path}</path><path_rejected>Rejected by policy</path_rejected>`,
             };
         }
-        const fileContent = await readFile(path, "utf8");
+        const fileContent = await readFile(path, 'utf8');
         const result = await replaceInFile(fileContent, diff);
-        await writeFile(path, result, "utf8");
+        await writeFile(path, result, 'utf8');
         return {
             type: ToolResponseType.Reply,
             message: `<replace_in_file_path>${path}</replace_in_file_path>`,
@@ -190,12 +190,12 @@ ${stderrText}
         // TODO
         return {
             type: ToolResponseType.Exit,
-            message: "Files searched successfully",
+            message: 'Files searched successfully',
         };
     },
     listFiles: async (options, args) => {
-        const path = getString(args, "path");
-        const depth = getInt(args, "depth", 5);
+        const path = getString(args, 'path');
+        const depth = getInt(args, 'depth', 5);
         const isPathAllowed = (await options.files?.isPathAllowed?.(path)) ?? true;
         if (!isPathAllowed) {
             return {
@@ -210,7 +210,7 @@ ${stderrText}
             message: `<list_files_path>${path}</list_files_path>
 <list_files_depth>${depth}</list_files_depth>
 <list_files_files>
-${files.join("\n")}
+${files.join('\n')}
 </list_files_files>
 <list_files_truncated>${limitReached}</list_files_truncated>`,
         };
@@ -219,20 +219,20 @@ ${files.join("\n")}
         // TODO
         return {
             type: ToolResponseType.Exit,
-            message: "Code definition names listed successfully",
+            message: 'Code definition names listed successfully',
         };
     },
     askFollowupQuestion: async (options, args) => {
         // TODO
         return {
             type: ToolResponseType.Exit,
-            message: "Followup question asked successfully",
+            message: 'Followup question asked successfully',
         };
     },
     attemptCompletion: async (options, args) => {
         return {
             type: ToolResponseType.Exit,
-            message: "<completed>true</completed>",
+            message: '<completed>true</completed>',
         };
     },
 };
@@ -251,4 +251,4 @@ export const getToolHandler = (options) => {
         return handler(options, args);
     };
 };
-//# sourceMappingURL=handlers.js.map
+//# sourceMappingURL=handlers.js.map

package/dist/index.js

@@ -1,19 +1,19 @@
-import { Command } from "commander";
-import { loadConfig } from "./config.js";
-import { runTask } from "./runTask.js";
-const { version } = await import("../package.json");
+import { Command } from 'commander';
+import { loadConfig } from './config';
+import { runTask } from './runTask';
+const { version } = await import('../package.json');
 const program = new Command();
-program.name("polka").description("Polka Codes CLI").version(version);
+program.name('polka').description('Polka Codes CLI').version(version);
 // Main command for executing tasks
 program
-    .argument("[task]", "The task to execute")
-    .option("-c --config <path>", "Path to config file")
-    .option("--api-provider <provider>", "API provider")
-    .option("--model-id <model>", "Model ID")
-    .option("--api-key <key>", "API key")
+    .argument('[task]', 'The task to execute')
+    .option('-c --config <path>', 'Path to config file')
+    .option('--api-provider <provider>', 'API provider')
+    .option('--model-id <model>', 'Model ID')
+    .option('--api-key <key>', 'API key')
     .action(async (taskArg, options) => {
     const config = loadConfig(options.config ?? options.c) ?? {};
-    const provider = options.apiProvider || process.env.POLKA_API_PROVIDER || config.provider || "anthropic";
+    const provider = options.apiProvider || process.env.POLKA_API_PROVIDER || config.provider || 'anthropic';
     const model = options.modelId || process.env.POLKA_MODEL_ID || config.model;
     const apiKey = options.apiKey || process.env.POLKA_API_KEY || config.apiKey;
     await runTask({
@@ -25,4 +25,4 @@ program
     });
 });
 program.parse();
-//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/runTask.d.ts

@@ -1,4 +1,4 @@
-import type { Config } from "./config.js";
+import type { Config } from './config';
 export type RunTaskOptions = {
     task: string;
     provider: string;
@@ -7,4 +7,4 @@ export type RunTaskOptions = {
     config: Config;
 };
 export declare const runTask: (options: RunTaskOptions) => Promise<void>;
-//# sourceMappingURL=runTask.d.ts.map
+//# sourceMappingURL=runTask.d.ts.map

package/dist/runTask.js

@@ -1,10 +1,10 @@
-import os from "node:os";
-import { CoderAgent, createService, createServiceLogger } from "@polka-codes/core";
-import { getToolHandler } from "./handlers.js";
-import { listFiles } from "./utils/listFiles.js";
-const logger = createServiceLogger("cli/runTask");
+import os from 'node:os';
+import { CoderAgent, createService, createServiceLogger } from '@polka-codes/core';
+import { getToolHandler } from './handlers';
+import { listFiles } from './utils/listFiles';
+const logger = createServiceLogger('cli/runTask');
 export const runTask = async (options) => {
-    logger.debug(options, "Running task");
+    logger.debug(options, 'Running task');
     const { task, provider, model, apiKey, config } = options;
     // Create service with the configured options
     const service = createService(provider, {
@@ -12,7 +12,7 @@ export const runTask = async (options) => {
         modelId: model,
     });
     let rules = config.rules;
-    if (typeof rules === "string") {
+    if (typeof rules === 'string') {
         rules = [rules];
     }
     const agent = new CoderAgent({
@@ -32,13 +32,13 @@ export const runTask = async (options) => {
                 },
                 commandStarted(command, child) {
                     console.log(`$ >>>> $ ${command}`);
-                    child.stdout.on("data", (data) => {
+                    child.stdout.on('data', (data) => {
                         process.stdout.write(data.toString());
                     });
-                    child.stderr.on("data", (data) => {
+                    child.stderr.on('data', (data) => {
                         process.stderr.write(data.toString());
                     });
-                    child.on("close", (code) => {
+                    child.on('close', (code) => {
                         console.log(`$ <<<< $ Command exited with code: ${code}`);
                     });
                 },
@@ -48,7 +48,7 @@ export const runTask = async (options) => {
     const cwd = process.cwd();
     const [fileList, limited] = await listFiles(cwd, true, 100, cwd);
     const fileContext = `<files>
-${fileList.join("\n")}${limited ? "\n<files_truncated>true</files_truncated>" : ""}
+${fileList.join('\n')}${limited ? '\n<files_truncated>true</files_truncated>' : ''}
 </files>`;
     const usage = {
         inputTokens: 0,
@@ -62,29 +62,29 @@ ${fileList.join("\n")}${limited ? "\n<files_truncated>true</files_truncated>" :
         context: `<now_date>${new Date().toISOString()}</now_date>${fileContext}`,
         maxIterations: 10,
         callback: (event) => {
-            if (event.kind === "usage") {
+            if (event.kind === 'usage') {
                 usage.inputTokens += event.info.inputTokens;
                 usage.outputTokens += event.info.outputTokens;
                 usage.cacheWriteTokens += event.info.cacheWriteTokens ?? 0;
                 usage.cacheReadTokens += event.info.cacheReadTokens ?? 0;
                 usage.totalCost += event.info.totalCost ?? 0;
             }
-            if (event.kind === "start_request") {
-                console.log(">>>>");
+            if (event.kind === 'start_request') {
+                console.log('>>>>');
                 const { userContent } = event;
                 if (userContent) {
                     for (const content of userContent) {
-                        console.log(content.type === "text" ? content.text : "");
+                        console.log(content.type === 'text' ? content.text : '');
                     }
                 }
-                console.log("====");
+                console.log('====');
             }
             if (event.newText) {
                 process.stdout.write(event.newText);
             }
-            if (event.kind === "end_request") {
-                process.stdout.write("\n");
-                console.log("<<<<");
+            if (event.kind === 'end_request') {
+                process.stdout.write('\n');
+                console.log('<<<<');
             }
         },
     });
@@ -95,13 +95,13 @@ ${fileList.join("\n")}${limited ? "\n<files_truncated>true</files_truncated>" :
         const outputCost = (modelInfo.outputPrice ?? 0) * usage.outputTokens;
         const cacheReadCost = (modelInfo.cacheReadsPrice ?? 0) * usage.cacheReadTokens;
         const cacheWriteCost = (modelInfo.cacheWritesPrice ?? 0) * usage.cacheWriteTokens;
-        usage.totalCost = (inputCost + outputCost + cacheReadCost + cacheWriteCost) / 1000000;
+        usage.totalCost = (inputCost + outputCost + cacheReadCost + cacheWriteCost) / 1_000_000;
     }
-    console.log("Usages:");
+    console.log('Usages:');
     console.log(`Input tokens: ${usage.inputTokens}`);
     console.log(`Output tokens: ${usage.outputTokens}`);
     console.log(`Cache read tokens: ${usage.cacheReadTokens}`);
     console.log(`Cache write tokens: ${usage.cacheWriteTokens}`);
     console.log(`Total cost: ${usage.totalCost}`);
 };
-//# sourceMappingURL=runTask.js.map
+//# sourceMappingURL=runTask.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/cli",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "license": "AGPL-3.0",
   "type": "module",
   "bin": {