mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-25 01:20:11 +08:00
@@ -242,7 +242,7 @@ def _load_dify_retrieval_module(monkeypatch):
|
||||
monkeypatch.setitem(sys.modules, "api.db.joint_services.tenant_model_service", tenant_model_service_mod)
|
||||
|
||||
module_name = "test_dify_retrieval_routes_unit_module"
|
||||
module_path = repo_root / "api" / "apps" / "sdk" / "dify_retrieval.py"
|
||||
module_path = repo_root / "api" / "apps" / "restful_apis" / "dify_retrieval_api.py"
|
||||
spec = importlib.util.spec_from_file_location(module_name, module_path)
|
||||
module = importlib.util.module_from_spec(spec)
|
||||
module.manager = _DummyManager()
|
||||
|
||||
@@ -84,7 +84,7 @@ def test_multi_dataset_search_with_metadata_filter(rest_client, ensure_parsed_do
|
||||
@pytest.mark.p2
|
||||
def test_retrieval_compatibility_endpoint(rest_client, ensure_parsed_document):
|
||||
dataset_id, _ = ensure_parsed_document()
|
||||
# /api/v1/retrieval is SDK compatibility endpoint from api/apps/sdk/doc.py.
|
||||
# /api/v1/retrieval is SDK compatibility endpoint registered from chunk_api.py.
|
||||
res = rest_client.post(
|
||||
"/retrieval",
|
||||
json={"dataset_ids": [dataset_id], "question": "test TXT file", "top_k": 5},
|
||||
|
||||
Reference in New Issue
Block a user