Spaces:
Sleeping
Sleeping
File size: 2,738 Bytes
8b02e7c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | """Gemini CLI adapter."""
import asyncio
import subprocess
from typing import Any
from .base import CLIAdapter
class GeminiAdapter(CLIAdapter):
"""Adapter for Gemini CLI."""
async def execute(self, task: str, progress_callback: Any = None, timeout: int | None = None, **kwargs: Any) -> tuple[str, str, int]:
"""Execute task using Gemini CLI with optional streaming."""
# Use streaming method from base class if progress_callback provided
if progress_callback:
return await self.execute_streaming(task, progress_callback, timeout, **kwargs)
# Otherwise use buffered execution (legacy)
cmd = self.format_task(task, **kwargs)
resolved_cmd = self.resolve_command(cmd)
process = await asyncio.create_subprocess_exec(
*resolved_cmd,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
env={**subprocess.os.environ, **self.get_env()},
)
try:
effective_timeout = timeout or self.get_timeout()
stdout, stderr = await asyncio.wait_for(
process.communicate(), timeout=effective_timeout
)
return (
stdout.decode("utf-8", errors="replace"),
stderr.decode("utf-8", errors="replace"),
process.returncode or 0,
)
except asyncio.TimeoutError:
process.kill()
await process.wait()
raise TimeoutError(f"Gemini CLI timed out after {effective_timeout}s")
def validate(self) -> bool:
"""Validate Gemini CLI is available."""
try:
subprocess.run(
["which", "gemini"] if subprocess.os.name != "nt" else ["where", "gemini"],
capture_output=True,
check=True,
)
return True
except subprocess.CalledProcessError:
return False
def format_task(self, task: str, **kwargs: Any) -> list[str]:
"""Format task for Gemini CLI."""
cmd = ["gemini"]
# Add model if specified in kwargs or config
if model := kwargs.get("model") or self.config.get("model"):
cmd.extend(["-m", model])
# Add allowed tools if specified
if tools := kwargs.get("allowed_tools"):
tool_list = tools if isinstance(tools, str) else ",".join(tools)
cmd.extend(["--allowed-tools", tool_list])
# Add any custom args from config
for arg in self.get_args():
if arg not in cmd: # Avoid duplicates
cmd.append(arg)
# Add task as final argument (quoted)
cmd.append(task)
return cmd
|