File size: 386 Bytes
6c0aeb9
 
 
 
892c58b
6c0aeb9
 
 
 
892c58b
6c0aeb9
 
 
 
 
 
892c58b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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)