gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix linting

Browse source
This commit is contained in:
yangdx 2025-07-15 23:57:24 +08:00
parent 5f7cb437e8
commit b1276a079f
1 changed files with 15 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lightrag/operate.py
View file

@ -1981,17 +1981,23 @@ async def _build_query_context(
max_entity_tokens = getattr( max_entity_tokens = getattr(
query_param, query_param,
"max_entity_tokens", "max_entity_tokens",
text_chunks_db.global_config.get("max_entity_tokens", DEFAULT_MAX_ENTITY_TOKENS), text_chunks_db.global_config.get(
"max_entity_tokens", DEFAULT_MAX_ENTITY_TOKENS
),
) )
max_relation_tokens = getattr( max_relation_tokens = getattr(
query_param, query_param,
"max_relation_tokens", "max_relation_tokens",
text_chunks_db.global_config.get("max_relation_tokens", DEFAULT_MAX_RELATION_TOKENS), text_chunks_db.global_config.get(
"max_relation_tokens", DEFAULT_MAX_RELATION_TOKENS
),
) )
max_total_tokens = getattr( max_total_tokens = getattr(
query_param, query_param,
"max_total_tokens", "max_total_tokens",
text_chunks_db.global_config.get("max_total_tokens", DEFAULT_MAX_TOTAL_TOKENS), text_chunks_db.global_config.get(
"max_total_tokens", DEFAULT_MAX_TOTAL_TOKENS
),
) )
# Truncate entities based on complete JSON serialization # Truncate entities based on complete JSON serialization
@ -2350,7 +2356,9 @@ async def _find_most_related_text_unit_from_entities(
text_units = [ text_units = [
split_string_by_multi_markers(dp["source_id"], [GRAPH_FIELD_SEP])[ split_string_by_multi_markers(dp["source_id"], [GRAPH_FIELD_SEP])[
: text_chunks_db.global_config.get("related_chunk_number", DEFAULT_RELATED_CHUNK_NUMBER) : text_chunks_db.global_config.get(
"related_chunk_number", DEFAULT_RELATED_CHUNK_NUMBER
)
] ]
for dp in node_datas for dp in node_datas
if dp["source_id"] is not None if dp["source_id"] is not None
@ -2660,7 +2668,9 @@ async def _find_related_text_unit_from_relationships(
text_units = [ text_units = [
split_string_by_multi_markers(dp["source_id"], [GRAPH_FIELD_SEP])[ split_string_by_multi_markers(dp["source_id"], [GRAPH_FIELD_SEP])[
: text_chunks_db.global_config.get("related_chunk_number", DEFAULT_RELATED_CHUNK_NUMBER) : text_chunks_db.global_config.get(
"related_chunk_number", DEFAULT_RELATED_CHUNK_NUMBER
)
] ]
for dp in edge_datas for dp in edge_datas
if dp["source_id"] is not None if dp["source_id"] is not None