zrguo
commited on
Commit
·
53d7ba8
1
Parent(s):
0e48aab
Update operate.py
Browse files- 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 |
-
|
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
|