feat: log api status to mlflow (#860)

* log api status to mlflow to frontend tracking

* fix CI

---------

Co-authored-by: Xu Yang <xuyang1@microsoft.com>
This commit is contained in:
Xu Yang
2025-05-09 12:20:49 +08:00
committed by GitHub
parent 5fecefb802
commit 1fd8abce0f
3 changed files with 20 additions and 1 deletions
+3 -1
View File
@@ -75,10 +75,12 @@ class RDAgentTimer:
class RDAgentTimerWrapper(SingletonBaseClass):
def __init__(self) -> None:
self.timer: RDAgentTimer = RDAgentTimer()
self.api_fail_count: int = 0
self.latest_api_fail_time: datetime | None = None
def replace_timer(self, timer: RDAgentTimer) -> None:
self.timer = timer
logger.info("Timer replaced successfully.")
RD_Agent_TIMER_wrapper = RDAgentTimerWrapper()
RD_Agent_TIMER_wrapper: RDAgentTimerWrapper = RDAgentTimerWrapper()