@polka-codes/core 0.7.11 → 0.7.13

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

@@ -7,8 +7,16 @@ declare abstract class AgentBase {
     protected readonly config: Readonly<AgentBaseConfig>;
     protected readonly handlers: Record<string, FullToolInfo>;
     protected readonly messages: MessageParam[];
-    constructor(name: string, ai: AiServiceBase, config: AgentBaseConfig);
+    constructor(name: string, ai: AiServiceBase, config: AgentBaseConfig, messages?: MessageParam[]);
     start(prompt: string): Promise<ExitReason>;
+    step(promp: string): Promise<AssistantMessageContent[] | {
+        type: string;
+    }>;
+    handleStepResponse(response: AssistantMessageContent[]): Promise<{
+        replay: string;
+    } | {
+        exit: ExitReason;
+    }>;
     continueTask(userMessage: string): Promise<ExitReason>;
     protected abstract onBeforeInvokeTool(name: string, args: Record<string, string>): Promise<ToolResponse | undefined>;
     get model(): {
@@ -1422,6 +1430,7 @@ export { getAvailableTools as getAvailableTools_alias_1 }
 declare const getBoolean: <T extends string>(args: Partial<Record<T, string>>, name: T, defaultValue?: boolean) => boolean;
 export { getBoolean }
 export { getBoolean as getBoolean_alias_1 }
+export { getBoolean as getBoolean_alias_2 }
 
 declare type GetInput<T> = T extends AiToolDefinition<infer Input, any> ? Input : never;
 export { GetInput }
@@ -1431,6 +1440,7 @@ export { GetInput as GetInput_alias_2 }
 declare const getInt: <T extends string>(args: Partial<Record<T, string>>, name: T, defaultValue?: number) => number;
 export { getInt }
 export { getInt as getInt_alias_1 }
+export { getInt as getInt_alias_2 }
 
 declare type GetOutput<T> = T extends AiToolDefinition<any, infer Output> ? Output : never;
 export { GetOutput }
@@ -1440,10 +1450,12 @@ export { GetOutput as GetOutput_alias_2 }
 declare const getString: <T extends string>(args: Partial<Partial<Record<T, string>>>, name: T, defaultValue?: string) => string;
 export { getString }
 export { getString as getString_alias_1 }
+export { getString as getString_alias_2 }
 
 declare const getStringArray: <T extends string>(args: Partial<Record<T, string>>, name: T, defaultValue?: string[]) => string[];
 export { getStringArray }
 export { getStringArray as getStringArray_alias_1 }
+export { getStringArray as getStringArray_alias_2 }
 
 export declare const handler: ToolHandler<typeof toolInfo, InteractionProvider>;
 
@@ -1753,6 +1765,7 @@ export { PermissionLevel }
 export { PermissionLevel as PermissionLevel_alias_1 }
 
 declare const replaceInFile_2: (fileContent: string, diff: string) => Promise<string>;
+export { replaceInFile_2 as replaceInFileHelper }
 export { replaceInFile_2 as replaceInFile_alias_3 }
 export { replaceInFile_2 as replaceInFile_alias_4 }
 
@@ -2387,6 +2400,7 @@ export { ToolResponseError as ToolResponseError_alias_1 }
 declare type ToolResponseExit = {
     type: ToolResponseType.Exit;
     message: string;
+    object?: any;
 };
 export { ToolResponseExit }
 export { ToolResponseExit as ToolResponseExit_alias_1 }

package/dist/index.d.ts

@@ -1,3 +1,4 @@
+export { replaceInFileHelper } from './_tsup-dts-rollup.js';
 export { AiServiceProvider } from './_tsup-dts-rollup.js';
 export { defaultModels } from './_tsup-dts-rollup.js';
 export { createService } from './_tsup-dts-rollup.js';
@@ -104,3 +105,7 @@ 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 { getString } from './_tsup-dts-rollup.js';
+export { getStringArray } from './_tsup-dts-rollup.js';
+export { getBoolean } from './_tsup-dts-rollup.js';
+export { getInt } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -2185,9 +2185,10 @@ var AgentBase = class {
   ai;
   config;
   handlers;
-  messages = [];
-  constructor(name, ai, config) {
+  messages;
+  constructor(name, ai, config, messages = []) {
     this.ai = ai;
+    this.messages = messages;
     if (config.agents && config.agents.length > 0) {
       const agents = agentsPrompt(config.agents, name);
       config.systemPrompt += `
@@ -2207,6 +2208,19 @@ ${agents}`;
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
     return await this.#processLoop(prompt5);
   }
+  async step(promp) {
+    if (this.messages.length === 0) {
+      this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
+    }
+    if (this.ai.usageMeter.isLimitExceeded().result) {
+      this.#callback({ kind: "UsageExceeded" /* UsageExceeded */, agent: this });
+      return { type: "UsageExceeded" };
+    }
+    return await this.#request(promp);
+  }
+  async handleStepResponse(response) {
+    return this.#handleResponse(response);
+  }
   async #processLoop(userMessage) {
     let nextRequest = userMessage;
     while (true) {
@@ -3407,6 +3421,10 @@ export {
   generateGithubPullRequestDetails,
   generateProjectConfig,
   getAvailableTools,
+  getBoolean,
+  getInt,
+  getString,
+  getStringArray,
   handOver_default as handOver,
   listCodeDefinitionNames_default as listCodeDefinitionNames,
   listFiles_default as listFiles,
@@ -3418,6 +3436,7 @@ export {
   removeFile_default as removeFile,
   renameFile_default as renameFile,
   replaceInFile_default as replaceInFile,
+  replaceInFile as replaceInFileHelper,
   responsePrompts,
   searchFiles_default as searchFiles,
   systemInformation,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.7.11",
+  "version": "0.7.13",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",