feat: adicionar mais opções de prompts ao gerar ementa

_utils/gerar_documento_utils/GerarDocumento.py

@@ -329,3 +329,49 @@ class GerarDocumento:
             )
 
         return documento_gerado
+
+    async def gerar_ementa_final(
+        self,
+        llm_ultimas_requests: str,
+        prompt_primeira_etapa: str,
+        context_primeiro_prompt: str,
+    ):
+
+        llm = self.select_model_for_last_requests(llm_ultimas_requests)  # type: ignore
+        prompt_instance = Prompt()
+
+        documento_gerado = await self.checar_se_resposta_vazia_do_documento_final(
+            llm_ultimas_requests, prompt_primeira_etapa
+        )
+
+        texto_final_juntando_as_etapas = ""
+        resposta_primeira_etapa = documento_gerado
+        texto_final_juntando_as_etapas += resposta_primeira_etapa
+        self.axiom_instance.send_axiom(f"RESULTADO ETAPA 1: {resposta_primeira_etapa}")
+
+        if self.prompt_gerar_documento_etapa_2:
+            self.axiom_instance.send_axiom("GERANDO DOCUMENTO - COMEÇANDO ETAPA 2")
+            prompt_etapa_2 = prompt_instance.create_and_invoke_prompt(
+                self.prompt_gerar_documento_etapa_2,
+                dynamic_dict={"context": context_primeiro_prompt},
+            )
+            documento_gerado = llm.invoke(prompt_etapa_2).content
+            resposta_segunda_etapa = documento_gerado
+            texto_final_juntando_as_etapas += (
+                f"\n\nresposta_segunda_etapa:{resposta_segunda_etapa}"
+            )
+            self.axiom_instance.send_axiom(f"RESULTADO ETAPA 2: {documento_gerado}")
+
+        if self.prompt_gerar_documento_etapa_3:
+            self.axiom_instance.send_axiom("GERANDO DOCUMENTO - COMEÇANDO ETAPA 3")
+            prompt_etapa_3 = prompt_instance.create_and_invoke_prompt(
+                self.prompt_gerar_documento_etapa_3,
+                dynamic_dict={
+                    "context": f"{resposta_primeira_etapa}\n\n{resposta_segunda_etapa}"
+                },
+            )
+            documento_gerado = llm.invoke(prompt_etapa_3).content
+            texto_final_juntando_as_etapas += f"\n\n{documento_gerado}"
+            self.axiom_instance.send_axiom(f"RESULTADO ETAPA 3: {documento_gerado}")
+
+        return texto_final_juntando_as_etapas

gerar_documento/views.py

@@ -167,9 +167,12 @@ class GerarEmentaView(AsyncAPIView):
                             input_variables=["context"],
                             template=obj.prompt_gerar_documento,
                         )
-                        texto_da_parte = await gerar_documento_instance.checar_se_resposta_vazia_do_documento_final(
-                            obj.llm_ultimas_requests,
-                            prompt_template.format(context=text),
+                        texto_da_parte = (
+                            await gerar_documento_instance.gerar_ementa_final(
+                                obj.llm_ultimas_requests,
+                                prompt_template.format(context=text),
+                                text,
+                            )
                         )
 
                         texto_completo += texto_da_parte
@@ -271,9 +274,10 @@ class GerarEmentaComPDFProprioView(AsyncAPIView):
                     input_variables=["context"],
                     template=serializer_obj.prompt_gerar_documento,
                 )
-                texto_da_parte = await gerar_documento_instance.checar_se_resposta_vazia_do_documento_final(
+                texto_da_parte = await gerar_documento_instance.gerar_ementa_final(
                     serializer_obj.llm_ultimas_requests,
                     prompt_template.format(context=text),
+                    text,
                 )
 
                 texto_completo += texto_da_parte