Files
FastDeploy/fastdeploy/trace/trace_logger.py
T
gongweibao a6351dea0b [BugFix][Optimization] Replace silent failures with catchable exceptions and informative error messages (#6533)
* init

* init

* fix format

* add

* add files

* add ut

* fix some

* add ut

* add more

* add

* fix pre-commit

* fix pre-commit

* fix cover

* skip long seq

* add

* add

* fix

* remove not need

* fix set attr

* fix comments

* fix comments

* fix failed tests

---------

Co-authored-by: gongweibao <gognweibao@baidu.com>
2026-03-16 21:32:43 +08:00

49 lines
1.6 KiB
Python

# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from fastdeploy.metrics.trace import get_trace_info_for_request
from fastdeploy.trace.constants import EVENT_TO_STAGE_MAP
from fastdeploy.utils import trace_logger
def print(event, request_id, user):
"""
Records task tracking log information, including task name, start time, end time, etc.
Args:
event: Event to be recorded.
request_id: Request ID.
user: User identifier.
"""
try:
attributes = {
"request_id": f"{request_id}",
"user_id": f"{user}",
"event": event.value,
"stage": EVENT_TO_STAGE_MAP.get(event).value,
}
trace_info = get_trace_info_for_request(request_id)
if trace_info:
attributes["trace_id"] = trace_info["trace_id"]
else:
attributes["trace_id"] = "unknown"
trace_logger.info(
"",
extra={"attributes": attributes},
stacklevel=2,
)
except Exception as e:
trace_logger.debug(f"Failed to log trace event: {e}")