@polka-codes/core 0.7.15 → 0.7.17

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;
@@ -11,9 +12,11 @@ declare abstract class AgentBase {
     start(prompt: string): Promise<ExitReason>;
     step(promp: string, messages?: MessageParam[]): Promise<AssistantMessageContent[]>;
     handleStepResponse(response: AssistantMessageContent[], messages?: MessageParam[]): Promise<{
-        replay: string;
+        type: "reply";
+        message: string;
     } | {
-        exit: ExitReason;
+        type: "exit";
+        reason: ExitReason;
     }>;
     continueTask(userMessage: string): Promise<ExitReason>;
     protected abstract onBeforeInvokeTool(name: string, args: Record<string, string>): Promise<ToolResponse | undefined>;
@@ -390,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[];

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

@@ -2257,11 +2257,11 @@ ${agents}`;
       }
       const response = await this.#request(nextRequest);
       const resp = await this.#handleResponse(response);
-      if ("exit" in resp) {
-        this.#callback({ kind: "EndTask" /* EndTask */, agent: this, exitReason: resp.exit });
-        return resp.exit;
+      if (resp.type === "exit") {
+        this.#callback({ kind: "EndTask" /* EndTask */, agent: this, exitReason: resp.reason });
+        return resp.reason;
       }
-      nextRequest = resp.replay;
+      nextRequest = resp.message;
     }
   }
   async continueTask(userMessage) {
@@ -2333,7 +2333,7 @@ ${agents}`;
               if (toolReponses.length > 0) {
                 break outer;
               }
-              return { exit: toolResp };
+              return { type: "exit", reason: toolResp };
             case "Invalid" /* Invalid */:
               await this.#callback({ kind: "ToolInvalid" /* ToolInvalid */, agent: this, tool: content.name });
               toolReponses.push({ tool: content.name, response: toolResp.message });
@@ -2344,7 +2344,7 @@ ${agents}`;
               break outer;
             case "Interrupted" /* Interrupted */:
               await this.#callback({ kind: "ToolInterrupted" /* ToolInterrupted */, agent: this, tool: content.name });
-              return { exit: toolResp };
+              return { type: "exit", reason: toolResp };
             case "HandOver" /* HandOver */:
               if (toolReponses.length > 0) {
                 break outer;
@@ -2358,7 +2358,7 @@ ${agents}`;
                 context: toolResp.context,
                 files: toolResp.files
               });
-              return { exit: toolResp };
+              return { type: "exit", reason: toolResp };
             case "Delegate" /* Delegate */:
               if (toolReponses.length > 0) {
                 continue;
@@ -2372,17 +2372,17 @@ ${agents}`;
                 context: toolResp.context,
                 files: toolResp.files
               });
-              return { exit: toolResp };
+              return { type: "exit", reason: toolResp };
           }
           break;
         }
       }
     }
     if (toolReponses.length === 0) {
-      return { replay: responsePrompts.requireUseTool };
+      return { type: "reply", message: responsePrompts.requireUseTool };
     }
     const finalResp = toolReponses.map(({ tool, response: response2 }) => responsePrompts.toolResults(tool, response2)).join("\n\n");
-    return { replay: finalResp };
+    return { type: "reply", message: finalResp };
   }
   async #invokeTool(name, args) {
     try {
@@ -3407,6 +3407,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 +3492,7 @@ export {
   capabilities,
   codeFixerAgentInfo,
   coderAgentInfo,
+  configSchema,
   createNewProject,
   createService,
   customInstructions,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.7.15",
+  "version": "0.7.17",
   "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"
   }
 }