Test: Added test cases for Retrieval Chunks HTTP API (#6649)

### What problem does this PR solve?

cover [retrieval
chunk](https://ragflow.io/docs/v0.17.2/http_api_reference#retrieve-chunks)
endpoints

### Type of change

- [x]  add test cases
This commit is contained in:
liu an
2025-03-31 10:05:35 +08:00
committed by GitHub
parent 9aa047257a
commit aca4cf4369
2 changed files with 309 additions and 0 deletions

View File

@ -172,6 +172,12 @@ def delete_chunks(auth, dataset_id, document_id, payload=None):
return res.json()
def retrieval_chunks(auth, payload=None):
url = f"{HOST_ADDRESS}/api/v1/retrieval"
res = requests.post(url=url, headers=HEADERS, auth=auth, json=payload)
return res.json()
def batch_add_chunks(auth, dataset_id, document_id, num):
chunk_ids = []
for i in range(num):