fix: Add node_name and node_type to context search

This commit is contained in:
Igor Ilic 2025-08-29 13:38:52 +02:00
parent e6ee182d78
commit b9fa285c1a

View file

@ -228,7 +228,15 @@ async def specific_search_by_context(
"""
async def _search_by_context(
dataset, user, query_type, query_text, system_prompt_path, top_k, last_k
dataset,
user,
query_type,
query_text,
system_prompt_path,
top_k,
node_type: Optional[Type] = NodeSet,
node_name: Optional[List[str]] = None,
last_k: Optional[int] = None,
):
# Set database configuration in async context for each dataset user has access for
await set_database_global_context_variables(dataset.id, dataset.owner_id)