@polka-codes/core 0.7.16 → 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;
@@ -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[];

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

@@ -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.16",
+  "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"
   }
 }