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

Fix: model list (#15860)

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

Remove tenant_llm call in rag.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Lynn
2026-06-10 14:59:57 +08:00
committed by GitHub
parent 899f76af6b
commit 478c9846a1
12 changed files with 64 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/testcases/restful_api/test_user_tenant_routes_unit.py
View File

@@ -1036,23 +1036,14 @@ def test_logout_setting_profile_matrix_unit(monkeypatch):
def test_registration_helpers_and_register_route_matrix_unit(monkeypatch):
module = _load_user_app(monkeypatch)
deleted = {"user": 0, "tenant": 0, "user_tenant": 0, "tenant_llm": 0}
deleted = {"user": 0, "tenant": 0, "user_tenant": 0}
monkeypatch.setattr(module.UserService, "delete_by_id", lambda _user_id: deleted.__setitem__("user", deleted["user"] + 1))
monkeypatch.setattr(module.TenantService, "delete_by_id", lambda _tenant_id: deleted.__setitem__("tenant", deleted["tenant"] + 1))
monkeypatch.setattr(module.UserTenantService, "query", lambda **_kwargs: [SimpleNamespace(id="ut-1")])
monkeypatch.setattr(module.UserTenantService, "delete_by_id", lambda _ut_id: deleted.__setitem__("user_tenant", deleted["user_tenant"] + 1))
class _DeleteQuery:
def where(self, *_args, **_kwargs):
return self
def execute(self):
deleted["tenant_llm"] += 1
return 1
monkeypatch.setattr(module.TenantLLM, "delete", lambda: _DeleteQuery())
module.rollback_user_registration("user-1")
assert deleted == {"user": 1, "tenant": 1, "user_tenant": 1, "tenant_llm": 1}, deleted
assert deleted == {"user": 1, "tenant": 1, "user_tenant": 1}, deleted
monkeypatch.setattr(module.UserService, "delete_by_id", lambda _user_id: (_ for _ in ()).throw(RuntimeError("u boom")))
monkeypatch.setattr(module.TenantService, "delete_by_id", lambda _tenant_id: (_ for _ in ()).throw(RuntimeError("t boom")))
@@ -1062,7 +1053,6 @@ def test_registration_helpers_and_register_route_matrix_unit(monkeypatch):
def where(self, *_args, **_kwargs):
raise RuntimeError("llm boom")
monkeypatch.setattr(module.TenantLLM, "delete", lambda: _RaisingDeleteQuery())
module.rollback_user_registration("user-2")
monkeypatch.setattr(module.UserService, "save", lambda **_kwargs: False)