@polka-codes/core 0.7.12 → 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(): {
@@ -2392,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.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) {

package/package.json

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