From eb09692c317f4f79e2ab3cf98dbf1b53488181ff Mon Sep 17 00:00:00 2001 From: Andrew Brookins Date: Wed, 12 Feb 2025 05:18:47 -0800 Subject: [PATCH] Use set_client() instead of passing in a client instance (#30) --- backend/arxivsearch/db/load.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/arxivsearch/db/load.py b/backend/arxivsearch/db/load.py index d47c1bd..a7b0f36 100644 --- a/backend/arxivsearch/db/load.py +++ b/backend/arxivsearch/db/load.py @@ -65,7 +65,7 @@ async def preprocess_paper(paper: dict) -> dict: _ = await index.load( data=papers, preprocess=preprocess_paper, - concurrency=config.WRITE_CONCURRENCY, + concurrency=int(config.WRITE_CONCURRENCY), id_field="id", ) @@ -74,7 +74,8 @@ async def preprocess_paper(paper: dict) -> dict: async def load_data(): # Load schema specs and create index in Redis - index = AsyncSearchIndex(redis_helpers.schema, redis_helpers.client) + index = AsyncSearchIndex(redis_helpers.schema) + await index.set_client(redis_helpers.client) # Load dataset and create index try: