LarFii commited on
Commit
9049e56
·
1 Parent(s): 720f8fb

fix linting errors

Browse files
Files changed (1) hide show
  1. lightrag/operate.py +18 -5
lightrag/operate.py CHANGED
@@ -4,7 +4,6 @@ import re
4
  from tqdm.asyncio import tqdm as tqdm_async
5
  from typing import Union
6
  from collections import Counter, defaultdict
7
- import warnings
8
  from .utils import (
9
  logger,
10
  clean_str,
@@ -605,10 +604,16 @@ async def kg_query(
605
  logger.warning("low_level_keywords and high_level_keywords is empty")
606
  return PROMPTS["fail_response"]
607
  if ll_keywords == [] and query_param.mode in ["local", "hybrid"]:
608
- logger.warning("low_level_keywords is empty, switching from %s mode to global mode", query_param.mode)
 
 
 
609
  query_param.mode = "global"
610
  if hl_keywords == [] and query_param.mode in ["global", "hybrid"]:
611
- logger.warning("high_level_keywords is empty, switching from %s mode to local mode", query_param.mode)
 
 
 
612
  query_param.mode = "local"
613
 
614
  ll_keywords = ", ".join(ll_keywords) if ll_keywords else ""
@@ -699,14 +704,22 @@ async def _build_query_context(
699
  query_param,
700
  )
701
  else: # hybrid mode
702
- ll_entities_context, ll_relations_context, ll_text_units_context = await _get_node_data(
 
 
 
 
703
  ll_keywords,
704
  knowledge_graph_inst,
705
  entities_vdb,
706
  text_chunks_db,
707
  query_param,
708
  )
709
- hl_entities_context, hl_relations_context, hl_text_units_context = await _get_edge_data(
 
 
 
 
710
  hl_keywords,
711
  knowledge_graph_inst,
712
  relationships_vdb,
 
4
  from tqdm.asyncio import tqdm as tqdm_async
5
  from typing import Union
6
  from collections import Counter, defaultdict
 
7
  from .utils import (
8
  logger,
9
  clean_str,
 
604
  logger.warning("low_level_keywords and high_level_keywords is empty")
605
  return PROMPTS["fail_response"]
606
  if ll_keywords == [] and query_param.mode in ["local", "hybrid"]:
607
+ logger.warning(
608
+ "low_level_keywords is empty, switching from %s mode to global mode",
609
+ query_param.mode,
610
+ )
611
  query_param.mode = "global"
612
  if hl_keywords == [] and query_param.mode in ["global", "hybrid"]:
613
+ logger.warning(
614
+ "high_level_keywords is empty, switching from %s mode to local mode",
615
+ query_param.mode,
616
+ )
617
  query_param.mode = "local"
618
 
619
  ll_keywords = ", ".join(ll_keywords) if ll_keywords else ""
 
704
  query_param,
705
  )
706
  else: # hybrid mode
707
+ (
708
+ ll_entities_context,
709
+ ll_relations_context,
710
+ ll_text_units_context,
711
+ ) = await _get_node_data(
712
  ll_keywords,
713
  knowledge_graph_inst,
714
  entities_vdb,
715
  text_chunks_db,
716
  query_param,
717
  )
718
+ (
719
+ hl_entities_context,
720
+ hl_relations_context,
721
+ hl_text_units_context,
722
+ ) = await _get_edge_data(
723
  hl_keywords,
724
  knowledge_graph_inst,
725
  relationships_vdb,