yangdx commited on
Commit
dadfdd4
·
1 Parent(s): bdc25f8

Update logger message

Browse files
Files changed (1) hide show
  1. lightrag/operate.py +10 -0
lightrag/operate.py CHANGED
@@ -1185,6 +1185,13 @@ async def mix_kg_vector_query(
1185
  tokenizer=tokenizer,
1186
  )
1187
 
 
 
 
 
 
 
 
1188
  if not maybe_trun_chunks:
1189
  return None
1190
 
@@ -2013,6 +2020,9 @@ async def naive_query(
2013
  logger.debug(
2014
  f"Truncate chunks from {len(chunks)} to {len(maybe_trun_chunks)} (max tokens:{query_param.max_token_for_text_unit})"
2015
  )
 
 
 
2016
 
2017
  section = "\n--New Chunk--\n".join(
2018
  [
 
1185
  tokenizer=tokenizer,
1186
  )
1187
 
1188
+ logger.debug(
1189
+ f"Truncate chunks from {len(valid_chunks)} to {len(maybe_trun_chunks)} (max tokens:{query_param.max_token_for_text_unit})"
1190
+ )
1191
+ logger.info(
1192
+ f"Naive query: {len(maybe_trun_chunks)} chunks, top_k: {mix_topk}"
1193
+ )
1194
+
1195
  if not maybe_trun_chunks:
1196
  return None
1197
 
 
2020
  logger.debug(
2021
  f"Truncate chunks from {len(chunks)} to {len(maybe_trun_chunks)} (max tokens:{query_param.max_token_for_text_unit})"
2022
  )
2023
+ logger.info(
2024
+ f"Naive query: {len(maybe_trun_chunks)} chunks, top_k: {query_param.top_k}"
2025
+ )
2026
 
2027
  section = "\n--New Chunk--\n".join(
2028
  [