zrguo commited on
Commit
53d7ba8
·
1 Parent(s): 0e48aab

Update operate.py

Browse files
Files changed (1) hide show
  1. lightrag/operate.py +4 -4
lightrag/operate.py CHANGED
@@ -1929,10 +1929,7 @@ async def _build_query_context(
1929
  (hl_entities_context, hl_relations_context, hl_chunks) = hl_data
1930
 
1931
  # Collect chunks from entity and relationship sources
1932
- all_chunks.extend(ll_chunks)
1933
- all_chunks.extend(hl_chunks)
1934
-
1935
- # Get vector chunks if in mix mode
1936
  if query_param.mode == "mix" and chunks_vdb:
1937
  vector_chunks = await _get_vector_context(
1938
  query,
@@ -1941,6 +1938,9 @@ async def _build_query_context(
1941
  )
1942
  all_chunks.extend(vector_chunks)
1943
 
 
 
 
1944
  # Combine entities and relations contexts
1945
  entities_context = process_combine_contexts(
1946
  ll_entities_context, hl_entities_context
 
1929
  (hl_entities_context, hl_relations_context, hl_chunks) = hl_data
1930
 
1931
  # Collect chunks from entity and relationship sources
1932
+ # Get vector chunks first if in mix mode
 
 
 
1933
  if query_param.mode == "mix" and chunks_vdb:
1934
  vector_chunks = await _get_vector_context(
1935
  query,
 
1938
  )
1939
  all_chunks.extend(vector_chunks)
1940
 
1941
+ all_chunks.extend(ll_chunks)
1942
+ all_chunks.extend(hl_chunks)
1943
+
1944
  # Combine entities and relations contexts
1945
  entities_context = process_combine_contexts(
1946
  ll_entities_context, hl_entities_context