| 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}") |
|
|
