setup context manger for SignalClient and LLMClient

2026-04-17 04:58:19 -04:00 · 2026-03-09 11:39:02 -04:00
parent f762f12bd2
commit c73aa5c98a
3 changed files with 23 additions and 9 deletions
--- a/python/signal_bot/main.py
+++ b/python/signal_bot/main.py
@@ -165,15 +165,13 @@ def main(
        inventory_file=inventory_file,
    )

-    signal = SignalClient(config.signal_api_url, config.phone_number)
-    llm = LLMClient(model=llm_model, host=llm_host, port=llm_port)
-    registry = DeviceRegistry(signal, Path(registry_file))

-    try:
+    with (
+        SignalClient(config.signal_api_url, config.phone_number) as signal,
+        LLMClient(model=llm_model, host=llm_host, port=llm_port) as llm,
+    ):
+        registry = DeviceRegistry(signal, Path(registry_file))
        run_loop(config, signal, llm, registry)
-    finally:
-        signal.close()
-        llm.close()


 if __name__ == "__main__":