Spaces:
Running
Running
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 |
-
|
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 = []
|