From 1e03888cef9225d0c554f375d04ed156e6da7252 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 7 May 2025 10:57:33 +0800 Subject: [PATCH] Change function name get_kg_context to _get_kg_context --- lightrag/operate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lightrag/operate.py b/lightrag/operate.py index 2f1a8a57..ce1e76f6 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -1156,7 +1156,7 @@ async def mix_kg_vector_query( ) # 2. Execute knowledge graph and vector searches in parallel - async def get_kg_context(): + async def _get_kg_context(): try: hl_keywords, ll_keywords = await get_keywords_from_query( query, query_param, global_config, hashing_kv @@ -1194,13 +1194,13 @@ async def mix_kg_vector_query( return context except Exception as e: - logger.error(f"Error in get_kg_context: {str(e)}") + logger.error(f"Error in _get_kg_context: {str(e)}") traceback.print_exc() return None # 3. Execute both retrievals in parallel kg_context, vector_context = await asyncio.gather( - get_kg_context(), _get_vector_context(query, chunks_vdb, query_param, tokenizer) + _get_kg_context(), _get_vector_context(query, chunks_vdb, query_param, tokenizer) ) # 4. Merge contexts