diff --git a/src/main/java/org/dromara/easyai/entity/TypeMapping.java b/src/main/java/org/dromara/easyai/entity/TypeMapping.java
new file mode 100644
index 0000000..2cdf425
--- /dev/null
+++ b/src/main/java/org/dromara/easyai/entity/TypeMapping.java
@@ -0,0 +1,22 @@
+package org.dromara.easyai.entity;
+
+public class TypeMapping {
+    private int type;
+    private int mapping;
+
+    public int getType() {
+        return type;
+    }
+
+    public void setType(int type) {
+        this.type = type;
+    }
+
+    public int getMapping() {
+        return mapping;
+    }
+
+    public void setMapping(int mapping) {
+        this.mapping = mapping;
+    }
+}