Update kig_core/graph_operations.py

kig_core/graph_operations.py

@@ -47,7 +47,7 @@ def generate_cypher_auto(question: str) -> str:
         | StrOutputParser()
         | extract_cypher
     )
-    return llm_invoke(chain,question)
+    return invoke_llm(chain,question)
 
 def generate_cypher_guided(question: str, plan_step: int) -> str:
     """Generates Cypher using the 'guided' method based on concepts."""
@@ -64,7 +64,7 @@ def generate_cypher_guided(question: str, plan_step: int) -> str:
             | concept_llm
             | StrOutputParser()
         )
-        selected_concept = llm_invoke(concept_chain,{
+        selected_concept = invoke_llm(concept_chain,{
             "question": question,
             "concepts": "\n".join(concepts)
         }).strip()
@@ -179,7 +179,7 @@ def evaluate_documents(
             formatted_doc = format_doc_for_llm(doc)
             if not formatted_doc.strip(): continue
             try:
-                result = llm_invoke(binary_grader,{"question": query, "document": formatted_doc})
+                result = invoke_llm(binary_grader,{"question": query, "document": formatted_doc})
                 logger.debug(f"Binary grader result for doc '{doc.get('title', 'N/A')}': {result}")
                 if result and 'yes' in result.lower():
                     valid_docs_with_scores.append((doc, 1.0)) # Score 1.0 for relevant
@@ -193,7 +193,7 @@ def evaluate_documents(
             formatted_doc = format_doc_for_llm(doc)
             if not formatted_doc.strip(): continue
             try:
-                result: GradeDocumentsScore = llm_invoke(score_grader,{"query": query, "document": formatted_doc})
+                result: GradeDocumentsScore = invoke_llm(score_grader,{"query": query, "document": formatted_doc})
                 logger.debug(f"Score grader result for doc '{doc.get('title', 'N/A')}': Score={result.score}, Rationale={result.rationale}")
                 if result.score >= settings.eval_threshold:
                     valid_docs_with_scores.append((doc, result.score))