diff --git a/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java b/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java
index 280dca1..8a9c959 100644
--- a/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java
+++ b/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java
@@ -27,6 +27,9 @@ public class AiCloudDoc implements Serializable {
@Schema(description = "云目录ID")
private String categoryId;
+ @Schema(description = "项目ID")
+ private Integer projectId;
+
@Schema(description = "单位ID")
private Integer companyId;
diff --git a/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml b/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml
index be5a5a5..08d04a1 100644
--- a/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml
+++ b/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml
@@ -13,6 +13,9 @@
AND a.category_id = #{param.categoryId}
+
+ AND a.project_id = #{param.projectId}
+
AND a.company_id = #{param.companyId}
diff --git a/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java b/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java
index f09aade..c5802c2 100644
--- a/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java
+++ b/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java
@@ -27,6 +27,10 @@ public class AiCloudDocParam extends BaseParam {
@Schema(description = "云目录ID")
@QueryField(type = QueryType.EQ)
private String categoryId;
+
+ @Schema(description = "项目ID")
+ @QueryField(type = QueryType.EQ)
+ private Integer projectId = 0;
@Schema(description = "单位ID")
@QueryField(type = QueryType.EQ)