_utils/axiom_logs.py

from dataclasses import dataclass

from setup.logging import Axiom


@dataclass
class AxiomLogs:
    axiom: Axiom

    def texto_completo_pdf(self, full_text_as_array):
        self.axiom.send_axiom(
            f"INÍCIO DO TEXTO COMPLETO DOS PDFS: {full_text_as_array[0:5]}"
        )

    def resumo_inicial_processo(self, response_auxiliar_summary):
        self.axiom.send_axiom(
            f"RESUMO INICIAL DO PROCESSO: {response_auxiliar_summary}"
        )

    def inicio_requisicao_contextual(self):
        self.axiom.send_axiom("COMEÇANDO A FAZER AS REQUISIÇÕES DO CONTEXTUAL")

    def fim_requisicao_contextual(self):
        self.axiom.send_axiom("TERMINOU DE FAZER TODAS AS REQUISIÇÕES DO CONTEXTUAL")

    def chunks_inicialmente(self, chunks_processados):
        self.axiom.send_axiom(f"CHUNKS PROCESSADOS INICIALMENTE: {chunks_processados}")