diff --git a/src/main/java/ai/devchat/cli/DevChatConfig.java b/src/main/java/ai/devchat/cli/DevChatConfig.java index fdfe44a..f174ddc 100644 --- a/src/main/java/ai/devchat/cli/DevChatConfig.java +++ b/src/main/java/ai/devchat/cli/DevChatConfig.java @@ -1,11 +1,11 @@ package ai.devchat.cli; -import java.io.File; -import java.util.Map; - import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.dataformat.yaml.YAMLFactory; +import java.io.File; +import java.util.Map; + /* * default_model: gpt-3.5-turbo * models: @@ -75,6 +75,30 @@ public void writeDefaultConfig() { setProvider("devchat.ai"); setStream(true); } + }, + + "gpt-3.5-turbo-16k", + new ModelConfig() { + { + setProvider("devchat.ai"); + setStream(true); + } + }, + + "gpt-4", + new ModelConfig() { + { + setProvider("devchat.ai"); + setStream(true); + } + }, + + "claude-2", + new ModelConfig() { + { + setProvider("general"); + setStream(true); + } }); this.save(); }