From 67e48c59be6229e60cb8c7b020be6099fe69e6a2 Mon Sep 17 00:00:00 2001
From: b4rtaz <b4rtaz@gmail.com>
Date: Sun, 9 Feb 2025 20:02:54 +0100
Subject: [PATCH] build fix.

---
 src/app.hpp        | 5 -----
 src/dllama-api.cpp | 5 ++---
 2 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/src/app.hpp b/src/app.hpp
index e95253a..b066dd6 100644
--- a/src/app.hpp
+++ b/src/app.hpp
@@ -97,9 +97,4 @@ typedef struct {
 void runInferenceApp(AppCliArgs *args, void (*handler)(AppInferenceContext *context));
 void runWorkerApp(AppCliArgs *args);
 
-class BadArgumentException : public std::runtime_error {
-public:
-    explicit BadArgumentException(const std::string& message) : std::runtime_error(message) {}
-};
-
 #endif
diff --git a/src/dllama-api.cpp b/src/dllama-api.cpp
index e67ac1f..8c67430 100644
--- a/src/dllama-api.cpp
+++ b/src/dllama-api.cpp
@@ -564,9 +564,8 @@ int main(int argc, char *argv[]) {
             return EXIT_SUCCESS;
         }
         runInferenceApp(&args, server);
-    } catch (const BadArgumentException& e) {
-        fprintf(stderr, "%s\n\n", e.what());
-        usage();
+    } catch (std::exception &e) {
+        printf("🚨 Critical error: %s\n", e.what());
         cleanupSockets();
         return EXIT_FAILURE;
     }