@polka-codes/core 0.7.5 → 0.7.6

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

@@ -8,10 +8,7 @@ declare abstract class AgentBase {
     protected readonly handlers: Record<string, FullToolInfo>;
     protected readonly messages: MessageParam[];
     constructor(name: string, ai: AiServiceBase, config: AgentBaseConfig);
-    startTask({ task, context }: {
-        task: string;
-        context?: string;
-    }): Promise<ExitReason>;
+    start(prompt: string): Promise<ExitReason>;
     continueTask(userMessage: string): Promise<ExitReason>;
     protected abstract onBeforeInvokeTool(name: string, args: Record<string, string>): Promise<ToolResponse | undefined>;
     get model(): {
@@ -241,9 +238,15 @@ export declare class AnthropicService extends AiServiceBase {
     sendImpl(systemPrompt: string, messages: MessageParam[]): ApiStream;
 }
 
-export declare type ApiStream = AsyncGenerator<ApiStreamChunk>;
+declare type ApiStream = AsyncGenerator<ApiStreamChunk>;
+export { ApiStream }
+export { ApiStream as ApiStream_alias_1 }
+export { ApiStream as ApiStream_alias_2 }
 
-export declare type ApiStreamChunk = ApiStreamTextChunk | ApiStreamUsageChunk | ApiStreamReasoningTextChunk;
+declare type ApiStreamChunk = ApiStreamTextChunk | ApiStreamUsageChunk | ApiStreamReasoningTextChunk;
+export { ApiStreamChunk }
+export { ApiStreamChunk as ApiStreamChunk_alias_1 }
+export { ApiStreamChunk as ApiStreamChunk_alias_2 }
 
 export declare interface ApiStreamReasoningTextChunk {
     type: 'reasoning';
@@ -1637,7 +1640,6 @@ declare class MultiAgent {
     startTask(options: {
         agentName: string;
         task: string;
-        context?: string;
     }): Promise<ExitReason>;
     continueTask(userMessage: string): Promise<ExitReason>;
     get hasActiveAgent(): boolean;
@@ -1648,7 +1650,7 @@ export { MultiAgent as MultiAgent_alias_2 }
 
 declare type MultiAgentConfig = {
     createAgent: (name: string) => Promise<AgentBase>;
-    getContext?: (name: string, context?: string, files?: string[]) => Promise<string>;
+    getPrompt?: (name: string, task: string, context?: string, files?: string[]) => Promise<string>;
 };
 export { MultiAgentConfig }
 export { MultiAgentConfig as MultiAgentConfig_alias_1 }

package/dist/index.d.ts

@@ -7,6 +7,8 @@ export { AiServiceOptions } from './_tsup-dts-rollup.js';
 export { AiServiceBase } from './_tsup-dts-rollup.js';
 export { ModelInfo } from './_tsup-dts-rollup.js';
 export { ApiUsage } from './_tsup-dts-rollup.js';
+export { ApiStream } from './_tsup-dts-rollup.js';
+export { ApiStreamChunk } from './_tsup-dts-rollup.js';
 export { allAgents } from './_tsup-dts-rollup.js';
 export { AgentNameType } from './_tsup-dts-rollup.js';
 export { TaskEventKind } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -2209,14 +2209,9 @@ ${agents}`;
   async #callback(event) {
     await this.config.callback?.(event);
   }
-  async startTask({ task, context }) {
-    let text = `<task>${task}</task>`;
-    if (context) {
-      text += `
-<context>${context}</context>`;
-    }
+  async start(prompt5) {
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
-    return await this.#processLoop(text);
+    return await this.#processLoop(prompt5);
   }
   async #processLoop(userMessage) {
     let nextRequest = userMessage;
@@ -2944,12 +2939,12 @@ var MultiAgent = class {
     switch (exitReason.type) {
       case "HandOver" /* HandOver */: {
         this.#agents.pop();
-        const newContext = await this.#config.getContext?.(exitReason.agentName, exitReason.context, exitReason.files);
-        return await this.#startTask(exitReason.agentName, exitReason.task, newContext);
+        const prompt5 = await this.#config.getPrompt?.(exitReason.agentName, exitReason.task, exitReason.context, exitReason.files) ?? exitReason.task;
+        return await this.#startTask(exitReason.agentName, prompt5);
       }
       case "Delegate" /* Delegate */: {
-        const newContext = await this.#config.getContext?.(exitReason.agentName, exitReason.context, exitReason.files);
-        const delegateResult = await this.#startTask(exitReason.agentName, exitReason.task, newContext);
+        const prompt5 = await this.#config.getPrompt?.(exitReason.agentName, exitReason.task, exitReason.context, exitReason.files) ?? exitReason.task;
+        const delegateResult = await this.#startTask(exitReason.agentName, prompt5);
         switch (delegateResult.type) {
           case "HandOver" /* HandOver */:
           case "Delegate" /* Delegate */:
@@ -2970,20 +2965,17 @@ var MultiAgent = class {
         return exitReason;
     }
   }
-  async #startTask(agentName, task, context) {
+  async #startTask(agentName, task) {
     const newAgent = await this.#config.createAgent(agentName);
     this.#agents.push(newAgent);
-    const exitReason = await newAgent.startTask({
-      task,
-      context
-    });
+    const exitReason = await newAgent.start(task);
     return await this.#handleTaskResult(exitReason);
   }
   async startTask(options) {
     if (this.#agents.length > 0) {
       throw new Error("An active agent already exists");
     }
-    return this.#startTask(options.agentName, options.task, options.context);
+    return this.#startTask(options.agentName, options.task);
   }
   async continueTask(userMessage) {
     if (!this.#agents.length) {
@@ -3358,8 +3350,8 @@ var executeAgentTool = async (definition, agent, params) => {
   }
   const exitReason = await agent.startTask({
     agentName: definition.agent,
-    task: definition.prompt,
-    context: definition.formatInput(params)
+    task: `<task>${definition.prompt}</task>
+<context>${definition.formatInput(params)}</context>`
   });
   if (exitReason.type === "Exit" /* Exit */) {
     return definition.parseOutput(exitReason.message);

package/package.json

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