File tree Expand file tree Collapse file tree 1 file changed +15
-3
lines changed Expand file tree Collapse file tree 1 file changed +15
-3
lines changed Original file line number Diff line number Diff line change 45
45
46
46
logger = logging .getLogger (__name__ )
47
47
48
- # Register Starlette User as an implementation of a2a.auth.user.User
49
- A2AUser .register (BaseUser )
48
+
49
+ class StarletteUserProxy (A2AUser ):
50
+ """Adapts the Starlette User class to the A2A user representation."""
51
+
52
+ def __init__ (self , user : BaseUser ):
53
+ self ._user = user
54
+
55
+ @property
56
+ def is_authenticated (self ):
57
+ return self ._user .is_authenticated
58
+
59
+ @property
60
+ def user_name (self ):
61
+ return self ._user .display_name
50
62
51
63
52
64
class CallContextBuilder (ABC ):
@@ -64,7 +76,7 @@ def build(self, request: Request) -> ServerCallContext:
64
76
user = UnauthenticatedUser ()
65
77
state = {}
66
78
with contextlib .suppress (Exception ):
67
- user = request .user
79
+ user = StarletteUserProxy ( request .user )
68
80
state ['auth' ] = request .auth
69
81
return ServerCallContext (user = user , state = state )
70
82
You can’t perform that action at this time.
0 commit comments