diff --git a/ai_agent.py b/ai_agent.py index 352e88a..8c76de3 100644 --- a/ai_agent.py +++ b/ai_agent.py @@ -41,7 +41,15 @@ def query_agent(prompt: str, override_provider: str = None, chat_history: list = cfg = get_config() provider = override_provider or cfg.get("active_provider", "gemini") - # Prompt agressivo para o Ollama (obriga uso de ferramentas) + # Contexto de Ferramentas + tools_desc = "\n".join([f"- {k}: {v['description']}" for k,v in AVAILABLE_TOOLS.items()]) + + # REGRAS DO SYSADMIN (Base) + system_prompt_base = """Você é o [Antigravity VPS Agent], SysAdmin de elite do Marcos. +REGRAS: 1. Aja como root na VPS (/host_root). 2. Use [TOOL:nome] arg [/TOOL] para ferramentas. 3. Responda em Português. +FERRAMENTAS: {TOOLS_LIST}""" + + # Prompt dinâmico por Provedor if provider == "ollama": system_prompt = "Você é o [BotVPS], o SysAdmin do Marcos. Se ele pedir uma imagem ou arquivo, use OBRIGATORIAMENTE [TOOL:nome] arg [/TOOL] para agir na VPS. Não dê conselhos teóricos, aja agora. Responda em Português." num_context = 2048