Spaces:
Sleeping
Sleeping
Update _utils/gerar_relatorio_modelo_usuario/EnhancedDocumentSummarizer.py
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 |
-
|
| 179 |
-
|
| 180 |
-
)
|
| 181 |
-
|
| 182 |
-
documento_gerado = cast(
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
)
|
| 191 |
|
| 192 |
# Split the response into paragraphs
|
| 193 |
-
summaries = [
|
|
|
|
|
|
|
| 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 = []
|