Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,7 @@ def basic_search(
alternate_db_session=search_db_session,
retrieved_sections_callback=callback_container.append,
skip_query_analysis=True,
original_query=rewritten_query,
),
):
# get retrieved docs to send to the rest of the graph
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -293,7 +293,7 @@ def run(
self, override_kwargs: SearchToolOverrideKwargs | None = None, **llm_kwargs: Any
) -> Generator[ToolResponse, None, None]:
query = cast(str, llm_kwargs[QUERY_FIELD])
original_query = None
original_query = query
precomputed_query_embedding = None
precomputed_is_keyword = None
precomputed_keywords = None
Expand All @@ -312,7 +312,7 @@ def run(
kg_sources = None
kg_chunk_id_zero_only = False
if override_kwargs:
original_query = override_kwargs.original_query
original_query = override_kwargs.original_query or query
precomputed_is_keyword = override_kwargs.precomputed_is_keyword
precomputed_keywords = override_kwargs.precomputed_keywords
precomputed_query_embedding = override_kwargs.precomputed_query_embedding
Expand Down
Loading