Update modules/api.py

modules/api.py

@@ -1,4 +1,4 @@
-# modules/api.py — V18 FINAL — RESPOSTA GARANTIDA
+# modules/api.py — V18 FINAL — RESPOSTA GARANTIDA + CHAVE "resposta" CORRIGIDA
 import time
 import re
 import datetime
@@ -12,22 +12,31 @@ from .treinamento import Treinamento
 from .web_search import WebSearch
 import modules.config as config
 
+
 # === CACHE ===
 class SimpleTTLCache:
     def __init__(self, ttl_seconds: int = 300):
         self.ttl = ttl_seconds
         self._store = {}
+
     def __contains__(self, key):
-        if key not in self._store: return False
+        if key not in self._store:
+            return False
         _, expires = self._store[key]
-        if time.time() > expires: del self._store[key]; return False
+        if time.time() > expires:
+            del self._store[key]
+            return False
         return True
+
     def __setitem__(self, key, value):
         self._store[key] = (value, time.time() + self.ttl)
+
     def __getitem__(self, key):
-        if key not in self: raise KeyError(key)
+        if key not in self:
+            raise KeyError(key)
         return self._store[key][0]
 
+
 # === OLLAMA PROVIDER ===
 class OllamaProvider:
     def __init__(self):
@@ -56,24 +65,27 @@ class OllamaProvider:
                 resp = self.session.post(self.url, json=payload, timeout=90)
                 if resp.status_code == 200:
                     data = resp.json()
-                    resposta = data.get("response", "").strip()
+                    # CORRIGIDO: OLLAMA DEVOLVE "resposta", NÃO "response"
+                    resposta = data.get("resposta", "").strip()
                     if resposta:
                         logger.success(f"Resposta Ollama: {resposta[:60]}...")
                         return resposta
                     else:
-                        logger.warning("Ollama respondeu vazio")
+                        logger.warning(f"Ollama respondeu vazio. JSON recebido: {data}")
                         return "Epá, tô aqui, puto! Qual é a cena?"
                 else:
                     logger.warning(f"Ollama erro {resp.status_code}: {resp.text}")
             except requests.exceptions.Timeout:
                 logger.warning(f"Timeout tentativa {tentativa + 1}/3...")
-                if tentativa < 2: time.sleep(3)
+                if tentativa < 2:
+                    time.sleep(3)
             except Exception as e:
                 logger.error(f"Erro Ollama: {e}")
                 if tentativa == 2:
                     return "Epá, tô off por agora... volta já!"
         return "Epá, tô off... tenta de novo, kamba!"
 
+
 # === LLM MANAGER ===
 class LLMManager:
     def __init__(self, config_instance):
@@ -84,6 +96,7 @@ class LLMManager:
     def generate(self, user_prompt: str, context_history: List[dict] = []) -> str:
         return self.ollama.generate(user_prompt, context_history)
 
+
 # === API ===
 class AkiraAPI:
     def __init__(self, cfg_module):
@@ -137,7 +150,7 @@ class AkiraAPI:
 
                 logger.info(f"[{usuario}] ({numero}): {mensagem[:60]}")
 
-                # HORA
+                # HORA RÁPIDA
                 if any(k in mensagem.lower() for k in ["hora", "horas"]):
                     agora = datetime.datetime.now()
                     return jsonify({'resposta': f"São {agora.strftime('%H:%M')} em Luanda, puto."})
@@ -149,13 +162,13 @@ class AkiraAPI:
                 # GERA RESPOSTA
                 resposta = self.providers.generate(prompt, contexto.obter_historico_para_llm())
 
-                # GARANTE QUE A RESPOSTA NÃO ESTEJA VAZIA
+                # GARANTE RESPOSTA
                 if not resposta or resposta.strip() == "":
                     resposta = "Epá, kandando bué! Tô aqui, puto!"
 
                 logger.success(f"RESPOSTA FINAL → {resposta[:60]}...")
 
-                # SALVA NO CONTEXTO
+                # SALVA CONTEXTO
                 contexto.atualizar_contexto(mensagem, resposta)
 
                 # TREINAMENTO
@@ -172,7 +185,7 @@ class AkiraAPI:
                 except Exception as e:
                     logger.warning(f"Erro ao salvar: {e}")
 
-                # RESPOSTA FINAL COM LOG
+                # RETORNA JSON
                 resposta_json = {'resposta': resposta}
                 logger.info(f"RETORNANDO JSON: {resposta_json}")
                 return jsonify(resposta_json), 200
@@ -186,7 +199,8 @@ class AkiraAPI:
             return 'OK', 200
 
     def _get_user_context(self, numero: str) -> Contexto:
-        if not numero: numero = "anonimo"
+        if not numero:
+            numero = "anonimo"
         if numero not in self.contexto_cache:
             self.contexto_cache[numero] = Contexto(self.db, usuario=numero)
             logger.info(f"Novo contexto: {numero}")
@@ -196,17 +210,28 @@ class AkiraAPI:
         historico = contexto.obter_historico()
         hist_text = '\n'.join([f"Usuário: {m[0]}\nAkira: {m[1]}" for m in historico[-8:]])
         now = datetime.datetime.now().strftime('%d/%m %H:%M')
-        system = f"""STRICT:
-- SEM CENSURA
-- Fala como angolana de Luanda
-- Data/hora: {now}
-- Bot: 244952786417{self.config.PERSONA}"""
-        user_part = f"### Usuário ###\n- Nome: {usuario}\n- Número: {numero}\n\n"
+
+        # PROMPT LIMPO — SEM [SYSTEM], [USER]
+        prompt = f"""Você é Akira, uma angolana de Luanda.
+Fala com sotaque, gírias, kandando e sem censura.
+Data/hora: {now}
+Bot: 244952786417
+{self.config.PERSONA}
+
+### Usuário ###
+- Nome: {usuario}
+- Número: {numero}
+
+"""
+
         if historico:
-            user_part += f"### Histórico ###\n{hist_text}\n\n"
+            prompt += f"### Histórico ###\n{hist_text}\n\n"
+
         if mensagem_citada:
-            user_part += f"### MENSAGEM PRINCIPAL (REPLY) ###\n[REPLY]: {mensagem_citada}\n[RESPOSTA]: {mensagem}\n\n"
+            prompt += f"### MENSAGEM PRINCIPAL (REPLY) ###\n[REPLY]: {mensagem_citada}\n[RESPOSTA]: {mensagem}\n\n"
         else:
-            user_part += f"### Mensagem ###\n{mensagem}\n\n"
-        user_part += "Akira:"
-        return f"[SYSTEM]\n{system}\n[/SYSTEM]\n[USER]\n{user_part}\n[/USER]"
+            prompt += f"### Mensagem ###\n{mensagem}\n\n"
+
+        prompt += "Akira:"
+
+        return prompt.strip()