heymenn commited on
Commit
1071982
·
verified ·
1 Parent(s): 4180edf

Update kig_core/processing.py

Browse files
Files changed (1) hide show
  1. kig_core/processing.py +1 -1
kig_core/processing.py CHANGED
@@ -57,12 +57,12 @@ def summarize_document(doc_content: str) -> str:
57
  summarize_chain = SUMMARIZER_PROMPT | summarize_llm | StrOutputParser()
58
  summary = invoke_llm(summarize_chain, {"document": doc_content})
59
  logger.debug("Summarization complete.")
 
60
  return summary
61
  except Exception as e:
62
  logger.error(f"Summarization failed: {e}", exc_info=True)
63
  return f"Error during summarization: {e}" # Return error message instead of failing
64
 
65
-
66
  def compress_document(doc_content: str) -> str:
67
  """Compresses a single document using LLMLingua."""
68
  logger.debug(f"Compressing document using method: {settings.compression_method}...")
 
57
  summarize_chain = SUMMARIZER_PROMPT | summarize_llm | StrOutputParser()
58
  summary = invoke_llm(summarize_chain, {"document": doc_content})
59
  logger.debug("Summarization complete.")
60
+ print(f"summary is : {summary}")
61
  return summary
62
  except Exception as e:
63
  logger.error(f"Summarization failed: {e}", exc_info=True)
64
  return f"Error during summarization: {e}" # Return error message instead of failing
65
 
 
66
  def compress_document(doc_content: str) -> str:
67
  """Compresses a single document using LLMLingua."""
68
  logger.debug(f"Compressing document using method: {settings.compression_method}...")