luanpoppe commited on
Commit
4167f1d
·
1 Parent(s): 4d3bceb

fix: adicionando o retorno do summary de iterative refinement

Browse files
Files changed (1) hide show
  1. langchain_backend/main.py +1 -0
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