aiagent-runner 0.1.3__py3-none-any.whl

aiagent_runner/prompt_builder.py

@@ -0,0 +1,120 @@
+# src/aiagent_runner/prompt_builder.py
+# Build prompts for CLI execution from task information
+# Reference: docs/plan/PHASE3_PULL_ARCHITECTURE.md - Phase 3-5
+
+from typing import Optional
+from aiagent_runner.mcp_client import TaskInfo
+
+
+class PromptBuilder:
+    """Builds prompts for CLI execution from task information."""
+
+    def __init__(self, agent_id: str, agent_name: Optional[str] = None):
+        """Initialize prompt builder.
+
+        Args:
+            agent_id: Agent ID executing the task
+            agent_name: Human-readable agent name (optional)
+        """
+        self.agent_id = agent_id
+        self.agent_name = agent_name or agent_id
+
+    def build(
+        self,
+        task: TaskInfo,
+        context: Optional[dict] = None,
+        handoff: Optional[dict] = None
+    ) -> str:
+        """Build a prompt from task information.
+
+        Args:
+            task: Task information
+            context: Previous context (optional, overrides task.context)
+            handoff: Handoff information (optional, overrides task.handoff)
+
+        Returns:
+            Complete prompt string for CLI execution
+        """
+        # Use provided context/handoff or fall back to task's
+        ctx = context or task.context
+        ho = handoff or task.handoff
+
+        sections = [
+            self._build_header(task),
+            self._build_identification(task),
+            self._build_description(task),
+        ]
+
+        if task.working_directory:
+            sections.append(self._build_working_directory(task))
+
+        if ctx:
+            sections.append(self._build_context(ctx))
+
+        if ho:
+            sections.append(self._build_handoff(ho))
+
+        sections.append(self._build_instructions(task))
+
+        return "\n\n".join(sections)
+
+    def _build_header(self, task: TaskInfo) -> str:
+        """Build header section."""
+        return f"# Task: {task.title}"
+
+    def _build_identification(self, task: TaskInfo) -> str:
+        """Build identification section with IDs."""
+        return f"""## Identification
+- Task ID: {task.task_id}
+- Project ID: {task.project_id}
+- Agent ID: {self.agent_id}
+- Agent Name: {self.agent_name}
+- Priority: {task.priority}"""
+
+    def _build_description(self, task: TaskInfo) -> str:
+        """Build description section."""
+        return f"""## Description
+{task.description}"""
+
+    def _build_working_directory(self, task: TaskInfo) -> str:
+        """Build working directory section."""
+        return f"""## Working Directory
+Path: {task.working_directory}"""
+
+    def _build_context(self, context: dict) -> str:
+        """Build previous context section."""
+        lines = ["## Previous Context"]
+        if context.get("progress"):
+            lines.append(f"**Progress**: {context['progress']}")
+        if context.get("findings"):
+            lines.append(f"**Findings**: {context['findings']}")
+        if context.get("blockers"):
+            lines.append(f"**Blockers**: {context['blockers']}")
+        if context.get("next_steps"):
+            lines.append(f"**Next Steps**: {context['next_steps']}")
+        return "\n".join(lines)
+
+    def _build_handoff(self, handoff: dict) -> str:
+        """Build handoff information section."""
+        lines = ["## Handoff Information"]
+        if handoff.get("from_agent") or handoff.get("from_agent_id"):
+            from_agent = handoff.get("from_agent") or handoff.get("from_agent_id")
+            lines.append(f"**From Agent**: {from_agent}")
+        if handoff.get("summary"):
+            lines.append(f"**Summary**: {handoff['summary']}")
+        if handoff.get("context"):
+            lines.append(f"**Context**: {handoff['context']}")
+        if handoff.get("recommendations"):
+            lines.append(f"**Recommendations**: {handoff['recommendations']}")
+        return "\n".join(lines)
+
+    def _build_instructions(self, task: TaskInfo) -> str:
+        """Build instructions section."""
+        return f"""## Instructions
+1. Complete the task as described above
+2. Save your progress regularly using:
+   save_context(task_id="{task.task_id}", progress="...", findings="...", next_steps="...")
+3. When done, update the task status using:
+   update_task_status(task_id="{task.task_id}", status="done")
+4. If you need to hand off to another agent, use:
+   create_handoff(task_id="{task.task_id}", from_agent_id="{self.agent_id}", summary="...", recommendations="...")"""

aiagent_runner/runner.py

@@ -0,0 +1,236 @@
+# src/aiagent_runner/runner.py
+# Main runner loop for AI Agent PM
+# Reference: docs/plan/PHASE3_PULL_ARCHITECTURE.md - Phase 3-5
+
+import asyncio
+import logging
+import os
+from datetime import datetime
+from pathlib import Path
+from typing import Optional
+
+from aiagent_runner.config import RunnerConfig
+from aiagent_runner.executor import CLIExecutor, ExecutionResult
+from aiagent_runner.mcp_client import (
+    AuthenticationError,
+    MCPClient,
+    MCPError,
+    SessionExpiredError,
+    TaskInfo,
+)
+from aiagent_runner.prompt_builder import PromptBuilder
+
+logger = logging.getLogger(__name__)
+
+
+class Runner:
+    """Main runner that polls for tasks and executes them.
+
+    The runner operates in a polling loop:
+    1. Authenticate with MCP server
+    2. Poll for pending tasks
+    3. Execute tasks using CLI
+    4. Report execution results
+    5. Wait for polling interval
+    6. Repeat
+    """
+
+    def __init__(self, config: RunnerConfig):
+        """Initialize runner.
+
+        Args:
+            config: Runner configuration
+        """
+        self.config = config
+        self.mcp_client = MCPClient(config.mcp_socket_path)
+        self.executor = CLIExecutor(config.cli_command, config.cli_args)
+        self.prompt_builder: Optional[PromptBuilder] = None
+
+        self._running = False
+        self._authenticated = False
+        self._agent_name: Optional[str] = None
+
+    @property
+    def log_directory(self) -> Path:
+        """Get log directory, creating if needed."""
+        if self.config.log_directory:
+            log_dir = Path(self.config.log_directory)
+        else:
+            log_dir = Path.home() / ".aiagent-runner" / "logs"
+        log_dir.mkdir(parents=True, exist_ok=True)
+        return log_dir
+
+    @property
+    def working_directory(self) -> str:
+        """Get working directory."""
+        return self.config.working_directory or os.getcwd()
+
+    async def start(self) -> None:
+        """Start the runner loop.
+
+        Runs until stop() is called or an unrecoverable error occurs.
+        """
+        logger.info(
+            f"Starting runner for agent {self.config.agent_id}, "
+            f"polling every {self.config.polling_interval}s"
+        )
+
+        self._running = True
+
+        while self._running:
+            try:
+                await self._run_once()
+            except AuthenticationError as e:
+                logger.error(f"Authentication failed: {e}")
+                self._authenticated = False
+                # Wait before retrying
+                await asyncio.sleep(self.config.polling_interval)
+            except MCPError as e:
+                logger.error(f"MCP error: {e}")
+                # Wait before retrying
+                await asyncio.sleep(self.config.polling_interval)
+            except Exception as e:
+                logger.exception(f"Unexpected error: {e}")
+                # Wait before retrying
+                await asyncio.sleep(self.config.polling_interval)
+
+            if self._running:
+                await asyncio.sleep(self.config.polling_interval)
+
+    def stop(self) -> None:
+        """Stop the runner loop."""
+        logger.info("Stopping runner")
+        self._running = False
+
+    async def _run_once(self) -> None:
+        """Run one iteration of the polling loop."""
+        # Ensure authenticated
+        await self._ensure_authenticated()
+
+        # Get pending tasks (agent_id is derived from session token)
+        try:
+            tasks = await self.mcp_client.get_pending_tasks()
+        except SessionExpiredError:
+            logger.info("Session expired, re-authenticating")
+            self._authenticated = False
+            await self._ensure_authenticated()
+            tasks = await self.mcp_client.get_pending_tasks()
+
+        if not tasks:
+            logger.debug("No pending tasks")
+            return
+
+        logger.info(f"Found {len(tasks)} pending task(s)")
+
+        # Process first task (one at a time)
+        task = tasks[0]
+        await self._process_task(task)
+
+    async def _ensure_authenticated(self) -> None:
+        """Ensure we have a valid session."""
+        if self._authenticated:
+            return
+
+        logger.info(f"Authenticating agent {self.config.agent_id} for project {self.config.project_id}")
+        result = await self.mcp_client.authenticate(
+            self.config.agent_id,
+            self.config.passkey,
+            self.config.project_id
+        )
+        self._authenticated = True
+        self._agent_name = result.agent_name
+        self.prompt_builder = PromptBuilder(
+            self.config.agent_id,
+            self._agent_name
+        )
+        logger.info(
+            f"Authenticated as {self._agent_name or self.config.agent_id}, "
+            f"session expires in {result.expires_in}s"
+        )
+
+    async def _process_task(self, task: TaskInfo) -> None:
+        """Process a single task.
+
+        Args:
+            task: Task to process
+        """
+        logger.info(f"Processing task {task.task_id}: {task.title}")
+
+        # Report execution start (agent_id is derived from session token)
+        try:
+            start_result = await self.mcp_client.report_execution_start(
+                task.task_id
+            )
+            execution_id = start_result.execution_id
+        except MCPError as e:
+            logger.error(f"Failed to report execution start: {e}")
+            return
+
+        # Build prompt
+        prompt = self.prompt_builder.build(task)
+
+        # Determine working directory
+        work_dir = task.working_directory or self.working_directory
+
+        # Generate log file path
+        log_file = self._generate_log_path(task.task_id)
+
+        # Execute CLI
+        logger.info(f"Executing {self.config.cli_command} for task {task.task_id}")
+        result = self.executor.execute(prompt, work_dir, log_file)
+
+        # Report execution complete
+        error_message = None
+        if result.exit_code != 0:
+            error_message = f"CLI exited with code {result.exit_code}"
+            logger.warning(
+                f"Task {task.task_id} execution failed: {error_message}"
+            )
+        else:
+            logger.info(
+                f"Task {task.task_id} execution completed in "
+                f"{result.duration_seconds:.1f}s"
+            )
+
+        try:
+            await self.mcp_client.report_execution_complete(
+                execution_id,
+                result.exit_code,
+                result.duration_seconds,
+                result.log_file,
+                error_message
+            )
+        except MCPError as e:
+            logger.error(f"Failed to report execution complete: {e}")
+
+    def _generate_log_path(self, task_id: str) -> str:
+        """Generate a log file path for a task execution.
+
+        Args:
+            task_id: Task ID
+
+        Returns:
+            Path to log file
+        """
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"{task_id}_{timestamp}.log"
+        return str(self.log_directory / filename)
+
+
+async def run_async(config: RunnerConfig) -> None:
+    """Run the runner asynchronously.
+
+    Args:
+        config: Runner configuration
+    """
+    runner = Runner(config)
+    await runner.start()
+
+
+def run(config: RunnerConfig) -> None:
+    """Run the runner synchronously.
+
+    Args:
+        config: Runner configuration
+    """
+    asyncio.run(run_async(config))

aiagent_runner-0.1.3.dist-info/METADATA

@@ -0,0 +1,185 @@
+Metadata-Version: 2.4
+Name: aiagent-runner
+Version: 0.1.3
+Summary: Runner for AI Agent PM - executes tasks via MCP and CLI
+Author: AI Agent PM Team
+License-Expression: MIT
+Keywords: agent,ai,automation,mcp,orchestration,task-management
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: pyyaml>=6.0
+Provides-Extra: dev
+Requires-Dist: aiohttp>=3.9; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Provides-Extra: http
+Requires-Dist: aiohttp>=3.9; extra == 'http'
+Description-Content-Type: text/markdown
+
+# AI Agent PM Coordinator
+
+全エージェントのタスク実行を統合管理するオーケストレーションデーモン。
+
+---
+
+## 🚨 致命的な設計違反警告 (2026-01-09)
+
+**現在の実装には致命的な設計違反があります。修正が必要です。**
+
+### 問題点
+
+Coordinator/Runnerは**純粋なMCPクライアント**であるべきですが、現在の実装は以下の不正な情報を保持しています：
+
+| 設定項目 | 問題 | 対応 |
+|---------|------|------|
+| `mcp_server_command` | サーバー起動コマンドを保持 | **削除必須** |
+| `mcp_database_path` | DBパス（内部実装詳細）を保持 | **削除必須** |
+
+### 設計原則
+
+```
+✅ 正しい設計: Coordinatorはソケット接続のみ行う純粋なクライアント
+❌ 現在の実装: Coordinatorがサーバー起動やDB接続情報を管理
+```
+
+### 影響
+
+- Agent Instanceごとに新しいMCPサーバーをstdio transportで起動
+- 複数のMCPサーバーが同じDBに同時アクセス → データ整合性の問題
+- テストと本番で異なるDB設定が必要 → 設定の複雑化
+
+### 正しいアーキテクチャ
+
+```
+[アプリ] → MCPデーモン起動（唯一のサーバー）
+[Coordinator] → Unix Socketで接続（クライアントのみ）
+[Agent Instance] → 同じUnix Socketで接続（クライアントのみ）
+```
+
+詳細は `docs/plan/PHASE4_COORDINATOR_ARCHITECTURE.md` の警告セクションを参照。
+
+---
+
+## クイックスタート
+
+```bash
+cd runner
+pip install -e .
+
+# デフォルト設定で起動（config/coordinator_default.yaml を使用）
+python -m aiagent_runner --coordinator
+```
+
+---
+
+## Coordinatorモード（推奨）
+
+単一のCoordinatorが全ての(agent_id, project_id)ペアを管理します。
+
+### 基本起動
+
+```bash
+# デフォルト設定で起動
+python -m aiagent_runner --coordinator
+
+# 詳細ログ出力
+python -m aiagent_runner --coordinator -v
+```
+
+### カスタム設定ファイル
+
+```bash
+python -m aiagent_runner --coordinator -c /path/to/config.yaml
+```
+
+### 設定ファイル例
+
+```yaml
+# config/coordinator_default.yaml がデフォルトで読み込まれます
+# カスタム設定で上書き可能
+
+polling_interval: 10
+max_concurrent: 3
+
+# AI providers
+ai_providers:
+  claude:
+    cli_command: claude
+    cli_args:
+      - "--dangerously-skip-permissions"
+      - "--max-turns"
+      - "50"
+
+# Agents (passkeyのみ - ai_type等はMCPから取得)
+agents:
+  agt_developer:
+    passkey: secret123
+  agt_reviewer:
+    passkey: ${REVIEWER_PASSKEY}  # 環境変数展開対応
+
+log_directory: /tmp/coordinator_logs
+```
+
+### バックグラウンド実行
+
+```bash
+nohup python -m aiagent_runner --coordinator -v > coordinator.log 2>&1 &
+```
+
+---
+
+## 設定の優先順位
+
+1. **コマンドライン引数** (`--polling-interval` 等)
+2. **指定した設定ファイル** (`-c /path/to/config.yaml`)
+3. **デフォルト設定** (`runner/config/coordinator_default.yaml`)
+4. **組み込みデフォルト値**
+
+---
+
+## 動作フロー
+
+1. MCPサーバーに接続（Unixソケット）
+2. `list_active_projects_with_agents()` で全プロジェクト・エージェントを取得
+3. 各(agent_id, project_id)ペアに対して `get_agent_action()` を呼び出し
+4. 作業が必要な場合、Agent Instance（Claude CLI等）をスポーン
+5. Agent Instanceが `authenticate` → `get_my_task` → 実行 → `report_completed`
+6. 待機して2に戻る
+
+---
+
+## 前提条件
+
+- MCPサーバーが起動していること
+- エージェントがアプリで登録済みで、passkeyが設定されていること
+- 該当エージェントがプロジェクトに割り当てられていること
+- タスクが `in_progress` ステータスであること
+
+---
+
+## Legacy Runnerモード（非推奨）
+
+1エージェント = 1デーモン の旧アーキテクチャ。
+
+```bash
+# 非推奨: Coordinatorモードを使用してください
+aiagent-runner --agent-id <AGENT_ID> --passkey <PASSKEY> --project-id <PROJECT_ID>
+```
+
+---
+
+## 開発
+
+```bash
+pip install -e ".[dev]"
+pytest
+```

aiagent_runner-0.1.3.dist-info/RECORD

@@ -0,0 +1,13 @@
+aiagent_runner/__init__.py,sha256=dniIduQc_cgSMFCXg283ytOOFUxrlO5imMKOxLkTCT0,314
+aiagent_runner/__main__.py,sha256=xxBGs0gZdMEVSh21UOWVJg6ekud5OcwmCIflH0eSYcI,8711
+aiagent_runner/config.py,sha256=3Wu5XPuKjPpj6SJcGDoycrK7DkA2bWZ_1OiEUJm2l7g,3539
+aiagent_runner/coordinator.py,sha256=Gg3CbWGe2fH1dqVobqSMGGgXHQ1UrOHqcWZuKQHl43E,26338
+aiagent_runner/coordinator_config.py,sha256=DVGleEeGU93xLSSOONSgFLckwV-3bc53WiwWz6GIIAU,7208
+aiagent_runner/executor.py,sha256=kUyXFb0j6Qw7P41S8KkznI3JSYJT0dxL8gGQdk7kkcA,2914
+aiagent_runner/mcp_client.py,sha256=Y70DDzhQXBirui8OQ_MmOTwBZLWlRK7c3TSHHAYsT4o,23360
+aiagent_runner/prompt_builder.py,sha256=IwPOeXDqYED6iRkLETlQg6tarG-U2VUtnaQaiGwgEQk,4402
+aiagent_runner/runner.py,sha256=0re8QZXrbqYLWGo2mTh15R9PSvg5d7xTnHtIuJHu1Rg,7369
+aiagent_runner-0.1.3.dist-info/METADATA,sha256=GHUpqrzBmmm3rIUsgYczgT18uWA1PvcLrd9vv8-S02w,5179
+aiagent_runner-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiagent_runner-0.1.3.dist-info/entry_points.txt,sha256=UwbGpHt0NTjnTFlVQhcZLR2GWgG15v2r1n-OYvxHzK4,64
+aiagent_runner-0.1.3.dist-info/RECORD,,

aiagent_runner-0.1.3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

aiagent_runner-0.1.3.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+aiagent-runner = aiagent_runner.__main__:main