1 files changed, 22 insertions, 1 deletions
diff --git a/server/src/main/java/org/onap/usecaseui/server/bean/intent/IntentModel.java b/server/src/main/java/org/onap/usecaseui/server/bean/intent/IntentModel.java
index d8033bb7..1e21df25 100644
--- a/server/src/main/java/org/onap/usecaseui/server/bean/intent/IntentModel.java
+++ b/server/src/main/java/org/onap/usecaseui/server/bean/intent/IntentModel.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2017 CTC, Inc. and others. All rights reserved.
+ * Copyright (C) 2021 CTC, Inc. and others. All rights reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -47,6 +47,9 @@ public class IntentModel implements Serializable {
     @Column(name = "active")
     private Integer active;
 
+    @Column(name = "model_type")
+    private Integer modelType;
+
     public IntentModel() {
 
     }
@@ -98,4 +101,22 @@ public class IntentModel implements Serializable {
     public void setActive(Integer active) {
         this.active = active;
     }
+
+    public String getModelType() {
+        if (this.modelType == 1) {
+            return "ccvpn";
+        }
+        else {
+            return "5gs";
+        }
+    }
+
+    public void setModelType(String modelType) {
+        if ("ccvpn".equals(modelType)) {
+            this.modelType = 1;
+        }
+        else {
+            this.modelType = 0;
+        }
+    }
 }
 \ No newline at end of file