fix: adicionando mais regexs aos contextuals

_utils/gerar_documento_utils/contextual_retriever.py

@@ -257,12 +257,14 @@ class ContextualRetrieverUtils:
 
     def check_regex_patterns(self, context: str, lista_de_document_ids: List[int]):
         patterns = [
-            # r"\[*([\d.\-]+)\]*\s*---\s*\[*([^]]+)\]*\s*---\s*\[*([^]]+)\]*\s*</chunk_context>", # PRIMEIRO DE TODOS
-            # r"<chunk_context>\s*([\d.\-]+)\s*---\s*([^<]+)\s*---\s*([^<]+)\s*</chunk_context>",
             r"\[(.*?)\] --- \[(.*?)\] --- \[(.*?)\](?=\n|\s*$)",
             r"\[([^\[\]]+?)\]\s*---\s*\[([^\[\]]+?)\]\s*---\s*(.*?)</chunk_context>",
             r"<chunk_context>\s*(\d+)(?:\s*-\s*Pág\.\s*\d+)?\s*---\s*([^-\n]+)\s*---\s*([^<]+)</chunk_context>",
             r"<chunk_context>\s*(?:\[*([\d]+)\]*\s*[-–]*\s*(?:Pág\.\s*\d+\s*[-–]*)?)?\s*\[*([^\]]+)\]*\s*[-–]*\s*\[*([^\]]+)\]*\s*[-–]*\s*\[*([^\]]+)\]*\s*</chunk_context>",
+            r"<chunk_context>\s*(.*?)\s*---\s*(.*?)\s*---\s*(.*?)\s*</chunk_context>",
+            # -------------- ABAIXO SÃO OS ANTIGOS
+            # r"\[*([\d.\-]+)\]*\s*---\s*\[*([^]]+)\]*\s*---\s*\[*([^]]+)\]*\s*</chunk_context>", # PRIMEIRO DE TODOS
+            # r"<chunk_context>\s*([\d.\-]+)\s*---\s*([^<]+)\s*---\s*([^<]+)\s*</chunk_context>",
             # r"\[([\d.\-]+)\]\s*---\s*\[([^]]+)\]\s*---\s*\[([^]]+)\]\s*</chunk_context>",
             # r"<chunk_context>\s*\[?([\d.\-]+)\]?\s*---\s*\[?([^\]\[]+?)\]?\s*---\s*\[?([^<]+?)\]?\s*</chunk_context>",
             # r"<chunk_context>\s*\[([\d.\-]+)\]\s*---\s*\[([^\]]+)\]\s*---\s*\[([^\]]+)\]\s*</chunk_context>"
@@ -272,6 +274,7 @@ class ContextualRetrieverUtils:
         resultado = None
         for pattern in patterns:
             matches: List[str] = re.findall(pattern, context, re.DOTALL)
+            print("\n\nmatches", matches)
             condition_tuples_3_items = all(len(m) == 3 for m in matches)
             if len(matches) == len(lista_de_document_ids) and condition_tuples_3_items:
                 print("\n--------------- REGEX DO CONTEXTUAL FUNCIONOU")
@@ -286,6 +289,38 @@ class ContextualRetrieverUtils:
                     resultado.append((first_item, m[1], m[2]))
                 break
 
+        if not resultado:
+            context = (
+                context.replace("</final_output>", "")
+                .replace("<final_output>", "")
+                .strip()
+            )
+            raw_chunks = context.split("</chunk_context>")[0:20]
+
+            resultado_temporario = []
+            for r in raw_chunks:
+                lista_3_itens = r.split("---")
+                page_id = re.search(r"Num\.\s*(\d+)\s*-", lista_3_itens[0].strip())
+                page_id_tentativa_2 = re.search(
+                    r"\d+\.\s+(\d+)\s+-\s+Pág\.", lista_3_itens[0].strip()
+                )
+                if page_id:
+                    first_item = page_id.group(1)
+                elif page_id_tentativa_2:
+                    first_item = page_id_tentativa_2.group(1)
+                else:
+                    first_item = "0"
+                resultado_temporario.append(
+                    (first_item, lista_3_itens[1], lista_3_itens[2])
+                )
+
+            condition_tuples_3_items = all(len(m) == 3 for m in resultado_temporario)
+            if (
+                len(resultado_temporario) == len(lista_de_document_ids)
+                and condition_tuples_3_items
+            ):
+                resultado = resultado_temporario
+
         return resultado
 
 

_utils/gerar_documento_utils/tests/fixtures/resposta_llm_com_20_chunks.py

@@ -1,4 +1,12 @@
 resposta_llm_com_20_chunks_fixture_array = [
+    """1. <chunk_context> 72909022 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a procuração outorgada, com poderes específicos para atuar em ações cíveis e procedimentos administrativos, incluindo a capacidade de receber valores e praticar atos necessários ao cumprimento do mandato. A procuração tem validade até 31 de dezembro de 2023 e é assinada digitalmente. </chunk_context> 2. <chunk_context> 72909148 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta informações sobre a plataforma de assinatura digital NEOENERGIA, incluindo links e códigos para verificação das assinaturas digitais no documento. Detalha os signatários e seus status em 09/08/2023, com Luciana Maximino Maia e Renato De Almeida Rocha representando a COELBA, e Alana Cristina Braz Duarte como autorizador. </chunk_context> 3. <chunk_context> 72914243 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a certidão de triagem inicial, emitida pela Diretoria de Distribuição do 2º Grau do Tribunal de Justiça da Bahia, confirmando a regularidade dos dados cadastrais do processo, como classe, partes, advogados e assuntos. A certidão atesta que foram realizadas alterações necessárias nos dados. </chunk_context> 4. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do Desembargador Cássio José Barbosa Miranda, relator do Agravo de Instrumento, que trata da antecipação dos efeitos da tutela concedida em primeira instância. A decisão de primeira instância determinou o restabelecimento do serviço de energia elétrica, com base na essencialidade do serviço e na inscrição da autora no Programa de Baixa Renda Quilombola. </chunk_context> 5. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua a decisão do Desembargador, detalhando os argumentos da COELBA no Agravo de Instrumento, que questiona a decisão de primeira instância por ser genérica e permitir o fornecimento de energia sem garantia de pagamento. A COELBA também contesta o valor da multa imposta. O trecho também apresenta os artigos do CPC que fundamentam a decisão. </chunk_context> 6. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho da decisão do Desembargador Cássio Miranda analisa os requisitos para a concessão do efeito suspensivo. O relator entende que os requisitos foram parcialmente preenchidos, considerando a inscrição da agravada no Programa de Baixa Renda Quilombola, o ajuizamento de ações e o relatório que atesta erros no medidor. </chunk_context> 7. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho da decisão do Desembargador Cássio Miranda estabelece as condições para o restabelecimento do serviço de energia elétrica, condicionando-o ao pagamento das parcelas vincendas e reduzindo o valor da multa diária. O trecho também determina a intimação da parte agravada para apresentar contrarrazões e o envio de ofício ao juízo de primeira instância. </chunk_context> 8. <chunk_context> 72976135 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a certidão de juntada da decisão nos autos de primeiro grau, emitida pela Secretaria da Quinta Câmara Cível. </chunk_context> 9. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta as contrarrazões ao Agravo de Instrumento, interposto por Isailde Maria da Silva, através de sua advogada, Paloma Oliveira de Jesus Jambeiro. O documento solicita o recebimento das contrarrazões e a remessa ao Tribunal de Justiça para que seja negado provimento ao recurso. </chunk_context> 10. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho das contrarrazões apresenta uma síntese do recurso, com os argumentos da agravada, Isailde Maria da Silva, que ajuizou ação para restabelecer o serviço de energia elétrica. A agravada argumenta que a multa aplicada é proporcional e que a COELBA tenta distorcer os fatos. </chunk_context> 11. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho das contrarrazões argumenta sobre a impossibilidade de reforma da decisão que deferiu a tutela de urgência, com base na atuação irregular da COELBA e na condição de vulnerabilidade da agravada. A agravada também defende que a liminar deferida é provisória e visa garantir a continuidade do serviço. </chunk_context> 12. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho das contrarrazões aborda a alegação de enriquecimento ilícito e o descumprimento do prazo da obrigação de fazer. A agravada argumenta que não há enriquecimento ilícito e que a multa é proporcional ao descumprimento da decisão judicial. </chunk_context> 13. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho das contrarrazões aborda o histórico de falhas na prestação do serviço pela COELBA, o tempo sem energia elétrica devido ao erro da concessionária e a legalidade e razoabilidade da multa imposta. A agravada defende a manutenção da multa. </chunk_context> 14. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho das contrarrazões apresenta a conclusão, requerendo o não provimento do Agravo de Instrumento e a manutenção da decisão de primeira instância, com a reafirmação da multa imposta. </chunk_context> 15. <chunk_context> 73079323 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a qualificação da advogada da agravada, Paloma Oliveira de Jesus Jambeiro, e a data da assinatura das contrarrazões. </chunk_context> 16. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do Desembargador Cássio José Barbosa Miranda, que trata da antecipação dos efeitos da tutela concedida em primeira instância. A decisão de primeira instância determinou o restabelecimento do serviço de energia elétrica, com base na essencialidade do serviço e na inscrição da autora no Programa de Baixa Renda Quilombola. </chunk_context> 17. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua a decisão do Desembargador, detalhando os argumentos da COELBA no Agravo de Instrumento, que questiona a decisão de primeira instância por ser genérica e permitir o fornecimento de energia sem garantia de pagamento. A COELBA também contesta o valor da multa imposta. O trecho também apresenta os artigos do CPC que fundamentam a decisão. </chunk_context> 18. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho da decisão do Desembargador Cássio Miranda analisa os requisitos para a concessão do efeito suspensivo. O relator entende que os requisitos foram parcialmente preenchidos, considerando a inscrição da agravada no Programa de Baixa Renda Quilombola, o ajuizamento de ações e o relatório que atesta erros no medidor. </chunk_context> 19. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho da decisão do Desembargador Cássio Miranda estabelece as condições para o restabelecimento do serviço de energia elétrica, condicionando-o ao pagamento das parcelas vincendas e reduzindo o valor da multa diária. O trecho também determina a intimação da parte agravada para apresentar contrarrazões e o envio de ofício ao juízo de primeira instância. </chunk_context> 20. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho da decisão do Desembargador Cássio Miranda estabelece as condições para o restabelecimento do serviço de energia elétrica, condicionando-o ao pagamento das parcelas vincendas e reduzindo o valor da multa diária. O trecho também determina a intimação da parte agravada para apresentar contrarrazões e o envio de ofício ao juízo de primeira instância. </chunk_context>
+    """,
+    """<final_output> <chunk_context> 1. 72909020 - Pág. 1 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta os dados iniciais do processo de Agravo de Instrumento (nº 8068804-29.2024.8.05.0000) interposto pela COELBA, incluindo as partes envolvidas, os advogados, os assuntos e o pedido de liminar. Ele se encaixa na introdução do processo, conforme descrito no resumo, e estabelece o contexto geral do caso. </chunk_context> 2. 72909020 - Pág. 2 --- AGRAVO DE INSTRUMENTO --- Esta parte do documento continua com a qualificação da COELBA como agravante, indicando seu endereço e a decisão interlocutória que motivou o agravo. Menciona o artigo 1.015 do CPC como base legal para a interposição do recurso e solicita efeito suspensivo. O trecho se alinha com a descrição da ação da COELBA no resumo, que busca reverter a decisão de primeira instância. </chunk_context> 3. 72909020 - Pág. 3 --- AGRAVO DE INSTRUMENTO --- O trecho apresenta as "RAZÕES DO RECURSO", com a identificação das partes (agravante e agravado) e um resumo da decisão de primeira instância que concedeu a antecipação dos efeitos da tutela. Este trecho é fundamental, pois detalha o cerne da disputa e a decisão que a COELBA busca modificar, conforme mencionado no resumo. </chunk_context> 4. 72909020 - Pág. 4 --- AGRAVO DE INSTRUMENTO --- Esta seção continua a detalhar a decisão de primeira instância, que determinou o restabelecimento do serviço de energia elétrica e impôs multa. O texto argumenta sobre a tempestividade do recurso, com base no artigo 219 do CPC. O trecho se encaixa na argumentação da COELBA contra a decisão, conforme o resumo. </chunk_context> 5. 72909020 - Pág. 5 --- AGRAVO DE INSTRUMENTO --- O texto discute o cabimento do recurso, mencionando a ação declaratória de inexistência de débito e a liminar concedida. Cita o artigo 1.015 do CPC, que trata do agravo de instrumento contra decisões interlocutórias. O trecho detalha os argumentos da COELBA sobre a inadequação da liminar, alinhando-se com o resumo. </chunk_context> 6. 72909020 - Pág. 6 --- AGRAVO DE INSTRUMENTO --- O texto continua a argumentar sobre a necessidade de revogação da liminar, alegando que ela é genérica e não especifica as faturas em questão. O trecho critica a decisão de primeira instância por não abordar a contraprestação do cliente e por decidir o mérito em sede de tutela de urgência. Isso se encaixa na crítica da COELBA à decisão, conforme o resumo. </chunk_context> 7. 72909020 - Pág. 7 --- AGRAVO DE INSTRUMENTO --- O texto argumenta sobre a falta dos requisitos para manter a liminar, mencionando a ausência de verossimilhança e de prova inequívoca. Cita o artigo 14, §3º, do CDC, sobre a responsabilidade do fornecedor. O trecho defende a legalidade da conduta da COELBA e a necessidade de provimento do agravo, conforme o resumo. </chunk_context> 8. 72909020 - Pág. 8 --- AGRAVO DE INSTRUMENTO --- O texto cita o artigo 356 da Resolução da ANEEL, que trata da suspensão do fornecimento de energia por inadimplemento. Menciona o parecer do Ministro do STF, Eros Roberto Grau, sobre a legalidade da suspensão. O trecho reforça a argumentação da COELBA com base em normas e jurisprudência, conforme o resumo. </chunk_context> 9. 72909020 - Pág. 9 --- AGRAVO DE INSTRUMENTO --- O texto cita jurisprudência do TJSP sobre corte de energia por inadimplemento e conclui pela necessidade de revogação da liminar. O trecho se alinha com a argumentação da COELBA, que busca a suspensão da liminar, conforme o resumo. </chunk_context> 10. 72909020 - Pág. 10 --- AGRAVO DE INSTRUMENTO --- O texto discute o valor excessivo das astreintes (multa) e a necessidade de sua exclusão ou redução, citando o artigo 461, §6º, do CPC. Menciona a doutrina de Luiz Guilherme Marinoni sobre a função da multa. O trecho se encaixa na argumentação da COELBA sobre a necessidade de reduzir a multa, conforme o resumo. </chunk_context> 11. 72909020 - Pág. 11 --- AGRAVO DE INSTRUMENTO --- O texto continua a argumentar sobre a necessidade de redução da multa, citando jurisprudência do STJ e de tribunais estaduais. O trecho reforça a argumentação da COELBA sobre a necessidade de reduzir a multa, conforme o resumo. </chunk_context> 12. 72909020 - Pág. 12 --- AGRAVO DE INSTRUMENTO --- O texto conclui a argumentação sobre a multa e passa a discutir a atribuição do efeito suspensivo ao agravo de instrumento, com base no artigo 1.019, I, do CPC. O trecho se encaixa na argumentação da COELBA sobre a necessidade de suspender os efeitos da liminar, conforme o resumo. </chunk_context> 13. 72909020 - Pág. 13 --- AGRAVO DE INSTRUMENTO --- O texto continua a argumentar sobre o efeito suspensivo, citando Humberto Theodoro Júnior. Apresenta os pedidos da COELBA, incluindo a revogação da decisão liminar ou a atribuição de efeito suspensivo. O trecho resume os pedidos da COELBA, conforme o resumo. </chunk_context> 14. 72909020 - Pág. 14 --- AGRAVO DE INSTRUMENTO --- O texto apresenta os pedidos da COELBA, incluindo a declaração de nulidade da decisão agravada, a atribuição de efeito ativo, a revogação da decisão agravada, a restrição dos efeitos da liminar, o pagamento das faturas vincendas, a retirada ou minoração da multa, a comunicação ao juiz a quo e a intimação do patrono do agravado. O trecho detalha os pedidos da COELBA, conforme o resumo. </chunk_context> 15. 72909021 - Pág. 1 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta os documentos de arrecadação judicial e extrajudicial (DAJE) referentes ao processo, indicando os valores pagos e as datas de vencimento. O trecho demonstra o pagamento das custas processuais pela COELBA, conforme o resumo. </chunk_context> 16. 72909021 - Pág. 2 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta os documentos de arrecadação judicial e extrajudicial (DAJE) referentes ao processo, indicando os valores pagos e as datas de vencimento. O trecho demonstra o pagamento das custas processuais pela COELBA, conforme o resumo. </chunk_context> 17. 72909021 - Pág. 3 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta os documentos de arrecadação judicial e extrajudicial (DAJE) referentes ao processo, indicando os valores pagos e as datas de vencimento. O trecho demonstra o pagamento das custas processuais pela COELBA, conforme o resumo. </chunk_context> 18. 72909021 - Pág. 4 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta os documentos de arrecadação judicial e extrajudicial (DAJE) referentes ao processo, indicando os valores pagos e as datas de vencimento. O trecho demonstra o pagamento das custas processuais pela COELBA, conforme o resumo. </chunk_context> 19. 72909022 - Pág. 1 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta a procuração da COELBA, com a qualificação dos advogados outorgados. O trecho demonstra a representação da COELBA no processo, conforme o resumo. </chunk_context> 20. 72909022 - Pág. 2 --- AGRAVO DE INSTRUMENTO --- Este trecho continua a apresentar a procuração da COELBA, com a qualificação dos advogados outorgados. O trecho demonstra a representação da COELBA no processo, conforme o resumo. </chunk_context> </final_output>
+    """,
+    """1. <chunk_context> 72909022 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a procuração outorgada, com poderes específicos para atuar em ações cíveis e procedimentos administrativos, incluindo a representação da COELBA. A procuração é um documento relevante para o processo, pois define os limites da atuação dos representantes da empresa. </chunk_context> 2. <chunk_context> 72909148 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta informações sobre a plataforma de assinatura digital utilizada para o documento, incluindo os nomes dos signatários da COELBA e a data das assinaturas. Ele corrobora a validade e autenticidade do documento, conforme o resumo do processo. </chunk_context> 3. <chunk_context> 72914243 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a certidão de triagem inicial, que confirma a conferência dos dados cadastrais do processo, como classe do recurso, partes, advogados e assuntos. Ele demonstra a regularidade formal do processo, conforme o resumo do processo. </chunk_context> 4. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho reproduz a decisão do juízo de primeira instância que deferiu a antecipação dos efeitos da tutela, determinando o restabelecimento do serviço de energia elétrica. Ele estabelece o contexto da decisão agravada, conforme o resumo do processo. </chunk_context> 5. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua a reproduzir a decisão de primeira instância, incluindo a determinação de multa pelo descumprimento da decisão. Ele detalha os termos da decisão agravada, conforme o resumo do processo. </chunk_context> 6. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a análise do relator sobre os requisitos para a concessão do efeito suspensivo, mencionando o "fumus boni iuris" e o "periculum in mora". Ele introduz a fundamentação da decisão do relator, conforme o resumo do processo. </chunk_context> 7. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do relator que defere parcialmente o efeito suspensivo, condicionando o restabelecimento do serviço ao pagamento das parcelas vincendas e reduzindo a multa. Ele apresenta a conclusão da decisão do relator, conforme o resumo do processo. </chunk_context> 8. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho finaliza a decisão do relator, determinando a intimação da parte agravada para apresentar contrarrazões e oficiando o juiz a quo. Ele encerra a decisão do relator, conforme o resumo do processo. </chunk_context> 9. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta as contrarrazões ao agravo de instrumento, interpostas pela agravada, com a identificação da advogada e o pedido de não provimento do recurso. Ele introduz a defesa da agravada, conforme o resumo do processo. </chunk_context> 10. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a síntese do recurso, a impossibilidade de reforma da decisão e a alegação de que a liminar não é genérica. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 11. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua com os argumentos da agravada, abordando a inexistência de enriquecimento ilícito e a legalidade da multa imposta. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 12. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho aborda o descumprimento do prazo da obrigação de fazer e o histórico de falhas na prestação do serviço. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 13. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho aborda o tempo sem energia elétrica devido ao erro da concessionária e a legalidade e razoabilidade da multa imposta. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 14. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a conclusão das contrarrazões, com o pedido de não provimento do agravo de instrumento. Ele encerra a defesa da agravada, conforme o resumo do processo. </chunk_context> 15. <chunk_context> 8068804-29.2024.8.05.0000 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta informações sobre o processo, como número, classe, órgão julgador, partes e advogados. Ele fornece uma visão geral do processo, conforme o resumo do processo. </chunk_context> 16. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho reproduz a decisão do juízo de primeira instância, que deferiu a antecipação dos efeitos da tutela, determinando o restabelecimento do serviço de energia elétrica. Ele estabelece o contexto da decisão agravada, conforme o resumo do processo. </chunk_context> 17. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua a reproduzir a decisão de primeira instância, incluindo a determinação de multa pelo descumprimento da decisão. Ele detalha os termos da decisão agravada, conforme o resumo do processo. </chunk_context> 18. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a análise do relator sobre os requisitos para a concessão do efeito suspensivo, mencionando o "fumus boni iuris" e o "periculum in mora". Ele introduz a fundamentação da decisão do relator, conforme o resumo do processo. </chunk_context> 19. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do relator que defere parcialmente o efeito suspensivo, condicionando o restabelecimento do serviço ao pagamento das parcelas vincendas e reduzindo a multa. Ele apresenta a conclusão da decisão do relator, conforme o resumo do processo. </chunk_context> 20. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do relator que defere parcialmente o efeito suspensivo, condicionando o restabelecimento do serviço ao pagamento das parcelas vincendas e reduzindo a multa. Ele apresenta a conclusão da decisão do relator, conforme o resumo do processo. </chunk_context>
+    """,
+    """1. <chunk_context> 72909022 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a procuração outorgada, com poderes específicos para atuar em ações cíveis e procedimentos administrativos, incluindo a representação da COELBA. A procuração é um documento relevante para o processo, pois define os limites da atuação dos representantes da empresa. </chunk_context> 2. <chunk_context> 72909148 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta informações sobre a plataforma de assinatura digital utilizada para o documento, incluindo os nomes dos signatários da COELBA e a data das assinaturas. Ele corrobora a validade e autenticidade do documento, conforme o resumo do processo. </chunk_context> 3. <chunk_context> 72914243 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a certidão de triagem inicial, que confirma a conferência dos dados cadastrais do processo, como classe do recurso, partes, advogados e assuntos. Ele demonstra a regularidade formal do processo, conforme o resumo do processo. </chunk_context> 4. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho reproduz a decisão do juízo de primeira instância que deferiu a antecipação dos efeitos da tutela, determinando o restabelecimento do serviço de energia elétrica. Ele estabelece o contexto da decisão agravada, conforme o resumo do processo. </chunk_context> 5. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua a reproduzir a decisão de primeira instância, incluindo a determinação de multa pelo descumprimento da decisão. Ele detalha os termos da decisão agravada, conforme o resumo do processo. </chunk_context> 6. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a análise do relator sobre os requisitos para a concessão do efeito suspensivo, mencionando o "fumus boni iuris" e o "periculum in mora". Ele introduz a fundamentação da decisão do relator, conforme o resumo do processo. </chunk_context> 7. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do relator que defere parcialmente o efeito suspensivo, condicionando o restabelecimento do serviço ao pagamento das parcelas vincendas e reduzindo a multa. Ele apresenta a conclusão da decisão do relator, conforme o resumo do processo. </chunk_context> 8. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho finaliza a decisão do relator, determinando a intimação da parte agravada para apresentar contrarrazões e oficiando o juiz a quo. Ele encerra a decisão do relator, conforme o resumo do processo. </chunk_context> 9. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta as contrarrazões ao agravo de instrumento, interpostas pela agravada, com a identificação da advogada e o pedido de não provimento do recurso. Ele introduz a defesa da agravada, conforme o resumo do processo. </chunk_context> 10. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a síntese do recurso, a impossibilidade de reforma da decisão e a alegação de que a liminar não é genérica. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 11. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua com os argumentos da agravada, abordando a inexistência de enriquecimento ilícito e a legalidade da multa imposta. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 12. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho aborda o descumprimento do prazo da obrigação de fazer e o histórico de falhas na prestação do serviço. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 13. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho aborda o tempo sem energia elétrica devido ao erro da concessionária e a legalidade e razoabilidade da multa imposta. Ele detalha os argumentos da agravada em suas contrarrazões, conforme o resumo do processo. </chunk_context> 14. <chunk_context> 73079321 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a conclusão das contrarrazões, com o pedido de não provimento do agravo de instrumento. Ele encerra a defesa da agravada, conforme o resumo do processo. </chunk_context> 15. <chunk_context> 8068804-29.2024.8.05.0000 --- AGRAVO DE INSTRUMENTO --- Este trecho apresenta informações sobre o processo, como número, classe, órgão julgador, partes e advogados. Ele fornece uma visão geral do processo, conforme o resumo do processo. </chunk_context> 16. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho reproduz a decisão do juízo de primeira instância, que deferiu a antecipação dos efeitos da tutela, determinando o restabelecimento do serviço de energia elétrica. Ele estabelece o contexto da decisão agravada, conforme o resumo do processo. </chunk_context> 17. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho continua a reproduzir a decisão de primeira instância, incluindo a determinação de multa pelo descumprimento da decisão. Ele detalha os termos da decisão agravada, conforme o resumo do processo. </chunk_context> 18. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a análise do relator sobre os requisitos para a concessão do efeito suspensivo, mencionando o "fumus boni iuris" e o "periculum in mora". Ele introduz a fundamentação da decisão do relator, conforme o resumo do processo. </chunk_context> 19. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do relator que defere parcialmente o efeito suspensivo, condicionando o restabelecimento do serviço ao pagamento das parcelas vincendas e reduzindo a multa. Ele apresenta a conclusão da decisão do relator, conforme o resumo do processo. </chunk_context> 20. <chunk_context> 72933123 --- AGRAVO DE INSTRUMENTO n. 8068804-29.2024.8.05.0000 --- Este trecho apresenta a decisão do relator que defere parcialmente o efeito suspensivo, condicionando o restabelecimento do serviço ao pagamento das parcelas vincendas e reduzindo a multa. Ele apresenta a conclusão da decisão do relator, conforme o resumo do processo. </chunk_context>
+    """,
     """1.  <chunk_context>
 [76915] --- [APELAÇÃO CRIMINAL] --- Este chunk apresenta os dados iniciais do processo de Apelação Criminal, incluindo o número, classe, órgão julgador, partes envolvidas (Estado da Bahia, Defensor Dativo, e Adenilton da Conceição Samuel), e os assuntos tratados (Honorários Advocatícios, Crimes de Trânsito). Ele se encaixa na descrição do resumo do processo, que trata da apelação do Estado da Bahia sobre os honorários do defensor dativo em um caso de crimes de trânsito.
 </chunk_context>

_utils/gerar_documento_utils/utils.py

@@ -9,6 +9,7 @@ from _utils.gerar_documento_utils.prompts import (
     create_prompt_auxiliar_do_contextual_prompt,
 )
 
+from _utils.models.gerar_documento import DocumentChunk
 from gerar_documento.serializer import GerarDocumentoSerializerData
 
 
@@ -91,7 +92,7 @@ async def get_full_text_and_all_PDFs_chunks(
     should_use_llama_parse: bool,
     isBubble: bool,
 ):
-    all_PDFs_chunks = []
+    all_PDFs_chunks: List[DocumentChunk] = []
 
     pages: List[str] = []
 

gerar_documento/urls.py

@@ -3,6 +3,8 @@ from django.urls import path
 from gerar_documento.views import (
     GerarDocumentoView,
     GerarDocumentoComPDFProprioView,
+    GerarEmentaComPDFProprioView,
+    GerarEmentaView,
 )
 
 urlpatterns = [
@@ -16,4 +18,14 @@ urlpatterns = [
         GerarDocumentoComPDFProprioView.as_view(),
         name="generate-document-with-own-pdf",
     ),
+    path(
+        "gerar-ementa",
+        GerarEmentaView.as_view(),
+        name="generate-ementa",
+    ),
+    path(
+        "gerar-ementa/pdf",
+        GerarEmentaComPDFProprioView.as_view(),
+        name="generate-ementa-with-own-pdf",
+    ),
 ]

gerar_documento/views.py

@@ -112,34 +112,46 @@ class GerarDocumentoComPDFProprioView(AsyncAPIView):
 
 class GerarEmentaView(AsyncAPIView):
     serializer = {}
+    axiom_instance = Axiom()
 
     @extend_schema(
         request=GerarDocumentoSerializer,
     )
     async def post(self, request):
+        self.axiom_instance.generate_new_uuid()
         print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
-        print("request.data: ", request.data)
-        serializer = GerarEmentaSerializer(data=request.data)
+        self.axiom_instance.send_axiom(
+            f"COMEÇOU NOVA REQUISIÇÃO - request.data: {request.data}"
+        )
+        serializer = GerarDocumentoSerializer(data=request.data)
         if serializer.is_valid(raise_exception=True):
             if not serializer.validated_data:
                 raise ValueError("Erro no validated_data")
 
             async def proccess_data_after_response():
                 data = cast(Dict[str, Any], serializer.validated_data)
+                serializer_obj = serializer.get_obj()
                 self.serializer = data
 
                 listaPDFs = [l["link_arquivo"] for l in data["files"]]
-                print("\n\nlistaPDFs: ", listaPDFs)
+                self.axiom_instance.send_axiom(
+                    f"listaPDFs: {listaPDFs}",
+                )
 
                 all_PDFs_chunks, full_text_as_array = (
                     await get_full_text_and_all_PDFs_chunks(
                         listaPDFs,
-                        Splitter(data["chunk_size"], data["chunk_overlap"]),
+                        Splitter(
+                            serializer_obj.chunk_size, serializer_obj.chunk_overlap
+                        ),
                         False,
                         True,
                     )
                 )
                 full_text = "".join(full_text_as_array)
+                self.axiom_instance.send_axiom(
+                    f"full_text gerado dos pdfs: {full_text}"
+                )
 
                 llm = LLM()
                 prompt_template = PromptTemplate(
@@ -165,33 +177,44 @@ class GerarEmentaView(AsyncAPIView):
 class GerarEmentaComPDFProprioView(AsyncAPIView):
     parser_classes = [MultiPartParser]
     serializer = {}
+    axiom_instance = Axiom()
 
     @extend_schema(
         request=GerarDocumentoComPDFProprioSerializer,
     )
     async def post(self, request):
+        self.axiom_instance.generate_new_uuid()
         print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
+        self.axiom_instance.send_axiom(
+            f"COMEÇOU NOVA REQUISIÇÃO - request.data: {request.data}"
+        )
         serializer = GerarDocumentoComPDFProprioSerializer(data=request.data)
         if serializer.is_valid(raise_exception=True):
             data = cast(Dict[str, Any], serializer.validated_data)
-            print("\n\ndata: ", data)
+            self.axiom_instance.send_axiom(f"data: {data}")
             self.serializer = data
 
             data = cast(Dict[str, Any], serializer.validated_data)
+            serializer_obj = serializer.get_obj()
             self.serializer = data
 
-            listaPDFs = [l["link_arquivo"] for l in data["files"]]
-            print("\n\nlistaPDFs: ", listaPDFs)
+            listaPDFs = handle_pdf_files_from_serializer(
+                data["files"], self.axiom_instance
+            )
+            self.axiom_instance.send_axiom(f"listaPDFs: {listaPDFs}")
 
             all_PDFs_chunks, full_text_as_array = (
                 await get_full_text_and_all_PDFs_chunks(
                     listaPDFs,
-                    Splitter(data["chunk_size"], data["chunk_overlap"]),
+                    Splitter(serializer_obj.chunk_size, serializer_obj.chunk_overlap),
                     False,
                     True,
                 )
             )
             full_text = "".join(full_text_as_array)
+            self.axiom_instance.send_axiom(
+                f"full_text[0:100] gerado do PDF: {full_text[0:100]}"
+            )
 
             llm = LLM()
             prompt_template = PromptTemplate(
@@ -201,8 +224,10 @@ class GerarEmentaComPDFProprioView(AsyncAPIView):
                 prompt_template.format(context=full_text)
             )
 
-            print("\n\nresposta_llm: ", response.content)
+            self.axiom_instance.send_axiom(f"resposta_llm: {response.content}")
 
             remove_pdf_temp_files(listaPDFs)
-            print("PRÓXIMA LINHA ENVIA A RESPOSTA A QUEM FEZ A REQUISIÇÃO")
+            self.axiom_instance.send_axiom(
+                "PRÓXIMA LINHA ENVIA A RESPOSTA A QUEM FEZ A REQUISIÇÃO"
+            )
             return Response({"resposta": response.content})