Spaces:
Running
Running
luanpoppe
commited on
Commit
·
4167f1d
1
Parent(s):
4d3bceb
fix: adicionando o retorno do summary de iterative refinement
Browse files
langchain_backend/main.py
CHANGED
@@ -52,5 +52,6 @@ def get_llm_answer_summary(system_prompt, user_prompt, pdf_url, model, isIterati
|
|
52 |
chain = load_summarize_chain(create_llm(model), "refine", True)
|
53 |
result = chain.invoke({"input_documents": pages})
|
54 |
print('result: ', result)
|
|
|
55 |
# Obs --> Para passar informações personalizadas --> chain = load_summarize_chain(llm, "refine", True, question_prompt=initial_prompt, refine_prompt=PromptTemplate.from_template(refine_prompt))
|
56 |
# Para ver mais opções --> Acessa a origem da função load_summarize_chain , e nela acessa a origem da função _load_refine_chain --> As opções são os parâmetros que esta última função recebe
|
|
|
52 |
chain = load_summarize_chain(create_llm(model), "refine", True)
|
53 |
result = chain.invoke({"input_documents": pages})
|
54 |
print('result: ', result)
|
55 |
+
return result
|
56 |
# Obs --> Para passar informações personalizadas --> chain = load_summarize_chain(llm, "refine", True, question_prompt=initial_prompt, refine_prompt=PromptTemplate.from_template(refine_prompt))
|
57 |
# Para ver mais opções --> Acessa a origem da função load_summarize_chain , e nela acessa a origem da função _load_refine_chain --> As opções são os parâmetros que esta última função recebe
|