@polka-codes/core 0.9.14 → 0.9.16

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

@@ -176,6 +176,12 @@ export { AnalyzerAgentOptions }
 export { AnalyzerAgentOptions as AnalyzerAgentOptions_alias_1 }
 export { AnalyzerAgentOptions as AnalyzerAgentOptions_alias_2 }
 
+/**
+ * Annotate a diff with line numbers for additions and deletions
+ * Adds [Line N] for additions and [Line N removed] for deletions
+ */
+export declare function annotateDiffWithLineNumbers(diff: string): string;
+
 declare class ArchitectAgent extends AgentBase {
     constructor(options: ArchitectAgentOptions);
     onBeforeInvokeTool(): Promise<undefined>;
@@ -743,7 +749,7 @@ export declare const default_alias_3: {
 export declare const default_alias_4: {
     readonly name: "reviewDiff";
     readonly description: "Reviews a git diff";
-    readonly prompt: "\n# Code Review Prompt\n\nYou are a senior software engineer reviewing code changes.\n\n## Critical Instructions\n**ONLY review the actual changes shown in the diff.** Do not comment on existing code that wasn't modified.\n\n## Viewing Changes\n- **Use git_diff** to inspect the actual code changes for each relevant file.\n  - **Pull request**: use the provided commit range for the git_diff tool.\n  - **Local changes**: diff staged or unstaged files using the git_diff tool.\n- If a pull request is present you may receive:\n  - <pr_title>\n  - <pr_description>\n  - <commit_messages>\n- A <review_instructions> tag tells you the focus of the review.\n- File status information is provided in <file_status> - use this to understand which files were modified, added, deleted, or renamed.\n\n## Review Guidelines\nFocus exclusively on the changed lines (+ additions, - deletions, modified lines):\n- **Specific issues**: Point to exact problems in the changed code with line references\n- **Actionable fixes**: Provide concrete solutions, not vague suggestions\n- **Clear reasoning**: Explain why each issue matters and how to fix it\n- **Avoid generic advice**: No generic suggestions like \"add more tests\", \"improve documentation\", or \"follow best practices\" unless directly related to a specific problem in the diff\n\n## What NOT to review\n- Existing unchanged code\n- Overall project structure or architecture (unless directly impacted by changes)\n- Generic best practices unrelated to the specific changes\n- Missing features or functionality not part of this diff\n\n## Output Format\nDo **not** include praise or positive feedback. Ignore generated files such as lock files.\nOnly include reviews for actual issues found in the changed code.\n\nReturn your review as a JSON object inside a ```json block, wrapped like:\n<tool_attempt_completion>\n<tool_parameter_result>\n```json\n{\n  \"overview\": \"Summary of specific issues found in the diff changes, or 'No issues found' if the changes look good.\",\n  \"specificReviews\": [\n    {\n      \"file\": \"path/filename.ext\",\n      \"lines\": \"N or N-M\",\n      \"review\": \"Specific issue with the changed code and exact actionable fix.\"\n    }\n  ]\n}\n```\n</tool_parameter_result>\n</tool_attempt_completion>\n";
+    readonly prompt: "\n# Code Review Prompt\n\nYou are a senior software engineer reviewing code changes.\n\n## Critical Instructions\n**ONLY review the actual changes shown in the diff.** Do not comment on existing code that wasn't modified.\n\n## Viewing Changes\n- **Use git_diff** to inspect the actual code changes for each relevant file.\n  - **Pull request**: use the provided commit range for the git_diff tool with contextLines: 5 and includeLineNumbers: true\n  - **Local changes**: diff staged or unstaged files using the git_diff tool with contextLines: 5 and includeLineNumbers: true\n- The diff will include line number annotations: [Line N] for additions and [Line N removed] for deletions\n- If a pull request is present you may receive:\n  - <pr_title>\n  - <pr_description>\n  - <commit_messages>\n- A <review_instructions> tag tells you the focus of the review.\n- File status information is provided in <file_status> - use this to understand which files were modified, added, deleted, or renamed.\n\n## Line Number Reporting\n- **IMPORTANT**: Use the line numbers from the annotations in the diff output\n- For additions: Look for [Line N] annotations after the + lines\n- For deletions: Look for [Line N removed] annotations after the - lines\n- For modifications: Report the line number of the new/current code (from [Line N] annotations)\n- Report single lines as \"N\" and ranges as \"N-M\"\n\n## Review Guidelines\nFocus exclusively on the changed lines (+ additions, - deletions, modified lines):\n- **Specific issues**: Point to exact problems in the changed code with accurate line references from the annotations\n- **Actionable fixes**: Provide concrete solutions, not vague suggestions\n- **Clear reasoning**: Explain why each issue matters and how to fix it\n- **Avoid generic advice**: No generic suggestions like \"add more tests\", \"improve documentation\", or \"follow best practices\" unless directly related to a specific problem in the diff\n\n## What NOT to review\n- Existing unchanged code\n- Overall project structure or architecture (unless directly impacted by changes)\n- Generic best practices unrelated to the specific changes\n- Missing features or functionality not part of this diff\n\n## Output Format\nDo **not** include praise or positive feedback. Ignore generated files such as lock files.\nOnly include reviews for actual issues found in the changed code.\n\nReturn your review as a JSON object inside a ```json block, wrapped like:\n<tool_attempt_completion>\n<tool_parameter_result>\n```json\n{\n  \"overview\": \"Summary of specific issues found in the diff changes, or 'No issues found' if the changes look good.\",\n  \"specificReviews\": [\n    {\n      \"file\": \"path/filename.ext\",\n      \"lines\": \"N or N-M\",\n      \"review\": \"Specific issue with the changed code and exact actionable fix.\"\n    }\n  ]\n}\n```\n</tool_parameter_result>\n</tool_attempt_completion>\n";
     readonly formatInput: (params: Input_2) => string;
     readonly parseOutput: (output: string) => Output_2;
     readonly agent: (options: SharedAgentOptions) => AnalyzerAgent;
@@ -757,6 +763,8 @@ export declare const default_alias_5: {
             staged: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
             commitRange: z.ZodOptional<z.ZodString>;
             file: z.ZodOptional<z.ZodString>;
+            contextLines: z.ZodDefault<z.ZodOptional<z.ZodCoercedNumber<unknown>>>;
+            includeLineNumbers: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
         }, z.core.$strip>;
         readonly permissionLevel: PermissionLevel.Read;
     }, CommandProvider>;
@@ -767,10 +775,20 @@ export declare const default_alias_5: {
         staged: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
         commitRange: z.ZodOptional<z.ZodString>;
         file: z.ZodOptional<z.ZodString>;
+        contextLines: z.ZodDefault<z.ZodOptional<z.ZodCoercedNumber<unknown>>>;
+        includeLineNumbers: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
     }, z.core.$strip>;
     permissionLevel: PermissionLevel.Read;
 };
 
+export declare interface DiffHunk {
+    oldStart: number;
+    oldCount: number;
+    newStart: number;
+    newCount: number;
+    header: string;
+}
+
 /**
  * Returns the directory portion of a path string.
  * Strips trailing slashes, then takes everything up to the last slash.
@@ -811,6 +829,16 @@ export { ExitReason }
 export { ExitReason as ExitReason_alias_1 }
 export { ExitReason as ExitReason_alias_2 }
 
+/**
+ * Extract line ranges from a diff for a specific file
+ * Returns an array of line ranges that were modified
+ */
+export declare function extractModifiedLineRanges(diff: string, filePath?: string): Array<{
+    start: number;
+    end: number;
+    type: 'added' | 'removed' | 'modified';
+}>;
+
 declare type FilesystemProvider = {
     readFile?: (path: string, includeIgnored: boolean) => Promise<string | undefined>;
     writeFile?: (path: string, content: string) => Promise<void>;
@@ -1157,6 +1185,12 @@ export { parseAssistantMessage }
 export { parseAssistantMessage as parseAssistantMessage_alias_1 }
 export { parseAssistantMessage as parseAssistantMessage_alias_2 }
 
+/**
+ * Parse a unified diff hunk header
+ * Format: @@ -oldStart,oldCount +newStart,newCount @@ optional context
+ */
+export declare function parseHunkHeader(header: string): DiffHunk | null;
+
 declare enum PermissionLevel {
     None = 0,
     Read = 1,
@@ -1460,6 +1494,8 @@ export declare const toolInfo: {
         staged: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
         commitRange: z.ZodOptional<z.ZodString>;
         file: z.ZodOptional<z.ZodString>;
+        contextLines: z.ZodDefault<z.ZodOptional<z.ZodCoercedNumber<unknown>>>;
+        includeLineNumbers: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
     }, z.core.$strip>;
     readonly permissionLevel: PermissionLevel.Read;
 };

package/dist/index.js

@@ -3363,6 +3363,63 @@ var generateProjectConfig_default = {
 
 // src/AiTool/tools/gitDiff.ts
 import { z as z16 } from "zod";
+
+// src/AiTool/tools/utils/diffLineNumbers.ts
+function parseHunkHeader(header) {
+  const match = header.match(/^@@ -(\d+)(?:,(\d+))? \+(\d+)(?:,(\d+))? @@/);
+  if (!match) return null;
+  return {
+    oldStart: parseInt(match[1], 10),
+    oldCount: match[2] ? parseInt(match[2], 10) : 1,
+    newStart: parseInt(match[3], 10),
+    newCount: match[4] ? parseInt(match[4], 10) : 1,
+    header
+  };
+}
+function annotateDiffWithLineNumbers(diff) {
+  const lines = diff.split("\n");
+  const annotatedLines = [];
+  let currentNewLine = 0;
+  let currentOldLine = 0;
+  let inHunk = false;
+  for (const line of lines) {
+    if (line.startsWith("@@")) {
+      const hunk = parseHunkHeader(line);
+      if (hunk) {
+        currentOldLine = hunk.oldStart;
+        currentNewLine = hunk.newStart;
+        inHunk = true;
+      }
+      annotatedLines.push(line);
+      continue;
+    }
+    if (line.startsWith("diff --git") || line.startsWith("index ") || line.startsWith("---") || line.startsWith("+++")) {
+      annotatedLines.push(line);
+      inHunk = false;
+      continue;
+    }
+    if (!inHunk) {
+      annotatedLines.push(line);
+      continue;
+    }
+    if (line.startsWith("+") && !line.startsWith("+++")) {
+      annotatedLines.push(`${line} [Line ${currentNewLine}]`);
+      currentNewLine++;
+    } else if (line.startsWith("-") && !line.startsWith("---")) {
+      annotatedLines.push(`${line} [Line ${currentOldLine} removed]`);
+      currentOldLine++;
+    } else if (line.startsWith(" ")) {
+      annotatedLines.push(line);
+      currentOldLine++;
+      currentNewLine++;
+    } else {
+      annotatedLines.push(line);
+    }
+  }
+  return annotatedLines.join("\n");
+}
+
+// src/AiTool/tools/gitDiff.ts
 var toolInfo14 = {
   name: "git_diff",
   description: "Get the git diff for the current repository. Can be used to get staged changes, unstaged changes, or changes between commits. By default, it returns unstaged changes.",
@@ -3376,7 +3433,16 @@ var toolInfo14 = {
       return val;
     }, z16.boolean().optional().default(false)).describe("Get staged changes instead of unstaged changes."),
     commitRange: z16.string().optional().describe('The commit range to get the diff for (e.g., "main...HEAD").'),
-    file: z16.string().optional().describe("Get the diff for a specific file.")
+    file: z16.string().optional().describe("Get the diff for a specific file."),
+    contextLines: z16.coerce.number().optional().default(5).describe("Number of context lines to include around changes."),
+    includeLineNumbers: z16.preprocess((val) => {
+      if (typeof val === "string") {
+        const lower = val.toLowerCase();
+        if (lower === "false") return false;
+        if (lower === "true") return true;
+      }
+      return val;
+    }, z16.boolean().optional().default(false)).describe("Annotate the diff with line numbers for additions and deletions.")
   }),
   permissionLevel: 1 /* Read */
 };
@@ -3387,8 +3453,8 @@ var handler14 = async (provider, args) => {
       message: "Not possible to execute command. Abort."
     };
   }
-  const { staged, file, commitRange } = toolInfo14.parameters.parse(args);
-  const commandParts = ["git", "diff", "--no-color", "-U50"];
+  const { staged, file, commitRange, contextLines, includeLineNumbers } = toolInfo14.parameters.parse(args);
+  const commandParts = ["git", "diff", "--no-color", `-U${contextLines}`];
   if (staged) {
     commandParts.push("--staged");
   }
@@ -3408,10 +3474,14 @@ var handler14 = async (provider, args) => {
           message: "No diff found."
         };
       }
+      let diffOutput = result.stdout;
+      if (includeLineNumbers) {
+        diffOutput = annotateDiffWithLineNumbers(diffOutput);
+      }
       return {
         type: "Reply" /* Reply */,
         message: `<diff file="${file ?? "all"}">
-${result.stdout}
+${diffOutput}
 </diff>`
       };
     }
@@ -3447,8 +3517,9 @@ You are a senior software engineer reviewing code changes.
 
 ## Viewing Changes
 - **Use git_diff** to inspect the actual code changes for each relevant file.
-  - **Pull request**: use the provided commit range for the git_diff tool.
-  - **Local changes**: diff staged or unstaged files using the git_diff tool.
+  - **Pull request**: use the provided commit range for the git_diff tool with contextLines: 5 and includeLineNumbers: true
+  - **Local changes**: diff staged or unstaged files using the git_diff tool with contextLines: 5 and includeLineNumbers: true
+- The diff will include line number annotations: [Line N] for additions and [Line N removed] for deletions
 - If a pull request is present you may receive:
   - <pr_title>
   - <pr_description>
@@ -3456,9 +3527,16 @@ You are a senior software engineer reviewing code changes.
 - A <review_instructions> tag tells you the focus of the review.
 - File status information is provided in <file_status> - use this to understand which files were modified, added, deleted, or renamed.
 
+## Line Number Reporting
+- **IMPORTANT**: Use the line numbers from the annotations in the diff output
+- For additions: Look for [Line N] annotations after the + lines
+- For deletions: Look for [Line N removed] annotations after the - lines
+- For modifications: Report the line number of the new/current code (from [Line N] annotations)
+- Report single lines as "N" and ranges as "N-M"
+
 ## Review Guidelines
 Focus exclusively on the changed lines (+ additions, - deletions, modified lines):
-- **Specific issues**: Point to exact problems in the changed code with line references
+- **Specific issues**: Point to exact problems in the changed code with accurate line references from the annotations
 - **Actionable fixes**: Provide concrete solutions, not vague suggestions
 - **Clear reasoning**: Explain why each issue matters and how to fix it
 - **Avoid generic advice**: No generic suggestions like "add more tests", "improve documentation", or "follow best practices" unless directly related to a specific problem in the diff
@@ -3520,11 +3598,11 @@ ${fileList}
     }
     let instructions = "";
     if (params.commitRange) {
-      instructions = `Review the pull request. Use the git_diff tool with commit range '${params.commitRange}' to inspect the actual code changes. File status information is already provided above.`;
+      instructions = `Review the pull request. Use the git_diff tool with commit range '${params.commitRange}', contextLines: 5, and includeLineNumbers: true to inspect the actual code changes. The diff will include line number annotations to help you report accurate line numbers. File status information is already provided above.`;
     } else if (params.staged) {
-      instructions = "Review the staged changes. Use the git_diff tool with staged: true to inspect the actual code changes. File status information is already provided above.";
+      instructions = "Review the staged changes. Use the git_diff tool with staged: true, contextLines: 5, and includeLineNumbers: true to inspect the actual code changes. The diff will include line number annotations to help you report accurate line numbers. File status information is already provided above.";
     } else {
-      instructions = "Review the unstaged changes. Use the git_diff tool to inspect the actual code changes. File status information is already provided above.";
+      instructions = "Review the unstaged changes. Use the git_diff tool with contextLines: 5 and includeLineNumbers: true to inspect the actual code changes. The diff will include line number annotations to help you report accurate line numbers. File status information is already provided above.";
     }
     parts.push(`<review_instructions>
 ${instructions}

package/package.json

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