Hotfix NPE risks.

fairyto2 · Dec 1, 2023 · db1dbb7 · db1dbb7
1 parent 5fb2f5c
commit db1dbb7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/main/java/com/zhongan/devpilot/enums/ModelServiceEnum.java b/src/main/java/com/zhongan/devpilot/enums/ModelServiceEnum.java
@@ -22,14 +22,14 @@ public String getName() {
 
     public static ModelServiceEnum fromName(String name) {
         if (name == null) {
-            return null;
+            return OPENAI;
         }
         for (ModelServiceEnum type : ModelServiceEnum.values()) {
             if (type.getName().equals(name)) {
                 return type;
             }
         }
-        return null;
+        return OPENAI;
     }
 
     @Override

diff --git a/src/main/java/com/zhongan/devpilot/enums/ModelTypeEnum.java b/src/main/java/com/zhongan/devpilot/enums/ModelTypeEnum.java
@@ -31,14 +31,14 @@ public String getName() {
 
     public static ModelTypeEnum fromName(String name) {
         if (name == null) {
-            return null;
+            return GPT3_5;
         }
         for (ModelTypeEnum type : ModelTypeEnum.values()) {
             if (type.getName().equals(name)) {
                 return type;
             }
         }
-        return null;
+        return GPT3_5;
     }
 
     @Override

diff --git a/src/main/java/com/zhongan/devpilot/enums/SessionTypeEnum.java b/src/main/java/com/zhongan/devpilot/enums/SessionTypeEnum.java
@@ -26,13 +26,13 @@ public String getDesc() {
 
     public static SessionTypeEnum getEnumByCode(Integer code) {
         if (Objects.isNull(code)) {
-            return null;
+            return MULTI_TURN;
         }
         for (SessionTypeEnum type : SessionTypeEnum.values()) {
             if (type.getCode().equals(code)) {
                 return type;
             }
         }
-        return null;
+        return MULTI_TURN;
     }
 }