feat: adicionar sistema de feedback por comentário do agente

2026-05-16 08:35:54 +02:00
parent 9288f1e92b
commit 364a510272
17 changed files with 406 additions and 181 deletions
@@ -0,0 +1,96 @@
+import os
+import json
+import redis
+import argparse
+import subprocess
+from anthropic import Anthropic
+from key_manager import KeyManager
+
+# Configurações de Caminho
+BASE_DIR = "/home/deploy/projetos"
+INSTRUCOES_DIR = "/home/deploy/instrucoes"
+
+class SwarmAgent:
+    def __init__(self, task_data):
+        self.data = task_data
+        self.repo_name = task_data.get('repository')
+        self.repo_url = f"http://10.99.0.3:3000/{self.repo_name}.git"
+        self.local_path = os.path.join(BASE_DIR, self.repo_name.split('/')[-1])
+        self.km = KeyManager()
+        
+    def setup_workspace(self):
+        """Garante que o código e as instruções estão atualizados"""
+        print(f" [>] Sincronizando repositório: {self.repo_name}")
+        os.makedirs(BASE_DIR, exist_ok=True)
+        
+        if not os.path.exists(self.local_path):
+            subprocess.run(["git", "clone", self.repo_url, self.local_path])
+        else:
+            subprocess.run(["git", "-C", self.local_path, "pull"])
+            
+        # Sincroniza instruções globais
+        subprocess.run(["git", "-C", INSTRUCOES_DIR, "pull"])
+
+    def get_system_prompt(self):
+        """Carrega a doutrina e as instruções de cada VPS"""
+        with open(f"{INSTRUCOES_DIR}/global/doutrina_agentes.md", "r") as f:
+            doutrina = f.read()
+        with open(f"{INSTRUCOES_DIR}/global/claude.md", "r") as f:
+            especifico = f.read()
+            
+        ip_vps = subprocess.getoutput("hostname -I").split()[0]
+        
+        return f"""
+        {doutrina}
+        {especifico}
+        
+        CONTEXTO ATUAL:
+        - Você está na VPS com IP: {ip_vps}
+        - Projeto: {self.repo_name}
+        - Tarefa (Issue): {json.dumps(self.data.get('data', {}).get('issue', {}))}
+        """
+
+    def post_comment(self, message):
+        """Posta um comentário de feedback na Issue do Gitea"""
+        issue_id = self.data.get('data', {}).get('issue', {}).get('number')
+        if not issue_id:
+            return
+
+        url = f"http://10.99.0.3:3000/api/v1/repos/{self.repo_name}/issues/{issue_id}/comments"
+        auth = ('ruicesar', 'h$tg@g5aga$ra1E3$C-yHW$-BA@DF2@Grfa!3#')
+        
+        try:
+            import requests
+            response = requests.post(url, auth=auth, json={"body": message})
+            if response.status_code == 201:
+                print(f" [✓] Comentário postado na Issue #{issue_id}")
+            else:
+                print(f" [X] Erro ao postar comentário: {response.status_code}")
+        except Exception as e:
+            print(f" [E] Falha na comunicação com Gitea: {e}")
+
+    def execute(self):
+        self.setup_workspace()
+        
+        api_key = self.km.get_key('anthropic')
+        if not api_key:
+            return
+
+        client = Anthropic(api_key=api_key)
+        prompt = self.get_system_prompt()
+
+        print(" [!] IA Pensando na solução...")
+        
+        # Simulação de resposta da IA para o teste
+        feedback = "### 🤖 [VPS 4 - LABORATÓRIO]:\n* **Status**: Tarefa recebida e ambiente preparado.\n* **Ação**: Sincronização de arquivos concluída.\n* **Próximo**: Iniciando correção do erro de regressão de versão."
+        
+        self.post_comment(feedback)
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--data", help="Dados do Webhook JSON")
+    args = parser.parse_args()
+    
+    if args.data:
+        agent = SwarmAgent(json.loads(args.data))
+        agent.execute()