Fix test cases (#3718)

### What problem does this PR solve?

Fix test cases

### Type of change

- [x] Other (please describe): Fix error cases

---------

Signed-off-by: jinhai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2024-11-28 17:37:46 +08:00
committed by GitHub
parent 964a6f4ec4
commit cdae8d28fe
8 changed files with 20 additions and 19 deletions

View File

@ -190,4 +190,7 @@ def test_retrieve_chunks(get_api_key_fixture):
docs = ds.upload_documents(documents)
doc = docs[0]
doc.add_chunk(content="This is a chunk addition test")
rag.retrieve(dataset_ids=[ds.id],document_ids=[doc.id])
rag.retrieve(dataset_ids=[ds.id],document_ids=[doc.id])
rag.delete_datasets(ids=[ds.id])
# test different parameters for the retrieval