File size: 856 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
"""Built-in tools available to the agent."""

from __future__ import annotations

from ycode.tools.base import Tool, ToolContext, ToolRegistry, ToolResult
from ycode.tools.filesystem import EditFileTool, ListDirectoryTool, ReadFileTool, WriteFileTool
from ycode.tools.git import GitBranchTool, GitDiffTool, GitLogTool, GitStatusTool
from ycode.tools.search import FindFilesTool, SearchFilesTool
from ycode.tools.terminal import RunCommandTool

__all__ = ["Tool", "ToolContext", "ToolRegistry", "ToolResult", "default_tools"]


def default_tools() -> list[Tool]:
    return [
        ReadFileTool(),
        WriteFileTool(),
        EditFileTool(),
        ListDirectoryTool(),
        SearchFilesTool(),
        FindFilesTool(),
        RunCommandTool(),
        GitStatusTool(),
        GitDiffTool(),
        GitLogTool(),
        GitBranchTool(),
    ]