feat: Project Indexer + Context Manager + GeminiCaller 구현 및 테스트 #task-187 #task-188 #task-189

core/gemini_caller.py

@@ -0,0 +1,87 @@
+"""GeminiCaller — gemini -p 역할별 headless 호출.
+
+cmd /c 래핑으로 PowerShell 실행 정책 우회.
+"""
+
+import asyncio
+import json
+import time
+from pathlib import Path
+
+
+ROLE_PROMPTS_DIR = Path(__file__).parent.parent / "prompts"
+
+
+class GeminiCaller:
+    """Gemini CLI headless 호출을 관리합니다."""
+
+    def __init__(self, project_path: str = None):
+        self.project_path = project_path
+        self.call_count = 0
+        self.last_call_time = 0.0
+
+    async def call(self, role: str, context: str, timeout: int = 120) -> str:
+        """역할별 프롬프트로 gemini -p 호출.
+
+        Args:
+            role: 프롬프트 파일명 (planner, coder, reviewer, tester)
+            context: 전달할 컨텍스트
+            timeout: 최대 대기 시간 (초)
+        """
+        # 시스템 프롬프트 로드
+        prompt_file = ROLE_PROMPTS_DIR / f"{role}.md"
+        if prompt_file.exists():
+            system_prompt = prompt_file.read_text(encoding="utf-8")
+        else:
+            system_prompt = f"You are a {role}. Respond in Korean."
+
+        # cmd 구성
+        cmd_parts = ["gemini", "-p", context]
+
+        if system_prompt:
+            cmd_parts.extend(["--system", system_prompt])
+
+        cmd_parts.extend(["--approval-mode", "yolo"])
+
+        if self.project_path:
+            cmd_parts.extend(["--include-directories", self.project_path])
+
+        # cmd /c 래핑 (PowerShell 실행 정책 우회)
+        escaped_context = context.replace('"', '\\"')
+        cmd_str = f'gemini -p "{escaped_context}" --approval-mode yolo'
+        if self.project_path:
+            cmd_str += f' --include-directories "{self.project_path}"'
+
+        try:
+            proc = await asyncio.create_subprocess_shell(
+                f'cmd /c {cmd_str}',
+                stdout=asyncio.subprocess.PIPE,
+                stderr=asyncio.subprocess.PIPE,
+            )
+            stdout, stderr = await asyncio.wait_for(
+                proc.communicate(), timeout=timeout
+            )
+
+            self.call_count += 1
+            self.last_call_time = time.time()
+
+            output = stdout.decode("utf-8", errors="replace").strip()
+
+            # YOLO 모드 메시지 제거
+            lines = output.splitlines()
+            cleaned = []
+            for line in lines:
+                if "YOLO mode" in line or "Loaded cached" in line:
+                    continue
+                cleaned.append(line)
+
+            return "\n".join(cleaned).strip()
+
+        except asyncio.TimeoutError:
+            return f"[ERROR] Gemini CLI timeout after {timeout}s"
+        except Exception as e:
+            return f"[ERROR] Gemini CLI call failed: {e}"
+
+    async def call_simple(self, prompt: str, timeout: int = 60) -> str:
+        """시스템 프롬프트 없이 단순 호출."""
+        return await self.call("default", prompt, timeout)