@polka-codes/core 0.7.0 → 0.7.2

package/dist/index.js

@@ -13420,6 +13420,7 @@ class OpenRouterService extends AiServiceBase {
   #client;
   #apiKey;
   #options;
+  #modelProviderInfo;
   model;
   constructor(options) {
     super(options.usageMeter);
@@ -13444,6 +13445,9 @@ class OpenRouterService extends AiServiceBase {
       id: options.model,
       info: {}
     };
+    fetch(`https://openrouter.ai/api/v1/models/${this.model.id}/endpoints`).then((res) => res.json()).then((data) => {
+      this.#modelProviderInfo = data.data;
+    });
   }
   async* sendImpl(systemPrompt, messages) {
     const openAiMessages = [
@@ -13564,12 +13568,20 @@ class OpenRouterService extends AiServiceBase {
         signal: controller.signal
       });
       const responseBody = await response.json();
-      const generation = responseBody.data;
+      const generation = responseBody.data ?? {};
+      let totalCost = generation.total_cost || 0;
+      if (generation.is_byok && this.#modelProviderInfo) {
+        const price = this.#modelProviderInfo.endpoints.find((e) => e.provider_name === generation.provider_name)?.pricing;
+        if (price) {
+          totalCost += (generation.native_tokens_prompt || 0) * price.request;
+          totalCost += (generation.native_tokens_completion || 0) * price.completion;
+        }
+      }
       yield {
         type: "usage",
-        inputTokens: generation?.native_tokens_prompt || 0,
-        outputTokens: generation?.native_tokens_completion || 0,
-        totalCost: generation?.total_cost || 0
+        inputTokens: generation.native_tokens_prompt || 0,
+        outputTokens: generation.native_tokens_completion || 0,
+        totalCost
       };
     } catch (error) {
       console.error("Error fetching OpenRouter generation details:", error);
@@ -15072,7 +15084,7 @@ ${agents}`;
       content: userMessage
     });
     let currentAssistantMessage = "";
-    const retryCount = 3;
+    const retryCount = 5;
     for (let i = 0;i < retryCount; i++) {
       currentAssistantMessage = "";
       const stream = this.ai.send(this.config.systemPrompt, this.messages);
@@ -15848,14 +15860,38 @@ var prompt = `You are an AiTool designed to assist users in creating new project
        - ".env.*"
      \`\`\`
 
-5. **Handover to Coder Agent:**
+5. **Generate Essential Project Files:**
+   - Create a .gitattributes file with appropriate configurations:
+     - Mark lock files as generated and exclude them from diffs
+     - Example for different package managers:
+       
+       # For Bun
+       bun.lock linguist-generated=true
+       bun.lock -diff
+       
+       # For npm
+       package-lock.json linguist-generated=true
+       package-lock.json -diff
+       
+       # For Yarn
+       yarn.lock linguist-generated=true
+       yarn.lock -diff
+       
+       # For pnpm
+       pnpm-lock.yaml linguist-generated=true
+       pnpm-lock.yaml -diff
+       
+     - Include other common configurations as needed based on project type
+
+6. **Handover to Coder Agent:**
    - Once all required information is collected and validated by the user, compile:
      1. The final project specifications
      2. The .polkacodes.yml configuration content
    - Clearly hand over these details to the coder agent, instructing them to:
      1. Create the new project based on the confirmed specifications
      2. Include the .polkacodes.yml file in the project root
-     3. Ensure all specified tools and configurations are properly set up`;
+     3. Include the .gitattributes file with appropriate configurations
+     4. Ensure all specified tools and configurations are properly set up`;
 var createNewProject_default = {
   name: "createNewProject",
   description: "Creates a new project",

package/package.json

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