autopilot-code 0.2.5 → 0.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "autopilot-code",
-  "version": "0.2.5",
+  "version": "0.3.0",
   "private": false,
   "description": "Repo-issue–driven autopilot runner",
   "license": "MIT",

package/scripts/issue_runner/agents/__init__.py

@@ -0,0 +1,3 @@
+from .base import BaseAgent, AgentResult
+
+__all__ = ["BaseAgent", "AgentResult"]

package/scripts/issue_runner/agents/base.py

@@ -0,0 +1,130 @@
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+from typing import Optional
+from pathlib import Path
+
+
+@dataclass
+class AgentResult:
+    success: bool
+    session_id: Optional[str]
+    output: str
+    error: Optional[str] = None
+
+
+class BaseAgent(ABC):
+    def __init__(self, config: dict):
+        self.config = config
+        self._binary_path: Optional[str] = None
+
+    @property
+    @abstractmethod
+    def name(self) -> str:
+        pass
+
+    @property
+    @abstractmethod
+    def supports_sessions(self) -> bool:
+        pass
+
+    @abstractmethod
+    def find_binary(self) -> str:
+        pass
+
+    @property
+    def binary_path(self) -> str:
+        if self._binary_path is None:
+            self._binary_path = self.find_binary()
+        return self._binary_path
+
+    @abstractmethod
+    def run(
+        self, worktree: Path, prompt: str, session_id: Optional[str] = None
+    ) -> AgentResult:
+        pass
+
+    def run_planning(
+        self,
+        worktree: Path,
+        issue_number: int,
+        issue_title: str,
+        issue_body: str,
+        session_id: Optional[str] = None,
+    ) -> AgentResult:
+        prompt = f"""Please analyze the following GitHub issue and provide a detailed implementation plan. Do NOT make any code changes yet - only analyze and plan.
+
+Issue #{issue_number}: {issue_title}
+
+{issue_body}
+
+Your task:
+1. Analyze the issue requirements thoroughly
+2. Explore the codebase to understand the relevant components
+3. Identify the files that need to be modified or created
+4. Outline the step-by-step approach for implementing this change
+5. Note any potential issues, dependencies, or risks
+6. List any assumptions you're making
+
+Provide your analysis and plan in a clear, structured format."""
+
+        return self.run(worktree, prompt, session_id)
+
+    def run_implementation(
+        self,
+        worktree: Path,
+        issue_number: int,
+        issue_title: str,
+        issue_body: str,
+        branch: str,
+        session_id: Optional[str] = None,
+    ) -> AgentResult:
+        prompt = f"""Please implement the following GitHub issue.
+
+Issue #{issue_number}: {issue_title}
+
+{issue_body}
+
+Work rules:
+- Make the necessary code changes.
+- Commit with message: "autopilot: work for issue #{issue_number}".
+- Push your changes to the remote branch {branch}.
+- If the issue is a simple file-addition, just do it directly (no extra refactors)."""
+
+        return self.run(worktree, prompt, session_id)
+
+    def run_conflict_resolution(
+        self, worktree: Path, session_id: Optional[str] = None
+    ) -> AgentResult:
+        prompt = """This PR has merge conflicts. Please resolve them:
+
+1. Run 'git status' to see all conflicted files
+2. For each conflicted file, examine the conflict markers (<<<<<, =======, >>>>>)
+3. Resolve the conflicts by choosing the appropriate code
+4. Stage the resolved files with 'git add <file>'
+5. The goal is to make the branch mergeable with main
+
+After resolving all conflicts, report the files that were resolved."""
+
+        return self.run(worktree, prompt, session_id)
+
+    def run_ci_fix(
+        self,
+        worktree: Path,
+        failure_context: str,
+        attempt: int,
+        max_attempts: int,
+        branch: str,
+        session_id: Optional[str] = None,
+    ) -> AgentResult:
+        prompt = f"""The PR checks have failed. Please analyze the CI failures and fix the issues.
+
+{failure_context}
+
+Work rules:
+- Examine the failed checks and identify the root cause
+- Make the necessary code changes to fix the failures
+- Commit with message: "autopilot: fix CI check failures (attempt {attempt}/{max_attempts})"
+- Push your changes to the branch {branch}
+- Focus only on fixing the CI failures, do not make unrelated changes"""
+
+        return self.run(worktree, prompt, session_id)