@polka-codes/core 0.7.16 → 0.7.18

package/dist/_tsup-dts-rollup.d.ts

@@ -1,5 +1,6 @@
 import type { Anthropic } from '@anthropic-ai/sdk';
 import type OpenAI from 'openai';
+import { z } from 'zod';
 
 declare abstract class AgentBase {
     #private;
@@ -392,6 +393,293 @@ export { CommandProvider }
 export { CommandProvider as CommandProvider_alias_1 }
 export { CommandProvider as CommandProvider_alias_2 }
 
+declare type Config = z.infer<typeof configSchema>;
+export { Config }
+export { Config as Config_alias_1 }
+
+declare const configSchema: z.ZodObject<{
+    agent: z.ZodOptional<z.ZodString>;
+    prices: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodObject<{
+        inputPrice: z.ZodOptional<z.ZodNumber>;
+        outputPrice: z.ZodOptional<z.ZodNumber>;
+        cacheWritesPrice: z.ZodOptional<z.ZodNumber>;
+        cacheReadsPrice: z.ZodOptional<z.ZodNumber>;
+    }, "strip", z.ZodTypeAny, {
+        inputPrice?: number | undefined;
+        outputPrice?: number | undefined;
+        cacheWritesPrice?: number | undefined;
+        cacheReadsPrice?: number | undefined;
+    }, {
+        inputPrice?: number | undefined;
+        outputPrice?: number | undefined;
+        cacheWritesPrice?: number | undefined;
+        cacheReadsPrice?: number | undefined;
+    }>>>>;
+    providers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        apiKey: z.ZodOptional<z.ZodString>;
+        defaultModel: z.ZodOptional<z.ZodString>;
+        defaultParameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, "strip", z.ZodTypeAny, {
+        apiKey?: string | undefined;
+        defaultModel?: string | undefined;
+        defaultParameters?: Record<string, any> | undefined;
+    }, {
+        apiKey?: string | undefined;
+        defaultModel?: string | undefined;
+        defaultParameters?: Record<string, any> | undefined;
+    }>>>;
+    defaultProvider: z.ZodOptional<z.ZodString>;
+    defaultModel: z.ZodOptional<z.ZodString>;
+    defaultParameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    maxMessageCount: z.ZodOptional<z.ZodNumber>;
+    budget: z.ZodOptional<z.ZodNumber>;
+    scripts: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodString, z.ZodObject<{
+        command: z.ZodString;
+        description: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        command: string;
+        description: string;
+    }, {
+        command: string;
+        description: string;
+    }>]>>>;
+    agents: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<z.objectUtil.extendShape<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, {
+        initialContext: z.ZodOptional<z.ZodObject<{
+            maxFileCount: z.ZodOptional<z.ZodNumber>;
+            excludes: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+        }, "strip", z.ZodTypeAny, {
+            maxFileCount?: number | undefined;
+            excludes?: string[] | undefined;
+        }, {
+            maxFileCount?: number | undefined;
+            excludes?: string[] | undefined;
+        }>>;
+    }>, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+        initialContext?: {
+            maxFileCount?: number | undefined;
+            excludes?: string[] | undefined;
+        } | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+        initialContext?: {
+            maxFileCount?: number | undefined;
+            excludes?: string[] | undefined;
+        } | undefined;
+    }>>>;
+    commands: z.ZodOptional<z.ZodObject<{
+        default: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodOptional<z.ZodString>;
+            model: z.ZodOptional<z.ZodString>;
+            parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+        }, "strip", z.ZodTypeAny, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }>>;
+    }, "strip", z.ZodObject<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }>, z.objectOutputType<{
+        default: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodOptional<z.ZodString>;
+            model: z.ZodOptional<z.ZodString>;
+            parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+        }, "strip", z.ZodTypeAny, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }>>;
+    }, z.ZodObject<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }>, "strip">, z.objectInputType<{
+        default: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodOptional<z.ZodString>;
+            model: z.ZodOptional<z.ZodString>;
+            parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+        }, "strip", z.ZodTypeAny, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }>>;
+    }, z.ZodObject<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }>, "strip">>>;
+    rules: z.ZodOptional<z.ZodUnion<[z.ZodOptional<z.ZodArray<z.ZodString, "many">>, z.ZodString]>>;
+    excludeFiles: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+}, "strict", z.ZodTypeAny, {
+    agent?: string | undefined;
+    prices?: Record<string, Record<string, {
+        inputPrice?: number | undefined;
+        outputPrice?: number | undefined;
+        cacheWritesPrice?: number | undefined;
+        cacheReadsPrice?: number | undefined;
+    }>> | undefined;
+    defaultModel?: string | undefined;
+    defaultParameters?: Record<string, any> | undefined;
+    providers?: Record<string, {
+        apiKey?: string | undefined;
+        defaultModel?: string | undefined;
+        defaultParameters?: Record<string, any> | undefined;
+    }> | undefined;
+    defaultProvider?: string | undefined;
+    maxMessageCount?: number | undefined;
+    budget?: number | undefined;
+    scripts?: Record<string, string | {
+        command: string;
+        description: string;
+    }> | undefined;
+    agents?: Record<string, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+        initialContext?: {
+            maxFileCount?: number | undefined;
+            excludes?: string[] | undefined;
+        } | undefined;
+    }> | undefined;
+    commands?: z.objectOutputType<{
+        default: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodOptional<z.ZodString>;
+            model: z.ZodOptional<z.ZodString>;
+            parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+        }, "strip", z.ZodTypeAny, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }>>;
+    }, z.ZodObject<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }>, "strip"> | undefined;
+    rules?: string | string[] | undefined;
+    excludeFiles?: string[] | undefined;
+}, {
+    agent?: string | undefined;
+    prices?: Record<string, Record<string, {
+        inputPrice?: number | undefined;
+        outputPrice?: number | undefined;
+        cacheWritesPrice?: number | undefined;
+        cacheReadsPrice?: number | undefined;
+    }>> | undefined;
+    defaultModel?: string | undefined;
+    defaultParameters?: Record<string, any> | undefined;
+    providers?: Record<string, {
+        apiKey?: string | undefined;
+        defaultModel?: string | undefined;
+        defaultParameters?: Record<string, any> | undefined;
+    }> | undefined;
+    defaultProvider?: string | undefined;
+    maxMessageCount?: number | undefined;
+    budget?: number | undefined;
+    scripts?: Record<string, string | {
+        command: string;
+        description: string;
+    }> | undefined;
+    agents?: Record<string, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+        initialContext?: {
+            maxFileCount?: number | undefined;
+            excludes?: string[] | undefined;
+        } | undefined;
+    }> | undefined;
+    commands?: z.objectInputType<{
+        default: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodOptional<z.ZodString>;
+            model: z.ZodOptional<z.ZodString>;
+            parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+        }, "strip", z.ZodTypeAny, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }, {
+            provider?: string | undefined;
+            model?: string | undefined;
+            parameters?: Record<string, any> | undefined;
+        }>>;
+    }, z.ZodObject<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        parameters?: Record<string, any> | undefined;
+    }>, "strip"> | undefined;
+    rules?: string | string[] | undefined;
+    excludeFiles?: string[] | undefined;
+}>;
+export { configSchema }
+export { configSchema as configSchema_alias_1 }
+
 export declare function convertToAnthropicMessage(completion: OpenAI.Chat.Completions.ChatCompletion): Anthropic.Messages.Message;
 
 export declare function convertToOpenAiMessages(anthropicMessages: Anthropic.Messages.MessageParam[]): OpenAI.Chat.ChatCompletionMessageParam[];
@@ -1684,7 +1972,7 @@ export { MultiAgent as MultiAgent_alias_2 }
 
 declare type MultiAgentConfig = {
     createAgent: (name: string) => Promise<AgentBase>;
-    getPrompt?: (name: string, task: string, context?: string, files?: string[]) => Promise<string>;
+    getPrompt?: (name: string, task: string, context?: string, files?: string[], originalTask?: string) => Promise<string>;
 };
 export { MultiAgentConfig }
 export { MultiAgentConfig as MultiAgentConfig_alias_1 }
@@ -1944,6 +2232,7 @@ declare interface TaskEventToolHandOverDelegate extends TaskEventBase {
     task: string;
     context?: string;
     files?: string[];
+    originalTask?: string;
 }
 export { TaskEventToolHandOverDelegate }
 export { TaskEventToolHandOverDelegate as TaskEventToolHandOverDelegate_alias_1 }
@@ -2401,6 +2690,7 @@ declare type ToolResponseDelegate = {
     task: string;
     context?: string;
     files?: string[];
+    originalTask?: string;
 };
 export { ToolResponseDelegate }
 export { ToolResponseDelegate as ToolResponseDelegate_alias_1 }
@@ -2427,6 +2717,7 @@ declare type ToolResponseHandOver = {
     task: string;
     context?: string;
     files?: string[];
+    originalTask?: string;
 };
 export { ToolResponseHandOver }
 export { ToolResponseHandOver as ToolResponseHandOver_alias_1 }

package/dist/index.d.ts

@@ -109,6 +109,8 @@ export { ToolResponseDelegate } from './_tsup-dts-rollup.js';
 export { ToolResponse } from './_tsup-dts-rollup.js';
 export { ToolHandler } from './_tsup-dts-rollup.js';
 export { getAvailableTools_alias_1 as getAvailableTools } from './_tsup-dts-rollup.js';
+export { configSchema_alias_1 as configSchema } from './_tsup-dts-rollup.js';
+export { Config_alias_1 as Config } from './_tsup-dts-rollup.js';
 export { getString } from './_tsup-dts-rollup.js';
 export { getStringArray } from './_tsup-dts-rollup.js';
 export { getBoolean } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -1219,6 +1219,7 @@ var handler3 = async (_provider, args) => {
     task,
     context,
     files
+    // originalTask will be set by AgentBase
   };
 };
 var isAvailable3 = (_provider) => {
@@ -1814,6 +1815,7 @@ var handler11 = async (_provider, args) => {
     task,
     context,
     files
+    // originalTask will be set by AgentBase
   };
 };
 var isAvailable11 = (_provider) => {
@@ -2208,6 +2210,7 @@ var AgentBase = class {
   config;
   handlers;
   #messages;
+  #originalTask;
   constructor(name, ai, config, messages = []) {
     this.ai = ai;
     this.#messages = messages;
@@ -2230,6 +2233,7 @@ ${agents}`;
     await this.config.callback?.(event);
   }
   async start(prompt5) {
+    this.#originalTask = prompt5;
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
     return await this.#processLoop(prompt5);
   }
@@ -2345,34 +2349,46 @@ ${agents}`;
             case "Interrupted" /* Interrupted */:
               await this.#callback({ kind: "ToolInterrupted" /* ToolInterrupted */, agent: this, tool: content.name });
               return { type: "exit", reason: toolResp };
-            case "HandOver" /* HandOver */:
+            case "HandOver" /* HandOver */: {
               if (toolReponses.length > 0) {
                 break outer;
               }
+              const handOverResp = {
+                ...toolResp,
+                originalTask: this.#originalTask
+              };
               await this.#callback({
                 kind: "ToolHandOver" /* ToolHandOver */,
                 agent: this,
                 tool: content.name,
-                agentName: toolResp.agentName,
-                task: toolResp.task,
-                context: toolResp.context,
-                files: toolResp.files
+                agentName: handOverResp.agentName,
+                task: handOverResp.task,
+                context: handOverResp.context,
+                files: handOverResp.files,
+                originalTask: handOverResp.originalTask
               });
-              return { type: "exit", reason: toolResp };
-            case "Delegate" /* Delegate */:
+              return { type: "exit", reason: handOverResp };
+            }
+            case "Delegate" /* Delegate */: {
               if (toolReponses.length > 0) {
                 continue;
               }
+              const delegateResp = {
+                ...toolResp,
+                originalTask: this.#originalTask
+              };
               await this.#callback({
                 kind: "ToolDelegate" /* ToolDelegate */,
                 agent: this,
                 tool: content.name,
-                agentName: toolResp.agentName,
-                task: toolResp.task,
-                context: toolResp.context,
-                files: toolResp.files
+                agentName: delegateResp.agentName,
+                task: delegateResp.task,
+                context: delegateResp.context,
+                files: delegateResp.files,
+                originalTask: delegateResp.originalTask
               });
-              return { type: "exit", reason: toolResp };
+              return { type: "exit", reason: delegateResp };
+            }
           }
           break;
         }
@@ -2974,11 +2990,23 @@ var MultiAgent = class {
     switch (exitReason.type) {
       case "HandOver" /* HandOver */: {
         this.#agents.pop();
-        const prompt5 = await this.#config.getPrompt?.(exitReason.agentName, exitReason.task, exitReason.context, exitReason.files) ?? exitReason.task;
+        const prompt5 = await this.#config.getPrompt?.(
+          exitReason.agentName,
+          exitReason.task,
+          exitReason.context,
+          exitReason.files,
+          exitReason.originalTask
+        ) ?? exitReason.task;
         return await this.#startTask(exitReason.agentName, prompt5);
       }
       case "Delegate" /* Delegate */: {
-        const prompt5 = await this.#config.getPrompt?.(exitReason.agentName, exitReason.task, exitReason.context, exitReason.files) ?? exitReason.task;
+        const prompt5 = await this.#config.getPrompt?.(
+          exitReason.agentName,
+          exitReason.task,
+          exitReason.context,
+          exitReason.files,
+          exitReason.originalTask
+        ) ?? exitReason.task;
         const delegateResult = await this.#startTask(exitReason.agentName, prompt5);
         switch (delegateResult.type) {
           case "HandOver" /* HandOver */:
@@ -3407,6 +3435,65 @@ var generateGitCommitMessage = makeTool(generateGitCommitMessage_default);
 var generateGithubPullRequestDetails = makeTool(generateGithubPullRequestDetails_default);
 var generateProjectConfig = makeAgentTool(generateProjectConfig_default);
 var createNewProject = makeAgentTool(createNewProject_default);
+
+// src/config.ts
+import { z } from "zod";
+var providerModelSchema = z.object({
+  provider: z.string().optional(),
+  model: z.string().optional(),
+  parameters: z.record(z.string(), z.any()).optional()
+});
+var agentSchema = providerModelSchema.extend({
+  initialContext: z.object({
+    maxFileCount: z.number().int().positive().optional(),
+    excludes: z.array(z.string()).optional()
+  }).optional()
+});
+var configSchema = z.object({
+  agent: z.string().optional(),
+  prices: z.record(
+    z.string(),
+    // provider
+    z.record(
+      z.string(),
+      // model
+      z.object({
+        inputPrice: z.number().optional(),
+        outputPrice: z.number().optional(),
+        cacheWritesPrice: z.number().optional(),
+        cacheReadsPrice: z.number().optional()
+      })
+    )
+  ).optional(),
+  providers: z.record(
+    z.string(),
+    z.object({
+      apiKey: z.string().optional(),
+      defaultModel: z.string().optional(),
+      defaultParameters: z.record(z.string(), z.any()).optional()
+    })
+  ).optional(),
+  defaultProvider: z.string().optional(),
+  defaultModel: z.string().optional(),
+  defaultParameters: z.record(z.string(), z.any()).optional(),
+  maxMessageCount: z.number().int().positive().optional(),
+  budget: z.number().positive().optional(),
+  scripts: z.record(
+    z.string(),
+    z.string().or(
+      z.object({
+        command: z.string(),
+        description: z.string()
+      })
+    )
+  ).optional(),
+  agents: z.record(z.string(), agentSchema).optional(),
+  commands: z.object({
+    default: providerModelSchema.optional()
+  }).catchall(providerModelSchema).optional(),
+  rules: z.array(z.string()).optional().or(z.string()).optional(),
+  excludeFiles: z.array(z.string()).optional()
+}).strict();
 export {
   AgentBase,
   AiServiceBase,
@@ -3433,6 +3520,7 @@ export {
   capabilities,
   codeFixerAgentInfo,
   coderAgentInfo,
+  configSchema,
   createNewProject,
   createService,
   customInstructions,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.7.16",
+  "version": "0.7.18",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -17,6 +17,7 @@
   "dependencies": {
     "@anthropic-ai/sdk": "0.39.0",
     "lodash": "^4.17.21",
-    "openai": "^4.80.0"
+    "openai": "^4.80.0",
+    "zod": "^3.24.2"
   }
 }