1 files changed, 43 insertions, 0 deletions
diff --git a/Omni/Llamacpp.py b/Omni/Llamacpp.py
new file mode 100644
index 0000000..66b57d8
--- /dev/null
+++ b/Omni/Llamacpp.py
@@ -0,0 +1,43 @@
+"""
+Test that llama.cpp can build and exec in the omni repo.
+
+Note that this does not test if llama-cpp can actually execute any models. I
+(currently) use ollama for running and managing models, but I'd like to make
+sure llama-cpp still works in case I need/want to switch at some point.
+"""
+
+# : out llamacpp-test
+# : run llama-cpp
+
+import os
+import sys
+import unittest
+
+
+class TestLlamaCpp(unittest.TestCase):
+    """Test that llama.cpp is available."""
+
+    def test_in_path(self) -> None:
+        """Test that llama.cpp is in $PATH."""
+        self.assertIn("llama-cpp", os.environ.get("PATH", ""))
+
+
+def test() -> None:
+    """Run this module's test suite."""
+    suite = unittest.TestSuite()
+    suite.addTests(
+        unittest.defaultTestLoader.loadTestsFromTestCase(TestLlamaCpp),
+    )
+    unittest.TextTestRunner().run(suite)
+
+
+def main() -> None:
+    """Entrypoint."""
+    if sys.argv[1] == "test":
+        test()
+    else:
+        sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()