|
|
@@ -32,13 +32,14 @@ class R2R:
|
|
|
|
|
|
async def search(self, query: str, filters: dict[str, Any]):
|
|
|
await self._check_login()
|
|
|
- return await self.client.retrieval.search(
|
|
|
+ search_response = await self.client.retrieval.search(
|
|
|
query=query,
|
|
|
search_settings={
|
|
|
"filters": filters,
|
|
|
"limit": tool_settings.R2R_SEARCH_LIMIT,
|
|
|
},
|
|
|
)
|
|
|
+ return search_response.get("results").get("chunk_search_results")
|
|
|
|
|
|
async def _check_login(self):
|
|
|
if not self.auth_enabled:
|
