fix(api): update user retrieval logic in get_user function

2025-09-17 23:23:58 +08:00 · 2025-09-17 23:23:58 +08:00 · eed82f7ca7
parent e386f350d1
commit eed82f7ca7
1 changed files with 10 additions and 1 deletions
--- a/api/controllers/inner_api/plugin/wraps.py
+++ b/api/controllers/inner_api/plugin/wraps.py
@ -32,11 +32,20 @@ def get_user(tenant_id: str, user_id: str | None) -> EndUser:
            user_model = (
                session.query(EndUser)
                .where(
-                    EndUser.session_id == user_id,
+                    EndUser.id == user_id,
                    EndUser.tenant_id == tenant_id,
                )
                .first()
            )
+            if not user_model:
+                user_model = (
+                    session.query(EndUser)
+                    .where(
+                        EndUser.session_id == user_id,
+                        EndUser.tenant_id == tenant_id,
+                    )
+                    .first()
+                )
            if not user_model:
                user_model = EndUser(
                    tenant_id=tenant_id,