feat: melhorando logs do contextual

_utils/gerar_documento_utils/contextual_retriever.py

@@ -69,11 +69,11 @@ class ContextualRetriever:
         utils = self.contextual_retriever_utils
 
         try:
-            print("\n\nCOMEÇOU A REQUISIÇÃO")
             prompt = contextual_prompt(
                 resumo_auxiliar, all_chunks_contents, len(lista_com_20_chunks)
             )
             result = None
+
             for attempt in range(4):
                 if attempt != 0:
                     send_axiom(
@@ -86,7 +86,7 @@ class ContextualRetriever:
                 raw_response = await agemini_answer(prompt, "gemini-2.0-flash-lite")
                 response = cast(str, raw_response)
                 send_axiom(
-                    f"TERMINOU UMA REQUISIÇÃO DO CONTEXTUAL - TENTATIVA {attempt + 1} ------- RESPOSTA LLM DE UM ITEM COM 20 CHUNKS EM TEXTO DA LISTA : {response}"
+                    f"TERMINOU UMA REQUISIÇÃO DO CONTEXTUAL - TENTATIVA {attempt + 1}"
                 )
 
                 matches = utils.validate_many_chunks_in_one_request(
@@ -94,7 +94,9 @@ class ContextualRetriever:
                 )
 
                 if matches:
-                    send_axiom(f"CONTEXTUAL FUNCIONOU NA TENTATIVA {attempt + 1}")
+                    send_axiom(
+                        f"VALIDAÇÃO DO CONTEXTUAL FUNCIONOU NA TENTATIVA {attempt + 1} (ou seja, a função validate_many_chunks_in_one_request)"
+                    )
 
                     result = utils.get_info_from_validated_chunks(matches)
 
@@ -102,7 +104,7 @@ class ContextualRetriever:
 
             if result is None:
                 axiom_instance.send_axiom(
-                    f"-------------- UMA LISTA DE 20 CHUNKS FALHOU AS 4x NA FORMATAÇÃO ------------- {response}"
+                    f"-------------- UMA LISTA DE 20 CHUNKS FALHOU AS 4x NA FORMATAÇÃO ------------- ÚLTIMO RETORNO ERRADO: {response}"
                 )
                 result = [[""]]  # default value if no iteration succeeded
 
@@ -141,7 +143,7 @@ class ContextualRetriever:
                 )
         except BaseException as e:
             axiom_instance.send_axiom(
-                f"ERRO EM UMA LISTA COM 20 CHUNKS CONTEXTUALS --------- lista: {lista_com_20_chunks} ------------ resultado da llm_call_uma_lista_de_20_chunks: {result} ------- erro: {e}"
+                f"ERRO EM UMA LISTA COM 20 CHUNKS CONTEXTUALS --------- lista: {lista_com_20_chunks} ------------ ERRO: {e}"
             )
 
         return lista_chunks
@@ -175,11 +177,13 @@ class ContextualRetriever:
                 for lista_com_20_chunks in lista_de_listas_cada_com_20_chunks
             ]
 
-        # contextualized_chunks = [task.result() for task in tasks]
         contextualized_chunks: List[ContextualizedChunk] = []
         for task in tasks:
             contextualized_chunks = contextualized_chunks + task.result()
 
+        axiom_instance.send_axiom(
+            "TERMINOU COM SUCESSO DE PROCESSAR TUDO DOS CONTEXTUALS"
+        )
         return contextualized_chunks