@polka-codes/core 0.4.0 → 0.4.2

package/dist/index.js

@@ -2351,6 +2351,35 @@ Anthropic.Beta = Beta;
 
 // src/AiService/AiServiceBase.ts
 class AiServiceBase {
+  async request(systemPrompt, messages) {
+    const stream = this.send(systemPrompt, messages);
+    const usage = {
+      inputTokens: 0,
+      outputTokens: 0,
+      cacheWriteTokens: 0,
+      cacheReadTokens: 0,
+      totalCost: 0
+    };
+    let resp = "";
+    for await (const chunk of stream) {
+      switch (chunk.type) {
+        case "usage":
+          usage.inputTokens = chunk.inputTokens;
+          usage.outputTokens = chunk.outputTokens;
+          usage.cacheWriteTokens = chunk.cacheWriteTokens ?? 0;
+          usage.cacheReadTokens = chunk.cacheReadTokens ?? 0;
+          usage.totalCost = chunk.totalCost;
+          break;
+        case "text":
+          resp += chunk.text;
+          break;
+      }
+    }
+    return {
+      response: resp,
+      usage
+    };
+  }
 }
 
 // src/AiService/ModelInfo.ts
@@ -8190,7 +8219,12 @@ var handler6 = async (provider, args) => {
   const resp = [];
   for (const path of paths) {
     const fileContent = await provider.readFile(path);
-    resp.push(`<read_file_file_conten path="${path}">${fileContent}</read_file_file_content>`);
+    const isEmpty = fileContent.trim().length === 0;
+    if (isEmpty) {
+      resp.push(`<read_file_file_content path="${path}" is_empty="true" />`);
+    } else {
+      resp.push(`<read_file_file_conten path="${path}">${fileContent}</read_file_file_content>`);
+    }
   }
   return {
     type: "Reply" /* Reply */,
@@ -8662,14 +8696,91 @@ class CoderAgent extends AgentBase {
     });
   }
 }
+// src/AiTool/generateGitCommitMessage.ts
+var prompt = `
+You are an advanced assistant specialized in creating concise and accurate Git commit messages. When you receive:
+- A Git diff inside the <tool_input> tag.
+- Additional user-supplied context inside the <tool_input_context> tag (if any).
+
+You will produce a single commit message enclosed within <tool_output> tags. The commit message must accurately reflect the changes shown in the diff and should be clear, descriptive, and devoid of unnecessary or repeated information. If a context is provided, it MUST be incorporated into the commit message.
+
+Here’s an example of the input and the expected output format:
+
+\`\`\`
+<tool_input>
+--- a/example_file.py
++++ b/example_file.py
+@@ -10,7 +10,7 @@ def example_function():
+-    print("Old behavior")
++    print("New behavior")
+</tool_input>
+<tool_input_context>
+Changing print statement to update the user-facing message.
+</tool_input_context>
+\`\`\`
+
+Example Output:
+
+\`\`\`
+<tool_output>
+Update print statement for revised user-facing message
+</tool_output>
+\`\`\`
+
+Follow the same structure for any new input. Never repeat questions; focus on generating a concise commit message that captures the essence of the changes.
+`;
+var generateGitCommitMessage_default = {
+  name: "generateGitCommitMessage",
+  description: "Generates git commit messages from git diff output",
+  prompt,
+  formatInput: (params) => {
+    let ret = `<tool_input>
+${params.diff}
+</tool_input>`;
+    if (params.context) {
+      ret += `
+<tool_input_context>
+${params.context}
+</tool_input_context>`;
+    }
+    return ret;
+  },
+  parseOutput: (output) => {
+    const regex = /<tool_output>([\s\S]*)<\/tool_output>/gm;
+    const match = regex.exec(output);
+    if (match) {
+      return match[1];
+    }
+    throw new Error(`Could not parse output:
+${output}`);
+  }
+};
+
+// src/AiTool/index.ts
+var executeTool = async (definition, ai, params) => {
+  const { response, usage } = await ai.request(definition.prompt, [{ role: "user", content: definition.formatInput(params) }]);
+  return {
+    response: definition.parseOutput(response),
+    usage
+  };
+};
+var makeTool = (definition) => {
+  return async (ai, params) => {
+    return executeTool(definition, ai, params);
+  };
+};
+var generateGitCommitMessage = makeTool(generateGitCommitMessage_default);
 export {
   writeToFile_default as writeToFile,
   searchFiles_default as searchFiles,
   replaceInFile_default as replaceInFile,
   readFile_default as readFile,
   openAiModelInfoSaneDefaults,
+  makeTool,
   listFiles_default as listFiles,
   listCodeDefinitionNames_default as listCodeDefinitionNames,
+  generateGitCommitMessage,
+  executeTool,
   executeCommand_default as executeCommand,
   defaultModels,
   deepSeekModels,

package/package.json

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