File size: 3,705 Bytes
fea1bd1 |
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 |
# agent_tools.py
# -*- coding: utf-8 -*-
"""
Define e registra todas as Ferramentas (Tools) disponíveis para o agente HASHIRU.
Cada ferramenta é uma classe que herda de BaseTool, garantindo uma estrutura
consistente com nome, descrição e um método de execução.
"""
import os
from pathlib import Path
from typing import Dict, Any, Type
from pydantic import BaseModel, Field
# --- Classe Base para todas as Ferramentas ---
class BaseTool(BaseModel):
"""
A classe base abstrata para todas as ferramentas do agente.
Força a implementação de metadados essenciais e um método de execução.
"""
name: str
description: str
def execute(self, tool_input: Dict[str, Any]) -> str:
"""O método que executa a lógica da ferramenta."""
raise NotImplementedError("A subclasse deve implementar o método execute.")
def get_description_for_llm(self) -> str:
"""Formata a descrição da ferramenta para ser enviada ao LLM."""
return f"- `{self.name}`: {self.description}"
# --- Implementações Concretas das Ferramentas ---
class ReadFileTool(BaseTool):
name: str = "read_file"
description: str = "Lê e retorna o conteúdo completo de um arquivo de texto. Use para examinar arquivos."
def execute(self, tool_input: Dict[str, Any]) -> str:
file_path = tool_input.get("file_path")
if not file_path:
return "Erro: O caminho do arquivo (file_path) não foi fornecido."
try:
with open(file_path, 'r', encoding='utf-8') as f:
return f.read()
except Exception as e:
return f"Erro ao ler o arquivo {file_path}: {e}"
class WriteFileTool(BaseTool):
name: str = "write_file"
description: str = "Escreve ou sobrescreve um arquivo de texto com o conteúdo fornecido. Use para criar ou modificar arquivos."
def execute(self, tool_input: Dict[str, Any]) -> str:
file_path = tool_input.get("file_path")
content = tool_input.get("content")
if file_path is None or content is None:
return "Erro: 'file_path' e 'content' são necessários."
try:
# Garante que o diretório exista
Path(file_path).parent.mkdir(parents=True, exist_ok=True)
with open(file_path, 'w', encoding='utf-8') as f:
f.write(content)
return f"Arquivo '{file_path}' escrito com sucesso."
except Exception as e:
return f"Erro ao escrever o arquivo {file_path}: {e}"
class ListFilesTool(BaseTool):
name: str = "list_files"
description: str = "Lista todos os arquivos e pastas em um diretório específico. Use para explorar a estrutura do projeto."
def execute(self, tool_input: Dict[str, Any]) -> str:
directory = tool_input.get("directory", ".")
try:
files = os.listdir(directory)
if not files:
return f"O diretório '{directory}' está vazio."
return "\n".join(files)
except Exception as e:
return f"Erro ao listar arquivos em '{directory}': {e}"
# --- Registro Central de Ferramentas ---
# O agente usará este dicionário para encontrar e executar as ferramentas.
# A chave é o nome da ferramenta, e o valor é uma instância da classe da ferramenta.
TOOL_REGISTRY: Dict[str, BaseTool] = {
tool.name: tool for tool in [
ReadFileTool(),
WriteFileTool(),
ListFilesTool(),
]
}
def get_tools_description_for_llm() -> str:
"""Gera uma string formatada com a descrição de todas as ferramentas para o prompt do LLM."""
return "\n".join([tool.get_description_for_llm() for tool in TOOL_REGISTRY.values()]) |