Update operate.py
This commit is contained in:
parent
7edf087baa
commit
86a0a4872e
1 changed files with 4 additions and 4 deletions
|
|
@ -1929,10 +1929,7 @@ async def _build_query_context(
|
||||||
(hl_entities_context, hl_relations_context, hl_chunks) = hl_data
|
(hl_entities_context, hl_relations_context, hl_chunks) = hl_data
|
||||||
|
|
||||||
# Collect chunks from entity and relationship sources
|
# Collect chunks from entity and relationship sources
|
||||||
all_chunks.extend(ll_chunks)
|
# Get vector chunks first if in mix mode
|
||||||
all_chunks.extend(hl_chunks)
|
|
||||||
|
|
||||||
# Get vector chunks if in mix mode
|
|
||||||
if query_param.mode == "mix" and chunks_vdb:
|
if query_param.mode == "mix" and chunks_vdb:
|
||||||
vector_chunks = await _get_vector_context(
|
vector_chunks = await _get_vector_context(
|
||||||
query,
|
query,
|
||||||
|
|
@ -1941,6 +1938,9 @@ async def _build_query_context(
|
||||||
)
|
)
|
||||||
all_chunks.extend(vector_chunks)
|
all_chunks.extend(vector_chunks)
|
||||||
|
|
||||||
|
all_chunks.extend(ll_chunks)
|
||||||
|
all_chunks.extend(hl_chunks)
|
||||||
|
|
||||||
# Combine entities and relations contexts
|
# Combine entities and relations contexts
|
||||||
entities_context = process_combine_contexts(
|
entities_context = process_combine_contexts(
|
||||||
ll_entities_context, hl_entities_context
|
ll_entities_context, hl_entities_context
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue