diff --git a/erp/custom.go b/erp/custom.go
index f2a38c8..7f6a880 100644
--- a/erp/custom.go
+++ b/erp/custom.go
@@ -2,11 +2,12 @@ package erp
 
 import (
 	"context"
+	"time"
+
 	client2 "git.kumo.work/shama/service/client"
 	custom2 "git.kumo.work/shama/service/erp/custom"
 	"git.kumo.work/shama/service/lib/bean"
 	"github.com/shopspring/decimal"
-	"time"
 )
 
 type custom struct {
@@ -112,6 +113,7 @@ type CustomAdd struct {
 	MarkText         string          // 唛头文字
 	MarkImg          string          // 唛头图片
 	ImgFilePaths     []string        // 图片地址集合
+	DepartmentIds    []int64         // 部门id
 }
 
 // Add @TITLE 添加客户
@@ -154,6 +156,7 @@ type ReplyCustomInfo struct {
 	MarkText         string          `json:"markText"`         // 唛头文字
 	MarkImg          string          `json:"markImg"`          // 唛头图片
 	ImgFilePaths     []string        `json:"imgFilePaths"`     // 图片地址集合
+	DepartmentIds    []int64         `json:"departmentIds"`    // 部门ID集合
 	Status           int64           `json:"status"`
 	CreatedStaffId   int64           `json:"createdStaffId"`
 	CreatedAt        *time.Time      `json:"createdAt"`