luanpoppe commited on
Commit
43e7781
·
1 Parent(s): 9ef64c5

feat: mudando para apenas um prompt final

Browse files
_utils/gerar_relatorio_modelo_usuario/EnhancedDocumentSummarizer.py CHANGED
@@ -174,23 +174,25 @@ class EnhancedDocumentSummarizer(DocumentSummarizer):
174
 
175
  self.resumo_gerado = cast(str, resumo_auxiliar_do_documento.content)
176
 
177
- prompt_gerar_documento = PromptTemplate(
178
- template=self.prompt_gerar_documento,
179
- input_variables=["context"],
180
- )
181
-
182
- documento_gerado = cast(
183
- str,
184
- llm.invoke(
185
- prompt_gerar_documento.format(
186
- context=self.resumo_gerado,
187
- # modelo_usuario=serializer.data["modelo"],
188
- )
189
- ).content,
190
- )
191
 
192
  # Split the response into paragraphs
193
- summaries = [p.strip() for p in documento_gerado.split("\n\n") if p.strip()]
 
 
194
 
195
  # Create structured output
196
  structured_output = []
 
174
 
175
  self.resumo_gerado = cast(str, resumo_auxiliar_do_documento.content)
176
 
177
+ # prompt_gerar_documento = PromptTemplate(
178
+ # template=self.prompt_gerar_documento,
179
+ # input_variables=["context"],
180
+ # )
181
+
182
+ # documento_gerado = cast(
183
+ # str,
184
+ # llm.invoke(
185
+ # prompt_gerar_documento.format(
186
+ # context=self.resumo_gerado,
187
+ # # modelo_usuario=serializer.data["modelo"],
188
+ # )
189
+ # ).content,
190
+ # )
191
 
192
  # Split the response into paragraphs
193
+ summaries = [
194
+ p.strip() for p in self.resumo_gerado.split("\n\n") if p.strip()
195
+ ]
196
 
197
  # Create structured output
198
  structured_output = []