Update modules/treinamento.py

modules/treinamento.py

@@ -6,6 +6,7 @@
 ✅ Detecção de padrões de conversa
 ✅ Compatível com STT/TTS
 ✅ Otimizado para produção
+✅ CORREÇÃO: Suporte para nivel_transicao adicionado
 """
 
 import json
@@ -54,7 +55,7 @@ class Treinamento:
         logger.info(f"✅ Treinamento inicializado (intervalo: {interval_hours}h)")
 
     # ========================================================================
-    # 📝 REGISTRO DE INTERAÇÕES (ADAPTADO AO INDEX.JS)
+    # 📝 REGISTRO DE INTERAÇÕES (ADAPTADO AO INDEX.JS) - CORRIGIDO
     # ========================================================================
     
     def registrar_interacao(
@@ -69,7 +70,8 @@ class Treinamento:
         tipo_conversa: str = 'pv',
         tipo_mensagem: str = 'texto',
         reply_to_bot: bool = False,
-        reply_metadata: Optional[Dict] = None
+        reply_metadata: Optional[Dict] = None,
+        nivel_transicao: int = 0  # NOVO PARÂMETRO ADICIONADO
     ):
         """
         Registra interação para treinamento - TOTALMENTE COMPATÍVEL
@@ -86,6 +88,7 @@ class Treinamento:
             tipo_mensagem: 'texto', 'audio', etc
             reply_to_bot: Se é reply ao bot
             reply_metadata: Metadata do reply (do index.js)
+            nivel_transicao: Nível de transição do usuário privilegiado
         """
         try:
             numero = str(numero).strip()
@@ -98,11 +101,14 @@ class Treinamento:
             if reply_metadata and reply_metadata.get('reply_to_bot') is not None:
                 reply_to_bot = reply_metadata.get('reply_to_bot', False)
             
+            # Extrai info_transicao do contexto
+            info_transicao = contexto.get('info_transicao', {})
+            
             # Determina emoção e qualidade
             emocao_detectada, confianca_emocao = self._detectar_emocao(mensagem)
             qualidade = self._calcular_qualidade_resposta(mensagem, resposta, tipo_mensagem)
             
-            # Salva no banco usando método CORRETO
+            # Salva no banco usando método CORRETO com nivel_transicao
             self.db.salvar_mensagem(
                 usuario=usuario,
                 mensagem=mensagem,
@@ -115,12 +121,25 @@ class Treinamento:
                 modo_resposta=contexto.get('modo_resposta', 'normal_ironico'),
                 emocao_detectada=emocao_detectada,
                 confianca_emocao=confianca_emocao,
+                nivel_transicao=nivel_transicao,  # PARÂMETRO ADICIONADO
+                info_transicao=info_transicao,  # INFO DE TRANSIÇÃO
                 tipo_mensagem=tipo_mensagem,
                 usuario_nome=usuario,
                 tipo_conversa=tipo_conversa,
                 reply_info_json=json.dumps(reply_metadata) if reply_metadata else None
             )
             
+            # Atualiza contexto com nivel_transicao
+            self.db.atualizar_contexto(
+                numero=numero,
+                humor_atual=contexto.get('humor_atualizado', 'normal_ironico'),
+                modo_resposta=contexto.get('modo_resposta', 'normal_ironico'),
+                nivel_transicao=nivel_transicao,
+                info_transicao=info_transicao,
+                tom=contexto.get('tom', 'normal'),
+                emocao_tendencia=emocao_detectada
+            )
+            
             # Adiciona ao dataset se qualidade boa
             if qualidade >= QUALIDADE_MINIMA:
                 self._adicionar_ao_dataset(
@@ -136,16 +155,18 @@ class Treinamento:
                     tipo_conversa=tipo_conversa,
                     is_reply=is_reply,
                     reply_to_bot=reply_to_bot,
-                    reply_metadata=reply_metadata
+                    reply_metadata=reply_metadata,
+                    nivel_transicao=nivel_transicao  # ADICIONADO
                 )
             
-            # Salva exemplo de treinamento
+            # Salva exemplo de treinamento com nivel_transicao
             if tipo_mensagem == 'texto' and len(resposta) > 10:
                 self.db.salvar_training_example(
                     input_text=mensagem,
                     output_text=resposta,
                     humor=contexto.get('humor_atualizado', 'normal_ironico'),
                     modo_resposta=contexto.get('modo_resposta', 'normal_ironico'),
+                    nivel_transicao=nivel_transicao,  # ADICIONADO
                     emocao_contexto=emocao_detectada,
                     qualidade_score=qualidade,
                     contexto_super_claro={
@@ -153,10 +174,26 @@ class Treinamento:
                         'reply_to_bot': reply_to_bot,
                         'tipo_conversa': tipo_conversa,
                         'tipo_mensagem': tipo_mensagem,
-                        'reply_metadata': reply_metadata
+                        'reply_metadata': reply_metadata,
+                        'nivel_transicao': nivel_transicao,
+                        'info_transicao': info_transicao
                     }
                 )
             
+            # Registra interação para treinamento
+            self.db.registrar_interacao(
+                numero=numero,
+                mensagem=mensagem,
+                resposta=resposta,
+                humor=contexto.get('humor_atualizado', 'normal_ironico'),
+                modo_resposta=contexto.get('modo_resposta', 'normal_ironico'),
+                nivel_transicao=nivel_transicao,  # PARÂMETRO ADICIONADO
+                emocao_detectada=emocao_detectada,
+                tipo_conversa=tipo_conversa,
+                reply_info_json=json.dumps(reply_metadata) if reply_metadata else None,
+                qualidade_score=qualidade
+            )
+            
             # Analisa padrões
             self._analisar_padroes_usuario(
                 numero=numero,
@@ -168,10 +205,11 @@ class Treinamento:
                 tipo_conversa=tipo_conversa,
                 is_reply=is_reply,
                 reply_to_bot=reply_to_bot,
-                reply_metadata=reply_metadata
+                reply_metadata=reply_metadata,
+                nivel_transicao=nivel_transicao  # ADICIONADO
             )
             
-            logger.debug(f"✅ Interação registrada: {usuario[:10]} | reply: {is_reply} | reply_to_bot: {reply_to_bot}")
+            logger.debug(f"✅ Interação registrada: {usuario[:10]} | Nível: {nivel_transicao} | reply: {is_reply}")
             
         except Exception as e:
             logger.error(f"❌ Erro ao registrar interação: {e}")
@@ -258,7 +296,7 @@ class Treinamento:
         return round(qualidade, 2)
 
     # ========================================================================
-    # 💾 ADICIONAR AO DATASET
+    # 💾 ADICIONAR AO DATASET (ATUALIZADO)
     # ========================================================================
     
     def _adicionar_ao_dataset(
@@ -275,10 +313,11 @@ class Treinamento:
         tipo_conversa: str,
         is_reply: bool,
         reply_to_bot: bool,
-        reply_metadata: Optional[Dict] = None
+        reply_metadata: Optional[Dict] = None,
+        nivel_transicao: int = 0  # NOVO PARÂMETRO
     ):
         """
-        Adiciona exemplo ao dataset de treinamento
+        Adiciona exemplo ao dataset de treinamento - ATUALIZADO
         
         Args:
             mensagem: Mensagem do usuário
@@ -294,6 +333,7 @@ class Treinamento:
             is_reply: Se é reply
             reply_to_bot: Se é reply ao bot
             reply_metadata: Metadata do reply
+            nivel_transicao: Nível de transição do usuário
         """
         try:
             humor = contexto.get("humor_atualizado", "normal_ironico")
@@ -312,12 +352,13 @@ class Treinamento:
             if modo == "casual_amigavel":
                 modo = "normal_ironico"
             
-            # Prepara metadados com reply_metadata
+            # Prepara metadados com reply_metadata e nivel_transicao
             metadata = {
                 "usuario": usuario[:20],
                 "numero_hash": hashlib.md5(numero.encode()).hexdigest()[:8],
                 "humor": humor,
                 "modo_resposta": modo,
+                "nivel_transicao": nivel_transicao,  # ADICIONADO
                 "emocao_detectada": emocao_detectada,
                 "confianca_emocao": confianca_emocao,
                 "qualidade_score": qualidade,
@@ -337,6 +378,15 @@ class Treinamento:
                     "reply_metadata_context": reply_metadata.get('context_hint', '')
                 })
             
+            # Adiciona info_transicao se disponível
+            info_transicao = contexto.get('info_transicao', {})
+            if info_transicao:
+                metadata.update({
+                    "info_transicao_desc": info_transicao.get('desc', ''),
+                    "info_transicao_modo": info_transicao.get('modo', ''),
+                    "info_transicao_deve_transicionar": info_transicao.get('deve_transicionar', False)
+                })
+            
             entry = {
                 "input": mensagem.strip(),
                 "output": resposta.strip(),
@@ -369,13 +419,13 @@ class Treinamento:
                 with open(DATASET_PATH, "w", encoding="utf-8") as f:
                     json.dump(dataset, f, ensure_ascii=False, indent=2)
                 
-                logger.debug(f"✅ Exemplo adicionado ao dataset | qualidade: {qualidade:.2f}")
+                logger.debug(f"✅ Exemplo adicionado ao dataset | nível: {nivel_transicao} | qualidade: {qualidade:.2f}")
         
         except Exception as e:
             logger.warning(f"⚠️ Erro ao adicionar ao dataset: {e}")
 
     # ========================================================================
-    # 🔍 ANÁLISE DE PADRÕES
+    # 🔍 ANÁLISE DE PADRÕES (ATUALIZADA)
     # ========================================================================
     
     def _analisar_padroes_usuario(
@@ -389,10 +439,11 @@ class Treinamento:
         tipo_conversa: str,
         is_reply: bool,
         reply_to_bot: bool,
-        reply_metadata: Optional[Dict] = None
+        reply_metadata: Optional[Dict] = None,
+        nivel_transicao: int = 0  # NOVO PARÂMETRO
     ):
         """
-        Analisa padrões de comunicação do usuário
+        Analisa padrões de comunicação do usuário - ATUALIZADA
         
         Args:
             numero: Número do usuário
@@ -405,6 +456,7 @@ class Treinamento:
             is_reply: Se é reply
             reply_to_bot: Se é reply ao bot
             reply_metadata: Metadata do reply
+            nivel_transicao: Nível de transição do usuário
         """
         try:
             # 1. REGISTRAR TOM
@@ -425,22 +477,25 @@ class Treinamento:
                 except Exception as e:
                     logger.warning(f"Erro ao salvar gíria: {e}")
             
-            # 3. REGISTRAR HUMOR
+            # 3. REGISTRAR TRANSIÇÃO DE HUMOR COM nivel_transicao
             if "humor_atualizado" in contexto:
                 humor = contexto["humor_atualizado"]
                 humor_atual = self.db.recuperar_humor_atual(numero)
-                if humor != humor_atual:
+                nivel_atual = self.db.recuperar_nivel_transicao(numero)
+                
+                if humor != humor_atual or nivel_transicao != nivel_atual:
                     self.db.salvar_transicao_humor(
                         numero=numero,
                         humor_anterior=humor_atual,
                         humor_novo=humor,
+                        nivel_transicao_anterior=nivel_atual,
+                        nivel_transicao_novo=nivel_transicao,
                         emocao_trigger=emocao_detectada,
                         confianca_emocao=contexto.get('confianca_emocao', 0.5),
-                        nivel_transicao=contexto.get('nivel_transicao', 0),
-                        razao=f"Interação {tipo_conversa}"
+                        razao=f"Transição nível {nivel_atual}→{nivel_transicao} | {tipo_conversa}"
                     )
             
-            # 4. APRENDER PADRÕES DE REPLY
+            # 4. APRENDER PADRÕES DE REPLY COM nivel_transicao
             if is_reply:
                 self._aprender_padrao_reply(
                     numero=numero,
@@ -449,7 +504,18 @@ class Treinamento:
                     resposta=resposta,
                     reply_to_bot=reply_to_bot,
                     tipo_conversa=tipo_conversa,
-                    reply_metadata=reply_metadata
+                    reply_metadata=reply_metadata,
+                    nivel_transicao=nivel_transicao  # ADICIONADO
+                )
+            
+            # 5. ANALISAR TRANSIÇÕES DE USUÁRIOS PRIVILEGIADOS
+            if nivel_transicao > 0:
+                self._analisar_transicao_privilegiado(
+                    numero=numero,
+                    usuario=usuario,
+                    nivel_transicao=nivel_transicao,
+                    mensagem=mensagem,
+                    contexto=contexto
                 )
             
         except Exception as e:
@@ -525,10 +591,11 @@ class Treinamento:
         resposta: str,
         reply_to_bot: bool,
         tipo_conversa: str,
-        reply_metadata: Optional[Dict] = None
+        reply_metadata: Optional[Dict] = None,
+        nivel_transicao: int = 0  # NOVO PARÂMETRO
     ):
         """
-        Aprende padrões de reply
+        Aprende padrões de reply - ATUALIZADO
         
         Args:
             numero: Número do usuário
@@ -538,6 +605,7 @@ class Treinamento:
             reply_to_bot: Se é reply ao bot
             tipo_conversa: Tipo da conversa
             reply_metadata: Metadata do reply
+            nivel_transicao: Nível de transição do usuário
         """
         try:
             # Define padrão com base no reply_metadata
@@ -561,10 +629,13 @@ class Treinamento:
                     tipo = "conversa_alheia"
                     contexto_extra = ""
             
+            # Adiciona info de transição se disponível
+            transicao_info = f"[Nível transição: {nivel_transicao}]" if nivel_transicao > 0 else ""
+            
             # Prepara texto com contexto
-            input_text_com_contexto = f"[CONTEXTO: {padrao.upper()}] {contexto_extra} {mensagem}"
+            input_text_com_contexto = f"[CONTEXTO: {padrao.upper()}] {transicao_info} {contexto_extra} {mensagem}"
             
-            # Salva aprendizado
+            # Salva aprendizado com nivel_transicao
             self.db.salvar_aprendizado_detalhado(
                 input_text=input_text_com_contexto,
                 output_text=resposta,
@@ -575,17 +646,64 @@ class Treinamento:
                     'reply_to_bot': reply_to_bot,
                     'tipo_conversa': tipo_conversa,
                     'tipo': tipo,
-                    'reply_metadata': reply_metadata
+                    'reply_metadata': reply_metadata,
+                    'nivel_transicao': nivel_transicao  # ADICIONADO
                 },
                 qualidade_score=0.8,
-                tipo_aprendizado=f"reply_{tipo}"
+                tipo_aprendizado=f"reply_{tipo}_nivel_{nivel_transicao}"
             )
             
-            logger.debug(f"��� Padrão de reply aprendido: {padrao}")
+            logger.debug(f"✅ Padrão de reply aprendido: {padrao} | Nível: {nivel_transicao}")
             
         except Exception as e:
             logger.warning(f"⚠️ Erro ao aprender padrão de reply: {e}")
 
+    def _analisar_transicao_privilegiado(
+        self,
+        numero: str,
+        usuario: str,
+        nivel_transicao: int,
+        mensagem: str,
+        contexto: Dict
+    ):
+        """
+        Analisa transições de usuários privilegiados
+        
+        Args:
+            numero: Número do usuário
+            usuario: Nome do usuário
+            nivel_transicao: Nível atual de transição
+            mensagem: Mensagem enviada
+            contexto: Contexto da conversa
+        """
+        try:
+            # Recupera histórico de transições
+            transicoes = self.db._execute_with_retry(
+                """
+                SELECT nivel_transicao_anterior, nivel_transicao_novo, timestamp, razao
+                FROM transicoes_humor
+                WHERE numero = ?
+                ORDER BY timestamp DESC
+                LIMIT 10
+                """,
+                (numero,),
+                fetch=True
+            )
+            
+            # Analisa padrão de transição
+            if len(transicoes) >= 3:
+                niveis = [t[1] for t in transicoes]  # Últimos níveis novos
+                mudancas = sum(1 for i in range(len(niveis)-1) if niveis[i] != niveis[i+1])
+                
+                # Se muitas mudanças, usuário é volátil
+                if mudancas >= 2:
+                    logger.info(f"⚠️ Usuário {usuario} é volátil em transições: {mudancas} mudanças")
+            
+            logger.debug(f"📊 Transição privilegiado: {usuario} → Nível {nivel_transicao}")
+            
+        except Exception as e:
+            logger.warning(f"⚠️ Erro na análise de transição: {e}")
+
     # ========================================================================
     # 🔄 TREINAMENTO PERIÓDICO
     # ========================================================================
@@ -636,7 +754,7 @@ class Treinamento:
                 logger.warning("⚠️ Nenhum exemplo com qualidade suficiente")
                 return
 
-            # Gera arquivo JSONL
+            # Gera arquivo JSONL com nivel_transicao
             with open("training_dataset_akira_v21.jsonl", "w", encoding="utf-8") as f:
                 for ex in exemplos[:500]:
                     if ex.get("score", 0) >= 0.7:
@@ -645,6 +763,7 @@ class Treinamento:
                             "output": ex.get("output", ""),
                             "humor": ex.get("humor", "normal_ironico"),
                             "modo": ex.get("modo", "normal_ironico"),
+                            "nivel_transicao": ex.get("nivel_transicao", 0),  # ADICIONADO
                             "metadata": {
                                 "score": ex.get("score", 0.5),
                                 "timestamp": time.time(),
@@ -652,7 +771,7 @@ class Treinamento:
                             }
                         }, ensure_ascii=False) + "\n")
             
-            logger.info(f"✅ Dataset gerado: {len(exemplos)} exemplos")
+            logger.info(f"✅ Dataset gerado: {len(exemplos)} exemplos (com nível transição)")
             self.db.marcar_examples_como_usados()
             
         except Exception as e:
@@ -667,7 +786,7 @@ class Treinamento:
             with open(DATASET_PATH, "r", encoding="utf-8") as f:
                 dataset = json.load(f)
 
-            # Análise estatística
+            # Análise estatística com nivel_transicao
             padroes = {
                 "total": len(dataset),
                 "reply_to_bot": 0,
@@ -676,7 +795,11 @@ class Treinamento:
                 "grupo": 0,
                 "pv": 0,
                 "audio": 0,
-                "texto": 0
+                "texto": 0,
+                "transicao_nivel_0": 0,
+                "transicao_nivel_1": 0,
+                "transicao_nivel_2": 0,
+                "transicao_nivel_3": 0
             }
             
             for e in dataset:
@@ -698,6 +821,17 @@ class Treinamento:
                     padroes["audio"] += 1
                 else:
                     padroes["texto"] += 1
+                
+                # Analisa nível de transição
+                nivel = meta.get("nivel_transicao", 0)
+                if nivel == 0:
+                    padroes["transicao_nivel_0"] += 1
+                elif nivel == 1:
+                    padroes["transicao_nivel_1"] += 1
+                elif nivel == 2:
+                    padroes["transicao_nivel_2"] += 1
+                elif nivel == 3:
+                    padroes["transicao_nivel_3"] += 1
             
             # Log estatísticas
             logger.info(f"📊 Estatísticas do dataset:")
@@ -707,6 +841,7 @@ class Treinamento:
             logger.info(f"  Com reply_metadata: {padroes['reply_with_metadata']}")
             logger.info(f"  Grupo: {padroes['grupo']} | PV: {padroes['pv']}")
             logger.info(f"  Áudio: {padroes['audio']} | Texto: {padroes['texto']}")
+            logger.info(f"  Níveis transição: 0={padroes['transicao_nivel_0']} | 1={padroes['transicao_nivel_1']} | 2={padroes['transicao_nivel_2']} | 3={padroes['transicao_nivel_3']}")
             
         except Exception as e:
             logger.error(f"❌ Erro na análise global: {e}")
@@ -721,12 +856,12 @@ class Treinamento:
             logger.warning(f"⚠️ Erro na otimização: {e}")
 
     # ========================================================================
-    # 🔧 FUNÇÃO PARA USO DIRETO DA API
+    # 🔧 FUNÇÃO PARA USO DIRETO DA API (ATUALIZADA)
     # ========================================================================
     
     def processar_interacao_api(self, payload: Dict, resposta: str) -> Dict:
         """
-        Processa interação da API para treinamento
+        Processa interação da API para treinamento - ATUALIZADA
         
         Args:
             payload: Payload da requisição
@@ -753,10 +888,11 @@ class Treinamento:
             
             is_reply = bool(payload.get('mensagem_citada')) or bool(reply_metadata)
             
-            # Contexto da análise
+            # Contexto da análise com nivel_transicao
             contexto_analise = payload.get('analise', {})
+            nivel_transicao = contexto_analise.get('nivel_transicao', 0)
             
-            # Registra interação
+            # Registra interação com nivel_transicao
             self.registrar_interacao(
                 usuario=usuario,
                 mensagem=mensagem,
@@ -768,13 +904,15 @@ class Treinamento:
                 tipo_conversa=tipo_conversa,
                 tipo_mensagem=tipo_mensagem,
                 reply_to_bot=reply_to_bot,
-                reply_metadata=reply_metadata
+                reply_metadata=reply_metadata,
+                nivel_transicao=nivel_transicao  # ADICIONADO
             )
             
             return {
                 'status': 'success',
                 'message': 'Interação registrada',
                 'usuario': usuario,
+                'nivel_transicao': nivel_transicao,
                 'timestamp': time.time()
             }
             
@@ -809,7 +947,7 @@ def get_treinamento_instance(db: Database = None):
     return _treinamento_instance
 
 # ============================================================================
-# 🎯 FUNÇÃO DE INTEGRAÇÃO RÁPIDA
+# 🎯 FUNÇÃO DE INTEGRAÇÃO RÁPIDA (ATUALIZADA)
 # ============================================================================
 def registrar_interacao_rapida(
     usuario: str,
@@ -821,10 +959,11 @@ def registrar_interacao_rapida(
     tipo_conversa: str = 'pv',
     tipo_mensagem: str = 'texto',
     contexto: Dict = None,
-    reply_metadata: Optional[Dict] = None
+    reply_metadata: Optional[Dict] = None,
+    nivel_transicao: int = 0  # NOVO PARÂMETRO
 ) -> bool:
     """
-    Registra interação rapidamente
+    Registra interação rapidamente - ATUALIZADA
     
     Args:
         usuario: Nome do usuário
@@ -837,6 +976,7 @@ def registrar_interacao_rapida(
         tipo_mensagem: Tipo da mensagem
         contexto: Contexto da conversa
         reply_metadata: Metadata do reply
+        nivel_transicao: Nível de transição do usuário
         
     Returns:
         True se sucesso, False caso contrário
@@ -853,9 +993,10 @@ def registrar_interacao_rapida(
             tipo_conversa=tipo_conversa,
             tipo_mensagem=tipo_mensagem,
             contexto=contexto,
-            reply_metadata=reply_metadata
+            reply_metadata=reply_metadata,
+            nivel_transicao=nivel_transicao  # ADICIONADO
         )
-        logger.debug(f"✅ Interação rápida registrada: {usuario[:10]}")
+        logger.debug(f"✅ Interação rápida registrada: {usuario[:10]} | Nível: {nivel_transicao}")
         return True
     except Exception as e:
         logger.error(f"❌ Erro no registro rápido: {e}")
@@ -866,7 +1007,7 @@ def registrar_interacao_rapida(
 # ============================================================================
 if __name__ == "__main__":
     print("=" * 80)
-    print("TESTANDO TREINAMENTO.PY - CORRIGIDO E COMPLETO")
+    print("TESTANDO TREINAMENTO.PY - COMPLETO COM nivel_transicao")
     print("=" * 80)
     
     from .database import Database
@@ -876,7 +1017,7 @@ if __name__ == "__main__":
         db = Database(":memory:")
         treinamento = Treinamento(db)
         
-        # Simula payload do api.py com reply_metadata
+        # Simula payload do api.py com reply_metadata e nivel_transicao
         payload_teste = {
             "usuario": "Isaac Teste",
             "numero": "244978787009",
@@ -891,20 +1032,27 @@ if __name__ == "__main__":
             },
             "analise": {
                 "humor_atualizado": "normal_ironico",
-                "modo_resposta": "normal_ironico"
+                "modo_resposta": "normal_ironico",
+                "nivel_transicao": 2,
+                "info_transicao": {
+                    "desc": "Nível 2 - Formal Relaxado",
+                    "modo": "tecnico_formal",
+                    "deve_transicionar": False
+                }
             }
         }
         
         resposta_teste = "Tudo e tu, puto?"
         
-        # Processa interação
+        # Processa interação com nivel_transicao
         resultado = treinamento.processar_interacao_api(payload_teste, resposta_teste)
         
         print(f"✅ Teste OK: {resultado}")
         print(f"📝 Mensagem: {payload_teste['mensagem']}")
         print(f"💬 Resposta: {resposta_teste}")
+        print(f"🎯 Nível transição: {payload_teste['analise']['nivel_transicao']}")
         
-        # Teste com registro rápido
+        # Teste com registro rápido com nivel_transicao
         sucesso = registrar_interacao_rapida(
             usuario="Teste 2",
             numero="244000000000",
@@ -912,7 +1060,8 @@ if __name__ == "__main__":
             resposta="Nada, cota.",
             is_reply=True,
             reply_to_bot=True,
-            reply_metadata={"quoted_author_name": "Akira", "is_reply": True}
+            reply_metadata={"quoted_author_name": "Akira", "is_reply": True},
+            nivel_transicao=3
         )
         
         print(f"✅ Registro rápido: {'Sucesso' if sucesso else 'Falhou'}")
@@ -923,5 +1072,5 @@ if __name__ == "__main__":
         traceback.print_exc()
     
     print("\n" + "=" * 80)
-    print("TREINAMENTO.PY - COMPLETO E PRONTO PARA USO")
+    print("TREINAMENTO.PY - COMPLETO COM SUPORTE A nivel_transicao")
     print("=" * 80)