Feat: new api getSetting/request

Signed-off-by: Daniel Hu <[email protected]>

api/setting.md

@@ -0,0 +1,93 @@
+## Get Setting
+
+### JS to Java
+
+```json
+{
+  "action": "getSetting/request",
+  "metadata": {
+    "callback": "responseFunctionName"
+  },
+  "payload": null
+}
+```
+
+### Java to JS
+
+- success
+
+```json
+{
+  "action": "getSetting/response",
+  "metadata": {
+    "status": "success",
+    "error": ""
+  },
+  "payload": {
+    "setting": {
+      "apiKey": "DC.abcdef123456",
+      "apiBase": "https://api.example.com/v1",
+      "currentModel": "gpt-4"
+    }
+  }
+}
+```
+
+- failed
+
+```json
+{
+  "action": "getSetting/response",
+  "metadata": {
+    "status": "Failed",
+    "error": "Failed to get setting."
+  },
+  "payload": null
+}
+```
+
+## Set or Update Setting
+
+### JS to Java
+
+```json
+{
+  "action": "updateSetting/request",
+  "metadata": {
+    "callback": "responseFunctionName"
+  },
+  "payload": {
+    "setting": {
+      "currentModel": "gpt-3.5"
+    }
+  }
+}
+```
+
+### Java to JS
+
+- success
+
+```json
+{
+  "action": "updateSetting/response",
+  "metadata": {
+    "status": "success",
+    "error": ""
+  },
+  "payload": null
+}
+```
+
+- failed
+
+```json
+{
+  "action": "updateSetting/response",
+  "metadata": {
+    "status": "Failed",
+    "error": "Failed to update setting."
+  },
+  "payload": null
+}
+```

src/main/java/ai/devchat/cli/DevChatWrapper.java

@@ -17,16 +17,18 @@
 public class DevChatWrapper {
     private String apiBase;
     private String apiKey;
+    private String currentModel;
     private String command;
     private final String DEFAULT_LOG_MAX_COUNT = "100";
 
     public DevChatWrapper(String command) {
         this.command = command;
     }
 
-    public DevChatWrapper(String apiBase, String apiKey, String command) {
+    public DevChatWrapper(String apiBase, String apiKey, String currentModel, String command) {
         this.apiBase = apiBase;
         this.apiKey = apiKey;
+        this.currentModel = currentModel;
         this.command = command;
     }
 

src/main/java/ai/devchat/devchat/ActionHandlerFactory.java

@@ -23,6 +23,7 @@ public class ActionHandlerFactory {
             put(DevChatActions.ADD_CONTEXT_REQUEST, AddContextRequestHandler.class);
             put(DevChatActions.GET_KEY_REQUEST, GetKeyRequestHandler.class);
             put(DevChatActions.COMMIT_CODE_REQUEST, CommitCodeRequestHandler.class);
+            put(DevChatActions.GET_SETTING_REQUEST, GetSettingRequestHandler.class);
         }
     };
 

src/main/java/ai/devchat/devchat/DevChatActions.java

@@ -28,4 +28,6 @@ public class DevChatActions {
     public static final String GET_KEY_RESPONSE = "getKey/response";
     public static final String COMMIT_CODE_REQUEST = "commitCode/request";
     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";
 }

src/main/java/ai/devchat/devchat/handler/GetKeyRequestHandler.java

@@ -2,6 +2,7 @@
 
 import ai.devchat.devchat.ActionHandler;
 import ai.devchat.devchat.DevChatActionHandler;
+import ai.devchat.devchat.DevChatActions;
 import ai.devchat.idea.storage.SensitiveDataStorage;
 import com.alibaba.fastjson.JSONObject;
 
@@ -20,13 +21,13 @@ public void executeAction() {
         String key = SensitiveDataStorage.getKey();
 
         if (key != null && !key.isEmpty()) {
-            devChatActionHandler.sendResponse("getKey/response", callbackFunc, (metadata, payload) -> {
+            devChatActionHandler.sendResponse(DevChatActions.GET_KEY_RESPONSE, callbackFunc, (metadata, payload) -> {
                 metadata.put("status", "success");
                 metadata.put("error", "");
                 payload.put("key", key);
             });
         } else {
-            devChatActionHandler.sendResponse("getKey/response", callbackFunc, (metadata, payload) -> {
+            devChatActionHandler.sendResponse(DevChatActions.GET_KEY_RESPONSE, callbackFunc, (metadata, payload) -> {
                 metadata.put("status", "error");
                 metadata.put("error", "key is empty");
             });

src/main/java/ai/devchat/devchat/handler/GetSettingRequestHandler.java

@@ -0,0 +1,56 @@
+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 ai.devchat.idea.storage.SensitiveDataStorage;
+import com.alibaba.fastjson.JSONObject;
+
+public class GetSettingRequestHandler implements ActionHandler {
+    private JSONObject metadata;
+    private JSONObject payload;
+    private final DevChatActionHandler devChatActionHandler;
+
+    public GetSettingRequestHandler(DevChatActionHandler devChatActionHandler) {
+        this.devChatActionHandler = devChatActionHandler;
+    }
+
+    @Override
+    public void executeAction() {
+        String callbackFunc = metadata.getString("callback");
+
+        DevChatSettingsState settings = DevChatSettingsState.getInstance();
+        String apiKey = SensitiveDataStorage.getKey();
+        if (settings.apiBase == null || settings.apiBase.isEmpty()) {
+            if (apiKey.startsWith("sk-")) {
+                settings.apiBase = "https://api.openai.com/v1";
+            } else if (apiKey.startsWith("DC.")) {
+                settings.apiBase = "https://api.devchat.ai/v1";
+            }
+        }
+
+        String apiBase = settings.apiBase;
+        String currentModel = settings.defaultModel;
+
+        devChatActionHandler.sendResponse(DevChatActions.GET_SETTING_RESPONSE, callbackFunc, (metadata, payload) -> {
+            metadata.put("status", "success");
+            metadata.put("error", "");
+            JSONObject setting = new JSONObject();
+            setting.put("apiKey", apiKey);
+            setting.put("apiBase", apiBase);
+            setting.put("currentModel", currentModel);
+            payload.put("setting", setting);
+        });
+    }
+
+    @Override
+    public void setMetadata(JSONObject metadata) {
+        this.metadata = metadata;
+    }
+
+    @Override
+    public void setPayload(JSONObject payload) {
+        this.payload = payload;
+    }
+}

src/main/java/ai/devchat/devchat/handler/ListModelsRequestHandler.java

@@ -28,6 +28,8 @@ public void executeAction() {
         List<String> modelList = new ArrayList<>();
         modelList.add("gpt-3.5-turbo");
         modelList.add("gpt-4");
+        modelList.add("gpt-3.5-turbo-16k");
+        modelList.add("claude-2");
 
         devChatActionHandler.sendResponse(DevChatActions.LIST_MODELS_RESPONSE, callbackFunc, (metadata, payload) -> {
             metadata.put("status", "success");

src/main/java/ai/devchat/devchat/handler/SendMessageRequestHandler.java

@@ -92,10 +92,12 @@ public void executeAction() {
             DevChatSettingsState settings = DevChatSettingsState.getInstance();
             if (settings.apiBase != null && !settings.apiBase.isEmpty()) {
                 apiBase = settings.apiBase;
+            } else {
+                settings.apiBase = apiBase;
             }
 
             DevChatResponseConsumer responseConsumer = getResponseConsumer(callbackFunc);
-            DevChatWrapper devchatWrapper = new DevChatWrapper(apiBase, apiKey, devchatCommandPath);
+            DevChatWrapper devchatWrapper = new DevChatWrapper(apiBase, apiKey, settings.defaultModel, devchatCommandPath);
             devchatWrapper.runPromptCommand(flags, message, responseConsumer);
         } catch (Exception e) {
             Log.error("Exception occurred while executing DevChat command. Exception message: " + e.getMessage());

src/main/java/ai/devchat/idea/setting/DevChatSettingsComponent.java

@@ -15,11 +15,13 @@ public class DevChatSettingsComponent {
     private final JPanel jPanel;
     private final JBTextField apiBaseText = new JBTextField();
     private final JBTextField apiKeyText = new JBTextField();
+    private final JBTextField defaultModelText = new JBTextField();
 
     public DevChatSettingsComponent() {
         jPanel = FormBuilder.createFormBuilder()
                 .addLabeledComponent(new JBLabel("api_base"), apiBaseText, 1, false)
                 .addLabeledComponent(new JBLabel("api_key"), apiKeyText, 2, false)
+                .addLabeledComponent(new JBLabel("default_model"), defaultModelText, 3, false)
                 .addComponentFillVertically(new JPanel(), 0)
                 .getPanel();
     }
@@ -41,11 +43,19 @@ public String getApiKey() {
         return apiKeyText.getText();
     }
 
+    public String getDefaultModel() {
+        return defaultModelText.getText();
+    }
+
     public void setApiBase(@NotNull String apiBase) {
         apiBaseText.setText(apiBase);
     }
 
     public void setApiKey(@NotNull String apiKey) {
         apiKeyText.setText(apiKey);
     }
+
+    public void setDefaultModel(@NotNull String defaultModel) {
+        defaultModelText.setText(defaultModel);
+    }
 }

src/main/java/ai/devchat/idea/setting/DevChatSettingsConfigurable.java

@@ -12,54 +12,57 @@
  */
 public class DevChatSettingsConfigurable implements Configurable {
 
-  private DevChatSettingsComponent devChatSettingsComponent;
+    private DevChatSettingsComponent devChatSettingsComponent;
 
-  // A default constructor with no arguments is required because this implementation
-  // is registered in an applicationConfigurable EP
+    // A default constructor with no arguments is required because this implementation
+    // is registered in an applicationConfigurable EP
 
-  @Nls(capitalization = Nls.Capitalization.Title)
-  @Override
-  public String getDisplayName() {
-    return "DevChat";
-  }
+    @Nls(capitalization = Nls.Capitalization.Title)
+    @Override
+    public String getDisplayName() {
+        return "DevChat";
+    }
 
-  @Override
-  public JComponent getPreferredFocusedComponent() {
-    return devChatSettingsComponent.getPreferredFocusedComponent();
-  }
+    @Override
+    public JComponent getPreferredFocusedComponent() {
+        return devChatSettingsComponent.getPreferredFocusedComponent();
+    }
 
-  @Nullable
-  @Override
-  public JComponent createComponent() {
-    devChatSettingsComponent = new DevChatSettingsComponent();
-    return devChatSettingsComponent.getPanel();
-  }
+    @Nullable
+    @Override
+    public JComponent createComponent() {
+        devChatSettingsComponent = new DevChatSettingsComponent();
+        return devChatSettingsComponent.getPanel();
+    }
 
-  @Override
-  public boolean isModified() {
-    DevChatSettingsState settings = DevChatSettingsState.getInstance();
-    return !devChatSettingsComponent.getApiBase().equals(settings.apiBase) ||
-            !devChatSettingsComponent.getApiKey().equals(settings.apiKey);
-  }
+    @Override
+    public boolean isModified() {
+        DevChatSettingsState settings = DevChatSettingsState.getInstance();
+        return !devChatSettingsComponent.getApiBase().equals(settings.apiBase) ||
+                !devChatSettingsComponent.getApiKey().equals(settings.apiKey) ||
+                !devChatSettingsComponent.getDefaultModel().equals(settings.defaultModel);
+    }
 
-  @Override
-  public void apply() {
-    DevChatSettingsState settings = DevChatSettingsState.getInstance();
-    settings.apiBase = devChatSettingsComponent.getApiBase();
-    settings.apiKey = devChatSettingsComponent.getApiKey();
-    SensitiveDataStorage.setKey(settings.apiKey);
-  }
+    @Override
+    public void apply() {
+        DevChatSettingsState settings = DevChatSettingsState.getInstance();
+        settings.apiBase = devChatSettingsComponent.getApiBase();
+        settings.apiKey = devChatSettingsComponent.getApiKey();
+        settings.defaultModel = devChatSettingsComponent.getDefaultModel();
+        SensitiveDataStorage.setKey(settings.apiKey);
+    }
 
-  @Override
-  public void reset() {
-    DevChatSettingsState settings = DevChatSettingsState.getInstance();
-    devChatSettingsComponent.setApiBase(settings.apiBase);
-    devChatSettingsComponent.setApiKey(settings.apiKey);
-    SensitiveDataStorage.setKey(settings.apiKey);
-  }
+    @Override
+    public void reset() {
+        DevChatSettingsState settings = DevChatSettingsState.getInstance();
+        devChatSettingsComponent.setApiBase(settings.apiBase);
+        devChatSettingsComponent.setApiKey(settings.apiKey);
+        devChatSettingsComponent.setDefaultModel(settings.defaultModel);
+        SensitiveDataStorage.setKey(settings.apiKey);
+    }
 
-  @Override
-  public void disposeUIResources() {
-    devChatSettingsComponent = null;
-  }
+    @Override
+    public void disposeUIResources() {
+        devChatSettingsComponent = null;
+    }
 }

src/main/java/ai/devchat/idea/setting/DevChatSettingsState.java

@@ -23,6 +23,8 @@ public class DevChatSettingsState implements PersistentStateComponent<DevChatSet
 
   public String apiKey = "change_me";
 
+  public String defaultModel = "gpt-3.5-turbo";
+
   public static DevChatSettingsState getInstance() {
     return ApplicationManager.getApplication().getService(DevChatSettingsState.class);
   }