Spaces:
Sleeping
Sleeping
Update kig_core/processing.py
Browse files- 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}...")
|