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}...")
|