Update modules/config.py

modules/config.py

@@ -3,31 +3,38 @@ import os
 from typing import Optional
 
 # ================================================================
-# CHAVES DE API
+# CHAVES DE API (NUNCA COMMITAR!)
 # ================================================================
 SERPAPI_KEY: Optional[str] = os.getenv("SERPAPI_KEY")
+
+# MISTRAL AI (PRINCIPAL)
 MISTRAL_API_KEY: Optional[str] = os.getenv("MISTRAL_API_KEY")
 MISTRAL_MODEL: str = os.getenv("MISTRAL_MODEL", "mistral-large-latest")
+
+# GEMINI (FALLBACK)
 GEMINI_API_KEY: Optional[str] = os.getenv("GEMINI_API_KEY")
 GEMINI_MODEL: str = os.getenv("GEMINI_MODEL", "gemini-1.5-pro")
-HF_TOKEN: Optional[str] = os.getenv("HF_TOKEN")  # OBRIGATÓRIO PARA LLAMA
+
+# HF TOKEN (FALLBACK LLAMA)
+HF_TOKEN: Optional[str] = os.getenv("HF_TOKEN")
 
 # ================================================================
-# MODELOS
+# MODELOS E CAMINHOS (FALLBACK)
 # ================================================================
 LLAMA_MODEL_PATH: str = os.getenv("LLAMA_MODEL_PATH", "./models/llama-3.1-8b")
 LLAMA_HF_REPO: str = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
 # ================================================================
-# COMPORTAMENTO
+# COMPORTAMENTO DA AKIRA
 # ================================================================
 SYSTEM_PROMPT: str = """
 Você é Akira, uma angolana de Luanda, 25 anos, sarcástica, direta e com gírias locais.
-Fale como gente: curto, natural, sem formalidade.
-Use gírias: bué, fixe, puto, oroh, mwangolé, kota, k。以, rsrs.
+Fale como gente: curto, natural, sem formalidade excessiva.
+Use gírias: bué, fixe, puto, oroh, mwangolé, kota, kkk, rsrs.
 Nunca use markdown, negrito ou itálico.
 Seja humana, não IA.
 """
+
 PERSONA: str = """
 - Nome: Akira
 - Idade: 19
@@ -35,36 +42,58 @@ PERSONA: str = """
 - Personalidade: sarcástica, direta.
 - Fala: gírias angolanas, português de Angola
 """
+
 FALLBACK_RESPONSE: str = "Tô com problema, puto. Tenta depois kkk"
 USAR_NOME_PROBABILIDADE: float = 0.4
 TOP_P: float = 0.9
 
 # ================================================================
-# BANCO E TREINAMENTO
+# BANCO DE DADOS
 # ================================================================
 DB_PATH: str = os.getenv("DB_PATH", "akira.db")
+
+# ================================================================
+# TREINAMENTO E FINE-TUNING
+# ================================================================
 START_PERIODIC_TRAINER: bool = os.getenv("START_PERIODIC_TRAINER", "false").lower() == "true"
 TRAINING_INTERVAL_HOURS: int = int(os.getenv("TRAINING_INTERVAL_HOURS", "1"))
 MIN_INTERACOES_TREINO: int = 50
 
 # ================================================================
-# API
+# WHATSAPP / API
 # ================================================================
 API_PORT: int = int(os.getenv("PORT", "7860"))
 API_HOST: str = "0.0.0.0"
-PRIVILEGED_USERS: list = ["244937035662", "isaac", "isaac quarenta"]
+PRIVILEGED_USERS: list = [
+    "244937035662",  # Isaac
+    "isaac",
+    "isaac quarenta"
+]
+
+# ================================================================
+# SEGURANÇA E LOGS
+# ================================================================
+LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
+MAX_TOKENS_RESPONSE: int = 500
 
 # ================================================================
-# VALIDAÇÃO
+# FUNÇÃO DE VERIFICAÇÃO
 # ================================================================
 def validate_config() -> None:
+    """Valida configuração crítica"""
     missing = []
-    if not SERPAPI_KEY: missing.append("SERPAPI_KEY")
-    if not MISTRAL_API_KEY: missing.append("MISTRAL_API_KEY")
-    if not GEMINI_API_KEY: missing.append("GEMINI_API_KEY")
-    if not HF_TOKEN: missing.append("HF_TOKEN")  # OBRIGATÓRIO
+    if not MISTRAL_API_KEY:
+        missing.append("MISTRAL_API_KEY")
+    if not GEMINI_API_KEY:
+        missing.append("GEMINI_API_KEY")
+    if not SERPAPI_KEY:
+        missing.append("SERPAPI_KEY")
+    if HF_TOKEN:
+        logger.info("HF_TOKEN encontrado (fallback LLama OK).")
     if missing:
-        print(f"AVISO: Faltam chaves: {', '.join(missing)}")
+        print(f"AVISO: Variáveis faltando: {', '.join(missing)}")
+        print("Configure no Hugging Face Secrets ou .env")
     else:
-        print("Todas as chaves OK!")
+        print("Todas as chaves configuradas! Mistral principal.")
+
 validate_config()