88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
"""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)
|