refactor: move system_distro, get_installed_programs to utils

ErikBjare · Nov 9, 2024 · 40ecd1d · 40ecd1d
1 parent 1fb1459
commit 40ecd1d
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/gptme/tools/shell.py b/gptme/tools/shell.py
@@ -22,9 +22,7 @@
 logger = logging.getLogger(__name__)
 
 
-@functools.lru_cache
-def get_installed_programs() -> set[str]:
-    candidates = [
+candidates = (
         # platform-specific
         "brew",
         "apt-get",
@@ -35,15 +33,10 @@ def get_installed_programs() -> set[str]:
         "pandoc",
         "git",
         "docker",
-    ]
-    installed = set()
-    for candidate in candidates:
-        if shutil.which(candidate) is not None:
-            installed.add(candidate)
-    return installed
+    )
 
 
-shell_programs_str = "\n".join(f"- {prog}" for prog in get_installed_programs())
+shell_programs_str = "\n".join(f"- {prog}" for prog in get_installed_programs(candidates))
 is_macos = sys.platform == "darwin"
 
 instructions = f"""