build fix.

b4rtaz · Feb 9, 2025 · 67e48c5 · 67e48c5
1 parent 969848a
commit 67e48c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git 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
@@ -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;
     }