File size: 670 Bytes
1b7e88c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import logging
import requests
from omagent_core.engine.task_client import TaskClient
class ConductorLogHandler(logging.Handler):
def __init__(self, task_client):
super().__init__()
self.task_client: TaskClient = task_client
self.task_id = None
def set_task_id(self, task_id):
self.task_id = task_id
def emit(self, record):
if not self.task_id:
return super().emit(record)
log_entry = self.format(record)
try:
self.task_client.log(log_entry, self.task_id)
except requests.exceptions.RequestException as e:
print(f"Failed to send log to Conductor: {e}")
|