fix wrong api endpoind - handler mapping

janhq · Nov 2, 2023 · af3bdb9 · af3bdb9
1 parent 3d81378
commit af3bdb9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/controllers/llamaCPP.cc b/controllers/llamaCPP.cc
@@ -348,12 +348,12 @@ void llamaCPP::loadModel(
 void inferences::llamaCPP::unloadModel(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
 {
   Json::Value jsonResp;
+  jsonResp["message"] = "No model loaded";
   if (model_loaded) {
     llama.unloadModel();
     model_loaded = false;
     jsonResp["message"] = "Model unloaded successfully";
   }
-  jsonResp["message"] = "No model loaded";
   auto resp = nitro_utils::nitroHttpJsonResponse(jsonResp);
   callback(resp);
 }
diff --git a/controllers/llamaCPP.h b/controllers/llamaCPP.h
@@ -1279,7 +1279,7 @@ class llamaCPP : public drogon::HttpController<llamaCPP> {
   METHOD_ADD(llamaCPP::chatCompletion, "chat_completion", Post);
   METHOD_ADD(llamaCPP::embedding, "embedding", Post);
   METHOD_ADD(llamaCPP::loadModel, "loadmodel", Post);
-  METHOD_ADD(llamaCPP::loadModel, "unloadmodel", Delete);
+  METHOD_ADD(llamaCPP::unloadModel, "unloadmodel", Delete);
   // PATH_ADD("/llama/chat_completion", Post);
   METHOD_LIST_END
   void chatCompletion(const HttpRequestPtr &req,