@polka-codes/core 0.8.14 → 0.8.16

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

@@ -210,8 +210,32 @@ export { AnthropicModelId }
 export { AnthropicModelId as AnthropicModelId_alias_1 }
 
 declare const anthropicModels: {
+    readonly 'claude-opus-4-20250514': {
+        readonly maxTokens: 32000;
+        readonly contextWindow: 200000;
+        readonly supportsImages: true;
+        readonly supportsComputerUse: true;
+        readonly supportsPromptCache: true;
+        readonly inputPrice: 15;
+        readonly outputPrice: 75;
+        readonly cacheWritesPrice: 18.75;
+        readonly cacheReadsPrice: 1.5;
+        readonly reasoning: true;
+    };
+    readonly 'claude-sonnet-4-20250514': {
+        readonly maxTokens: 64000;
+        readonly contextWindow: 200000;
+        readonly supportsImages: true;
+        readonly supportsComputerUse: true;
+        readonly supportsPromptCache: true;
+        readonly inputPrice: 3;
+        readonly outputPrice: 15;
+        readonly cacheWritesPrice: 3.75;
+        readonly cacheReadsPrice: 0.3;
+        readonly reasoning: true;
+    };
     readonly 'claude-3-7-sonnet-20250219': {
-        readonly maxTokens: 8192;
+        readonly maxTokens: 64000;
         readonly contextWindow: 200000;
         readonly supportsImages: true;
         readonly supportsComputerUse: true;
@@ -2406,8 +2430,32 @@ export { ModelInfo as ModelInfo_alias_2 }
 
 declare const modelInfos: {
     anthropic: {
+        readonly 'claude-opus-4-20250514': {
+            readonly maxTokens: 32000;
+            readonly contextWindow: 200000;
+            readonly supportsImages: true;
+            readonly supportsComputerUse: true;
+            readonly supportsPromptCache: true;
+            readonly inputPrice: 15;
+            readonly outputPrice: 75;
+            readonly cacheWritesPrice: 18.75;
+            readonly cacheReadsPrice: 1.5;
+            readonly reasoning: true;
+        };
+        readonly 'claude-sonnet-4-20250514': {
+            readonly maxTokens: 64000;
+            readonly contextWindow: 200000;
+            readonly supportsImages: true;
+            readonly supportsComputerUse: true;
+            readonly supportsPromptCache: true;
+            readonly inputPrice: 3;
+            readonly outputPrice: 15;
+            readonly cacheWritesPrice: 3.75;
+            readonly cacheReadsPrice: 0.3;
+            readonly reasoning: true;
+        };
         readonly 'claude-3-7-sonnet-20250219': {
-            readonly maxTokens: 8192;
+            readonly maxTokens: 64000;
             readonly contextWindow: 200000;
             readonly supportsImages: true;
             readonly supportsComputerUse: true;

package/dist/index.js

@@ -69,8 +69,32 @@ import { Anthropic } from "@anthropic-ai/sdk";
 // src/AiService/ModelInfo.ts
 var anthropicDefaultModelId = "claude-3-7-sonnet-20250219";
 var anthropicModels = {
+  "claude-opus-4-20250514": {
+    maxTokens: 32e3,
+    contextWindow: 2e5,
+    supportsImages: true,
+    supportsComputerUse: true,
+    supportsPromptCache: true,
+    inputPrice: 15,
+    outputPrice: 75,
+    cacheWritesPrice: 18.75,
+    cacheReadsPrice: 1.5,
+    reasoning: true
+  },
+  "claude-sonnet-4-20250514": {
+    maxTokens: 64e3,
+    contextWindow: 2e5,
+    supportsImages: true,
+    supportsComputerUse: true,
+    supportsPromptCache: true,
+    inputPrice: 3,
+    outputPrice: 15,
+    cacheWritesPrice: 3.75,
+    cacheReadsPrice: 0.3,
+    reasoning: true
+  },
   "claude-3-7-sonnet-20250219": {
-    maxTokens: 8192,
+    maxTokens: 64e3,
     contextWindow: 2e5,
     supportsImages: true,
     supportsComputerUse: true,
@@ -198,6 +222,7 @@ var AnthropicService = class extends AiServiceBase {
     }
     switch (modelId) {
       // 'latest' alias does not support cache_control
+      case "claude-sonnet-4-20250514":
       case "claude-3-7-sonnet-20250219":
       case "claude-3-5-sonnet-20241022":
       case "claude-3-5-haiku-20241022":
@@ -590,71 +615,32 @@ var OpenRouterService = class extends AiServiceBase {
       ...convertToOpenAiMessages(messages)
     ];
     const cacheControl = this.#options.enableCache ? { type: "ephemeral" } : void 0;
-    switch (this.model.id) {
-      case "anthropic/claude-3.7-sonnet:thinking":
-      case "anthropic/claude-3.7-sonnet":
-      case "anthropic/claude-3.7-sonnet:beta":
-      case "anthropic/claude-3-7-sonnet":
-      case "anthropic/claude-3-7-sonnet:beta":
-      case "anthropic/claude-3.5-sonnet":
-      case "anthropic/claude-3.5-sonnet:beta":
-      case "anthropic/claude-3.5-sonnet-20240620":
-      case "anthropic/claude-3.5-sonnet-20240620:beta":
-      case "anthropic/claude-3-5-haiku":
-      case "anthropic/claude-3-5-haiku:beta":
-      case "anthropic/claude-3-5-haiku-20241022":
-      case "anthropic/claude-3-5-haiku-20241022:beta":
-      case "anthropic/claude-3-haiku":
-      case "anthropic/claude-3-haiku:beta":
-      case "anthropic/claude-3-opus":
-      case "anthropic/claude-3-opus:beta": {
-        openAiMessages[0] = {
-          role: "system",
-          content: [
-            {
-              type: "text",
-              text: systemPrompt,
-              // @ts-ignore-next-line
-              cache_control: cacheControl
-            }
-          ]
-        };
-        const lastTwoUserMessages = openAiMessages.filter((msg) => msg.role === "user").slice(-2);
-        for (const msg of lastTwoUserMessages) {
-          if (typeof msg.content === "string") {
-            msg.content = [{ type: "text", text: msg.content }];
+    if (this.model.id.startsWith("anthropic/claude")) {
+      openAiMessages[0] = {
+        role: "system",
+        content: [
+          {
+            type: "text",
+            text: systemPrompt,
+            // @ts-ignore-next-line
+            cache_control: cacheControl
           }
-          if (Array.isArray(msg.content)) {
-            let lastTextPart = msg.content.filter((part) => part.type === "text").pop();
-            if (!lastTextPart) {
-              lastTextPart = { type: "text", text: "..." };
-              msg.content.push(lastTextPart);
-            }
-            lastTextPart.cache_control = cacheControl;
+        ]
+      };
+      const lastTwoUserMessages = openAiMessages.filter((msg) => msg.role === "user").slice(-2);
+      for (const msg of lastTwoUserMessages) {
+        if (typeof msg.content === "string") {
+          msg.content = [{ type: "text", text: msg.content }];
+        }
+        if (Array.isArray(msg.content)) {
+          let lastTextPart = msg.content.filter((part) => part.type === "text").pop();
+          if (!lastTextPart) {
+            lastTextPart = { type: "text", text: "..." };
+            msg.content.push(lastTextPart);
           }
+          lastTextPart.cache_control = cacheControl;
         }
-        break;
       }
-      default:
-        break;
-    }
-    let maxTokens;
-    switch (this.model.id) {
-      case "anthropic/claude-3.7-sonnet:thinking":
-      case "anthropic/claude-3.7-sonnet":
-      case "anthropic/claude-3.7-sonnet:beta":
-      case "anthropic/claude-3-7-sonnet":
-      case "anthropic/claude-3-7-sonnet:beta":
-      case "anthropic/claude-3.5-sonnet":
-      case "anthropic/claude-3.5-sonnet:beta":
-      case "anthropic/claude-3.5-sonnet-20240620":
-      case "anthropic/claude-3.5-sonnet-20240620:beta":
-      case "anthropic/claude-3-5-haiku":
-      case "anthropic/claude-3-5-haiku:beta":
-      case "anthropic/claude-3-5-haiku-20241022":
-      case "anthropic/claude-3-5-haiku-20241022:beta":
-        maxTokens = 8192;
-        break;
     }
     let reasoning = {};
     switch (this.model.id) {
@@ -662,7 +648,9 @@ var OpenRouterService = class extends AiServiceBase {
       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": {
+      case "anthropic/claude-3-7-sonnet:beta":
+      case "anthropic/claude-opus-4":
+      case "anthropic/claude-sonnet-4": {
         const budget_tokens = this.#options.parameters.thinkingBudgetTokens || 0;
         if (budget_tokens > 0) {
           reasoning = { max_tokens: budget_tokens };
@@ -676,7 +664,6 @@ var OpenRouterService = class extends AiServiceBase {
     }
     const stream = await this.#client.chat.completions.create({
       model: this.model.id,
-      max_completion_tokens: maxTokens,
       messages: openAiMessages,
       temperature: 0,
       stream: true,

package/package.json

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