zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-08 08:28:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: display embedding model name in list dataset api response (#17832)

Browse Source
This commit is contained in:
Lynn
2026-08-05 14:51:07 +08:00
committed by GitHub
parent 510e1197c6
commit 02923bfc46
4 changed files with 41 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/unit_test/api/apps/services/test_dataset_api_service_list_datasets.py
View File

@@ -82,6 +82,7 @@ def _load_list_datasets_module(monkeypatch, *, kbs, parsing_status_by_kb):
_stub(
monkeypatch,
"api.db.joint_services.tenant_model_service",
get_composite_model_name_by_ids=MagicMock(),
resolve_model_config=MagicMock(),
resolve_model_id=MagicMock(),
)
@@ -188,8 +189,8 @@ def _load_list_datasets_module(monkeypatch, *, kbs, parsing_status_by_kb):
def _stub_kbs():
return [
{"id": "kb-a", "tenant_id": "tenant-1", "name": "Alpha"},
{"id": "kb-b", "tenant_id": "tenant-1", "name": "Beta"},
{"id": "kb-a", "tenant_id": "tenant-1", "name": "Alpha", "embedding_model": "emb-a"},
{"id": "kb-b", "tenant_id": "tenant-1", "name": "Beta", "embedding_model": "emb-b"},
]

1
test/unit_test/api/apps/services/test_delete_datasets.py
View File

@@ -113,6 +113,7 @@ def _load_delete_datasets_module(monkeypatch, *, f2d_rows, file_filter_delete):
_stub(
monkeypatch,
"api.db.joint_services.tenant_model_service",
get_composite_model_name_by_ids=MagicMock(),
get_model_config_from_provider_instance=MagicMock(),
resolve_model_config=MagicMock(),
resolve_model_id=MagicMock(),