Update kig_core/processing.py

kig_core/processing.py

@@ -3,7 +3,7 @@ from typing import List, Dict, Any, Union, Optional
 from langchain_core.output_parsers import StrOutputParser
 
 from .config import settings
-from .llm_interface import get_llm
+from .llm_interface import get_llm, invoke_llm
 from .prompts import SUMMARIZER_PROMPT
 from .graph_operations import format_doc_for_llm # Reuse formatting
 
@@ -55,7 +55,7 @@ def summarize_document(doc_content: str) -> str:
     try:
         summarize_llm = get_llm(settings.summarize_llm_model)
         summarize_chain = SUMMARIZER_PROMPT | summarize_llm | StrOutputParser()
-        summary = summarize_chain.invoke({"document": doc_content})
+        summary = invoke_llm(summarize_chain, {"document": doc_content})
         logger.debug("Summarization complete.")
         return summary
     except Exception as e: