From 7865263c1c0418a4b58de7776f247de4f6eb460c Mon Sep 17 00:00:00 2001 From: SpudGunMan Date: Sat, 31 Aug 2024 23:46:12 -0700 Subject: [PATCH] Update mesh_bot.py --- mesh_bot.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/mesh_bot.py b/mesh_bot.py index 440c1a8..b149d80 100755 --- a/mesh_bot.py +++ b/mesh_bot.py @@ -23,7 +23,7 @@ def auto_response(message, snr, rssi, hop, message_from_id, channel_number, devi "wxc": lambda: handle_wxc(message_from_id, deviceID, 'wxc'), "wx": lambda: handle_wxc(message_from_id, deviceID, 'wx'), "wiki:": lambda: handle_wiki(message), - "ask:": lambda: handle_llm(message_from_id, channel_number, deviceID, message), + "ask:": lambda: handle_llm(message_from_id, channel_number, deviceID, message, publicChannel), "joke": tell_joke, "bbslist": bbs_list_messages, "bbspost": lambda: handle_bbspost(message, message_from_id, deviceID), @@ -103,7 +103,7 @@ def handle_wiki(message): else: return "Please add a search term example:wiki: travelling gnome" -def handle_llm(message_from_id, channel_number, deviceID, message): +def handle_llm(message_from_id, channel_number, deviceID, message, publicChannel): global llmRunCounter, llmTotalRuntime if "ask:" in message.lower(): user_input = message.split(":")[1] @@ -119,10 +119,16 @@ def handle_llm(message_from_id, channel_number, deviceID, message): averageRuntime = sum(llmTotalRuntime) / len(llmTotalRuntime) if averageRuntime > 25: msg = f"Please wait, average query time is: {int(averageRuntime)} seconds" - send_message(msg, channel_number, message_from_id, deviceID) + if channel_number == publicChannel: + send_message(msg, channel_number, message_from_id, deviceID) + else: + send_message(msg, channel_number, 0, deviceID) else: msg = "Please wait, response could take 3+ minutes. Fund the SysOp's GPU budget!" - send_message(msg, channel_number, message_from_id, deviceID) + if channel_number == publicChannel: + send_message(msg, channel_number, message_from_id, deviceID) + else: + send_message(msg, channel_number, 0, deviceID) start = time.time()