diff --git a/apps/dataset/models/data_set.py b/apps/dataset/models/data_set.py
index 4f46eda2a..dd2f1b7c9 100644
--- a/apps/dataset/models/data_set.py
+++ b/apps/dataset/models/data_set.py
@@ -82,6 +82,9 @@ class Type(models.TextChoices):
 
     web = 1, 'web站点类型'
 
+    lark = 2, '飞书类型'
+    yuque = 3, '语雀类型'
+
 
 class HitHandlingMethod(models.TextChoices):
     optimization = 'optimization', '模型优化'