@polka-codes/core 0.4.8 → 0.4.10

package/dist/index.js

@@ -3069,6 +3069,7 @@ class AnthropicService extends AiServiceBase {
   async* sendImpl(systemPrompt, messages) {
     let stream;
     const modelId = this.model.id;
+    const cacheControl = this.#options.enableCache ? { type: "ephemeral" } : undefined;
     switch (modelId) {
       case "claude-3-5-sonnet-20241022":
       case "claude-3-5-haiku-20241022":
@@ -3090,7 +3091,7 @@ class AnthropicService extends AiServiceBase {
             {
               text: systemPrompt,
               type: "text",
-              cache_control: { type: "ephemeral" }
+              cache_control: cacheControl
             }
           ],
           messages: messages.map((message, index) => {
@@ -3101,36 +3102,18 @@ class AnthropicService extends AiServiceBase {
                   {
                     type: "text",
                     text: message.content,
-                    cache_control: {
-                      type: "ephemeral"
-                    }
+                    cache_control: cacheControl
                   }
                 ] : message.content.map((content, contentIndex) => contentIndex === message.content.length - 1 ? {
                   ...content,
-                  cache_control: {
-                    type: "ephemeral"
-                  }
+                  cache_control: cacheControl
                 } : content)
               };
             }
             return message;
           }),
           stream: true
-        }, (() => {
-          switch (modelId) {
-            case "claude-3-5-sonnet-20241022":
-            case "claude-3-5-haiku-20241022":
-            case "claude-3-opus-20240229":
-            case "claude-3-haiku-20240307":
-              return {
-                headers: {
-                  "anthropic-beta": "prompt-caching-2024-07-31"
-                }
-              };
-            default:
-              return;
-          }
-        })());
+        });
         break;
       }
       default: {
@@ -8024,6 +8007,7 @@ class OllamaService extends AiServiceBase {
 class OpenRouterService extends AiServiceBase {
   #client;
   #apiKey;
+  #options;
   model;
   constructor(options) {
     super(options.usageMeter);
@@ -8042,6 +8026,7 @@ class OpenRouterService extends AiServiceBase {
         "X-Title": "Polka Codes"
       }
     });
+    this.#options = options;
     this.model = {
       id: options.model,
       info: {}
@@ -8052,6 +8037,7 @@ class OpenRouterService extends AiServiceBase {
       { role: "system", content: systemPrompt },
       ...convertToOpenAiMessages(messages)
     ];
+    const cacheControl = this.#options.enableCache ? { type: "ephemeral" } : undefined;
     switch (this.model.id) {
       case "anthropic/claude-3.5-sonnet":
       case "anthropic/claude-3.5-sonnet:beta":
@@ -8071,7 +8057,7 @@ class OpenRouterService extends AiServiceBase {
             {
               type: "text",
               text: systemPrompt,
-              cache_control: { type: "ephemeral" }
+              cache_control: cacheControl
             }
           ]
         };
@@ -8086,7 +8072,7 @@ class OpenRouterService extends AiServiceBase {
               lastTextPart = { type: "text", text: "..." };
               msg.content.push(lastTextPart);
             }
-            lastTextPart.cache_control = { type: "ephemeral" };
+            lastTextPart.cache_control = cacheControl;
           }
         }
         break;
@@ -9396,7 +9382,15 @@ var responsePrompts = {
 <tool_result>
 ${result}
 </tool_result>
-</tool_response>`
+</tool_response>`,
+  commandResult: (command, exitCode, stdout, stderr) => `<command>${command}</command>
+<command_exit_code>${exitCode}</command_exit_code>
+<command_stdout>
+${stdout}
+</command_stdout>
+<command_stderr>
+${stderr}
+</command_stderr>`
 };
 
 // src/Agent/AgentBase.ts
@@ -9580,6 +9574,10 @@ ${agents}`;
           canRetry: false
         };
       }
+      const resp = await this.onBeforeInvokeTool(name, args);
+      if (resp) {
+        return resp;
+      }
       return await handler13(this.config.provider, args);
     } catch (error) {
       return {
@@ -9651,9 +9649,13 @@ class AnalyzerAgent extends AgentBase {
       toolNamePrefix,
       provider: options.provider,
       interactive: options.interactive,
-      agents: options.agents
+      agents: options.agents,
+      scripts: options.scripts
     });
   }
+  onBeforeInvokeTool() {
+    return Promise.resolve(undefined);
+  }
 }
 var analyzerAgentInfo = {
   name: "analyzer",
@@ -9745,9 +9747,13 @@ class ArchitectAgent extends AgentBase {
       toolNamePrefix,
       provider: options.provider,
       interactive: options.interactive,
-      agents: options.agents
+      agents: options.agents,
+      scripts: options.scripts
     });
   }
+  onBeforeInvokeTool() {
+    return Promise.resolve(undefined);
+  }
 }
 var architectAgentInfo = {
   name: "architect",
@@ -9888,9 +9894,55 @@ class CoderAgent extends AgentBase {
       toolNamePrefix,
       provider: options.provider,
       interactive: options.interactive,
-      agents: options.agents
+      agents: options.agents,
+      scripts: options.scripts
     });
   }
+  async onBeforeInvokeTool(name, args) {
+    const executeCommand = this.config.provider.executeCommand;
+    if (!executeCommand) {
+      return;
+    }
+    const format = this.config.scripts?.format;
+    const formatCommand = typeof format === "string" ? format : format?.command;
+    if (formatCommand) {
+      try {
+        await executeCommand(formatCommand, false);
+      } catch (error) {
+        console.warn(`Failed to format code using command: ${formatCommand}`, error);
+      }
+    }
+    const check = this.config.scripts?.check;
+    const checkCommand = typeof check === "string" ? check : check?.command;
+    if (checkCommand) {
+      try {
+        const { exitCode, stdout, stderr } = await executeCommand(checkCommand, false);
+        if (exitCode !== 0) {
+          return {
+            type: "Reply" /* Reply */,
+            message: responsePrompts.commandResult(checkCommand, exitCode, stdout, stderr)
+          };
+        }
+      } catch (error) {
+        console.warn(`Failed to check code using command: ${checkCommand}`, error);
+      }
+    }
+    const test = this.config.scripts?.test;
+    const testCommand = typeof test === "string" ? test : test?.command;
+    if (testCommand) {
+      try {
+        const { exitCode, stdout, stderr } = await executeCommand(testCommand, false);
+        if (exitCode !== 0) {
+          return {
+            type: "Reply" /* Reply */,
+            message: responsePrompts.commandResult(testCommand, exitCode, stdout, stderr)
+          };
+        }
+      } catch (error) {
+        console.warn(`Failed to test code using command: ${testCommand}`, error);
+      }
+    }
+  }
 }
 var coderAgentInfo = {
   name: "coder",
@@ -10130,10 +10182,16 @@ var prompt3 = `You are an analyzer agent responsible for examining project files
    - Code style tools and rules
    - Project structure and conventions
    - Common development workflows
+   - .gitignore patterns and sensitive/generated files
 
 2. Generate a YAML configuration that captures:
-   - scripts section based on package.json scripts and CI workflows
+   - scripts section based on package.json scripts and CI workflows. If applicable, generate following scripts:
+     - format: Format the code using a code formatter tool
+     - check: Check the code for errors using a linter tool
+     - test: Run tests using a test runner tool
+   - include other relevant scripts based on project conventions, tools, and patterns
    - rules section based on project conventions, tools, and patterns
+   - excludeFiles section for common sensitive/generated files
 
 3. Use tool_attempt_completion to return the final configuration in this format:
 
@@ -10151,6 +10209,16 @@ rules:
   - "Use \`bun\` as package manager"
   - "Write tests using bun:test with snapshots"
   - "Follow Biome code style"
+
+excludeFiles:
+  # Sensitive files
+  - ".env"
+  - ".env.*"
+  - ".npmrc"
+  # Generated files
+  - "package-lock.json"
+  - "yarn.lock"
+  - "pnpm-lock.yaml"
 </tool_parameter_result>
 </tool_attempt_completion>
 
@@ -10160,6 +10228,13 @@ Focus on:
 - Code style and linting rules
 - File organization and naming conventions
 - Build and development workflows
+- Sensitive files that should not be committed:
+  - Environment files (.env*)
+  - Configuration files with potential secrets
+- Generated files to exclude:
+  - Lock files from package managers
+  - Build artifacts and outputs
+  - Generated file that are not ignored by .gitignore
 
 The configuration should accurately reflect the project's structure, tools, and conventions.
 `;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.4.8",
+  "version": "0.4.10",
   "license": "AGPL-3.0",
   "type": "module",
   "exports": {