deepdraft / src /deepengineer /logging_tools.py
Charles Azam
clean: run linting and formating on repo
892c58b
raw
history blame contribute delete
386 Bytes
import queue
from typing import Any
from smolagents import Tool
class LoggingTool(Tool):
"""
Base class for tools that can push logs to a queue.
"""
def __init__(self, log_queue: queue.Queue | None = None):
super().__init__()
self.log_queue = log_queue
def push_log(self, msg: str):
if self.log_queue:
self.log_queue.put(msg)