mtfyun/ragflow
1
0 Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-11-13 23:01:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Agent bot 2 permission bugs (#818) (#16909)

Browse Source
This commit is contained in:
Wang Qi authored and GitHub committed 2026-07-14 20:11:16 +08:00
1 parent 760203be42
commit 3087eb854c
5 files changed
+37 -52
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

test/unit_test/api/apps/restful_apis/test_attachment_download_missing_blob.py
+4 -1
View File
@@ -71,8 +71,11 @@ def _load_agent_api(monkeypatch, *, storage_get):
_stub(
monkeypatch,
"api.apps",
AUTH_JWT="jwt",
AUTH_API="api",
AUTH_BETA="beta",
current_user=SimpleNamespace(id="tenant-1"),
login_required=lambda func: func,
login_required=lambda func=None, **_kwargs: (lambda f: f) if func is None else func,
)
_stub(monkeypatch, "api.apps.services.canvas_replica_service", CanvasReplicaService=SimpleNamespace())
_stub(monkeypatch, "api.db", CanvasCategory=SimpleNamespace())
test/unit_test/api/apps/restful_apis/test_get_agent_session.py
+10 -1
View File
@@ -44,7 +44,16 @@ def _load_agent_api(monkeypatch, get_by_id_result, delete_calls=None):
delete_calls.append(session_id)
return True
_stub(monkeypatch, "api.apps", current_user=SimpleNamespace(id="tenant-1"), login_required=lambda func: func)
_stub(
monkeypatch,
"api.apps",
AUTH_JWT="jwt",
AUTH_API="api",
AUTH_BETA="beta",
QuartAuthUnauthorized=Exception,
current_user=SimpleNamespace(id="tenant-1", is_superuser=True),
login_required=lambda func=None, **_kwargs: (lambda f: f) if func is None else func,
)
_stub(monkeypatch, "api.apps.services.canvas_replica_service", CanvasReplicaService=SimpleNamespace())
_stub(monkeypatch, "api.db", CanvasCategory=SimpleNamespace())
_stub(monkeypatch, "api.db.db_models", Task=SimpleNamespace())
test/unit_test/api/apps/restful_apis/test_rerun_agent_authorization.py
+10 -1
View File
@@ -48,7 +48,16 @@ def _load_agent_api_for_rerun(monkeypatch, *, documents_info, accessible):
acc_fn = accessible if callable(accessible) else (lambda *_a, **_k: accessible)
_stub(monkeypatch, "api.apps", current_user=SimpleNamespace(id="user-owner"), login_required=lambda func: func)
_stub(
monkeypatch,
"api.apps",
AUTH_JWT="jwt",
AUTH_API="api",
AUTH_BETA="beta",
QuartAuthUnauthorized=Exception,
current_user=SimpleNamespace(id="user-owner", is_superuser=True),
login_required=lambda func=None, **_kwargs: (lambda f: f) if func is None else func,
)
_stub(monkeypatch, "api.apps.services.canvas_replica_service", CanvasReplicaService=SimpleNamespace())
_stub(monkeypatch, "api.db", CanvasCategory=SimpleNamespace())