pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-01-30 07:06:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: message write testcase (#12417)

Browse Source 
### What problem does this PR solve?

Write testcase for message web apis.

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

---------

Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
This commit is contained in:
Lynn
2026-01-04 16:52:44 +08:00
committed by GitHub
parent 21ba9e6d72
commit a2211c200d
5 changed files with 401 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
test/testcases/test_web_api/test_message_app/conftest.py
View File

@ -38,9 +38,54 @@ def add_empty_raw_type_memory(request, WebApiAuth):
res = create_memory(WebApiAuth, payload)
memory_id = res["data"]["id"]
request.cls.memory_id = memory_id
request.cls.memory_type = payload["memory_type"]
return memory_id
@pytest.fixture(scope="class")
def add_empty_multiple_type_memory(request, WebApiAuth):
def cleanup():
memory_list_res = list_memory(WebApiAuth)
exist_memory_ids = [memory["id"] for memory in memory_list_res["data"]["memory_list"]]
for _memory_id in exist_memory_ids:
delete_memory(WebApiAuth, _memory_id)
request.addfinalizer(cleanup)
payload = {
"name": "test_memory_0",
"memory_type": ["raw"] + random.choices(["semantic", "episodic", "procedural"], k=random.randint(1, 3)),
"embd_id": "BAAI/bge-small-en-v1.5@Builtin",
"llm_id": "glm-4-flash@ZHIPU-AI"
}
res = create_memory(WebApiAuth, payload)
memory_id = res["data"]["id"]
request.cls.memory_id = memory_id
request.cls.memory_type = payload["memory_type"]
return memory_id
@pytest.fixture(scope="class")
def add_2_multiple_type_memory(request, WebApiAuth):
def cleanup():
memory_list_res = list_memory(WebApiAuth)
exist_memory_ids = [memory["id"] for memory in memory_list_res["data"]["memory_list"]]
for _memory_id in exist_memory_ids:
delete_memory(WebApiAuth, _memory_id)
request.addfinalizer(cleanup)
memory_ids = []
for i in range(2):
payload = {
"name": f"test_memory_{i}",
"memory_type": ["raw"] + random.choices(["semantic", "episodic", "procedural"], k=random.randint(1, 3)),
"embd_id": "BAAI/bge-small-en-v1.5@Builtin",
"llm_id": "glm-4-flash@ZHIPU-AI"
}
res = create_memory(WebApiAuth, payload)
memory_ids.append(res["data"]["id"])
request.cls.memory_ids = memory_ids
return memory_ids
@pytest.fixture(scope="class")
def add_memory_with_multiple_type_message_func(request, WebApiAuth):
def cleanup():