Skip to content

Commit f3eadce

Browse files
committed
update retrievers
Signed-off-by: Mustafa <mustafa.cetin@intel.com>
1 parent 142fb76 commit f3eadce

2 files changed

Lines changed: 9 additions & 6 deletions

File tree

comps/dataprep/src/integrations/redis.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ async def ingest_chunks_to_redis(file_name: str, chunks: List, embedder, index_n
225225
await create_index(client)
226226

227227
try:
228-
await store_by_id(client, key=file_name, value="#".join(file_ids))
228+
await store_by_id(client, key=encode_filename(ingest_index_name) + "_" + file_name, value="#".join(file_ids))
229229
except Exception as e:
230230
if logflag:
231231
logger.info(f"[ redis ingest chunks ] {e}. Fail to store chunks of file {file_name}.")
@@ -389,7 +389,9 @@ async def ingest_files(
389389

390390
for file in files:
391391
encode_file = encode_filename(file.filename)
392-
doc_id = "file:" + encode_file
392+
index_name_id = encode_filename(INDEX_NAME if index_name is None else index_name)
393+
doc_id = "file:" + index_name_id + "_" + encode_file
394+
393395
if logflag:
394396
logger.info(f"[ redis ingest ] processing file {doc_id}")
395397

@@ -405,7 +407,7 @@ async def ingest_files(
405407
if key_ids:
406408
raise HTTPException(
407409
status_code=400,
408-
detail=f"Uploaded file {file.filename} already exists. Please change file name.",
410+
detail=f"Uploaded file {file.filename} already exists. Please change file name or index name.",
409411
)
410412

411413
save_path = upload_folder + encode_file
@@ -436,7 +438,8 @@ async def ingest_files(
436438
raise HTTPException(status_code=400, detail=f"Link_list {link_list} should be a list.")
437439
for link in link_list:
438440
encoded_link = encode_filename(link)
439-
doc_id = "file:" + encoded_link + ".txt"
441+
index_name_id = encode_filename(INDEX_NAME if index_name is None else index_name)
442+
doc_id = "file:" + index_name_id + "_" + encoded_link + ".txt"
440443
if logflag:
441444
logger.info(f"[ redis ingest] processing link {doc_id}")
442445

@@ -451,7 +454,7 @@ async def ingest_files(
451454
logger.info(f"[ redis ingest] Link {link} does not exist. Keep storing.")
452455
if key_ids:
453456
raise HTTPException(
454-
status_code=400, detail=f"Uploaded link {link} already exists. Please change another link."
457+
status_code=400, detail=f"Uploaded link {link} already exists. Please change another link or index_name."
455458
)
456459

457460
save_path = upload_folder + encoded_link + ".txt"

tests/dataprep/test_dataprep_redis.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ function main() {
9999

100100
build_docker_images
101101
start_service
102-
102+
103103
validate_microservice
104104

105105
stop_docker

0 commit comments

Comments
 (0)