File size: 3,464 Bytes
5ef3875 | 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 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 | """Wires configuration, context, security, tools, provider, agent and UI together."""
from __future__ import annotations
from pathlib import Path
from ycode import __version__
from ycode.agent.context import ProjectContext, detect_project
from ycode.agent.loop import Agent, TaskResult
from ycode.agent.prompts import build_system_prompt
from ycode.config import Config
from ycode.llm.base import LLMProvider
from ycode.security.permissions import PermissionManager
from ycode.tools import ToolContext, ToolRegistry, default_tools
from ycode.tools import git
from ycode.ui.banner import BannerInfo
from ycode.ui.console import ConsoleUI
class YCodeApp:
def __init__(
self,
*,
config: Config,
workspace: Path,
provider: LLMProvider,
ui: ConsoleUI,
context: ProjectContext | None = None,
) -> None:
self.config = config
self.workspace = workspace.resolve()
self.ui = ui
self.context = context or detect_project(self.workspace, config.ignore_dirs)
self.permissions = PermissionManager(
self.workspace,
mode=config.approval_mode,
approver=ui.ask_approval,
allow_outside_workspace=config.allow_outside_workspace,
)
self.tool_context = ToolContext(
workspace=self.workspace,
permissions=self.permissions,
ignore_dirs=config.ignore_dirs,
command_timeout=config.command_timeout,
max_output_chars=config.max_output_chars,
)
self.registry = ToolRegistry(default_tools())
self.agent = Agent(
provider=provider,
registry=self.registry,
tool_context=self.tool_context,
system_prompt=build_system_prompt(self.context),
max_steps=config.max_steps,
events=ui,
)
self.should_exit = False
@property
def provider(self) -> LLMProvider:
return self.agent.provider
def banner_info(self) -> BannerInfo:
return BannerInfo(
version=__version__,
model=self.provider.display_name,
directory=str(self.workspace),
branch=self.context.branch,
project_types=tuple(self.context.project_types),
approval_mode=self.permissions.mode,
instructions_loaded=bool(self.context.instructions),
)
def changes_for(self, files: list[str]) -> list[git.FileChange]:
"""Git status entries for files the task touched (or a plain list without git)."""
if not files:
return []
if not self.context.is_git_repo:
return [git.FileChange(" M", f) for f in files]
wanted = set(files)
entries = [e for e in git.status_entries(self.workspace) if e.path in wanted]
return entries
def run_task(self, task: str) -> TaskResult:
result = self.agent.run(task)
self.ui.print_task_summary(result, self.changes_for(result.changed_files))
return result
def handle_line(self, line: str) -> None:
"""Process one line of user input: a /command or a task for the agent."""
from ycode.commands import handle_command, is_command
text = line.strip()
if not text:
return
if is_command(text):
handle_command(self, text)
return
self.ui.console.print()
self.run_task(text)
|