fix: Resolve backwards compatibility of search

This commit is contained in:
Igor Ilic 2025-09-11 20:25:03 +02:00
parent 3cf1722527
commit 89207780e9

View file

@ -132,14 +132,31 @@ async def search(
], ],
) )
else: else:
return [ # This is for maintaining backwards compatibility
SearchResult( if os.getenv("ENABLE_BACKEND_ACCESS_CONTROL", "false").lower() == "true":
search_result=result, return_value = {}
dataset_id=datasets[min(index, len(datasets) - 1)].id if datasets else None, for search_result in search_results:
dataset_name=datasets[min(index, len(datasets) - 1)].name if datasets else None, result, context, datasets = search_result
) return_value[str(datasets[0].id)] = {
for index, (result, _, datasets) in enumerate(search_results) "search_result": result,
] "dataset_id": str(datasets[0].id),
}
return return_value
else:
return_value = []
for search_result in search_results:
result, context, datasets = search_result
return_value.append(result)
return return_value
# return [
# SearchResult(
# search_result=result,
# dataset_id=datasets[min(index, len(datasets) - 1)].id if datasets else None,
# dataset_name=datasets[min(index, len(datasets) - 1)].name if datasets else None,
# )
# for index, (result, _, datasets) in enumerate(search_results)
# ]
async def authorized_search( async def authorized_search(