diff --git a/ai_web_common/flask_sso/ext.py b/ai_web_common/flask_sso/ext.py index 2071ac229761ff881556a1a281a6d75113553083..555d19b7a6bf3466bac8fda462c22e6e14a8cb3b 100644 --- a/ai_web_common/flask_sso/ext.py +++ b/ai_web_common/flask_sso/ext.py @@ -14,6 +14,7 @@ from flask import request, session, redirect, make_response, g, \ from flask_talisman import ALLOW_FROM from opentelemetry import trace +from opentelemetry.semconv.trace import SpanAttributes def init_sso(app, talisman): @@ -148,7 +149,7 @@ def auth_required(unauth_action=_redirect_to_login_server): # Add authentication info to the current trace, if any. current_span = trace.get_current_span() if current_span: - current_span.set_attribute('auth.user', g.current_user) + current_span.set_attribute(SpanAttributes.ENDUSER_ID, g.current_user) return func(*args, **kwargs) return _auth_wrapper