fix: mudança no que passar em cada um dos prompts das etapas

_utils/gerar_relatorio_modelo_usuario/GerarDocumento.py

@@ -252,14 +252,14 @@ class GerarDocumento:
             prompt_instance = Prompt()
             documento_gerado = ""
             tentativas = 0
+            context_do_prompt_primeira_etapa = "\n\n".join(contexts)
+            prompt_primeira_etapa = prompt_gerar_documento.format(
+                context=context_do_prompt_primeira_etapa,
+            )
 
             while tentativas < 5 and not documento_gerado:
                 tentativas += 1
-                resposta = llm.invoke(
-                    prompt_gerar_documento.format(
-                        context="\n\n".join(contexts),
-                    )
-                )
+                resposta = llm.invoke(prompt_primeira_etapa)
                 if hasattr(resposta, "content") and resposta.content.strip():  # type: ignore
                     documento_gerado = resposta.content.strip()  # type: ignore
                 else:
@@ -270,27 +270,31 @@ class GerarDocumento:
                 self.axiom_instance.send_axiom(
                     "TENTANDO GERAR DOCUMENTO FINAL COM GPT 4o-mini COMO ÚLTIMA TENTATIVA"
                 )
-                prompt = prompt_gerar_documento.format(
-                    context="\n\n".join(contexts),
-                )
                 documento_gerado = (
-                    self.gerar_documento_utils.ultima_tentativa_requisicao(prompt)
+                    self.gerar_documento_utils.ultima_tentativa_requisicao(
+                        prompt_primeira_etapa
+                    )
                 )
 
+            resposta_primeira_etapa = documento_gerado
+
             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": documento_gerado},
+                    dynamic_dict={"context": context_do_prompt_primeira_etapa},
                 )
                 documento_gerado = llm.invoke(prompt_etapa_2).content
+                resposta_segunda_etapa = documento_gerado
                 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": documento_gerado},
+                    dynamic_dict={
+                        "context": f"{resposta_primeira_etapa}\n\n{resposta_segunda_etapa}"
+                    },
                 )
                 documento_gerado = llm.invoke(prompt_etapa_3).content
                 self.axiom_instance.send_axiom(f"RESULTADO ETAPA 3: {documento_gerado}")