fix linting errors
Browse files- 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(
|
|
|
|
|
|
|
|
|
|
| 609 |
query_param.mode = "global"
|
| 610 |
if hl_keywords == [] and query_param.mode in ["global", "hybrid"]:
|
| 611 |
-
logger.warning(
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 703 |
ll_keywords,
|
| 704 |
knowledge_graph_inst,
|
| 705 |
entities_vdb,
|
| 706 |
text_chunks_db,
|
| 707 |
query_param,
|
| 708 |
)
|
| 709 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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,
|