Feat: new api updateSetting/request

Signed-off-by: Daniel Hu <[email protected]>
devchat-ai · Nov 21, 2023 · 0cc3309 · 0cc3309
1 parent 7df12ae
commit 0cc3309
Show file tree

Hide file tree

Showing 5 changed files with 56 additions and 1 deletion.
diff --git a/api/setting.md b/api/setting.md
@@ -58,7 +58,7 @@
   },
   "payload": {
     "setting": {
-      "currentModel": "gpt-3.5"
+      "currentModel": "gpt-3.5-turbo"
     }
   }
 }

diff --git a/src/main/java/ai/devchat/devchat/ActionHandlerFactory.java b/src/main/java/ai/devchat/devchat/ActionHandlerFactory.java
@@ -24,6 +24,7 @@ public class ActionHandlerFactory {
             put(DevChatActions.GET_KEY_REQUEST, GetKeyRequestHandler.class);
             put(DevChatActions.COMMIT_CODE_REQUEST, CommitCodeRequestHandler.class);
             put(DevChatActions.GET_SETTING_REQUEST, GetSettingRequestHandler.class);
+            put(DevChatActions.UPDATE_SETTING_REQUEST, UpdateSettingRequestHandler.class);
         }
     };
 

diff --git a/src/main/java/ai/devchat/devchat/DevChatActions.java b/src/main/java/ai/devchat/devchat/DevChatActions.java
@@ -30,4 +30,6 @@ public class DevChatActions {
     public static final String COMMIT_CODE_RESPONSE = "commitCode/response";
     public static final String GET_SETTING_REQUEST = "getSetting/request";
     public static final String GET_SETTING_RESPONSE = "getSetting/response";
+    public static final String UPDATE_SETTING_REQUEST = "updateSetting/request";
+    public static final String UPDATE_SETTING_RESPONSE = "updateSetting/response";
 }
diff --git a/src/main/java/ai/devchat/devchat/handler/GetSettingRequestHandler.java b/src/main/java/ai/devchat/devchat/handler/GetSettingRequestHandler.java
@@ -1,5 +1,6 @@
 package ai.devchat.devchat.handler;
 
+import ai.devchat.common.Log;
 import ai.devchat.devchat.ActionHandler;
 import ai.devchat.devchat.DevChatActionHandler;
 import ai.devchat.devchat.DevChatActions;
@@ -18,6 +19,7 @@ public GetSettingRequestHandler(DevChatActionHandler devChatActionHandler) {
 
     @Override
     public void executeAction() {
+        Log.info("Handling getSetting request");
         String callbackFunc = metadata.getString("callback");
 
         DevChatSettingsState settings = DevChatSettingsState.getInstance();

diff --git a/src/main/java/ai/devchat/devchat/handler/UpdateSettingRequestHandler.java b/src/main/java/ai/devchat/devchat/handler/UpdateSettingRequestHandler.java
@@ -0,0 +1,50 @@
+package ai.devchat.devchat.handler;
+
+import ai.devchat.devchat.ActionHandler;
+import ai.devchat.devchat.DevChatActionHandler;
+import ai.devchat.devchat.DevChatActions;
+import ai.devchat.idea.setting.DevChatSettingsState;
+import com.alibaba.fastjson.JSONObject;
+
+public class UpdateSettingRequestHandler implements ActionHandler {
+    private JSONObject metadata;
+    private JSONObject payload;
+    private final DevChatActionHandler devChatActionHandler;
+
+    public UpdateSettingRequestHandler(DevChatActionHandler devChatActionHandler) {
+        this.devChatActionHandler = devChatActionHandler;
+    }
+
+    @Override
+    public void executeAction() {
+        String callbackFunc = metadata.getString("callback");
+        JSONObject setting = payload.getJSONObject("setting");
+
+        try {
+            DevChatSettingsState settings = DevChatSettingsState.getInstance();
+            if (setting.containsKey("currentModel")) {
+                settings.defaultModel = setting.getString("currentModel");
+            }
+
+            devChatActionHandler.sendResponse(DevChatActions.UPDATE_SETTING_RESPONSE, callbackFunc, (metadata, payload) -> {
+                metadata.put("status", "success");
+                metadata.put("error", "");
+            });
+        } catch (Exception e) {
+            devChatActionHandler.sendResponse(DevChatActions.UPDATE_SETTING_RESPONSE, callbackFunc, (metadata, payload) -> {
+                metadata.put("status", "Failed");
+                metadata.put("error", "Failed to update setting." + e.getMessage());
+            });
+        }
+    }
+
+    @Override
+    public void setMetadata(JSONObject metadata) {
+        this.metadata = metadata;
+    }
+
+    @Override
+    public void setPayload(JSONObject payload) {
+        this.payload = payload;
+    }
+}