diff --git a/telegram_inference_bot.py b/telegram_inference_bot.py index 04076d3..9aeb7e8 100644 --- a/telegram_inference_bot.py +++ b/telegram_inference_bot.py @@ -19,19 +19,22 @@ client = OpenAI() GPT_4O = "gpt-4o" GPT_4O_MINI = "gpt-4o-mini" -class NoDebugFilter(logging.Filter): - def filter(self, record): - return "DEBUG" not in record.getMessage() +class StringFilter(logging.Filter): + def __init__(self, string_to_filter): + super().__init__() + self.string_to_filter = string_to_filter -# Set up logging to console and file with filter + def filter(self, record): + return self.string_to_filter not in record.getMessage() + +string_to_filter = 'unwanted_string' # Change this to the specific string you want to filter out + +# Set up logging to console and file logging.basicConfig(level=logging.INFO, handlers=[ logging.StreamHandler(), logging.FileHandler('logs/output.log', mode='a') ]) - -# Add filter to the handlers -for handler in logging.getLogger().handlers: - handler.addFilter(NoDebugFilter()) +logging.getLogger().addFilter(StringFilter(string_to_filter)) # Set up Telegram bot TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN')