@polka-codes/core 0.7.4 → 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(): {
@@ -82,6 +79,7 @@ declare interface AiServiceOptions {
     baseUrl?: string;
     usageMeter: UsageMeter;
     enableCache?: boolean;
+    parameters: Record<string, any>;
 }
 export { AiServiceOptions }
 export { AiServiceOptions as AiServiceOptions_alias_1 }
@@ -182,6 +180,7 @@ declare const anthropicModels: {
         readonly outputPrice: 15;
         readonly cacheWritesPrice: 3.75;
         readonly cacheReadsPrice: 0.3;
+        readonly reasoning: true;
     };
     readonly 'claude-3-5-sonnet-20241022': {
         readonly maxTokens: 8192;
@@ -239,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';
@@ -1541,6 +1546,7 @@ declare interface ModelInfo {
     outputPrice?: number;
     cacheWritesPrice?: number;
     cacheReadsPrice?: number;
+    reasoning?: boolean;
 }
 export { ModelInfo }
 export { ModelInfo as ModelInfo_alias_1 }
@@ -1558,6 +1564,7 @@ declare const modelInfos: {
             readonly outputPrice: 15;
             readonly cacheWritesPrice: 3.75;
             readonly cacheReadsPrice: 0.3;
+            readonly reasoning: true;
         };
         readonly 'claude-3-5-sonnet-20241022': {
             readonly maxTokens: 8192;
@@ -1633,7 +1640,6 @@ declare class MultiAgent {
     startTask(options: {
         agentName: string;
         task: string;
-        context?: string;
     }): Promise<ExitReason>;
     continueTask(userMessage: string): Promise<ExitReason>;
     get hasActiveAgent(): boolean;
@@ -1644,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

@@ -74,7 +74,8 @@ var anthropicModels = {
     inputPrice: 3,
     outputPrice: 15,
     cacheWritesPrice: 3.75,
-    cacheReadsPrice: 0.3
+    cacheReadsPrice: 0.3,
+    reasoning: true
   },
   "claude-3-5-sonnet-20241022": {
     maxTokens: 8192,
@@ -183,6 +184,14 @@ var AnthropicService = class extends AiServiceBase {
     let stream;
     const modelId = this.model.id;
     const cacheControl = this.#options.enableCache ? { type: "ephemeral" } : void 0;
+    let temperature = 0;
+    let thinkingBudgetTokens = 0;
+    if (this.model.info.reasoning) {
+      thinkingBudgetTokens = this.#options.parameters.thinkingBudgetTokens ?? 0;
+    }
+    if (thinkingBudgetTokens > 0) {
+      temperature = void 0;
+    }
     switch (modelId) {
       // 'latest' alias does not support cache_control
       case "claude-3-7-sonnet-20250219":
@@ -201,7 +210,8 @@ var AnthropicService = class extends AiServiceBase {
         stream = await this.#client.messages.create({
           model: modelId,
           max_tokens: this.model.info.maxTokens || 8192,
-          temperature: 0,
+          thinking: thinkingBudgetTokens ? { type: "enabled", budget_tokens: thinkingBudgetTokens } : void 0,
+          temperature,
           system: [
             {
               text: systemPrompt,
@@ -241,8 +251,6 @@ var AnthropicService = class extends AiServiceBase {
           temperature: 0,
           system: [{ text: systemPrompt, type: "text" }],
           messages,
-          // tools,
-          // tool_choice: { type: "auto" },
           stream: true
         });
         break;
@@ -285,6 +293,18 @@ var AnthropicService = class extends AiServiceBase {
                 text: chunk.content_block.text
               };
               break;
+            case "thinking":
+              yield {
+                type: "reasoning",
+                text: chunk.content_block.thinking
+              };
+              break;
+            case "redacted_thinking":
+              yield {
+                type: "reasoning",
+                text: "[Redacted by providered]"
+              };
+              break;
           }
           break;
         case "content_block_delta":
@@ -295,6 +315,12 @@ var AnthropicService = class extends AiServiceBase {
                 text: chunk.delta.text
               };
               break;
+            case "thinking_delta":
+              yield {
+                type: "reasoning",
+                text: chunk.delta.thinking
+              };
+              break;
           }
           break;
         case "content_block_stop":
@@ -354,10 +380,18 @@ function convertToOpenAiMessages(anthropicMessages) {
             role: "user",
             content: nonToolMessages.map((part) => {
               if (part.type === "image") {
+                if (part.source.type === "base64") {
+                  return {
+                    type: "image_url",
+                    image_url: {
+                      url: `data:${part.source.media_type};base64,${part.source.data}`
+                    }
+                  };
+                }
                 return {
                   type: "image_url",
                   image_url: {
-                    url: `data:${part.source.media_type};base64,${part.source.data}`
+                    url: part.source.url
                   }
                 };
               }
@@ -618,6 +652,20 @@ var OpenRouterService = class extends AiServiceBase {
         maxTokens = 8192;
         break;
     }
+    let reasoning = {};
+    switch (this.model.id) {
+      case "anthropic/claude-3.7-sonnet":
+      case "anthropic/claude-3.7-sonnet:beta":
+      case "anthropic/claude-3.7-sonnet:thinking":
+      case "anthropic/claude-3-7-sonnet":
+      case "anthropic/claude-3-7-sonnet:beta": {
+        const budget_tokens = this.#options.parameters.thinkingBudgetTokens || 0;
+        if (budget_tokens > 0) {
+          reasoning = { max_tokens: budget_tokens };
+        }
+        break;
+      }
+    }
     let shouldApplyMiddleOutTransform = !this.model.info.supportsPromptCache;
     if (this.model.id === "deepseek/deepseek-chat") {
       shouldApplyMiddleOutTransform = true;
@@ -629,7 +677,8 @@ var OpenRouterService = class extends AiServiceBase {
       temperature: 0,
       stream: true,
       transforms: shouldApplyMiddleOutTransform ? ["middle-out"] : void 0,
-      include_reasoning: true
+      include_reasoning: true,
+      ...reasoning
     });
     let genId;
     for await (const chunk of stream) {
@@ -2160,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;
@@ -2202,19 +2246,23 @@ ${agents}`;
     for (let i = 0; i < retryCount; i++) {
       currentAssistantMessage = "";
       const stream = this.ai.send(this.config.systemPrompt, this.messages);
-      for await (const chunk of stream) {
-        switch (chunk.type) {
-          case "usage":
-            await this.#callback({ kind: "Usage" /* Usage */, agent: this });
-            break;
-          case "text":
-            currentAssistantMessage += chunk.text;
-            await this.#callback({ kind: "Text" /* Text */, agent: this, newText: chunk.text });
-            break;
-          case "reasoning":
-            await this.#callback({ kind: "Reasoning" /* Reasoning */, agent: this, newText: chunk.text });
-            break;
+      try {
+        for await (const chunk of stream) {
+          switch (chunk.type) {
+            case "usage":
+              await this.#callback({ kind: "Usage" /* Usage */, agent: this });
+              break;
+            case "text":
+              currentAssistantMessage += chunk.text;
+              await this.#callback({ kind: "Text" /* Text */, agent: this, newText: chunk.text });
+              break;
+            case "reasoning":
+              await this.#callback({ kind: "Reasoning" /* Reasoning */, agent: this, newText: chunk.text });
+              break;
+          }
         }
+      } catch (error) {
+        console.error("Error in stream:", error);
       }
       if (currentAssistantMessage) {
         break;
@@ -2891,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 */:
@@ -2917,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) {
@@ -3305,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.4",
+  "version": "0.7.6",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -15,7 +15,7 @@
     "build": "tsup src/index.ts --experimental-dts --format esm --clean"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.36.2",
+    "@anthropic-ai/sdk": "0.39.0",
     "lodash": "^4.17.21",
     "openai": "^4.80.0"
   }