@polka-codes/cli 0.4.6 → 0.4.7

package/dist/index.js

@@ -24629,7 +24629,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.4.6";
+var version = "0.4.7";
 
 // ../../node_modules/@anthropic-ai/sdk/version.mjs
 var VERSION = "0.36.2";
@@ -35149,9 +35149,9 @@ async function searchFiles(path2, regex, filePattern, cwd, excludeFiles) {
 }
 
 // src/provider.ts
-var getProvider = (options) => {
+var getProvider = (agentName, config, options) => {
   const ig = import_ignore2.default().add(options.excludeFiles ?? []);
-  return {
+  const provider2 = {
     readFile: async (path2) => {
       if (ig.ignores(path2)) {
         throw new Error(`Not allow to access file ${path2}`);
@@ -35215,8 +35215,31 @@ var getProvider = (options) => {
           reject(err);
         });
       });
+    },
+    attemptCompletion: async (result) => {
+      const cmd = (config.hooks?.agents?.[agentName]?.beforeCompletion ?? config.hooks?.agents?.default?.beforeCompletion)?.trim();
+      if (cmd) {
+        try {
+          const { exitCode, stdout, stderr } = await provider2.executeCommand(cmd, false);
+          if (exitCode !== 0) {
+            return `
+<command>${cmd}</command>
+<command_exit_code>${exitCode}</command_exit_code>
+<command_stdout>
+${stdout}
+</command_stdout>
+<command_stderr>
+${stderr}
+</command_stderr>`;
+          }
+        } catch (error) {
+          console.warn(`Failed to execute hook: ${error}`);
+        }
+      }
+      return;
     }
   };
+  return provider2;
 };
 
 // src/Runner.ts
@@ -35240,7 +35263,7 @@ class Runner {
     if (typeof rules2 === "string") {
       rules2 = [rules2];
     }
-    const provider2 = getProvider({
+    const providerOptions = {
       command: {
         onStarted(command) {
           console.log(`$ >>>> $ ${command}`);
@@ -35259,19 +35282,20 @@ class Runner {
         }
       },
       excludeFiles: options.config.excludeFiles
-    });
+    };
     const platform = os.platform();
     const agents = [coderAgentInfo, architectAgentInfo];
     this.#multiAgent = new MultiAgent({
       createAgent: async (name) => {
-        switch (name.trim().toLowerCase()) {
+        const agentName = name.trim().toLowerCase();
+        switch (agentName) {
           case coderAgentInfo.name.toLowerCase():
             return new CoderAgent({
               ai: service,
               os: platform,
               customInstructions: rules2,
               scripts: options.config.scripts,
-              provider: provider2,
+              provider: getProvider("coder", options.config, providerOptions),
               interactive: options.interactive,
               agents
             });
@@ -35281,7 +35305,7 @@ class Runner {
               os: platform,
               customInstructions: rules2,
               scripts: options.config.scripts,
-              provider: provider2,
+              provider: getProvider("architect", options.config, providerOptions),
               interactive: options.interactive,
               agents
             });
@@ -39420,6 +39444,8 @@ var z2 = /* @__PURE__ */ Object.freeze({
 });
 
 // src/config.ts
+var agentNames = z2.enum(["coder", "architect"]);
+var agentNameOrDefault = z2.union([agentNames, z2.literal("default")]);
 var providerModelSchema = z2.object({
   provider: z2.string().optional(),
   model: z2.string().optional()
@@ -39438,13 +39464,14 @@ var configSchema = z2.object({
   defaultProvider: z2.string().optional(),
   defaultModel: z2.string().optional(),
   maxIterations: z2.number().int().positive().optional(),
+  hooks: z2.object({
+    agents: z2.record(agentNameOrDefault, z2.object({ beforeCompletion: z2.string().optional() })).optional()
+  }).optional(),
   scripts: z2.record(z2.string(), z2.string().or(z2.object({
     command: z2.string(),
     description: z2.string()
   }))).optional(),
-  agents: z2.object({
-    default: agentSchema.optional()
-  }).catchall(agentSchema).optional(),
+  agents: z2.record(agentNameOrDefault, agentSchema).optional(),
   commands: z2.object({
     default: providerModelSchema.optional()
   }).catchall(providerModelSchema).optional(),

package/package.json

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