Merge branch 'master' of git.ascrm.cn:scrm/wechat-api

desc/all.api

@@ -19,4 +19,6 @@ import "./wechat/agent.api"
 import "./wechat/employee.api"
 import "./wechat/work_experience.api"
 import "./wechat/tutorial.api"
-import "./wechat/token.api"
+import "./wechat/token.api"
+import "./wechat/employee_config.api"
+import "./wechat/category.api"

desc/wechat/agent.api

@@ -19,6 +19,10 @@ type (
 
         // examples | 对话案例 
         Examples  *string `json:"examples,optional"`
+
+		DatasetId *string `json:"dataset_id,optional"`
+
+		CollectionId *string `json:"collection_id,optional"`
     }
 
     // The response data of agent list | Agent列表数据
@@ -61,6 +65,217 @@ type (
         // Agent information | Agent数据
         Data AgentInfo `json:"data"`
     }
+
+	VectorModel {
+		Model *string `json:"model"`
+		Name *string `json:"name"`
+		CharsPointsPrice *uint64 `json:"charsPointsPrice"`
+		DefaultToken *uint64 `json:"defaultToken"`
+		MaxToken *uint64 `json:"maxToken"`
+		Weight *uint64 `json:"weight"`
+	}
+
+	AgentModel {
+		Model *string `json:"model"`
+		Name *string `json:"name"`
+		MaxContext *uint64 `json:"maxContext"`
+		MaxResponse *uint64 `json:"maxResponse"`
+		CharsPointsPrice *uint64 `json:"charsPointsPrice"`
+	}
+
+	DatasetId {
+		ID *string  `json:"id"`
+		ParentID *string `json:"parentId"`
+		TeamId *string `json:"teamId"`
+		TmbId *string `json:"tmbId"`
+		Type *string `json:"type"`
+		Status *string `json:"status"`
+		Avatar *string `json:"avatar"`
+		Name *string `json:"name"`
+		VectorModel *string `json:"vectorModel"`
+		AgentModel *string `json:"agentModel"`
+		Intro *string `json:"intro"`
+	}
+
+	// Dataset info | 知识库详情
+	DatasetInfo {
+		ID *string  `json:"id"`
+		ParentID *string `json:"parentId"`
+		TeamId *string `json:"teamId"`
+		TmbId *string `json:"tmbId"`
+		Type *string `json:"type"`
+		Name *string `json:"name"`
+		Intro *string `json:"intro"`
+		Status *string `json:"status"`
+		Avatar *string `json:"avatar"`
+		VectorModel VectorModel `json:"vectorModel"`
+		AgentModel AgentModel `json:"agentModel"`
+		Permission *string `json:"permission"`
+		CanWrite *bool `json:"canWrite"`
+		IsOwner *bool `json:"isOwner"`
+	}
+
+	Index {
+		DefaultIndex *bool `json:"defaultIndex"`
+		Type *string `json:"type"`
+		DataId *string  `json:"dataId"`
+		Text *string `json:"text"`
+		ID *string `json:"id"`
+	}
+
+	// Collection Info | 集合详情
+	CollectionInfo {
+		ID *string  `json:"id"`
+		ParentID *string `json:"parentId"`
+		TmbId *string `json:"tmbId,optional"`
+		Type *string `json:"type"`
+		Name *string `json:"name"`
+		DataAmount *uint64 `json:"dataAmount,optional"`
+		TrainingAmount *uint64 `json:"trainingAmount,optional"`
+		TrainingType *string `json:"trainingType,optional"`
+		ChunkSize *uint64 `json:"chunkSize,optional"`
+		ChunkSplitter *string `json:"chunkSplitter,optional"`
+		QaPrompt *string `json:"qaPrompt,optional"`
+		RawTextLength *uint64 `json:"rawTextLength,optional"`
+		CanWrite *bool `json:"canWrite,optional"`
+		SourceName *string `json:"sourceName,optional"`
+		DatasetId DatasetId `json:"datasetId,optional"`
+	}
+
+	CollectionDetailReq {
+		ID *string `json:"id"`
+	}
+
+	CollectionInfoResp {
+		BaseDataInfo
+
+		Data CollectionInfo `json:"data"`
+	}
+
+	DataDetailReq {
+		ID *string `json:"id"`
+	}
+
+	DataDetailResp {
+		BaseDataInfo
+
+		Data DataInfo `json:"data"`
+	}
+
+	CollectionSimpleInfo {
+		ID *string  `json:"id"`
+		ParentID *string `json:"parentId"`
+		TmbId *string `json:"tmbId,optional"`
+		Type *string `json:"type"`
+		Name *string `json:"name"`
+		DataAmount *uint64 `json:"dataAmount,optional"`
+		TrainingAmount *uint64 `json:"trainingAmount,optional"`
+	}
+
+	DataInfo {
+		ID *string  `json:"id"`
+		Q *string `json:"q"`
+		A *string `json:"a"`
+		ChunkIndex *uint64 `json:"chunkIndex"`
+		Indexes []Index `json:"indexes"`
+		DatasetId *string `json:"datasetId"`
+		CollectionId *string `json:"collectionId"`
+		SourceName *string `json:"sourceName"`
+		CanWrite *bool `json:"canWrite"`
+		IsOwner *bool `json:"isOwner"`
+	}
+
+	DataSimpleInfo {
+		ID *string  `json:"id"`
+		Q *string `json:"q"`
+		A *string `json:"a"`
+		ChunkIndex *uint64 `json:"chunkIndex"`
+		DatasetId *string `json:"datasetId"`
+		CollectionId *string `json:"collectionId"`
+	}
+
+	// Get collection list request params | Collection列表请求参数
+	CollectionListReq {
+		PageNum   *int    `json:"pageNum" validate:"required,number,gt=0"`
+
+		PageSize  *int    `json:"pageSize" validate:"required,number,lt=100000"`
+
+		DatasetId  *string `json:"datasetId" validate:"required"`
+	}
+
+	// Collection list response | Collection List信息返回体
+	CollectionListResp {
+		BaseDataInfo
+
+		// Agent information | Agent数据
+		Data []CollectionSimpleInfo `json:"data"`
+
+		PageNum   *int    `json:"pageNum" validate:"required,number,gt=0"`
+
+		PageSize  *int    `json:"pageSize" validate:"required,number,lt=100000"`
+
+		Total     *int    `json:"total"`
+	}
+
+	// Get collection list request params | Collection列表请求参数
+	DataListReq {
+		PageNum   *int    `json:"pageNum" validate:"required,number,gt=0"`
+
+		PageSize  *int    `json:"pageSize" validate:"required,number,lt=100000"`
+
+		CollectionId  *string `json:"collectionId" validate:"required"`
+	}
+
+	// Data list response | Data List信息返回体
+	DataListResp {
+		BaseDataInfo
+
+		// Agent information | Agent数据
+		Data []DataSimpleInfo `json:"data"`
+
+		PageNum   *int    `json:"pageNum" validate:"required,number,gt=0"`
+
+		PageSize  *int    `json:"pageSize" validate:"required,number,lt=100000"`
+
+		Total     *int    `json:"total"`
+	}
+
+	// Data create request | 信息请求体
+	CreateDataInfoReq {
+		CollectionId *string `json:"collectionId" validate:"required"`
+
+		// Q
+		 Q *string `json:"q" validate:"required"`
+
+		// A
+		 A *string `json:"a" validate:"required"`
+
+		// Indexes | 索引
+		//Indexes []IndexSingle `json:"indexes,optional"`
+	}
+
+//	IndexSingle {
+//		Text A *string `json:"text"`
+//	}
+
+	// Data create request | 信息返回体
+	UpdateDataInfoReq {
+		// ID
+		DataId *string `json:"dataId" validate:"required"`
+
+		// Q
+		Q *string `json:"q" validate:"required"`
+
+		// A
+		A *string `json:"a" validate:"required"`
+
+		// Indexes | 索引
+		//Indexes []IndexSingle `json:"indexes,optional"`
+	}
+
+	DeleteDataReq {
+		ID *string `json:"id" validate:"required"`
+	}
 )
 
 @server(
@@ -86,7 +301,35 @@ service Wechat {
     @handler getAgentList
     post /agent/list (AgentListReq) returns (AgentListResp)
 
-    // Get agent by ID | 通过ID获取Agent
-    @handler getAgentById
-    post /agent (IDReq) returns (AgentInfoResp)
+	// Get agent by ID | 通过ID获取Agent
+	@handler getAgentById
+	post /agent (IDReq) returns (AgentInfoResp)
+
+	// Get collect list | 获取collection列表
+    @handler getAgentCollectionList
+    post /agent/collection/list (CollectionListReq) returns (CollectionListResp)
+
+	// Get collect detail | 获取collection详情
+	@handler getAgentCollectionInfo
+	post /agent/collection/detail (CollectionDetailReq) returns (CollectionInfoResp)
+
+	// Get data list | 获取data列表
+	@handler getAgentDataList
+	post /agent/data/list (DataListReq) returns (DataListResp)
+
+	// Get data detail | 获取data详情
+	@handler getAgentDataDetail
+	post /agent/data/detail (DataDetailReq) returns (DataDetailResp)
+
+	// Create data | 添加data
+	@handler createAgentData
+	post /agent/data/create (CreateDataInfoReq) returns (BaseDataInfo)
+
+	// Update data | 修改data
+	@handler updateAgentData
+	post /agent/data/update (UpdateDataInfoReq) returns (BaseDataInfo)
+
+	// Delete data | 删除data
+	@handler deleteAgentData
+	post /agent/data/delete (DeleteDataReq) returns (BaseDataInfo)
 }

desc/wechat/category.api

@@ -0,0 +1,74 @@
+import "../base.api"
+
+type (
+    // The data of category information | Category信息
+    CategoryInfo {
+        BaseIDInfo
+
+        // name | 角色名称 
+        Name  *string `json:"name,optional"`
+
+        // organization_id | 租户ID 
+        OrganizationId  *uint64 `json:"organizationId,optional"`
+    }
+
+    // The response data of category list | Category列表数据
+    CategoryListResp {
+        BaseDataInfo
+
+        // Category list data | Category列表数据
+        Data CategoryListInfo `json:"data"`
+    }
+
+    // Category list data | Category列表数据
+    CategoryListInfo {
+        BaseListInfo
+
+        // The API list data | Category列表数据
+        Data  []CategoryInfo  `json:"data"`
+    }
+
+    // Get category list request params | Category列表请求参数
+    CategoryListReq {
+        PageInfo
+
+        // name | 角色名称 
+        Name  *string `json:"name,optional"`
+    }
+
+    // Category information response | Category信息返回体
+    CategoryInfoResp {
+        BaseDataInfo
+
+        // Category information | Category数据
+        Data CategoryInfo `json:"data"`
+    }
+)
+
+@server(
+    jwt: Auth
+    group: category
+    middleware: Authority
+)
+
+service Wechat {
+    // Create category information | 创建Category
+    @handler createCategory
+    post /category/create (CategoryInfo) returns (BaseMsgResp)
+
+    // Update category information | 更新Category
+    @handler updateCategory
+    post /category/update (CategoryInfo) returns (BaseMsgResp)
+
+    // Delete category information | 删除Category信息
+    @handler deleteCategory
+    post /category/delete (IDsReq) returns (BaseMsgResp)
+
+    // Get category list | 获取Category列表
+    @handler getCategoryList
+    post /category/list (CategoryListReq) returns (CategoryListResp)
+
+    // Get category by ID | 通过ID获取Category
+    @handler getCategoryById
+    post /category (IDReq) returns (CategoryInfoResp)
+}

desc/wechat/employee.api

@@ -1,6 +1,7 @@
 import "../base.api"
 import "./work_experience.api"
 import "./tutorial.api"
+import "./employee_config.api"
 
 type (
     // The data of employee information | Employee信息
@@ -25,6 +26,9 @@ type (
         // achievement_count | 业绩单数 
         AchievementCount  *int `json:"achievementCount,optional"`
 
+		// category_id | 分类ID
+		CategoryId  *uint64 `json:"categoryId,optional"`
+
         // intro | 个人介绍 
         Intro  *string `json:"intro,optional"`
 
@@ -39,9 +43,11 @@ type (
 
         // scene | 使用场景 
         Scene  *string `json:"scene,optional"`
+		SceneList []EmployeeConfigInfo `json:"sceneList,optional"`
 
         // switch_in | 支持介入 
         SwitchIn  *string `json:"switchIn,optional"`
+		SwitchInList []EmployeeConfigInfo `json:"switchInList,optional"`
 
         Tutorial  []TutorialInfo `json:"tutorial,optional"`
 
@@ -77,6 +83,9 @@ type (
 
         // tags | 个人标签 
         Tags  *string `json:"tags,optional"`
+
+		// category_id | 分类ID
+		CategoryId  *uint64 `json:"categoryId,optional"`
     }
 
     // Employee information response | Employee信息返回体

desc/wechat/employee_config.api

@@ -0,0 +1,57 @@
+import "../base.api"
+
+type (
+    // The data of employee config information | EmployeeConfig信息
+    EmployeeConfigInfo {
+        BaseIDInfo
+
+        // 类型：scene-场景 switch_in-接入方式 
+        Stype  *string `json:"stype,optional"`
+
+        // 标题
+        Title  *string `json:"title,optional"`
+
+        // 图片地址 
+        Photo  *string `json:"photo,optional"`
+    }
+
+    // EmployeeConfig list data | EmployeeConfig列表数据
+    EmployeeConfigListInfo {
+        BaseListInfo
+
+        // The API list data | EmployeeConfig列表数据
+        Data  []EmployeeConfigInfo  `json:"data"`
+    }
+
+	// Employee config response | Employee配置信息返回体
+	EmployeeConfigListReq {
+		/// 类型：scene-场景 switch_in-接入方式
+		Stype  *string `json:"stype,optional"`
+	}
+
+    // EmployeeConfig information response | EmployeeConfig信息返回体
+	EmployeeConfig {
+		Scene []EmployeeConfigInfo `json:"scene"`
+		SwitchIn []EmployeeConfigInfo `json:"switchIn"`
+	}
+
+	// Employee config response | Employee配置信息返回体
+	EmployeeConfigListResp {
+		BaseDataInfo
+
+		// Employee information | Employee数据
+		Data EmployeeConfig `json:"data"`
+	}
+)
+
+@server(
+    jwt: Auth
+    group: employee_config
+    middleware: Authority
+)
+
+service Wechat {
+	// Get employee config list | 获取EmployeeConfig列表
+	@handler getEmployeeListConfig
+	get /employee_config/list (EmployeeConfigListReq) returns (EmployeeConfigListResp)
+}

desc/wechat/wx.api

@@ -50,6 +50,12 @@ type (
 
 		// 模式信息
 		AgentInfo *AgentInfo `json:"agentInfo,optional"`
+
+		// 大模型服务地址
+        ApiBase *string `json:"apiBase,optional"`
+
+        // 大模型服务密钥
+        ApiKey *string `json:"apiKey,optional"`
     }
 
     // The response data of wx list | Wx列表数据
@@ -78,6 +84,9 @@ type (
         // 租户id
         OrganizationId  *uint64 `json:"organizationId,optional"`
 
+        // 租户名称
+        OrganizationName  *string `json:"organizationName,optional"`
+
         // 端口号 
         Port  *string `json:"port,optional"`
 
@@ -108,13 +117,17 @@ service Wechat {
     @handler createWx
     post /wx/create (WxInfo) returns (BaseMsgResp)
 
+    // Check wx information | 检查并更新Wx状态
+    @handler checkWx
+    post /wx/check (WxInfo) returns (BaseMsgResp)
+
     // Update wx information | 更新Wx
     @handler updateWx
     post /wx/update (WxInfo) returns (BaseMsgResp)
 
     // Delete wx information | 删除Wx信息
     @handler deleteWx
-    post /wx/delete (IDsReq) returns (BaseMsgResp)
+    post /wx/delete (IDReq) returns (BaseMsgResp)
 
     // Get wx list | 获取Wx列表
     @handler getWxList

desc/wechat/wxhook.api

@@ -100,6 +100,10 @@ service Wechat {
     @handler logout
     post /wxhook/logout (IDReq) returns (BaseMsgResp)
 
+    // 结束微信
+    @handler terminateThisWeChat
+    post /wxhook/terminateThisWeChat (IDReq) returns (BaseMsgResp)
+
     // 获取好友和群信息
     @handler getFriendsAndGroups
     post /wxhook/getFriendsAndGroups (IDReq) returns (BaseMsgResp)

ent/agent.go

@@ -35,6 +35,10 @@ type Agent struct {
 	Examples string `json:"examples,omitempty"`
 	// organization_id | 租户ID
 	OrganizationID uint64 `json:"organization_id,omitempty"`
+	// dataset_id | 知识库ID
+	DatasetID string `json:"dataset_id,omitempty"`
+	// collection_id | 集合ID
+	CollectionID string `json:"collection_id,omitempty"`
 	// Edges holds the relations/edges for other nodes in the graph.
 	// The values are being populated by the AgentQuery when eager-loading is set.
 	Edges        AgentEdges `json:"edges"`
@@ -66,7 +70,7 @@ func (*Agent) scanValues(columns []string) ([]any, error) {
 		switch columns[i] {
 		case agent.FieldID, agent.FieldStatus, agent.FieldOrganizationID:
 			values[i] = new(sql.NullInt64)
-		case agent.FieldName, agent.FieldRole, agent.FieldBackground, agent.FieldExamples:
+		case agent.FieldName, agent.FieldRole, agent.FieldBackground, agent.FieldExamples, agent.FieldDatasetID, agent.FieldCollectionID:
 			values[i] = new(sql.NullString)
 		case agent.FieldCreatedAt, agent.FieldUpdatedAt, agent.FieldDeletedAt:
 			values[i] = new(sql.NullTime)
@@ -145,6 +149,18 @@ func (a *Agent) assignValues(columns []string, values []any) error {
 			} else if value.Valid {
 				a.OrganizationID = uint64(value.Int64)
 			}
+		case agent.FieldDatasetID:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field dataset_id", values[i])
+			} else if value.Valid {
+				a.DatasetID = value.String
+			}
+		case agent.FieldCollectionID:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field collection_id", values[i])
+			} else if value.Valid {
+				a.CollectionID = value.String
+			}
 		default:
 			a.selectValues.Set(columns[i], values[i])
 		}
@@ -212,6 +228,12 @@ func (a *Agent) String() string {
 	builder.WriteString(", ")
 	builder.WriteString("organization_id=")
 	builder.WriteString(fmt.Sprintf("%v", a.OrganizationID))
+	builder.WriteString(", ")
+	builder.WriteString("dataset_id=")
+	builder.WriteString(a.DatasetID)
+	builder.WriteString(", ")
+	builder.WriteString("collection_id=")
+	builder.WriteString(a.CollectionID)
 	builder.WriteByte(')')
 	return builder.String()
 }

ent/agent/agent.go

@@ -33,6 +33,10 @@ const (
 	FieldExamples = "examples"
 	// FieldOrganizationID holds the string denoting the organization_id field in the database.
 	FieldOrganizationID = "organization_id"
+	// FieldDatasetID holds the string denoting the dataset_id field in the database.
+	FieldDatasetID = "dataset_id"
+	// FieldCollectionID holds the string denoting the collection_id field in the database.
+	FieldCollectionID = "collection_id"
 	// EdgeWxAgent holds the string denoting the wx_agent edge name in mutations.
 	EdgeWxAgent = "wx_agent"
 	// Table holds the table name of the agent in the database.
@@ -58,6 +62,8 @@ var Columns = []string{
 	FieldBackground,
 	FieldExamples,
 	FieldOrganizationID,
+	FieldDatasetID,
+	FieldCollectionID,
 }
 
 // ValidColumn reports if the column name is valid (part of the table columns).
@@ -102,6 +108,14 @@ var (
 	ExamplesValidator func(string) error
 	// OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
 	OrganizationIDValidator func(uint64) error
+	// DefaultDatasetID holds the default value on creation for the "dataset_id" field.
+	DefaultDatasetID string
+	// DatasetIDValidator is a validator for the "dataset_id" field. It is called by the builders before save.
+	DatasetIDValidator func(string) error
+	// DefaultCollectionID holds the default value on creation for the "collection_id" field.
+	DefaultCollectionID string
+	// CollectionIDValidator is a validator for the "collection_id" field. It is called by the builders before save.
+	CollectionIDValidator func(string) error
 )
 
 // OrderOption defines the ordering options for the Agent queries.
@@ -157,6 +171,16 @@ func ByOrganizationID(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldOrganizationID, opts...).ToFunc()
 }
 
+// ByDatasetID orders the results by the dataset_id field.
+func ByDatasetID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldDatasetID, opts...).ToFunc()
+}
+
+// ByCollectionID orders the results by the collection_id field.
+func ByCollectionID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldCollectionID, opts...).ToFunc()
+}
+
 // ByWxAgentCount orders the results by wx_agent count.
 func ByWxAgentCount(opts ...sql.OrderTermOption) OrderOption {
 	return func(s *sql.Selector) {

ent/agent/where.go

@@ -100,6 +100,16 @@ func OrganizationID(v uint64) predicate.Agent {
 	return predicate.Agent(sql.FieldEQ(FieldOrganizationID, v))
 }
 
+// DatasetID applies equality check predicate on the "dataset_id" field. It's identical to DatasetIDEQ.
+func DatasetID(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldEQ(FieldDatasetID, v))
+}
+
+// CollectionID applies equality check predicate on the "collection_id" field. It's identical to CollectionIDEQ.
+func CollectionID(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldEQ(FieldCollectionID, v))
+}
+
 // CreatedAtEQ applies the EQ predicate on the "created_at" field.
 func CreatedAtEQ(v time.Time) predicate.Agent {
 	return predicate.Agent(sql.FieldEQ(FieldCreatedAt, v))
@@ -600,6 +610,136 @@ func OrganizationIDLTE(v uint64) predicate.Agent {
 	return predicate.Agent(sql.FieldLTE(FieldOrganizationID, v))
 }
 
+// DatasetIDEQ applies the EQ predicate on the "dataset_id" field.
+func DatasetIDEQ(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldEQ(FieldDatasetID, v))
+}
+
+// DatasetIDNEQ applies the NEQ predicate on the "dataset_id" field.
+func DatasetIDNEQ(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldNEQ(FieldDatasetID, v))
+}
+
+// DatasetIDIn applies the In predicate on the "dataset_id" field.
+func DatasetIDIn(vs ...string) predicate.Agent {
+	return predicate.Agent(sql.FieldIn(FieldDatasetID, vs...))
+}
+
+// DatasetIDNotIn applies the NotIn predicate on the "dataset_id" field.
+func DatasetIDNotIn(vs ...string) predicate.Agent {
+	return predicate.Agent(sql.FieldNotIn(FieldDatasetID, vs...))
+}
+
+// DatasetIDGT applies the GT predicate on the "dataset_id" field.
+func DatasetIDGT(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldGT(FieldDatasetID, v))
+}
+
+// DatasetIDGTE applies the GTE predicate on the "dataset_id" field.
+func DatasetIDGTE(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldGTE(FieldDatasetID, v))
+}
+
+// DatasetIDLT applies the LT predicate on the "dataset_id" field.
+func DatasetIDLT(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldLT(FieldDatasetID, v))
+}
+
+// DatasetIDLTE applies the LTE predicate on the "dataset_id" field.
+func DatasetIDLTE(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldLTE(FieldDatasetID, v))
+}
+
+// DatasetIDContains applies the Contains predicate on the "dataset_id" field.
+func DatasetIDContains(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldContains(FieldDatasetID, v))
+}
+
+// DatasetIDHasPrefix applies the HasPrefix predicate on the "dataset_id" field.
+func DatasetIDHasPrefix(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldHasPrefix(FieldDatasetID, v))
+}
+
+// DatasetIDHasSuffix applies the HasSuffix predicate on the "dataset_id" field.
+func DatasetIDHasSuffix(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldHasSuffix(FieldDatasetID, v))
+}
+
+// DatasetIDEqualFold applies the EqualFold predicate on the "dataset_id" field.
+func DatasetIDEqualFold(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldEqualFold(FieldDatasetID, v))
+}
+
+// DatasetIDContainsFold applies the ContainsFold predicate on the "dataset_id" field.
+func DatasetIDContainsFold(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldContainsFold(FieldDatasetID, v))
+}
+
+// CollectionIDEQ applies the EQ predicate on the "collection_id" field.
+func CollectionIDEQ(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldEQ(FieldCollectionID, v))
+}
+
+// CollectionIDNEQ applies the NEQ predicate on the "collection_id" field.
+func CollectionIDNEQ(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldNEQ(FieldCollectionID, v))
+}
+
+// CollectionIDIn applies the In predicate on the "collection_id" field.
+func CollectionIDIn(vs ...string) predicate.Agent {
+	return predicate.Agent(sql.FieldIn(FieldCollectionID, vs...))
+}
+
+// CollectionIDNotIn applies the NotIn predicate on the "collection_id" field.
+func CollectionIDNotIn(vs ...string) predicate.Agent {
+	return predicate.Agent(sql.FieldNotIn(FieldCollectionID, vs...))
+}
+
+// CollectionIDGT applies the GT predicate on the "collection_id" field.
+func CollectionIDGT(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldGT(FieldCollectionID, v))
+}
+
+// CollectionIDGTE applies the GTE predicate on the "collection_id" field.
+func CollectionIDGTE(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldGTE(FieldCollectionID, v))
+}
+
+// CollectionIDLT applies the LT predicate on the "collection_id" field.
+func CollectionIDLT(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldLT(FieldCollectionID, v))
+}
+
+// CollectionIDLTE applies the LTE predicate on the "collection_id" field.
+func CollectionIDLTE(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldLTE(FieldCollectionID, v))
+}
+
+// CollectionIDContains applies the Contains predicate on the "collection_id" field.
+func CollectionIDContains(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldContains(FieldCollectionID, v))
+}
+
+// CollectionIDHasPrefix applies the HasPrefix predicate on the "collection_id" field.
+func CollectionIDHasPrefix(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldHasPrefix(FieldCollectionID, v))
+}
+
+// CollectionIDHasSuffix applies the HasSuffix predicate on the "collection_id" field.
+func CollectionIDHasSuffix(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldHasSuffix(FieldCollectionID, v))
+}
+
+// CollectionIDEqualFold applies the EqualFold predicate on the "collection_id" field.
+func CollectionIDEqualFold(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldEqualFold(FieldCollectionID, v))
+}
+
+// CollectionIDContainsFold applies the ContainsFold predicate on the "collection_id" field.
+func CollectionIDContainsFold(v string) predicate.Agent {
+	return predicate.Agent(sql.FieldContainsFold(FieldCollectionID, v))
+}
+
 // HasWxAgent applies the HasEdge predicate on the "wx_agent" edge.
 func HasWxAgent() predicate.Agent {
 	return predicate.Agent(func(s *sql.Selector) {

ent/agent_create.go

@@ -125,6 +125,34 @@ func (ac *AgentCreate) SetOrganizationID(u uint64) *AgentCreate {
 	return ac
 }
 
+// SetDatasetID sets the "dataset_id" field.
+func (ac *AgentCreate) SetDatasetID(s string) *AgentCreate {
+	ac.mutation.SetDatasetID(s)
+	return ac
+}
+
+// SetNillableDatasetID sets the "dataset_id" field if the given value is not nil.
+func (ac *AgentCreate) SetNillableDatasetID(s *string) *AgentCreate {
+	if s != nil {
+		ac.SetDatasetID(*s)
+	}
+	return ac
+}
+
+// SetCollectionID sets the "collection_id" field.
+func (ac *AgentCreate) SetCollectionID(s string) *AgentCreate {
+	ac.mutation.SetCollectionID(s)
+	return ac
+}
+
+// SetNillableCollectionID sets the "collection_id" field if the given value is not nil.
+func (ac *AgentCreate) SetNillableCollectionID(s *string) *AgentCreate {
+	if s != nil {
+		ac.SetCollectionID(*s)
+	}
+	return ac
+}
+
 // SetID sets the "id" field.
 func (ac *AgentCreate) SetID(u uint64) *AgentCreate {
 	ac.mutation.SetID(u)
@@ -209,6 +237,14 @@ func (ac *AgentCreate) defaults() error {
 		v := agent.DefaultExamples
 		ac.mutation.SetExamples(v)
 	}
+	if _, ok := ac.mutation.DatasetID(); !ok {
+		v := agent.DefaultDatasetID
+		ac.mutation.SetDatasetID(v)
+	}
+	if _, ok := ac.mutation.CollectionID(); !ok {
+		v := agent.DefaultCollectionID
+		ac.mutation.SetCollectionID(v)
+	}
 	return nil
 }
 
@@ -259,6 +295,22 @@ func (ac *AgentCreate) check() error {
 			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Agent.organization_id": %w`, err)}
 		}
 	}
+	if _, ok := ac.mutation.DatasetID(); !ok {
+		return &ValidationError{Name: "dataset_id", err: errors.New(`ent: missing required field "Agent.dataset_id"`)}
+	}
+	if v, ok := ac.mutation.DatasetID(); ok {
+		if err := agent.DatasetIDValidator(v); err != nil {
+			return &ValidationError{Name: "dataset_id", err: fmt.Errorf(`ent: validator failed for field "Agent.dataset_id": %w`, err)}
+		}
+	}
+	if _, ok := ac.mutation.CollectionID(); !ok {
+		return &ValidationError{Name: "collection_id", err: errors.New(`ent: missing required field "Agent.collection_id"`)}
+	}
+	if v, ok := ac.mutation.CollectionID(); ok {
+		if err := agent.CollectionIDValidator(v); err != nil {
+			return &ValidationError{Name: "collection_id", err: fmt.Errorf(`ent: validator failed for field "Agent.collection_id": %w`, err)}
+		}
+	}
 	return nil
 }
 
@@ -328,6 +380,14 @@ func (ac *AgentCreate) createSpec() (*Agent, *sqlgraph.CreateSpec) {
 		_spec.SetField(agent.FieldOrganizationID, field.TypeUint64, value)
 		_node.OrganizationID = value
 	}
+	if value, ok := ac.mutation.DatasetID(); ok {
+		_spec.SetField(agent.FieldDatasetID, field.TypeString, value)
+		_node.DatasetID = value
+	}
+	if value, ok := ac.mutation.CollectionID(); ok {
+		_spec.SetField(agent.FieldCollectionID, field.TypeString, value)
+		_node.CollectionID = value
+	}
 	if nodes := ac.mutation.WxAgentIDs(); len(nodes) > 0 {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,
@@ -528,6 +588,30 @@ func (u *AgentUpsert) AddOrganizationID(v uint64) *AgentUpsert {
 	return u
 }
 
+// SetDatasetID sets the "dataset_id" field.
+func (u *AgentUpsert) SetDatasetID(v string) *AgentUpsert {
+	u.Set(agent.FieldDatasetID, v)
+	return u
+}
+
+// UpdateDatasetID sets the "dataset_id" field to the value that was provided on create.
+func (u *AgentUpsert) UpdateDatasetID() *AgentUpsert {
+	u.SetExcluded(agent.FieldDatasetID)
+	return u
+}
+
+// SetCollectionID sets the "collection_id" field.
+func (u *AgentUpsert) SetCollectionID(v string) *AgentUpsert {
+	u.Set(agent.FieldCollectionID, v)
+	return u
+}
+
+// UpdateCollectionID sets the "collection_id" field to the value that was provided on create.
+func (u *AgentUpsert) UpdateCollectionID() *AgentUpsert {
+	u.SetExcluded(agent.FieldCollectionID)
+	return u
+}
+
 // UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
 // Using this option is equivalent to using:
 //
@@ -733,6 +817,34 @@ func (u *AgentUpsertOne) UpdateOrganizationID() *AgentUpsertOne {
 	})
 }
 
+// SetDatasetID sets the "dataset_id" field.
+func (u *AgentUpsertOne) SetDatasetID(v string) *AgentUpsertOne {
+	return u.Update(func(s *AgentUpsert) {
+		s.SetDatasetID(v)
+	})
+}
+
+// UpdateDatasetID sets the "dataset_id" field to the value that was provided on create.
+func (u *AgentUpsertOne) UpdateDatasetID() *AgentUpsertOne {
+	return u.Update(func(s *AgentUpsert) {
+		s.UpdateDatasetID()
+	})
+}
+
+// SetCollectionID sets the "collection_id" field.
+func (u *AgentUpsertOne) SetCollectionID(v string) *AgentUpsertOne {
+	return u.Update(func(s *AgentUpsert) {
+		s.SetCollectionID(v)
+	})
+}
+
+// UpdateCollectionID sets the "collection_id" field to the value that was provided on create.
+func (u *AgentUpsertOne) UpdateCollectionID() *AgentUpsertOne {
+	return u.Update(func(s *AgentUpsert) {
+		s.UpdateCollectionID()
+	})
+}
+
 // Exec executes the query.
 func (u *AgentUpsertOne) Exec(ctx context.Context) error {
 	if len(u.create.conflict) == 0 {
@@ -1104,6 +1216,34 @@ func (u *AgentUpsertBulk) UpdateOrganizationID() *AgentUpsertBulk {
 	})
 }
 
+// SetDatasetID sets the "dataset_id" field.
+func (u *AgentUpsertBulk) SetDatasetID(v string) *AgentUpsertBulk {
+	return u.Update(func(s *AgentUpsert) {
+		s.SetDatasetID(v)
+	})
+}
+
+// UpdateDatasetID sets the "dataset_id" field to the value that was provided on create.
+func (u *AgentUpsertBulk) UpdateDatasetID() *AgentUpsertBulk {
+	return u.Update(func(s *AgentUpsert) {
+		s.UpdateDatasetID()
+	})
+}
+
+// SetCollectionID sets the "collection_id" field.
+func (u *AgentUpsertBulk) SetCollectionID(v string) *AgentUpsertBulk {
+	return u.Update(func(s *AgentUpsert) {
+		s.SetCollectionID(v)
+	})
+}
+
+// UpdateCollectionID sets the "collection_id" field to the value that was provided on create.
+func (u *AgentUpsertBulk) UpdateCollectionID() *AgentUpsertBulk {
+	return u.Update(func(s *AgentUpsert) {
+		s.UpdateCollectionID()
+	})
+}
+
 // Exec executes the query.
 func (u *AgentUpsertBulk) Exec(ctx context.Context) error {
 	if u.create.err != nil {

ent/agent_update.go

@@ -171,6 +171,34 @@ func (au *AgentUpdate) AddOrganizationID(u int64) *AgentUpdate {
 	return au
 }
 
+// SetDatasetID sets the "dataset_id" field.
+func (au *AgentUpdate) SetDatasetID(s string) *AgentUpdate {
+	au.mutation.SetDatasetID(s)
+	return au
+}
+
+// SetNillableDatasetID sets the "dataset_id" field if the given value is not nil.
+func (au *AgentUpdate) SetNillableDatasetID(s *string) *AgentUpdate {
+	if s != nil {
+		au.SetDatasetID(*s)
+	}
+	return au
+}
+
+// SetCollectionID sets the "collection_id" field.
+func (au *AgentUpdate) SetCollectionID(s string) *AgentUpdate {
+	au.mutation.SetCollectionID(s)
+	return au
+}
+
+// SetNillableCollectionID sets the "collection_id" field if the given value is not nil.
+func (au *AgentUpdate) SetNillableCollectionID(s *string) *AgentUpdate {
+	if s != nil {
+		au.SetCollectionID(*s)
+	}
+	return au
+}
+
 // AddWxAgentIDs adds the "wx_agent" edge to the Wx entity by IDs.
 func (au *AgentUpdate) AddWxAgentIDs(ids ...uint64) *AgentUpdate {
 	au.mutation.AddWxAgentIDs(ids...)
@@ -286,6 +314,16 @@ func (au *AgentUpdate) check() error {
 			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Agent.organization_id": %w`, err)}
 		}
 	}
+	if v, ok := au.mutation.DatasetID(); ok {
+		if err := agent.DatasetIDValidator(v); err != nil {
+			return &ValidationError{Name: "dataset_id", err: fmt.Errorf(`ent: validator failed for field "Agent.dataset_id": %w`, err)}
+		}
+	}
+	if v, ok := au.mutation.CollectionID(); ok {
+		if err := agent.CollectionIDValidator(v); err != nil {
+			return &ValidationError{Name: "collection_id", err: fmt.Errorf(`ent: validator failed for field "Agent.collection_id": %w`, err)}
+		}
+	}
 	return nil
 }
 
@@ -343,6 +381,12 @@ func (au *AgentUpdate) sqlSave(ctx context.Context) (n int, err error) {
 	if value, ok := au.mutation.AddedOrganizationID(); ok {
 		_spec.AddField(agent.FieldOrganizationID, field.TypeUint64, value)
 	}
+	if value, ok := au.mutation.DatasetID(); ok {
+		_spec.SetField(agent.FieldDatasetID, field.TypeString, value)
+	}
+	if value, ok := au.mutation.CollectionID(); ok {
+		_spec.SetField(agent.FieldCollectionID, field.TypeString, value)
+	}
 	if au.mutation.WxAgentCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,
@@ -550,6 +594,34 @@ func (auo *AgentUpdateOne) AddOrganizationID(u int64) *AgentUpdateOne {
 	return auo
 }
 
+// SetDatasetID sets the "dataset_id" field.
+func (auo *AgentUpdateOne) SetDatasetID(s string) *AgentUpdateOne {
+	auo.mutation.SetDatasetID(s)
+	return auo
+}
+
+// SetNillableDatasetID sets the "dataset_id" field if the given value is not nil.
+func (auo *AgentUpdateOne) SetNillableDatasetID(s *string) *AgentUpdateOne {
+	if s != nil {
+		auo.SetDatasetID(*s)
+	}
+	return auo
+}
+
+// SetCollectionID sets the "collection_id" field.
+func (auo *AgentUpdateOne) SetCollectionID(s string) *AgentUpdateOne {
+	auo.mutation.SetCollectionID(s)
+	return auo
+}
+
+// SetNillableCollectionID sets the "collection_id" field if the given value is not nil.
+func (auo *AgentUpdateOne) SetNillableCollectionID(s *string) *AgentUpdateOne {
+	if s != nil {
+		auo.SetCollectionID(*s)
+	}
+	return auo
+}
+
 // AddWxAgentIDs adds the "wx_agent" edge to the Wx entity by IDs.
 func (auo *AgentUpdateOne) AddWxAgentIDs(ids ...uint64) *AgentUpdateOne {
 	auo.mutation.AddWxAgentIDs(ids...)
@@ -678,6 +750,16 @@ func (auo *AgentUpdateOne) check() error {
 			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Agent.organization_id": %w`, err)}
 		}
 	}
+	if v, ok := auo.mutation.DatasetID(); ok {
+		if err := agent.DatasetIDValidator(v); err != nil {
+			return &ValidationError{Name: "dataset_id", err: fmt.Errorf(`ent: validator failed for field "Agent.dataset_id": %w`, err)}
+		}
+	}
+	if v, ok := auo.mutation.CollectionID(); ok {
+		if err := agent.CollectionIDValidator(v); err != nil {
+			return &ValidationError{Name: "collection_id", err: fmt.Errorf(`ent: validator failed for field "Agent.collection_id": %w`, err)}
+		}
+	}
 	return nil
 }
 
@@ -752,6 +834,12 @@ func (auo *AgentUpdateOne) sqlSave(ctx context.Context) (_node *Agent, err error
 	if value, ok := auo.mutation.AddedOrganizationID(); ok {
 		_spec.AddField(agent.FieldOrganizationID, field.TypeUint64, value)
 	}
+	if value, ok := auo.mutation.DatasetID(); ok {
+		_spec.SetField(agent.FieldDatasetID, field.TypeString, value)
+	}
+	if value, ok := auo.mutation.CollectionID(); ok {
+		_spec.SetField(agent.FieldCollectionID, field.TypeString, value)
+	}
 	if auo.mutation.WxAgentCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,

ent/category.go

@@ -0,0 +1,150 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"fmt"
+	"strings"
+	"time"
+	"wechat-api/ent/category"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/sql"
+)
+
+// Category is the model entity for the Category schema.
+type Category struct {
+	config `json:"-"`
+	// ID of the ent.
+	ID uint64 `json:"id,omitempty"`
+	// Create Time | 创建日期
+	CreatedAt time.Time `json:"created_at,omitempty"`
+	// Update Time | 修改日期
+	UpdatedAt time.Time `json:"updated_at,omitempty"`
+	// Delete Time | 删除日期
+	DeletedAt time.Time `json:"deleted_at,omitempty"`
+	// name | 角色名称
+	Name string `json:"name,omitempty"`
+	// organization_id | 租户ID
+	OrganizationID uint64 `json:"organization_id,omitempty"`
+	selectValues   sql.SelectValues
+}
+
+// scanValues returns the types for scanning values from sql.Rows.
+func (*Category) scanValues(columns []string) ([]any, error) {
+	values := make([]any, len(columns))
+	for i := range columns {
+		switch columns[i] {
+		case category.FieldID, category.FieldOrganizationID:
+			values[i] = new(sql.NullInt64)
+		case category.FieldName:
+			values[i] = new(sql.NullString)
+		case category.FieldCreatedAt, category.FieldUpdatedAt, category.FieldDeletedAt:
+			values[i] = new(sql.NullTime)
+		default:
+			values[i] = new(sql.UnknownType)
+		}
+	}
+	return values, nil
+}
+
+// assignValues assigns the values that were returned from sql.Rows (after scanning)
+// to the Category fields.
+func (c *Category) assignValues(columns []string, values []any) error {
+	if m, n := len(values), len(columns); m < n {
+		return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
+	}
+	for i := range columns {
+		switch columns[i] {
+		case category.FieldID:
+			value, ok := values[i].(*sql.NullInt64)
+			if !ok {
+				return fmt.Errorf("unexpected type %T for field id", value)
+			}
+			c.ID = uint64(value.Int64)
+		case category.FieldCreatedAt:
+			if value, ok := values[i].(*sql.NullTime); !ok {
+				return fmt.Errorf("unexpected type %T for field created_at", values[i])
+			} else if value.Valid {
+				c.CreatedAt = value.Time
+			}
+		case category.FieldUpdatedAt:
+			if value, ok := values[i].(*sql.NullTime); !ok {
+				return fmt.Errorf("unexpected type %T for field updated_at", values[i])
+			} else if value.Valid {
+				c.UpdatedAt = value.Time
+			}
+		case category.FieldDeletedAt:
+			if value, ok := values[i].(*sql.NullTime); !ok {
+				return fmt.Errorf("unexpected type %T for field deleted_at", values[i])
+			} else if value.Valid {
+				c.DeletedAt = value.Time
+			}
+		case category.FieldName:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field name", values[i])
+			} else if value.Valid {
+				c.Name = value.String
+			}
+		case category.FieldOrganizationID:
+			if value, ok := values[i].(*sql.NullInt64); !ok {
+				return fmt.Errorf("unexpected type %T for field organization_id", values[i])
+			} else if value.Valid {
+				c.OrganizationID = uint64(value.Int64)
+			}
+		default:
+			c.selectValues.Set(columns[i], values[i])
+		}
+	}
+	return nil
+}
+
+// Value returns the ent.Value that was dynamically selected and assigned to the Category.
+// This includes values selected through modifiers, order, etc.
+func (c *Category) Value(name string) (ent.Value, error) {
+	return c.selectValues.Get(name)
+}
+
+// Update returns a builder for updating this Category.
+// Note that you need to call Category.Unwrap() before calling this method if this Category
+// was returned from a transaction, and the transaction was committed or rolled back.
+func (c *Category) Update() *CategoryUpdateOne {
+	return NewCategoryClient(c.config).UpdateOne(c)
+}
+
+// Unwrap unwraps the Category entity that was returned from a transaction after it was closed,
+// so that all future queries will be executed through the driver which created the transaction.
+func (c *Category) Unwrap() *Category {
+	_tx, ok := c.config.driver.(*txDriver)
+	if !ok {
+		panic("ent: Category is not a transactional entity")
+	}
+	c.config.driver = _tx.drv
+	return c
+}
+
+// String implements the fmt.Stringer.
+func (c *Category) String() string {
+	var builder strings.Builder
+	builder.WriteString("Category(")
+	builder.WriteString(fmt.Sprintf("id=%v, ", c.ID))
+	builder.WriteString("created_at=")
+	builder.WriteString(c.CreatedAt.Format(time.ANSIC))
+	builder.WriteString(", ")
+	builder.WriteString("updated_at=")
+	builder.WriteString(c.UpdatedAt.Format(time.ANSIC))
+	builder.WriteString(", ")
+	builder.WriteString("deleted_at=")
+	builder.WriteString(c.DeletedAt.Format(time.ANSIC))
+	builder.WriteString(", ")
+	builder.WriteString("name=")
+	builder.WriteString(c.Name)
+	builder.WriteString(", ")
+	builder.WriteString("organization_id=")
+	builder.WriteString(fmt.Sprintf("%v", c.OrganizationID))
+	builder.WriteByte(')')
+	return builder.String()
+}
+
+// Categories is a parsable slice of Category.
+type Categories []*Category

ent/category/category.go

@@ -0,0 +1,102 @@
+// Code generated by ent, DO NOT EDIT.
+
+package category
+
+import (
+	"time"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/sql"
+)
+
+const (
+	// Label holds the string label denoting the category type in the database.
+	Label = "category"
+	// FieldID holds the string denoting the id field in the database.
+	FieldID = "id"
+	// FieldCreatedAt holds the string denoting the created_at field in the database.
+	FieldCreatedAt = "created_at"
+	// FieldUpdatedAt holds the string denoting the updated_at field in the database.
+	FieldUpdatedAt = "updated_at"
+	// FieldDeletedAt holds the string denoting the deleted_at field in the database.
+	FieldDeletedAt = "deleted_at"
+	// FieldName holds the string denoting the name field in the database.
+	FieldName = "name"
+	// FieldOrganizationID holds the string denoting the organization_id field in the database.
+	FieldOrganizationID = "organization_id"
+	// Table holds the table name of the category in the database.
+	Table = "category"
+)
+
+// Columns holds all SQL columns for category fields.
+var Columns = []string{
+	FieldID,
+	FieldCreatedAt,
+	FieldUpdatedAt,
+	FieldDeletedAt,
+	FieldName,
+	FieldOrganizationID,
+}
+
+// ValidColumn reports if the column name is valid (part of the table columns).
+func ValidColumn(column string) bool {
+	for i := range Columns {
+		if column == Columns[i] {
+			return true
+		}
+	}
+	return false
+}
+
+// Note that the variables below are initialized by the runtime
+// package on the initialization of the application. Therefore,
+// it should be imported in the main as follows:
+//
+//	import _ "wechat-api/ent/runtime"
+var (
+	Hooks        [1]ent.Hook
+	Interceptors [1]ent.Interceptor
+	// DefaultCreatedAt holds the default value on creation for the "created_at" field.
+	DefaultCreatedAt func() time.Time
+	// DefaultUpdatedAt holds the default value on creation for the "updated_at" field.
+	DefaultUpdatedAt func() time.Time
+	// UpdateDefaultUpdatedAt holds the default value on update for the "updated_at" field.
+	UpdateDefaultUpdatedAt func() time.Time
+	// NameValidator is a validator for the "name" field. It is called by the builders before save.
+	NameValidator func(string) error
+	// OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
+	OrganizationIDValidator func(uint64) error
+)
+
+// OrderOption defines the ordering options for the Category queries.
+type OrderOption func(*sql.Selector)
+
+// ByID orders the results by the id field.
+func ByID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldID, opts...).ToFunc()
+}
+
+// ByCreatedAt orders the results by the created_at field.
+func ByCreatedAt(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldCreatedAt, opts...).ToFunc()
+}
+
+// ByUpdatedAt orders the results by the updated_at field.
+func ByUpdatedAt(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldUpdatedAt, opts...).ToFunc()
+}
+
+// ByDeletedAt orders the results by the deleted_at field.
+func ByDeletedAt(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldDeletedAt, opts...).ToFunc()
+}
+
+// ByName orders the results by the name field.
+func ByName(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldName, opts...).ToFunc()
+}
+
+// ByOrganizationID orders the results by the organization_id field.
+func ByOrganizationID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldOrganizationID, opts...).ToFunc()
+}

ent/category/where.go

@@ -0,0 +1,330 @@
+// Code generated by ent, DO NOT EDIT.
+
+package category
+
+import (
+	"time"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+)
+
+// ID filters vertices based on their ID field.
+func ID(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldID, id))
+}
+
+// IDEQ applies the EQ predicate on the ID field.
+func IDEQ(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldID, id))
+}
+
+// IDNEQ applies the NEQ predicate on the ID field.
+func IDNEQ(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldNEQ(FieldID, id))
+}
+
+// IDIn applies the In predicate on the ID field.
+func IDIn(ids ...uint64) predicate.Category {
+	return predicate.Category(sql.FieldIn(FieldID, ids...))
+}
+
+// IDNotIn applies the NotIn predicate on the ID field.
+func IDNotIn(ids ...uint64) predicate.Category {
+	return predicate.Category(sql.FieldNotIn(FieldID, ids...))
+}
+
+// IDGT applies the GT predicate on the ID field.
+func IDGT(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldGT(FieldID, id))
+}
+
+// IDGTE applies the GTE predicate on the ID field.
+func IDGTE(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldGTE(FieldID, id))
+}
+
+// IDLT applies the LT predicate on the ID field.
+func IDLT(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldLT(FieldID, id))
+}
+
+// IDLTE applies the LTE predicate on the ID field.
+func IDLTE(id uint64) predicate.Category {
+	return predicate.Category(sql.FieldLTE(FieldID, id))
+}
+
+// CreatedAt applies equality check predicate on the "created_at" field. It's identical to CreatedAtEQ.
+func CreatedAt(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldCreatedAt, v))
+}
+
+// UpdatedAt applies equality check predicate on the "updated_at" field. It's identical to UpdatedAtEQ.
+func UpdatedAt(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldUpdatedAt, v))
+}
+
+// DeletedAt applies equality check predicate on the "deleted_at" field. It's identical to DeletedAtEQ.
+func DeletedAt(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldDeletedAt, v))
+}
+
+// Name applies equality check predicate on the "name" field. It's identical to NameEQ.
+func Name(v string) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldName, v))
+}
+
+// OrganizationID applies equality check predicate on the "organization_id" field. It's identical to OrganizationIDEQ.
+func OrganizationID(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldOrganizationID, v))
+}
+
+// CreatedAtEQ applies the EQ predicate on the "created_at" field.
+func CreatedAtEQ(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldCreatedAt, v))
+}
+
+// CreatedAtNEQ applies the NEQ predicate on the "created_at" field.
+func CreatedAtNEQ(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldNEQ(FieldCreatedAt, v))
+}
+
+// CreatedAtIn applies the In predicate on the "created_at" field.
+func CreatedAtIn(vs ...time.Time) predicate.Category {
+	return predicate.Category(sql.FieldIn(FieldCreatedAt, vs...))
+}
+
+// CreatedAtNotIn applies the NotIn predicate on the "created_at" field.
+func CreatedAtNotIn(vs ...time.Time) predicate.Category {
+	return predicate.Category(sql.FieldNotIn(FieldCreatedAt, vs...))
+}
+
+// CreatedAtGT applies the GT predicate on the "created_at" field.
+func CreatedAtGT(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldGT(FieldCreatedAt, v))
+}
+
+// CreatedAtGTE applies the GTE predicate on the "created_at" field.
+func CreatedAtGTE(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldGTE(FieldCreatedAt, v))
+}
+
+// CreatedAtLT applies the LT predicate on the "created_at" field.
+func CreatedAtLT(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldLT(FieldCreatedAt, v))
+}
+
+// CreatedAtLTE applies the LTE predicate on the "created_at" field.
+func CreatedAtLTE(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldLTE(FieldCreatedAt, v))
+}
+
+// UpdatedAtEQ applies the EQ predicate on the "updated_at" field.
+func UpdatedAtEQ(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldUpdatedAt, v))
+}
+
+// UpdatedAtNEQ applies the NEQ predicate on the "updated_at" field.
+func UpdatedAtNEQ(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldNEQ(FieldUpdatedAt, v))
+}
+
+// UpdatedAtIn applies the In predicate on the "updated_at" field.
+func UpdatedAtIn(vs ...time.Time) predicate.Category {
+	return predicate.Category(sql.FieldIn(FieldUpdatedAt, vs...))
+}
+
+// UpdatedAtNotIn applies the NotIn predicate on the "updated_at" field.
+func UpdatedAtNotIn(vs ...time.Time) predicate.Category {
+	return predicate.Category(sql.FieldNotIn(FieldUpdatedAt, vs...))
+}
+
+// UpdatedAtGT applies the GT predicate on the "updated_at" field.
+func UpdatedAtGT(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldGT(FieldUpdatedAt, v))
+}
+
+// UpdatedAtGTE applies the GTE predicate on the "updated_at" field.
+func UpdatedAtGTE(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldGTE(FieldUpdatedAt, v))
+}
+
+// UpdatedAtLT applies the LT predicate on the "updated_at" field.
+func UpdatedAtLT(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldLT(FieldUpdatedAt, v))
+}
+
+// UpdatedAtLTE applies the LTE predicate on the "updated_at" field.
+func UpdatedAtLTE(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldLTE(FieldUpdatedAt, v))
+}
+
+// DeletedAtEQ applies the EQ predicate on the "deleted_at" field.
+func DeletedAtEQ(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldDeletedAt, v))
+}
+
+// DeletedAtNEQ applies the NEQ predicate on the "deleted_at" field.
+func DeletedAtNEQ(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldNEQ(FieldDeletedAt, v))
+}
+
+// DeletedAtIn applies the In predicate on the "deleted_at" field.
+func DeletedAtIn(vs ...time.Time) predicate.Category {
+	return predicate.Category(sql.FieldIn(FieldDeletedAt, vs...))
+}
+
+// DeletedAtNotIn applies the NotIn predicate on the "deleted_at" field.
+func DeletedAtNotIn(vs ...time.Time) predicate.Category {
+	return predicate.Category(sql.FieldNotIn(FieldDeletedAt, vs...))
+}
+
+// DeletedAtGT applies the GT predicate on the "deleted_at" field.
+func DeletedAtGT(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldGT(FieldDeletedAt, v))
+}
+
+// DeletedAtGTE applies the GTE predicate on the "deleted_at" field.
+func DeletedAtGTE(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldGTE(FieldDeletedAt, v))
+}
+
+// DeletedAtLT applies the LT predicate on the "deleted_at" field.
+func DeletedAtLT(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldLT(FieldDeletedAt, v))
+}
+
+// DeletedAtLTE applies the LTE predicate on the "deleted_at" field.
+func DeletedAtLTE(v time.Time) predicate.Category {
+	return predicate.Category(sql.FieldLTE(FieldDeletedAt, v))
+}
+
+// DeletedAtIsNil applies the IsNil predicate on the "deleted_at" field.
+func DeletedAtIsNil() predicate.Category {
+	return predicate.Category(sql.FieldIsNull(FieldDeletedAt))
+}
+
+// DeletedAtNotNil applies the NotNil predicate on the "deleted_at" field.
+func DeletedAtNotNil() predicate.Category {
+	return predicate.Category(sql.FieldNotNull(FieldDeletedAt))
+}
+
+// NameEQ applies the EQ predicate on the "name" field.
+func NameEQ(v string) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldName, v))
+}
+
+// NameNEQ applies the NEQ predicate on the "name" field.
+func NameNEQ(v string) predicate.Category {
+	return predicate.Category(sql.FieldNEQ(FieldName, v))
+}
+
+// NameIn applies the In predicate on the "name" field.
+func NameIn(vs ...string) predicate.Category {
+	return predicate.Category(sql.FieldIn(FieldName, vs...))
+}
+
+// NameNotIn applies the NotIn predicate on the "name" field.
+func NameNotIn(vs ...string) predicate.Category {
+	return predicate.Category(sql.FieldNotIn(FieldName, vs...))
+}
+
+// NameGT applies the GT predicate on the "name" field.
+func NameGT(v string) predicate.Category {
+	return predicate.Category(sql.FieldGT(FieldName, v))
+}
+
+// NameGTE applies the GTE predicate on the "name" field.
+func NameGTE(v string) predicate.Category {
+	return predicate.Category(sql.FieldGTE(FieldName, v))
+}
+
+// NameLT applies the LT predicate on the "name" field.
+func NameLT(v string) predicate.Category {
+	return predicate.Category(sql.FieldLT(FieldName, v))
+}
+
+// NameLTE applies the LTE predicate on the "name" field.
+func NameLTE(v string) predicate.Category {
+	return predicate.Category(sql.FieldLTE(FieldName, v))
+}
+
+// NameContains applies the Contains predicate on the "name" field.
+func NameContains(v string) predicate.Category {
+	return predicate.Category(sql.FieldContains(FieldName, v))
+}
+
+// NameHasPrefix applies the HasPrefix predicate on the "name" field.
+func NameHasPrefix(v string) predicate.Category {
+	return predicate.Category(sql.FieldHasPrefix(FieldName, v))
+}
+
+// NameHasSuffix applies the HasSuffix predicate on the "name" field.
+func NameHasSuffix(v string) predicate.Category {
+	return predicate.Category(sql.FieldHasSuffix(FieldName, v))
+}
+
+// NameEqualFold applies the EqualFold predicate on the "name" field.
+func NameEqualFold(v string) predicate.Category {
+	return predicate.Category(sql.FieldEqualFold(FieldName, v))
+}
+
+// NameContainsFold applies the ContainsFold predicate on the "name" field.
+func NameContainsFold(v string) predicate.Category {
+	return predicate.Category(sql.FieldContainsFold(FieldName, v))
+}
+
+// OrganizationIDEQ applies the EQ predicate on the "organization_id" field.
+func OrganizationIDEQ(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldEQ(FieldOrganizationID, v))
+}
+
+// OrganizationIDNEQ applies the NEQ predicate on the "organization_id" field.
+func OrganizationIDNEQ(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldNEQ(FieldOrganizationID, v))
+}
+
+// OrganizationIDIn applies the In predicate on the "organization_id" field.
+func OrganizationIDIn(vs ...uint64) predicate.Category {
+	return predicate.Category(sql.FieldIn(FieldOrganizationID, vs...))
+}
+
+// OrganizationIDNotIn applies the NotIn predicate on the "organization_id" field.
+func OrganizationIDNotIn(vs ...uint64) predicate.Category {
+	return predicate.Category(sql.FieldNotIn(FieldOrganizationID, vs...))
+}
+
+// OrganizationIDGT applies the GT predicate on the "organization_id" field.
+func OrganizationIDGT(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldGT(FieldOrganizationID, v))
+}
+
+// OrganizationIDGTE applies the GTE predicate on the "organization_id" field.
+func OrganizationIDGTE(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldGTE(FieldOrganizationID, v))
+}
+
+// OrganizationIDLT applies the LT predicate on the "organization_id" field.
+func OrganizationIDLT(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldLT(FieldOrganizationID, v))
+}
+
+// OrganizationIDLTE applies the LTE predicate on the "organization_id" field.
+func OrganizationIDLTE(v uint64) predicate.Category {
+	return predicate.Category(sql.FieldLTE(FieldOrganizationID, v))
+}
+
+// And groups predicates with the AND operator between them.
+func And(predicates ...predicate.Category) predicate.Category {
+	return predicate.Category(sql.AndPredicates(predicates...))
+}
+
+// Or groups predicates with the OR operator between them.
+func Or(predicates ...predicate.Category) predicate.Category {
+	return predicate.Category(sql.OrPredicates(predicates...))
+}
+
+// Not applies the not operator on the given predicate.
+func Not(p predicate.Category) predicate.Category {
+	return predicate.Category(sql.NotPredicates(p))
+}

ent/category_create.go

@@ -0,0 +1,757 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+	"wechat-api/ent/category"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// CategoryCreate is the builder for creating a Category entity.
+type CategoryCreate struct {
+	config
+	mutation *CategoryMutation
+	hooks    []Hook
+	conflict []sql.ConflictOption
+}
+
+// SetCreatedAt sets the "created_at" field.
+func (cc *CategoryCreate) SetCreatedAt(t time.Time) *CategoryCreate {
+	cc.mutation.SetCreatedAt(t)
+	return cc
+}
+
+// SetNillableCreatedAt sets the "created_at" field if the given value is not nil.
+func (cc *CategoryCreate) SetNillableCreatedAt(t *time.Time) *CategoryCreate {
+	if t != nil {
+		cc.SetCreatedAt(*t)
+	}
+	return cc
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (cc *CategoryCreate) SetUpdatedAt(t time.Time) *CategoryCreate {
+	cc.mutation.SetUpdatedAt(t)
+	return cc
+}
+
+// SetNillableUpdatedAt sets the "updated_at" field if the given value is not nil.
+func (cc *CategoryCreate) SetNillableUpdatedAt(t *time.Time) *CategoryCreate {
+	if t != nil {
+		cc.SetUpdatedAt(*t)
+	}
+	return cc
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (cc *CategoryCreate) SetDeletedAt(t time.Time) *CategoryCreate {
+	cc.mutation.SetDeletedAt(t)
+	return cc
+}
+
+// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
+func (cc *CategoryCreate) SetNillableDeletedAt(t *time.Time) *CategoryCreate {
+	if t != nil {
+		cc.SetDeletedAt(*t)
+	}
+	return cc
+}
+
+// SetName sets the "name" field.
+func (cc *CategoryCreate) SetName(s string) *CategoryCreate {
+	cc.mutation.SetName(s)
+	return cc
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (cc *CategoryCreate) SetOrganizationID(u uint64) *CategoryCreate {
+	cc.mutation.SetOrganizationID(u)
+	return cc
+}
+
+// SetID sets the "id" field.
+func (cc *CategoryCreate) SetID(u uint64) *CategoryCreate {
+	cc.mutation.SetID(u)
+	return cc
+}
+
+// Mutation returns the CategoryMutation object of the builder.
+func (cc *CategoryCreate) Mutation() *CategoryMutation {
+	return cc.mutation
+}
+
+// Save creates the Category in the database.
+func (cc *CategoryCreate) Save(ctx context.Context) (*Category, error) {
+	if err := cc.defaults(); err != nil {
+		return nil, err
+	}
+	return withHooks(ctx, cc.sqlSave, cc.mutation, cc.hooks)
+}
+
+// SaveX calls Save and panics if Save returns an error.
+func (cc *CategoryCreate) SaveX(ctx context.Context) *Category {
+	v, err := cc.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return v
+}
+
+// Exec executes the query.
+func (cc *CategoryCreate) Exec(ctx context.Context) error {
+	_, err := cc.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (cc *CategoryCreate) ExecX(ctx context.Context) {
+	if err := cc.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// defaults sets the default values of the builder before save.
+func (cc *CategoryCreate) defaults() error {
+	if _, ok := cc.mutation.CreatedAt(); !ok {
+		if category.DefaultCreatedAt == nil {
+			return fmt.Errorf("ent: uninitialized category.DefaultCreatedAt (forgotten import ent/runtime?)")
+		}
+		v := category.DefaultCreatedAt()
+		cc.mutation.SetCreatedAt(v)
+	}
+	if _, ok := cc.mutation.UpdatedAt(); !ok {
+		if category.DefaultUpdatedAt == nil {
+			return fmt.Errorf("ent: uninitialized category.DefaultUpdatedAt (forgotten import ent/runtime?)")
+		}
+		v := category.DefaultUpdatedAt()
+		cc.mutation.SetUpdatedAt(v)
+	}
+	return nil
+}
+
+// check runs all checks and user-defined validators on the builder.
+func (cc *CategoryCreate) check() error {
+	if _, ok := cc.mutation.CreatedAt(); !ok {
+		return &ValidationError{Name: "created_at", err: errors.New(`ent: missing required field "Category.created_at"`)}
+	}
+	if _, ok := cc.mutation.UpdatedAt(); !ok {
+		return &ValidationError{Name: "updated_at", err: errors.New(`ent: missing required field "Category.updated_at"`)}
+	}
+	if _, ok := cc.mutation.Name(); !ok {
+		return &ValidationError{Name: "name", err: errors.New(`ent: missing required field "Category.name"`)}
+	}
+	if v, ok := cc.mutation.Name(); ok {
+		if err := category.NameValidator(v); err != nil {
+			return &ValidationError{Name: "name", err: fmt.Errorf(`ent: validator failed for field "Category.name": %w`, err)}
+		}
+	}
+	if _, ok := cc.mutation.OrganizationID(); !ok {
+		return &ValidationError{Name: "organization_id", err: errors.New(`ent: missing required field "Category.organization_id"`)}
+	}
+	if v, ok := cc.mutation.OrganizationID(); ok {
+		if err := category.OrganizationIDValidator(v); err != nil {
+			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Category.organization_id": %w`, err)}
+		}
+	}
+	return nil
+}
+
+func (cc *CategoryCreate) sqlSave(ctx context.Context) (*Category, error) {
+	if err := cc.check(); err != nil {
+		return nil, err
+	}
+	_node, _spec := cc.createSpec()
+	if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil {
+		if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return nil, err
+	}
+	if _spec.ID.Value != _node.ID {
+		id := _spec.ID.Value.(int64)
+		_node.ID = uint64(id)
+	}
+	cc.mutation.id = &_node.ID
+	cc.mutation.done = true
+	return _node, nil
+}
+
+func (cc *CategoryCreate) createSpec() (*Category, *sqlgraph.CreateSpec) {
+	var (
+		_node = &Category{config: cc.config}
+		_spec = sqlgraph.NewCreateSpec(category.Table, sqlgraph.NewFieldSpec(category.FieldID, field.TypeUint64))
+	)
+	_spec.OnConflict = cc.conflict
+	if id, ok := cc.mutation.ID(); ok {
+		_node.ID = id
+		_spec.ID.Value = id
+	}
+	if value, ok := cc.mutation.CreatedAt(); ok {
+		_spec.SetField(category.FieldCreatedAt, field.TypeTime, value)
+		_node.CreatedAt = value
+	}
+	if value, ok := cc.mutation.UpdatedAt(); ok {
+		_spec.SetField(category.FieldUpdatedAt, field.TypeTime, value)
+		_node.UpdatedAt = value
+	}
+	if value, ok := cc.mutation.DeletedAt(); ok {
+		_spec.SetField(category.FieldDeletedAt, field.TypeTime, value)
+		_node.DeletedAt = value
+	}
+	if value, ok := cc.mutation.Name(); ok {
+		_spec.SetField(category.FieldName, field.TypeString, value)
+		_node.Name = value
+	}
+	if value, ok := cc.mutation.OrganizationID(); ok {
+		_spec.SetField(category.FieldOrganizationID, field.TypeUint64, value)
+		_node.OrganizationID = value
+	}
+	return _node, _spec
+}
+
+// OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
+// of the `INSERT` statement. For example:
+//
+//	client.Category.Create().
+//		SetCreatedAt(v).
+//		OnConflict(
+//			// Update the row with the new values
+//			// the was proposed for insertion.
+//			sql.ResolveWithNewValues(),
+//		).
+//		// Override some of the fields with custom
+//		// update values.
+//		Update(func(u *ent.CategoryUpsert) {
+//			SetCreatedAt(v+v).
+//		}).
+//		Exec(ctx)
+func (cc *CategoryCreate) OnConflict(opts ...sql.ConflictOption) *CategoryUpsertOne {
+	cc.conflict = opts
+	return &CategoryUpsertOne{
+		create: cc,
+	}
+}
+
+// OnConflictColumns calls `OnConflict` and configures the columns
+// as conflict target. Using this option is equivalent to using:
+//
+//	client.Category.Create().
+//		OnConflict(sql.ConflictColumns(columns...)).
+//		Exec(ctx)
+func (cc *CategoryCreate) OnConflictColumns(columns ...string) *CategoryUpsertOne {
+	cc.conflict = append(cc.conflict, sql.ConflictColumns(columns...))
+	return &CategoryUpsertOne{
+		create: cc,
+	}
+}
+
+type (
+	// CategoryUpsertOne is the builder for "upsert"-ing
+	//  one Category node.
+	CategoryUpsertOne struct {
+		create *CategoryCreate
+	}
+
+	// CategoryUpsert is the "OnConflict" setter.
+	CategoryUpsert struct {
+		*sql.UpdateSet
+	}
+)
+
+// SetUpdatedAt sets the "updated_at" field.
+func (u *CategoryUpsert) SetUpdatedAt(v time.Time) *CategoryUpsert {
+	u.Set(category.FieldUpdatedAt, v)
+	return u
+}
+
+// UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
+func (u *CategoryUpsert) UpdateUpdatedAt() *CategoryUpsert {
+	u.SetExcluded(category.FieldUpdatedAt)
+	return u
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (u *CategoryUpsert) SetDeletedAt(v time.Time) *CategoryUpsert {
+	u.Set(category.FieldDeletedAt, v)
+	return u
+}
+
+// UpdateDeletedAt sets the "deleted_at" field to the value that was provided on create.
+func (u *CategoryUpsert) UpdateDeletedAt() *CategoryUpsert {
+	u.SetExcluded(category.FieldDeletedAt)
+	return u
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (u *CategoryUpsert) ClearDeletedAt() *CategoryUpsert {
+	u.SetNull(category.FieldDeletedAt)
+	return u
+}
+
+// SetName sets the "name" field.
+func (u *CategoryUpsert) SetName(v string) *CategoryUpsert {
+	u.Set(category.FieldName, v)
+	return u
+}
+
+// UpdateName sets the "name" field to the value that was provided on create.
+func (u *CategoryUpsert) UpdateName() *CategoryUpsert {
+	u.SetExcluded(category.FieldName)
+	return u
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (u *CategoryUpsert) SetOrganizationID(v uint64) *CategoryUpsert {
+	u.Set(category.FieldOrganizationID, v)
+	return u
+}
+
+// UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
+func (u *CategoryUpsert) UpdateOrganizationID() *CategoryUpsert {
+	u.SetExcluded(category.FieldOrganizationID)
+	return u
+}
+
+// AddOrganizationID adds v to the "organization_id" field.
+func (u *CategoryUpsert) AddOrganizationID(v uint64) *CategoryUpsert {
+	u.Add(category.FieldOrganizationID, v)
+	return u
+}
+
+// UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
+// Using this option is equivalent to using:
+//
+//	client.Category.Create().
+//		OnConflict(
+//			sql.ResolveWithNewValues(),
+//			sql.ResolveWith(func(u *sql.UpdateSet) {
+//				u.SetIgnore(category.FieldID)
+//			}),
+//		).
+//		Exec(ctx)
+func (u *CategoryUpsertOne) UpdateNewValues() *CategoryUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
+		if _, exists := u.create.mutation.ID(); exists {
+			s.SetIgnore(category.FieldID)
+		}
+		if _, exists := u.create.mutation.CreatedAt(); exists {
+			s.SetIgnore(category.FieldCreatedAt)
+		}
+	}))
+	return u
+}
+
+// Ignore sets each column to itself in case of conflict.
+// Using this option is equivalent to using:
+//
+//	client.Category.Create().
+//	    OnConflict(sql.ResolveWithIgnore()).
+//	    Exec(ctx)
+func (u *CategoryUpsertOne) Ignore() *CategoryUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
+	return u
+}
+
+// DoNothing configures the conflict_action to `DO NOTHING`.
+// Supported only by SQLite and PostgreSQL.
+func (u *CategoryUpsertOne) DoNothing() *CategoryUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.DoNothing())
+	return u
+}
+
+// Update allows overriding fields `UPDATE` values. See the CategoryCreate.OnConflict
+// documentation for more info.
+func (u *CategoryUpsertOne) Update(set func(*CategoryUpsert)) *CategoryUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
+		set(&CategoryUpsert{UpdateSet: update})
+	}))
+	return u
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (u *CategoryUpsertOne) SetUpdatedAt(v time.Time) *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetUpdatedAt(v)
+	})
+}
+
+// UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
+func (u *CategoryUpsertOne) UpdateUpdatedAt() *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateUpdatedAt()
+	})
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (u *CategoryUpsertOne) SetDeletedAt(v time.Time) *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetDeletedAt(v)
+	})
+}
+
+// UpdateDeletedAt sets the "deleted_at" field to the value that was provided on create.
+func (u *CategoryUpsertOne) UpdateDeletedAt() *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateDeletedAt()
+	})
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (u *CategoryUpsertOne) ClearDeletedAt() *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.ClearDeletedAt()
+	})
+}
+
+// SetName sets the "name" field.
+func (u *CategoryUpsertOne) SetName(v string) *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetName(v)
+	})
+}
+
+// UpdateName sets the "name" field to the value that was provided on create.
+func (u *CategoryUpsertOne) UpdateName() *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateName()
+	})
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (u *CategoryUpsertOne) SetOrganizationID(v uint64) *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetOrganizationID(v)
+	})
+}
+
+// AddOrganizationID adds v to the "organization_id" field.
+func (u *CategoryUpsertOne) AddOrganizationID(v uint64) *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.AddOrganizationID(v)
+	})
+}
+
+// UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
+func (u *CategoryUpsertOne) UpdateOrganizationID() *CategoryUpsertOne {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateOrganizationID()
+	})
+}
+
+// Exec executes the query.
+func (u *CategoryUpsertOne) Exec(ctx context.Context) error {
+	if len(u.create.conflict) == 0 {
+		return errors.New("ent: missing options for CategoryCreate.OnConflict")
+	}
+	return u.create.Exec(ctx)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (u *CategoryUpsertOne) ExecX(ctx context.Context) {
+	if err := u.create.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// Exec executes the UPSERT query and returns the inserted/updated ID.
+func (u *CategoryUpsertOne) ID(ctx context.Context) (id uint64, err error) {
+	node, err := u.create.Save(ctx)
+	if err != nil {
+		return id, err
+	}
+	return node.ID, nil
+}
+
+// IDX is like ID, but panics if an error occurs.
+func (u *CategoryUpsertOne) IDX(ctx context.Context) uint64 {
+	id, err := u.ID(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return id
+}
+
+// CategoryCreateBulk is the builder for creating many Category entities in bulk.
+type CategoryCreateBulk struct {
+	config
+	err      error
+	builders []*CategoryCreate
+	conflict []sql.ConflictOption
+}
+
+// Save creates the Category entities in the database.
+func (ccb *CategoryCreateBulk) Save(ctx context.Context) ([]*Category, error) {
+	if ccb.err != nil {
+		return nil, ccb.err
+	}
+	specs := make([]*sqlgraph.CreateSpec, len(ccb.builders))
+	nodes := make([]*Category, len(ccb.builders))
+	mutators := make([]Mutator, len(ccb.builders))
+	for i := range ccb.builders {
+		func(i int, root context.Context) {
+			builder := ccb.builders[i]
+			builder.defaults()
+			var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
+				mutation, ok := m.(*CategoryMutation)
+				if !ok {
+					return nil, fmt.Errorf("unexpected mutation type %T", m)
+				}
+				if err := builder.check(); err != nil {
+					return nil, err
+				}
+				builder.mutation = mutation
+				var err error
+				nodes[i], specs[i] = builder.createSpec()
+				if i < len(mutators)-1 {
+					_, err = mutators[i+1].Mutate(root, ccb.builders[i+1].mutation)
+				} else {
+					spec := &sqlgraph.BatchCreateSpec{Nodes: specs}
+					spec.OnConflict = ccb.conflict
+					// Invoke the actual operation on the latest mutation in the chain.
+					if err = sqlgraph.BatchCreate(ctx, ccb.driver, spec); err != nil {
+						if sqlgraph.IsConstraintError(err) {
+							err = &ConstraintError{msg: err.Error(), wrap: err}
+						}
+					}
+				}
+				if err != nil {
+					return nil, err
+				}
+				mutation.id = &nodes[i].ID
+				if specs[i].ID.Value != nil && nodes[i].ID == 0 {
+					id := specs[i].ID.Value.(int64)
+					nodes[i].ID = uint64(id)
+				}
+				mutation.done = true
+				return nodes[i], nil
+			})
+			for i := len(builder.hooks) - 1; i >= 0; i-- {
+				mut = builder.hooks[i](mut)
+			}
+			mutators[i] = mut
+		}(i, ctx)
+	}
+	if len(mutators) > 0 {
+		if _, err := mutators[0].Mutate(ctx, ccb.builders[0].mutation); err != nil {
+			return nil, err
+		}
+	}
+	return nodes, nil
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (ccb *CategoryCreateBulk) SaveX(ctx context.Context) []*Category {
+	v, err := ccb.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return v
+}
+
+// Exec executes the query.
+func (ccb *CategoryCreateBulk) Exec(ctx context.Context) error {
+	_, err := ccb.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (ccb *CategoryCreateBulk) ExecX(ctx context.Context) {
+	if err := ccb.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
+// of the `INSERT` statement. For example:
+//
+//	client.Category.CreateBulk(builders...).
+//		OnConflict(
+//			// Update the row with the new values
+//			// the was proposed for insertion.
+//			sql.ResolveWithNewValues(),
+//		).
+//		// Override some of the fields with custom
+//		// update values.
+//		Update(func(u *ent.CategoryUpsert) {
+//			SetCreatedAt(v+v).
+//		}).
+//		Exec(ctx)
+func (ccb *CategoryCreateBulk) OnConflict(opts ...sql.ConflictOption) *CategoryUpsertBulk {
+	ccb.conflict = opts
+	return &CategoryUpsertBulk{
+		create: ccb,
+	}
+}
+
+// OnConflictColumns calls `OnConflict` and configures the columns
+// as conflict target. Using this option is equivalent to using:
+//
+//	client.Category.Create().
+//		OnConflict(sql.ConflictColumns(columns...)).
+//		Exec(ctx)
+func (ccb *CategoryCreateBulk) OnConflictColumns(columns ...string) *CategoryUpsertBulk {
+	ccb.conflict = append(ccb.conflict, sql.ConflictColumns(columns...))
+	return &CategoryUpsertBulk{
+		create: ccb,
+	}
+}
+
+// CategoryUpsertBulk is the builder for "upsert"-ing
+// a bulk of Category nodes.
+type CategoryUpsertBulk struct {
+	create *CategoryCreateBulk
+}
+
+// UpdateNewValues updates the mutable fields using the new values that
+// were set on create. Using this option is equivalent to using:
+//
+//	client.Category.Create().
+//		OnConflict(
+//			sql.ResolveWithNewValues(),
+//			sql.ResolveWith(func(u *sql.UpdateSet) {
+//				u.SetIgnore(category.FieldID)
+//			}),
+//		).
+//		Exec(ctx)
+func (u *CategoryUpsertBulk) UpdateNewValues() *CategoryUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
+		for _, b := range u.create.builders {
+			if _, exists := b.mutation.ID(); exists {
+				s.SetIgnore(category.FieldID)
+			}
+			if _, exists := b.mutation.CreatedAt(); exists {
+				s.SetIgnore(category.FieldCreatedAt)
+			}
+		}
+	}))
+	return u
+}
+
+// Ignore sets each column to itself in case of conflict.
+// Using this option is equivalent to using:
+//
+//	client.Category.Create().
+//		OnConflict(sql.ResolveWithIgnore()).
+//		Exec(ctx)
+func (u *CategoryUpsertBulk) Ignore() *CategoryUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
+	return u
+}
+
+// DoNothing configures the conflict_action to `DO NOTHING`.
+// Supported only by SQLite and PostgreSQL.
+func (u *CategoryUpsertBulk) DoNothing() *CategoryUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.DoNothing())
+	return u
+}
+
+// Update allows overriding fields `UPDATE` values. See the CategoryCreateBulk.OnConflict
+// documentation for more info.
+func (u *CategoryUpsertBulk) Update(set func(*CategoryUpsert)) *CategoryUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
+		set(&CategoryUpsert{UpdateSet: update})
+	}))
+	return u
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (u *CategoryUpsertBulk) SetUpdatedAt(v time.Time) *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetUpdatedAt(v)
+	})
+}
+
+// UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
+func (u *CategoryUpsertBulk) UpdateUpdatedAt() *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateUpdatedAt()
+	})
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (u *CategoryUpsertBulk) SetDeletedAt(v time.Time) *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetDeletedAt(v)
+	})
+}
+
+// UpdateDeletedAt sets the "deleted_at" field to the value that was provided on create.
+func (u *CategoryUpsertBulk) UpdateDeletedAt() *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateDeletedAt()
+	})
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (u *CategoryUpsertBulk) ClearDeletedAt() *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.ClearDeletedAt()
+	})
+}
+
+// SetName sets the "name" field.
+func (u *CategoryUpsertBulk) SetName(v string) *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetName(v)
+	})
+}
+
+// UpdateName sets the "name" field to the value that was provided on create.
+func (u *CategoryUpsertBulk) UpdateName() *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateName()
+	})
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (u *CategoryUpsertBulk) SetOrganizationID(v uint64) *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.SetOrganizationID(v)
+	})
+}
+
+// AddOrganizationID adds v to the "organization_id" field.
+func (u *CategoryUpsertBulk) AddOrganizationID(v uint64) *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.AddOrganizationID(v)
+	})
+}
+
+// UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
+func (u *CategoryUpsertBulk) UpdateOrganizationID() *CategoryUpsertBulk {
+	return u.Update(func(s *CategoryUpsert) {
+		s.UpdateOrganizationID()
+	})
+}
+
+// Exec executes the query.
+func (u *CategoryUpsertBulk) Exec(ctx context.Context) error {
+	if u.create.err != nil {
+		return u.create.err
+	}
+	for i, b := range u.create.builders {
+		if len(b.conflict) != 0 {
+			return fmt.Errorf("ent: OnConflict was set for builder %d. Set it on the CategoryCreateBulk instead", i)
+		}
+	}
+	if len(u.create.conflict) == 0 {
+		return errors.New("ent: missing options for CategoryCreateBulk.OnConflict")
+	}
+	return u.create.Exec(ctx)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (u *CategoryUpsertBulk) ExecX(ctx context.Context) {
+	if err := u.create.Exec(ctx); err != nil {
+		panic(err)
+	}
+}

ent/category_delete.go

@@ -0,0 +1,88 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"wechat-api/ent/category"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// CategoryDelete is the builder for deleting a Category entity.
+type CategoryDelete struct {
+	config
+	hooks    []Hook
+	mutation *CategoryMutation
+}
+
+// Where appends a list predicates to the CategoryDelete builder.
+func (cd *CategoryDelete) Where(ps ...predicate.Category) *CategoryDelete {
+	cd.mutation.Where(ps...)
+	return cd
+}
+
+// Exec executes the deletion query and returns how many vertices were deleted.
+func (cd *CategoryDelete) Exec(ctx context.Context) (int, error) {
+	return withHooks(ctx, cd.sqlExec, cd.mutation, cd.hooks)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (cd *CategoryDelete) ExecX(ctx context.Context) int {
+	n, err := cd.Exec(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return n
+}
+
+func (cd *CategoryDelete) sqlExec(ctx context.Context) (int, error) {
+	_spec := sqlgraph.NewDeleteSpec(category.Table, sqlgraph.NewFieldSpec(category.FieldID, field.TypeUint64))
+	if ps := cd.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	affected, err := sqlgraph.DeleteNodes(ctx, cd.driver, _spec)
+	if err != nil && sqlgraph.IsConstraintError(err) {
+		err = &ConstraintError{msg: err.Error(), wrap: err}
+	}
+	cd.mutation.done = true
+	return affected, err
+}
+
+// CategoryDeleteOne is the builder for deleting a single Category entity.
+type CategoryDeleteOne struct {
+	cd *CategoryDelete
+}
+
+// Where appends a list predicates to the CategoryDelete builder.
+func (cdo *CategoryDeleteOne) Where(ps ...predicate.Category) *CategoryDeleteOne {
+	cdo.cd.mutation.Where(ps...)
+	return cdo
+}
+
+// Exec executes the deletion query.
+func (cdo *CategoryDeleteOne) Exec(ctx context.Context) error {
+	n, err := cdo.cd.Exec(ctx)
+	switch {
+	case err != nil:
+		return err
+	case n == 0:
+		return &NotFoundError{category.Label}
+	default:
+		return nil
+	}
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (cdo *CategoryDeleteOne) ExecX(ctx context.Context) {
+	if err := cdo.Exec(ctx); err != nil {
+		panic(err)
+	}
+}

ent/category_query.go

@@ -0,0 +1,526 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"fmt"
+	"math"
+	"wechat-api/ent/category"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// CategoryQuery is the builder for querying Category entities.
+type CategoryQuery struct {
+	config
+	ctx        *QueryContext
+	order      []category.OrderOption
+	inters     []Interceptor
+	predicates []predicate.Category
+	// intermediate query (i.e. traversal path).
+	sql  *sql.Selector
+	path func(context.Context) (*sql.Selector, error)
+}
+
+// Where adds a new predicate for the CategoryQuery builder.
+func (cq *CategoryQuery) Where(ps ...predicate.Category) *CategoryQuery {
+	cq.predicates = append(cq.predicates, ps...)
+	return cq
+}
+
+// Limit the number of records to be returned by this query.
+func (cq *CategoryQuery) Limit(limit int) *CategoryQuery {
+	cq.ctx.Limit = &limit
+	return cq
+}
+
+// Offset to start from.
+func (cq *CategoryQuery) Offset(offset int) *CategoryQuery {
+	cq.ctx.Offset = &offset
+	return cq
+}
+
+// Unique configures the query builder to filter duplicate records on query.
+// By default, unique is set to true, and can be disabled using this method.
+func (cq *CategoryQuery) Unique(unique bool) *CategoryQuery {
+	cq.ctx.Unique = &unique
+	return cq
+}
+
+// Order specifies how the records should be ordered.
+func (cq *CategoryQuery) Order(o ...category.OrderOption) *CategoryQuery {
+	cq.order = append(cq.order, o...)
+	return cq
+}
+
+// First returns the first Category entity from the query.
+// Returns a *NotFoundError when no Category was found.
+func (cq *CategoryQuery) First(ctx context.Context) (*Category, error) {
+	nodes, err := cq.Limit(1).All(setContextOp(ctx, cq.ctx, "First"))
+	if err != nil {
+		return nil, err
+	}
+	if len(nodes) == 0 {
+		return nil, &NotFoundError{category.Label}
+	}
+	return nodes[0], nil
+}
+
+// FirstX is like First, but panics if an error occurs.
+func (cq *CategoryQuery) FirstX(ctx context.Context) *Category {
+	node, err := cq.First(ctx)
+	if err != nil && !IsNotFound(err) {
+		panic(err)
+	}
+	return node
+}
+
+// FirstID returns the first Category ID from the query.
+// Returns a *NotFoundError when no Category ID was found.
+func (cq *CategoryQuery) FirstID(ctx context.Context) (id uint64, err error) {
+	var ids []uint64
+	if ids, err = cq.Limit(1).IDs(setContextOp(ctx, cq.ctx, "FirstID")); err != nil {
+		return
+	}
+	if len(ids) == 0 {
+		err = &NotFoundError{category.Label}
+		return
+	}
+	return ids[0], nil
+}
+
+// FirstIDX is like FirstID, but panics if an error occurs.
+func (cq *CategoryQuery) FirstIDX(ctx context.Context) uint64 {
+	id, err := cq.FirstID(ctx)
+	if err != nil && !IsNotFound(err) {
+		panic(err)
+	}
+	return id
+}
+
+// Only returns a single Category entity found by the query, ensuring it only returns one.
+// Returns a *NotSingularError when more than one Category entity is found.
+// Returns a *NotFoundError when no Category entities are found.
+func (cq *CategoryQuery) Only(ctx context.Context) (*Category, error) {
+	nodes, err := cq.Limit(2).All(setContextOp(ctx, cq.ctx, "Only"))
+	if err != nil {
+		return nil, err
+	}
+	switch len(nodes) {
+	case 1:
+		return nodes[0], nil
+	case 0:
+		return nil, &NotFoundError{category.Label}
+	default:
+		return nil, &NotSingularError{category.Label}
+	}
+}
+
+// OnlyX is like Only, but panics if an error occurs.
+func (cq *CategoryQuery) OnlyX(ctx context.Context) *Category {
+	node, err := cq.Only(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return node
+}
+
+// OnlyID is like Only, but returns the only Category ID in the query.
+// Returns a *NotSingularError when more than one Category ID is found.
+// Returns a *NotFoundError when no entities are found.
+func (cq *CategoryQuery) OnlyID(ctx context.Context) (id uint64, err error) {
+	var ids []uint64
+	if ids, err = cq.Limit(2).IDs(setContextOp(ctx, cq.ctx, "OnlyID")); err != nil {
+		return
+	}
+	switch len(ids) {
+	case 1:
+		id = ids[0]
+	case 0:
+		err = &NotFoundError{category.Label}
+	default:
+		err = &NotSingularError{category.Label}
+	}
+	return
+}
+
+// OnlyIDX is like OnlyID, but panics if an error occurs.
+func (cq *CategoryQuery) OnlyIDX(ctx context.Context) uint64 {
+	id, err := cq.OnlyID(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return id
+}
+
+// All executes the query and returns a list of Categories.
+func (cq *CategoryQuery) All(ctx context.Context) ([]*Category, error) {
+	ctx = setContextOp(ctx, cq.ctx, "All")
+	if err := cq.prepareQuery(ctx); err != nil {
+		return nil, err
+	}
+	qr := querierAll[[]*Category, *CategoryQuery]()
+	return withInterceptors[[]*Category](ctx, cq, qr, cq.inters)
+}
+
+// AllX is like All, but panics if an error occurs.
+func (cq *CategoryQuery) AllX(ctx context.Context) []*Category {
+	nodes, err := cq.All(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return nodes
+}
+
+// IDs executes the query and returns a list of Category IDs.
+func (cq *CategoryQuery) IDs(ctx context.Context) (ids []uint64, err error) {
+	if cq.ctx.Unique == nil && cq.path != nil {
+		cq.Unique(true)
+	}
+	ctx = setContextOp(ctx, cq.ctx, "IDs")
+	if err = cq.Select(category.FieldID).Scan(ctx, &ids); err != nil {
+		return nil, err
+	}
+	return ids, nil
+}
+
+// IDsX is like IDs, but panics if an error occurs.
+func (cq *CategoryQuery) IDsX(ctx context.Context) []uint64 {
+	ids, err := cq.IDs(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return ids
+}
+
+// Count returns the count of the given query.
+func (cq *CategoryQuery) Count(ctx context.Context) (int, error) {
+	ctx = setContextOp(ctx, cq.ctx, "Count")
+	if err := cq.prepareQuery(ctx); err != nil {
+		return 0, err
+	}
+	return withInterceptors[int](ctx, cq, querierCount[*CategoryQuery](), cq.inters)
+}
+
+// CountX is like Count, but panics if an error occurs.
+func (cq *CategoryQuery) CountX(ctx context.Context) int {
+	count, err := cq.Count(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return count
+}
+
+// Exist returns true if the query has elements in the graph.
+func (cq *CategoryQuery) Exist(ctx context.Context) (bool, error) {
+	ctx = setContextOp(ctx, cq.ctx, "Exist")
+	switch _, err := cq.FirstID(ctx); {
+	case IsNotFound(err):
+		return false, nil
+	case err != nil:
+		return false, fmt.Errorf("ent: check existence: %w", err)
+	default:
+		return true, nil
+	}
+}
+
+// ExistX is like Exist, but panics if an error occurs.
+func (cq *CategoryQuery) ExistX(ctx context.Context) bool {
+	exist, err := cq.Exist(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return exist
+}
+
+// Clone returns a duplicate of the CategoryQuery builder, including all associated steps. It can be
+// used to prepare common query builders and use them differently after the clone is made.
+func (cq *CategoryQuery) Clone() *CategoryQuery {
+	if cq == nil {
+		return nil
+	}
+	return &CategoryQuery{
+		config:     cq.config,
+		ctx:        cq.ctx.Clone(),
+		order:      append([]category.OrderOption{}, cq.order...),
+		inters:     append([]Interceptor{}, cq.inters...),
+		predicates: append([]predicate.Category{}, cq.predicates...),
+		// clone intermediate query.
+		sql:  cq.sql.Clone(),
+		path: cq.path,
+	}
+}
+
+// GroupBy is used to group vertices by one or more fields/columns.
+// It is often used with aggregate functions, like: count, max, mean, min, sum.
+//
+// Example:
+//
+//	var v []struct {
+//		CreatedAt time.Time `json:"created_at,omitempty"`
+//		Count int `json:"count,omitempty"`
+//	}
+//
+//	client.Category.Query().
+//		GroupBy(category.FieldCreatedAt).
+//		Aggregate(ent.Count()).
+//		Scan(ctx, &v)
+func (cq *CategoryQuery) GroupBy(field string, fields ...string) *CategoryGroupBy {
+	cq.ctx.Fields = append([]string{field}, fields...)
+	grbuild := &CategoryGroupBy{build: cq}
+	grbuild.flds = &cq.ctx.Fields
+	grbuild.label = category.Label
+	grbuild.scan = grbuild.Scan
+	return grbuild
+}
+
+// Select allows the selection one or more fields/columns for the given query,
+// instead of selecting all fields in the entity.
+//
+// Example:
+//
+//	var v []struct {
+//		CreatedAt time.Time `json:"created_at,omitempty"`
+//	}
+//
+//	client.Category.Query().
+//		Select(category.FieldCreatedAt).
+//		Scan(ctx, &v)
+func (cq *CategoryQuery) Select(fields ...string) *CategorySelect {
+	cq.ctx.Fields = append(cq.ctx.Fields, fields...)
+	sbuild := &CategorySelect{CategoryQuery: cq}
+	sbuild.label = category.Label
+	sbuild.flds, sbuild.scan = &cq.ctx.Fields, sbuild.Scan
+	return sbuild
+}
+
+// Aggregate returns a CategorySelect configured with the given aggregations.
+func (cq *CategoryQuery) Aggregate(fns ...AggregateFunc) *CategorySelect {
+	return cq.Select().Aggregate(fns...)
+}
+
+func (cq *CategoryQuery) prepareQuery(ctx context.Context) error {
+	for _, inter := range cq.inters {
+		if inter == nil {
+			return fmt.Errorf("ent: uninitialized interceptor (forgotten import ent/runtime?)")
+		}
+		if trv, ok := inter.(Traverser); ok {
+			if err := trv.Traverse(ctx, cq); err != nil {
+				return err
+			}
+		}
+	}
+	for _, f := range cq.ctx.Fields {
+		if !category.ValidColumn(f) {
+			return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
+		}
+	}
+	if cq.path != nil {
+		prev, err := cq.path(ctx)
+		if err != nil {
+			return err
+		}
+		cq.sql = prev
+	}
+	return nil
+}
+
+func (cq *CategoryQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Category, error) {
+	var (
+		nodes = []*Category{}
+		_spec = cq.querySpec()
+	)
+	_spec.ScanValues = func(columns []string) ([]any, error) {
+		return (*Category).scanValues(nil, columns)
+	}
+	_spec.Assign = func(columns []string, values []any) error {
+		node := &Category{config: cq.config}
+		nodes = append(nodes, node)
+		return node.assignValues(columns, values)
+	}
+	for i := range hooks {
+		hooks[i](ctx, _spec)
+	}
+	if err := sqlgraph.QueryNodes(ctx, cq.driver, _spec); err != nil {
+		return nil, err
+	}
+	if len(nodes) == 0 {
+		return nodes, nil
+	}
+	return nodes, nil
+}
+
+func (cq *CategoryQuery) sqlCount(ctx context.Context) (int, error) {
+	_spec := cq.querySpec()
+	_spec.Node.Columns = cq.ctx.Fields
+	if len(cq.ctx.Fields) > 0 {
+		_spec.Unique = cq.ctx.Unique != nil && *cq.ctx.Unique
+	}
+	return sqlgraph.CountNodes(ctx, cq.driver, _spec)
+}
+
+func (cq *CategoryQuery) querySpec() *sqlgraph.QuerySpec {
+	_spec := sqlgraph.NewQuerySpec(category.Table, category.Columns, sqlgraph.NewFieldSpec(category.FieldID, field.TypeUint64))
+	_spec.From = cq.sql
+	if unique := cq.ctx.Unique; unique != nil {
+		_spec.Unique = *unique
+	} else if cq.path != nil {
+		_spec.Unique = true
+	}
+	if fields := cq.ctx.Fields; len(fields) > 0 {
+		_spec.Node.Columns = make([]string, 0, len(fields))
+		_spec.Node.Columns = append(_spec.Node.Columns, category.FieldID)
+		for i := range fields {
+			if fields[i] != category.FieldID {
+				_spec.Node.Columns = append(_spec.Node.Columns, fields[i])
+			}
+		}
+	}
+	if ps := cq.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if limit := cq.ctx.Limit; limit != nil {
+		_spec.Limit = *limit
+	}
+	if offset := cq.ctx.Offset; offset != nil {
+		_spec.Offset = *offset
+	}
+	if ps := cq.order; len(ps) > 0 {
+		_spec.Order = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	return _spec
+}
+
+func (cq *CategoryQuery) sqlQuery(ctx context.Context) *sql.Selector {
+	builder := sql.Dialect(cq.driver.Dialect())
+	t1 := builder.Table(category.Table)
+	columns := cq.ctx.Fields
+	if len(columns) == 0 {
+		columns = category.Columns
+	}
+	selector := builder.Select(t1.Columns(columns...)...).From(t1)
+	if cq.sql != nil {
+		selector = cq.sql
+		selector.Select(selector.Columns(columns...)...)
+	}
+	if cq.ctx.Unique != nil && *cq.ctx.Unique {
+		selector.Distinct()
+	}
+	for _, p := range cq.predicates {
+		p(selector)
+	}
+	for _, p := range cq.order {
+		p(selector)
+	}
+	if offset := cq.ctx.Offset; offset != nil {
+		// limit is mandatory for offset clause. We start
+		// with default value, and override it below if needed.
+		selector.Offset(*offset).Limit(math.MaxInt32)
+	}
+	if limit := cq.ctx.Limit; limit != nil {
+		selector.Limit(*limit)
+	}
+	return selector
+}
+
+// CategoryGroupBy is the group-by builder for Category entities.
+type CategoryGroupBy struct {
+	selector
+	build *CategoryQuery
+}
+
+// Aggregate adds the given aggregation functions to the group-by query.
+func (cgb *CategoryGroupBy) Aggregate(fns ...AggregateFunc) *CategoryGroupBy {
+	cgb.fns = append(cgb.fns, fns...)
+	return cgb
+}
+
+// Scan applies the selector query and scans the result into the given value.
+func (cgb *CategoryGroupBy) Scan(ctx context.Context, v any) error {
+	ctx = setContextOp(ctx, cgb.build.ctx, "GroupBy")
+	if err := cgb.build.prepareQuery(ctx); err != nil {
+		return err
+	}
+	return scanWithInterceptors[*CategoryQuery, *CategoryGroupBy](ctx, cgb.build, cgb, cgb.build.inters, v)
+}
+
+func (cgb *CategoryGroupBy) sqlScan(ctx context.Context, root *CategoryQuery, v any) error {
+	selector := root.sqlQuery(ctx).Select()
+	aggregation := make([]string, 0, len(cgb.fns))
+	for _, fn := range cgb.fns {
+		aggregation = append(aggregation, fn(selector))
+	}
+	if len(selector.SelectedColumns()) == 0 {
+		columns := make([]string, 0, len(*cgb.flds)+len(cgb.fns))
+		for _, f := range *cgb.flds {
+			columns = append(columns, selector.C(f))
+		}
+		columns = append(columns, aggregation...)
+		selector.Select(columns...)
+	}
+	selector.GroupBy(selector.Columns(*cgb.flds...)...)
+	if err := selector.Err(); err != nil {
+		return err
+	}
+	rows := &sql.Rows{}
+	query, args := selector.Query()
+	if err := cgb.build.driver.Query(ctx, query, args, rows); err != nil {
+		return err
+	}
+	defer rows.Close()
+	return sql.ScanSlice(rows, v)
+}
+
+// CategorySelect is the builder for selecting fields of Category entities.
+type CategorySelect struct {
+	*CategoryQuery
+	selector
+}
+
+// Aggregate adds the given aggregation functions to the selector query.
+func (cs *CategorySelect) Aggregate(fns ...AggregateFunc) *CategorySelect {
+	cs.fns = append(cs.fns, fns...)
+	return cs
+}
+
+// Scan applies the selector query and scans the result into the given value.
+func (cs *CategorySelect) Scan(ctx context.Context, v any) error {
+	ctx = setContextOp(ctx, cs.ctx, "Select")
+	if err := cs.prepareQuery(ctx); err != nil {
+		return err
+	}
+	return scanWithInterceptors[*CategoryQuery, *CategorySelect](ctx, cs.CategoryQuery, cs, cs.inters, v)
+}
+
+func (cs *CategorySelect) sqlScan(ctx context.Context, root *CategoryQuery, v any) error {
+	selector := root.sqlQuery(ctx)
+	aggregation := make([]string, 0, len(cs.fns))
+	for _, fn := range cs.fns {
+		aggregation = append(aggregation, fn(selector))
+	}
+	switch n := len(*cs.selector.flds); {
+	case n == 0 && len(aggregation) > 0:
+		selector.Select(aggregation...)
+	case n != 0 && len(aggregation) > 0:
+		selector.AppendSelect(aggregation...)
+	}
+	rows := &sql.Rows{}
+	query, args := selector.Query()
+	if err := cs.driver.Query(ctx, query, args, rows); err != nil {
+		return err
+	}
+	defer rows.Close()
+	return sql.ScanSlice(rows, v)
+}

ent/category_update.go

@@ -0,0 +1,400 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+	"wechat-api/ent/category"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// CategoryUpdate is the builder for updating Category entities.
+type CategoryUpdate struct {
+	config
+	hooks    []Hook
+	mutation *CategoryMutation
+}
+
+// Where appends a list predicates to the CategoryUpdate builder.
+func (cu *CategoryUpdate) Where(ps ...predicate.Category) *CategoryUpdate {
+	cu.mutation.Where(ps...)
+	return cu
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (cu *CategoryUpdate) SetUpdatedAt(t time.Time) *CategoryUpdate {
+	cu.mutation.SetUpdatedAt(t)
+	return cu
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (cu *CategoryUpdate) SetDeletedAt(t time.Time) *CategoryUpdate {
+	cu.mutation.SetDeletedAt(t)
+	return cu
+}
+
+// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
+func (cu *CategoryUpdate) SetNillableDeletedAt(t *time.Time) *CategoryUpdate {
+	if t != nil {
+		cu.SetDeletedAt(*t)
+	}
+	return cu
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (cu *CategoryUpdate) ClearDeletedAt() *CategoryUpdate {
+	cu.mutation.ClearDeletedAt()
+	return cu
+}
+
+// SetName sets the "name" field.
+func (cu *CategoryUpdate) SetName(s string) *CategoryUpdate {
+	cu.mutation.SetName(s)
+	return cu
+}
+
+// SetNillableName sets the "name" field if the given value is not nil.
+func (cu *CategoryUpdate) SetNillableName(s *string) *CategoryUpdate {
+	if s != nil {
+		cu.SetName(*s)
+	}
+	return cu
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (cu *CategoryUpdate) SetOrganizationID(u uint64) *CategoryUpdate {
+	cu.mutation.ResetOrganizationID()
+	cu.mutation.SetOrganizationID(u)
+	return cu
+}
+
+// SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
+func (cu *CategoryUpdate) SetNillableOrganizationID(u *uint64) *CategoryUpdate {
+	if u != nil {
+		cu.SetOrganizationID(*u)
+	}
+	return cu
+}
+
+// AddOrganizationID adds u to the "organization_id" field.
+func (cu *CategoryUpdate) AddOrganizationID(u int64) *CategoryUpdate {
+	cu.mutation.AddOrganizationID(u)
+	return cu
+}
+
+// Mutation returns the CategoryMutation object of the builder.
+func (cu *CategoryUpdate) Mutation() *CategoryMutation {
+	return cu.mutation
+}
+
+// Save executes the query and returns the number of nodes affected by the update operation.
+func (cu *CategoryUpdate) Save(ctx context.Context) (int, error) {
+	if err := cu.defaults(); err != nil {
+		return 0, err
+	}
+	return withHooks(ctx, cu.sqlSave, cu.mutation, cu.hooks)
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (cu *CategoryUpdate) SaveX(ctx context.Context) int {
+	affected, err := cu.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return affected
+}
+
+// Exec executes the query.
+func (cu *CategoryUpdate) Exec(ctx context.Context) error {
+	_, err := cu.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (cu *CategoryUpdate) ExecX(ctx context.Context) {
+	if err := cu.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// defaults sets the default values of the builder before save.
+func (cu *CategoryUpdate) defaults() error {
+	if _, ok := cu.mutation.UpdatedAt(); !ok {
+		if category.UpdateDefaultUpdatedAt == nil {
+			return fmt.Errorf("ent: uninitialized category.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
+		}
+		v := category.UpdateDefaultUpdatedAt()
+		cu.mutation.SetUpdatedAt(v)
+	}
+	return nil
+}
+
+// check runs all checks and user-defined validators on the builder.
+func (cu *CategoryUpdate) check() error {
+	if v, ok := cu.mutation.Name(); ok {
+		if err := category.NameValidator(v); err != nil {
+			return &ValidationError{Name: "name", err: fmt.Errorf(`ent: validator failed for field "Category.name": %w`, err)}
+		}
+	}
+	if v, ok := cu.mutation.OrganizationID(); ok {
+		if err := category.OrganizationIDValidator(v); err != nil {
+			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Category.organization_id": %w`, err)}
+		}
+	}
+	return nil
+}
+
+func (cu *CategoryUpdate) sqlSave(ctx context.Context) (n int, err error) {
+	if err := cu.check(); err != nil {
+		return n, err
+	}
+	_spec := sqlgraph.NewUpdateSpec(category.Table, category.Columns, sqlgraph.NewFieldSpec(category.FieldID, field.TypeUint64))
+	if ps := cu.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if value, ok := cu.mutation.UpdatedAt(); ok {
+		_spec.SetField(category.FieldUpdatedAt, field.TypeTime, value)
+	}
+	if value, ok := cu.mutation.DeletedAt(); ok {
+		_spec.SetField(category.FieldDeletedAt, field.TypeTime, value)
+	}
+	if cu.mutation.DeletedAtCleared() {
+		_spec.ClearField(category.FieldDeletedAt, field.TypeTime)
+	}
+	if value, ok := cu.mutation.Name(); ok {
+		_spec.SetField(category.FieldName, field.TypeString, value)
+	}
+	if value, ok := cu.mutation.OrganizationID(); ok {
+		_spec.SetField(category.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if value, ok := cu.mutation.AddedOrganizationID(); ok {
+		_spec.AddField(category.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if n, err = sqlgraph.UpdateNodes(ctx, cu.driver, _spec); err != nil {
+		if _, ok := err.(*sqlgraph.NotFoundError); ok {
+			err = &NotFoundError{category.Label}
+		} else if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return 0, err
+	}
+	cu.mutation.done = true
+	return n, nil
+}
+
+// CategoryUpdateOne is the builder for updating a single Category entity.
+type CategoryUpdateOne struct {
+	config
+	fields   []string
+	hooks    []Hook
+	mutation *CategoryMutation
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (cuo *CategoryUpdateOne) SetUpdatedAt(t time.Time) *CategoryUpdateOne {
+	cuo.mutation.SetUpdatedAt(t)
+	return cuo
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (cuo *CategoryUpdateOne) SetDeletedAt(t time.Time) *CategoryUpdateOne {
+	cuo.mutation.SetDeletedAt(t)
+	return cuo
+}
+
+// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
+func (cuo *CategoryUpdateOne) SetNillableDeletedAt(t *time.Time) *CategoryUpdateOne {
+	if t != nil {
+		cuo.SetDeletedAt(*t)
+	}
+	return cuo
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (cuo *CategoryUpdateOne) ClearDeletedAt() *CategoryUpdateOne {
+	cuo.mutation.ClearDeletedAt()
+	return cuo
+}
+
+// SetName sets the "name" field.
+func (cuo *CategoryUpdateOne) SetName(s string) *CategoryUpdateOne {
+	cuo.mutation.SetName(s)
+	return cuo
+}
+
+// SetNillableName sets the "name" field if the given value is not nil.
+func (cuo *CategoryUpdateOne) SetNillableName(s *string) *CategoryUpdateOne {
+	if s != nil {
+		cuo.SetName(*s)
+	}
+	return cuo
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (cuo *CategoryUpdateOne) SetOrganizationID(u uint64) *CategoryUpdateOne {
+	cuo.mutation.ResetOrganizationID()
+	cuo.mutation.SetOrganizationID(u)
+	return cuo
+}
+
+// SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
+func (cuo *CategoryUpdateOne) SetNillableOrganizationID(u *uint64) *CategoryUpdateOne {
+	if u != nil {
+		cuo.SetOrganizationID(*u)
+	}
+	return cuo
+}
+
+// AddOrganizationID adds u to the "organization_id" field.
+func (cuo *CategoryUpdateOne) AddOrganizationID(u int64) *CategoryUpdateOne {
+	cuo.mutation.AddOrganizationID(u)
+	return cuo
+}
+
+// Mutation returns the CategoryMutation object of the builder.
+func (cuo *CategoryUpdateOne) Mutation() *CategoryMutation {
+	return cuo.mutation
+}
+
+// Where appends a list predicates to the CategoryUpdate builder.
+func (cuo *CategoryUpdateOne) Where(ps ...predicate.Category) *CategoryUpdateOne {
+	cuo.mutation.Where(ps...)
+	return cuo
+}
+
+// Select allows selecting one or more fields (columns) of the returned entity.
+// The default is selecting all fields defined in the entity schema.
+func (cuo *CategoryUpdateOne) Select(field string, fields ...string) *CategoryUpdateOne {
+	cuo.fields = append([]string{field}, fields...)
+	return cuo
+}
+
+// Save executes the query and returns the updated Category entity.
+func (cuo *CategoryUpdateOne) Save(ctx context.Context) (*Category, error) {
+	if err := cuo.defaults(); err != nil {
+		return nil, err
+	}
+	return withHooks(ctx, cuo.sqlSave, cuo.mutation, cuo.hooks)
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (cuo *CategoryUpdateOne) SaveX(ctx context.Context) *Category {
+	node, err := cuo.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return node
+}
+
+// Exec executes the query on the entity.
+func (cuo *CategoryUpdateOne) Exec(ctx context.Context) error {
+	_, err := cuo.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (cuo *CategoryUpdateOne) ExecX(ctx context.Context) {
+	if err := cuo.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// defaults sets the default values of the builder before save.
+func (cuo *CategoryUpdateOne) defaults() error {
+	if _, ok := cuo.mutation.UpdatedAt(); !ok {
+		if category.UpdateDefaultUpdatedAt == nil {
+			return fmt.Errorf("ent: uninitialized category.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
+		}
+		v := category.UpdateDefaultUpdatedAt()
+		cuo.mutation.SetUpdatedAt(v)
+	}
+	return nil
+}
+
+// check runs all checks and user-defined validators on the builder.
+func (cuo *CategoryUpdateOne) check() error {
+	if v, ok := cuo.mutation.Name(); ok {
+		if err := category.NameValidator(v); err != nil {
+			return &ValidationError{Name: "name", err: fmt.Errorf(`ent: validator failed for field "Category.name": %w`, err)}
+		}
+	}
+	if v, ok := cuo.mutation.OrganizationID(); ok {
+		if err := category.OrganizationIDValidator(v); err != nil {
+			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Category.organization_id": %w`, err)}
+		}
+	}
+	return nil
+}
+
+func (cuo *CategoryUpdateOne) sqlSave(ctx context.Context) (_node *Category, err error) {
+	if err := cuo.check(); err != nil {
+		return _node, err
+	}
+	_spec := sqlgraph.NewUpdateSpec(category.Table, category.Columns, sqlgraph.NewFieldSpec(category.FieldID, field.TypeUint64))
+	id, ok := cuo.mutation.ID()
+	if !ok {
+		return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "Category.id" for update`)}
+	}
+	_spec.Node.ID.Value = id
+	if fields := cuo.fields; len(fields) > 0 {
+		_spec.Node.Columns = make([]string, 0, len(fields))
+		_spec.Node.Columns = append(_spec.Node.Columns, category.FieldID)
+		for _, f := range fields {
+			if !category.ValidColumn(f) {
+				return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
+			}
+			if f != category.FieldID {
+				_spec.Node.Columns = append(_spec.Node.Columns, f)
+			}
+		}
+	}
+	if ps := cuo.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if value, ok := cuo.mutation.UpdatedAt(); ok {
+		_spec.SetField(category.FieldUpdatedAt, field.TypeTime, value)
+	}
+	if value, ok := cuo.mutation.DeletedAt(); ok {
+		_spec.SetField(category.FieldDeletedAt, field.TypeTime, value)
+	}
+	if cuo.mutation.DeletedAtCleared() {
+		_spec.ClearField(category.FieldDeletedAt, field.TypeTime)
+	}
+	if value, ok := cuo.mutation.Name(); ok {
+		_spec.SetField(category.FieldName, field.TypeString, value)
+	}
+	if value, ok := cuo.mutation.OrganizationID(); ok {
+		_spec.SetField(category.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if value, ok := cuo.mutation.AddedOrganizationID(); ok {
+		_spec.AddField(category.FieldOrganizationID, field.TypeUint64, value)
+	}
+	_node = &Category{config: cuo.config}
+	_spec.Assign = _node.assignValues
+	_spec.ScanValues = _node.scanValues
+	if err = sqlgraph.UpdateNode(ctx, cuo.driver, _spec); err != nil {
+		if _, ok := err.(*sqlgraph.NotFoundError); ok {
+			err = &NotFoundError{category.Label}
+		} else if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return nil, err
+	}
+	cuo.mutation.done = true
+	return _node, nil
+}

ent/client.go

@@ -13,8 +13,10 @@ import (
 
 	"wechat-api/ent/agent"
 	"wechat-api/ent/batchmsg"
+	"wechat-api/ent/category"
 	"wechat-api/ent/contact"
 	"wechat-api/ent/employee"
+	"wechat-api/ent/employeeconfig"
 	"wechat-api/ent/label"
 	"wechat-api/ent/labelrelationship"
 	"wechat-api/ent/message"
@@ -46,10 +48,14 @@ type Client struct {
 	Agent *AgentClient
 	// BatchMsg is the client for interacting with the BatchMsg builders.
 	BatchMsg *BatchMsgClient
+	// Category is the client for interacting with the Category builders.
+	Category *CategoryClient
 	// Contact is the client for interacting with the Contact builders.
 	Contact *ContactClient
 	// Employee is the client for interacting with the Employee builders.
 	Employee *EmployeeClient
+	// EmployeeConfig is the client for interacting with the EmployeeConfig builders.
+	EmployeeConfig *EmployeeConfigClient
 	// Label is the client for interacting with the Label builders.
 	Label *LabelClient
 	// LabelRelationship is the client for interacting with the LabelRelationship builders.
@@ -89,8 +95,10 @@ func (c *Client) init() {
 	c.Schema = migrate.NewSchema(c.driver)
 	c.Agent = NewAgentClient(c.config)
 	c.BatchMsg = NewBatchMsgClient(c.config)
+	c.Category = NewCategoryClient(c.config)
 	c.Contact = NewContactClient(c.config)
 	c.Employee = NewEmployeeClient(c.config)
+	c.EmployeeConfig = NewEmployeeConfigClient(c.config)
 	c.Label = NewLabelClient(c.config)
 	c.LabelRelationship = NewLabelRelationshipClient(c.config)
 	c.Message = NewMessageClient(c.config)
@@ -198,8 +206,10 @@ func (c *Client) Tx(ctx context.Context) (*Tx, error) {
 		config:            cfg,
 		Agent:             NewAgentClient(cfg),
 		BatchMsg:          NewBatchMsgClient(cfg),
+		Category:          NewCategoryClient(cfg),
 		Contact:           NewContactClient(cfg),
 		Employee:          NewEmployeeClient(cfg),
+		EmployeeConfig:    NewEmployeeConfigClient(cfg),
 		Label:             NewLabelClient(cfg),
 		LabelRelationship: NewLabelRelationshipClient(cfg),
 		Message:           NewMessageClient(cfg),
@@ -234,8 +244,10 @@ func (c *Client) BeginTx(ctx context.Context, opts *sql.TxOptions) (*Tx, error)
 		config:            cfg,
 		Agent:             NewAgentClient(cfg),
 		BatchMsg:          NewBatchMsgClient(cfg),
+		Category:          NewCategoryClient(cfg),
 		Contact:           NewContactClient(cfg),
 		Employee:          NewEmployeeClient(cfg),
+		EmployeeConfig:    NewEmployeeConfigClient(cfg),
 		Label:             NewLabelClient(cfg),
 		LabelRelationship: NewLabelRelationshipClient(cfg),
 		Message:           NewMessageClient(cfg),
@@ -278,9 +290,9 @@ func (c *Client) Close() error {
 // In order to add hooks to a specific client, call: `client.Node.Use(...)`.
 func (c *Client) Use(hooks ...Hook) {
 	for _, n := range []interface{ Use(...Hook) }{
-		c.Agent, c.BatchMsg, c.Contact, c.Employee, c.Label, c.LabelRelationship,
-		c.Message, c.MessageRecords, c.Msg, c.Server, c.SopNode, c.SopStage, c.SopTask,
-		c.Token, c.Tutorial, c.WorkExperience, c.Wx,
+		c.Agent, c.BatchMsg, c.Category, c.Contact, c.Employee, c.EmployeeConfig,
+		c.Label, c.LabelRelationship, c.Message, c.MessageRecords, c.Msg, c.Server,
+		c.SopNode, c.SopStage, c.SopTask, c.Token, c.Tutorial, c.WorkExperience, c.Wx,
 	} {
 		n.Use(hooks...)
 	}
@@ -290,9 +302,9 @@ func (c *Client) Use(hooks ...Hook) {
 // In order to add interceptors to a specific client, call: `client.Node.Intercept(...)`.
 func (c *Client) Intercept(interceptors ...Interceptor) {
 	for _, n := range []interface{ Intercept(...Interceptor) }{
-		c.Agent, c.BatchMsg, c.Contact, c.Employee, c.Label, c.LabelRelationship,
-		c.Message, c.MessageRecords, c.Msg, c.Server, c.SopNode, c.SopStage, c.SopTask,
-		c.Token, c.Tutorial, c.WorkExperience, c.Wx,
+		c.Agent, c.BatchMsg, c.Category, c.Contact, c.Employee, c.EmployeeConfig,
+		c.Label, c.LabelRelationship, c.Message, c.MessageRecords, c.Msg, c.Server,
+		c.SopNode, c.SopStage, c.SopTask, c.Token, c.Tutorial, c.WorkExperience, c.Wx,
 	} {
 		n.Intercept(interceptors...)
 	}
@@ -305,10 +317,14 @@ func (c *Client) Mutate(ctx context.Context, m Mutation) (Value, error) {
 		return c.Agent.mutate(ctx, m)
 	case *BatchMsgMutation:
 		return c.BatchMsg.mutate(ctx, m)
+	case *CategoryMutation:
+		return c.Category.mutate(ctx, m)
 	case *ContactMutation:
 		return c.Contact.mutate(ctx, m)
 	case *EmployeeMutation:
 		return c.Employee.mutate(ctx, m)
+	case *EmployeeConfigMutation:
+		return c.EmployeeConfig.mutate(ctx, m)
 	case *LabelMutation:
 		return c.Label.mutate(ctx, m)
 	case *LabelRelationshipMutation:
@@ -626,6 +642,141 @@ func (c *BatchMsgClient) mutate(ctx context.Context, m *BatchMsgMutation) (Value
 	}
 }
 
+// CategoryClient is a client for the Category schema.
+type CategoryClient struct {
+	config
+}
+
+// NewCategoryClient returns a client for the Category from the given config.
+func NewCategoryClient(c config) *CategoryClient {
+	return &CategoryClient{config: c}
+}
+
+// Use adds a list of mutation hooks to the hooks stack.
+// A call to `Use(f, g, h)` equals to `category.Hooks(f(g(h())))`.
+func (c *CategoryClient) Use(hooks ...Hook) {
+	c.hooks.Category = append(c.hooks.Category, hooks...)
+}
+
+// Intercept adds a list of query interceptors to the interceptors stack.
+// A call to `Intercept(f, g, h)` equals to `category.Intercept(f(g(h())))`.
+func (c *CategoryClient) Intercept(interceptors ...Interceptor) {
+	c.inters.Category = append(c.inters.Category, interceptors...)
+}
+
+// Create returns a builder for creating a Category entity.
+func (c *CategoryClient) Create() *CategoryCreate {
+	mutation := newCategoryMutation(c.config, OpCreate)
+	return &CategoryCreate{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// CreateBulk returns a builder for creating a bulk of Category entities.
+func (c *CategoryClient) CreateBulk(builders ...*CategoryCreate) *CategoryCreateBulk {
+	return &CategoryCreateBulk{config: c.config, builders: builders}
+}
+
+// MapCreateBulk creates a bulk creation builder from the given slice. For each item in the slice, the function creates
+// a builder and applies setFunc on it.
+func (c *CategoryClient) MapCreateBulk(slice any, setFunc func(*CategoryCreate, int)) *CategoryCreateBulk {
+	rv := reflect.ValueOf(slice)
+	if rv.Kind() != reflect.Slice {
+		return &CategoryCreateBulk{err: fmt.Errorf("calling to CategoryClient.MapCreateBulk with wrong type %T, need slice", slice)}
+	}
+	builders := make([]*CategoryCreate, rv.Len())
+	for i := 0; i < rv.Len(); i++ {
+		builders[i] = c.Create()
+		setFunc(builders[i], i)
+	}
+	return &CategoryCreateBulk{config: c.config, builders: builders}
+}
+
+// Update returns an update builder for Category.
+func (c *CategoryClient) Update() *CategoryUpdate {
+	mutation := newCategoryMutation(c.config, OpUpdate)
+	return &CategoryUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// UpdateOne returns an update builder for the given entity.
+func (c *CategoryClient) UpdateOne(ca *Category) *CategoryUpdateOne {
+	mutation := newCategoryMutation(c.config, OpUpdateOne, withCategory(ca))
+	return &CategoryUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// UpdateOneID returns an update builder for the given id.
+func (c *CategoryClient) UpdateOneID(id uint64) *CategoryUpdateOne {
+	mutation := newCategoryMutation(c.config, OpUpdateOne, withCategoryID(id))
+	return &CategoryUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// Delete returns a delete builder for Category.
+func (c *CategoryClient) Delete() *CategoryDelete {
+	mutation := newCategoryMutation(c.config, OpDelete)
+	return &CategoryDelete{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// DeleteOne returns a builder for deleting the given entity.
+func (c *CategoryClient) DeleteOne(ca *Category) *CategoryDeleteOne {
+	return c.DeleteOneID(ca.ID)
+}
+
+// DeleteOneID returns a builder for deleting the given entity by its id.
+func (c *CategoryClient) DeleteOneID(id uint64) *CategoryDeleteOne {
+	builder := c.Delete().Where(category.ID(id))
+	builder.mutation.id = &id
+	builder.mutation.op = OpDeleteOne
+	return &CategoryDeleteOne{builder}
+}
+
+// Query returns a query builder for Category.
+func (c *CategoryClient) Query() *CategoryQuery {
+	return &CategoryQuery{
+		config: c.config,
+		ctx:    &QueryContext{Type: TypeCategory},
+		inters: c.Interceptors(),
+	}
+}
+
+// Get returns a Category entity by its id.
+func (c *CategoryClient) Get(ctx context.Context, id uint64) (*Category, error) {
+	return c.Query().Where(category.ID(id)).Only(ctx)
+}
+
+// GetX is like Get, but panics if an error occurs.
+func (c *CategoryClient) GetX(ctx context.Context, id uint64) *Category {
+	obj, err := c.Get(ctx, id)
+	if err != nil {
+		panic(err)
+	}
+	return obj
+}
+
+// Hooks returns the client hooks.
+func (c *CategoryClient) Hooks() []Hook {
+	hooks := c.hooks.Category
+	return append(hooks[:len(hooks):len(hooks)], category.Hooks[:]...)
+}
+
+// Interceptors returns the client interceptors.
+func (c *CategoryClient) Interceptors() []Interceptor {
+	inters := c.inters.Category
+	return append(inters[:len(inters):len(inters)], category.Interceptors[:]...)
+}
+
+func (c *CategoryClient) mutate(ctx context.Context, m *CategoryMutation) (Value, error) {
+	switch m.Op() {
+	case OpCreate:
+		return (&CategoryCreate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+	case OpUpdate:
+		return (&CategoryUpdate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+	case OpUpdateOne:
+		return (&CategoryUpdateOne{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+	case OpDelete, OpDeleteOne:
+		return (&CategoryDelete{config: c.config, hooks: c.Hooks(), mutation: m}).Exec(ctx)
+	default:
+		return nil, fmt.Errorf("ent: unknown Category mutation op: %q", m.Op())
+	}
+}
+
 // ContactClient is a client for the Contact schema.
 type ContactClient struct {
 	config
@@ -960,6 +1111,141 @@ func (c *EmployeeClient) mutate(ctx context.Context, m *EmployeeMutation) (Value
 	}
 }
 
+// EmployeeConfigClient is a client for the EmployeeConfig schema.
+type EmployeeConfigClient struct {
+	config
+}
+
+// NewEmployeeConfigClient returns a client for the EmployeeConfig from the given config.
+func NewEmployeeConfigClient(c config) *EmployeeConfigClient {
+	return &EmployeeConfigClient{config: c}
+}
+
+// Use adds a list of mutation hooks to the hooks stack.
+// A call to `Use(f, g, h)` equals to `employeeconfig.Hooks(f(g(h())))`.
+func (c *EmployeeConfigClient) Use(hooks ...Hook) {
+	c.hooks.EmployeeConfig = append(c.hooks.EmployeeConfig, hooks...)
+}
+
+// Intercept adds a list of query interceptors to the interceptors stack.
+// A call to `Intercept(f, g, h)` equals to `employeeconfig.Intercept(f(g(h())))`.
+func (c *EmployeeConfigClient) Intercept(interceptors ...Interceptor) {
+	c.inters.EmployeeConfig = append(c.inters.EmployeeConfig, interceptors...)
+}
+
+// Create returns a builder for creating a EmployeeConfig entity.
+func (c *EmployeeConfigClient) Create() *EmployeeConfigCreate {
+	mutation := newEmployeeConfigMutation(c.config, OpCreate)
+	return &EmployeeConfigCreate{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// CreateBulk returns a builder for creating a bulk of EmployeeConfig entities.
+func (c *EmployeeConfigClient) CreateBulk(builders ...*EmployeeConfigCreate) *EmployeeConfigCreateBulk {
+	return &EmployeeConfigCreateBulk{config: c.config, builders: builders}
+}
+
+// MapCreateBulk creates a bulk creation builder from the given slice. For each item in the slice, the function creates
+// a builder and applies setFunc on it.
+func (c *EmployeeConfigClient) MapCreateBulk(slice any, setFunc func(*EmployeeConfigCreate, int)) *EmployeeConfigCreateBulk {
+	rv := reflect.ValueOf(slice)
+	if rv.Kind() != reflect.Slice {
+		return &EmployeeConfigCreateBulk{err: fmt.Errorf("calling to EmployeeConfigClient.MapCreateBulk with wrong type %T, need slice", slice)}
+	}
+	builders := make([]*EmployeeConfigCreate, rv.Len())
+	for i := 0; i < rv.Len(); i++ {
+		builders[i] = c.Create()
+		setFunc(builders[i], i)
+	}
+	return &EmployeeConfigCreateBulk{config: c.config, builders: builders}
+}
+
+// Update returns an update builder for EmployeeConfig.
+func (c *EmployeeConfigClient) Update() *EmployeeConfigUpdate {
+	mutation := newEmployeeConfigMutation(c.config, OpUpdate)
+	return &EmployeeConfigUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// UpdateOne returns an update builder for the given entity.
+func (c *EmployeeConfigClient) UpdateOne(ec *EmployeeConfig) *EmployeeConfigUpdateOne {
+	mutation := newEmployeeConfigMutation(c.config, OpUpdateOne, withEmployeeConfig(ec))
+	return &EmployeeConfigUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// UpdateOneID returns an update builder for the given id.
+func (c *EmployeeConfigClient) UpdateOneID(id uint64) *EmployeeConfigUpdateOne {
+	mutation := newEmployeeConfigMutation(c.config, OpUpdateOne, withEmployeeConfigID(id))
+	return &EmployeeConfigUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// Delete returns a delete builder for EmployeeConfig.
+func (c *EmployeeConfigClient) Delete() *EmployeeConfigDelete {
+	mutation := newEmployeeConfigMutation(c.config, OpDelete)
+	return &EmployeeConfigDelete{config: c.config, hooks: c.Hooks(), mutation: mutation}
+}
+
+// DeleteOne returns a builder for deleting the given entity.
+func (c *EmployeeConfigClient) DeleteOne(ec *EmployeeConfig) *EmployeeConfigDeleteOne {
+	return c.DeleteOneID(ec.ID)
+}
+
+// DeleteOneID returns a builder for deleting the given entity by its id.
+func (c *EmployeeConfigClient) DeleteOneID(id uint64) *EmployeeConfigDeleteOne {
+	builder := c.Delete().Where(employeeconfig.ID(id))
+	builder.mutation.id = &id
+	builder.mutation.op = OpDeleteOne
+	return &EmployeeConfigDeleteOne{builder}
+}
+
+// Query returns a query builder for EmployeeConfig.
+func (c *EmployeeConfigClient) Query() *EmployeeConfigQuery {
+	return &EmployeeConfigQuery{
+		config: c.config,
+		ctx:    &QueryContext{Type: TypeEmployeeConfig},
+		inters: c.Interceptors(),
+	}
+}
+
+// Get returns a EmployeeConfig entity by its id.
+func (c *EmployeeConfigClient) Get(ctx context.Context, id uint64) (*EmployeeConfig, error) {
+	return c.Query().Where(employeeconfig.ID(id)).Only(ctx)
+}
+
+// GetX is like Get, but panics if an error occurs.
+func (c *EmployeeConfigClient) GetX(ctx context.Context, id uint64) *EmployeeConfig {
+	obj, err := c.Get(ctx, id)
+	if err != nil {
+		panic(err)
+	}
+	return obj
+}
+
+// Hooks returns the client hooks.
+func (c *EmployeeConfigClient) Hooks() []Hook {
+	hooks := c.hooks.EmployeeConfig
+	return append(hooks[:len(hooks):len(hooks)], employeeconfig.Hooks[:]...)
+}
+
+// Interceptors returns the client interceptors.
+func (c *EmployeeConfigClient) Interceptors() []Interceptor {
+	inters := c.inters.EmployeeConfig
+	return append(inters[:len(inters):len(inters)], employeeconfig.Interceptors[:]...)
+}
+
+func (c *EmployeeConfigClient) mutate(ctx context.Context, m *EmployeeConfigMutation) (Value, error) {
+	switch m.Op() {
+	case OpCreate:
+		return (&EmployeeConfigCreate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+	case OpUpdate:
+		return (&EmployeeConfigUpdate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+	case OpUpdateOne:
+		return (&EmployeeConfigUpdateOne{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+	case OpDelete, OpDeleteOne:
+		return (&EmployeeConfigDelete{config: c.config, hooks: c.Hooks(), mutation: m}).Exec(ctx)
+	default:
+		return nil, fmt.Errorf("ent: unknown EmployeeConfig mutation op: %q", m.Op())
+	}
+}
+
 // LabelClient is a client for the Label schema.
 type LabelClient struct {
 	config
@@ -2986,14 +3272,14 @@ func (c *WxClient) mutate(ctx context.Context, m *WxMutation) (Value, error) {
 // hooks and interceptors per client, for fast access.
 type (
 	hooks struct {
-		Agent, BatchMsg, Contact, Employee, Label, LabelRelationship, Message,
-		MessageRecords, Msg, Server, SopNode, SopStage, SopTask, Token, Tutorial,
-		WorkExperience, Wx []ent.Hook
+		Agent, BatchMsg, Category, Contact, Employee, EmployeeConfig, Label,
+		LabelRelationship, Message, MessageRecords, Msg, Server, SopNode, SopStage,
+		SopTask, Token, Tutorial, WorkExperience, Wx []ent.Hook
 	}
 	inters struct {
-		Agent, BatchMsg, Contact, Employee, Label, LabelRelationship, Message,
-		MessageRecords, Msg, Server, SopNode, SopStage, SopTask, Token, Tutorial,
-		WorkExperience, Wx []ent.Interceptor
+		Agent, BatchMsg, Category, Contact, Employee, EmployeeConfig, Label,
+		LabelRelationship, Message, MessageRecords, Msg, Server, SopNode, SopStage,
+		SopTask, Token, Tutorial, WorkExperience, Wx []ent.Interceptor
 	}
 )
 

ent/employee.go

@@ -51,6 +51,8 @@ type Employee struct {
 	VideoURL string `json:"video_url,omitempty"`
 	// organization_id | 租户ID
 	OrganizationID uint64 `json:"organization_id,omitempty"`
+	// category_id | 分类ID
+	CategoryID uint64 `json:"category_id,omitempty"`
 	// Edges holds the relations/edges for other nodes in the graph.
 	// The values are being populated by the EmployeeQuery when eager-loading is set.
 	Edges        EmployeeEdges `json:"edges"`
@@ -91,7 +93,7 @@ func (*Employee) scanValues(columns []string) ([]any, error) {
 	values := make([]any, len(columns))
 	for i := range columns {
 		switch columns[i] {
-		case employee.FieldID, employee.FieldHireCount, employee.FieldServiceCount, employee.FieldAchievementCount, employee.FieldOrganizationID:
+		case employee.FieldID, employee.FieldHireCount, employee.FieldServiceCount, employee.FieldAchievementCount, employee.FieldOrganizationID, employee.FieldCategoryID:
 			values[i] = new(sql.NullInt64)
 		case employee.FieldTitle, employee.FieldAvatar, employee.FieldTags, employee.FieldIntro, employee.FieldEstimate, employee.FieldSkill, employee.FieldAbilityType, employee.FieldScene, employee.FieldSwitchIn, employee.FieldVideoURL:
 			values[i] = new(sql.NullString)
@@ -220,6 +222,12 @@ func (e *Employee) assignValues(columns []string, values []any) error {
 			} else if value.Valid {
 				e.OrganizationID = uint64(value.Int64)
 			}
+		case employee.FieldCategoryID:
+			if value, ok := values[i].(*sql.NullInt64); !ok {
+				return fmt.Errorf("unexpected type %T for field category_id", values[i])
+			} else if value.Valid {
+				e.CategoryID = uint64(value.Int64)
+			}
 		default:
 			e.selectValues.Set(columns[i], values[i])
 		}
@@ -316,6 +324,9 @@ func (e *Employee) String() string {
 	builder.WriteString(", ")
 	builder.WriteString("organization_id=")
 	builder.WriteString(fmt.Sprintf("%v", e.OrganizationID))
+	builder.WriteString(", ")
+	builder.WriteString("category_id=")
+	builder.WriteString(fmt.Sprintf("%v", e.CategoryID))
 	builder.WriteByte(')')
 	return builder.String()
 }

ent/employee/employee.go

@@ -49,6 +49,8 @@ const (
 	FieldVideoURL = "video_url"
 	// FieldOrganizationID holds the string denoting the organization_id field in the database.
 	FieldOrganizationID = "organization_id"
+	// FieldCategoryID holds the string denoting the category_id field in the database.
+	FieldCategoryID = "category_id"
 	// EdgeEmWorkExperiences holds the string denoting the em_work_experiences edge name in mutations.
 	EdgeEmWorkExperiences = "em_work_experiences"
 	// EdgeEmTutorial holds the string denoting the em_tutorial edge name in mutations.
@@ -91,6 +93,7 @@ var Columns = []string{
 	FieldSwitchIn,
 	FieldVideoURL,
 	FieldOrganizationID,
+	FieldCategoryID,
 }
 
 // ValidColumn reports if the column name is valid (part of the table columns).
@@ -159,6 +162,8 @@ var (
 	VideoURLValidator func(string) error
 	// OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
 	OrganizationIDValidator func(uint64) error
+	// CategoryIDValidator is a validator for the "category_id" field. It is called by the builders before save.
+	CategoryIDValidator func(uint64) error
 )
 
 // OrderOption defines the ordering options for the Employee queries.
@@ -254,6 +259,11 @@ func ByOrganizationID(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldOrganizationID, opts...).ToFunc()
 }
 
+// ByCategoryID orders the results by the category_id field.
+func ByCategoryID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldCategoryID, opts...).ToFunc()
+}
+
 // ByEmWorkExperiencesCount orders the results by em_work_experiences count.
 func ByEmWorkExperiencesCount(opts ...sql.OrderTermOption) OrderOption {
 	return func(s *sql.Selector) {

ent/employee/where.go

@@ -140,6 +140,11 @@ func OrganizationID(v uint64) predicate.Employee {
 	return predicate.Employee(sql.FieldEQ(FieldOrganizationID, v))
 }
 
+// CategoryID applies equality check predicate on the "category_id" field. It's identical to CategoryIDEQ.
+func CategoryID(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldEQ(FieldCategoryID, v))
+}
+
 // CreatedAtEQ applies the EQ predicate on the "created_at" field.
 func CreatedAtEQ(v time.Time) predicate.Employee {
 	return predicate.Employee(sql.FieldEQ(FieldCreatedAt, v))
@@ -1080,6 +1085,46 @@ func OrganizationIDLTE(v uint64) predicate.Employee {
 	return predicate.Employee(sql.FieldLTE(FieldOrganizationID, v))
 }
 
+// CategoryIDEQ applies the EQ predicate on the "category_id" field.
+func CategoryIDEQ(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldEQ(FieldCategoryID, v))
+}
+
+// CategoryIDNEQ applies the NEQ predicate on the "category_id" field.
+func CategoryIDNEQ(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldNEQ(FieldCategoryID, v))
+}
+
+// CategoryIDIn applies the In predicate on the "category_id" field.
+func CategoryIDIn(vs ...uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldIn(FieldCategoryID, vs...))
+}
+
+// CategoryIDNotIn applies the NotIn predicate on the "category_id" field.
+func CategoryIDNotIn(vs ...uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldNotIn(FieldCategoryID, vs...))
+}
+
+// CategoryIDGT applies the GT predicate on the "category_id" field.
+func CategoryIDGT(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldGT(FieldCategoryID, v))
+}
+
+// CategoryIDGTE applies the GTE predicate on the "category_id" field.
+func CategoryIDGTE(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldGTE(FieldCategoryID, v))
+}
+
+// CategoryIDLT applies the LT predicate on the "category_id" field.
+func CategoryIDLT(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldLT(FieldCategoryID, v))
+}
+
+// CategoryIDLTE applies the LTE predicate on the "category_id" field.
+func CategoryIDLTE(v uint64) predicate.Employee {
+	return predicate.Employee(sql.FieldLTE(FieldCategoryID, v))
+}
+
 // HasEmWorkExperiences applies the HasEdge predicate on the "em_work_experiences" edge.
 func HasEmWorkExperiences() predicate.Employee {
 	return predicate.Employee(func(s *sql.Selector) {

ent/employee_create.go

@@ -230,6 +230,12 @@ func (ec *EmployeeCreate) SetOrganizationID(u uint64) *EmployeeCreate {
 	return ec
 }
 
+// SetCategoryID sets the "category_id" field.
+func (ec *EmployeeCreate) SetCategoryID(u uint64) *EmployeeCreate {
+	ec.mutation.SetCategoryID(u)
+	return ec
+}
+
 // SetID sets the "id" field.
 func (ec *EmployeeCreate) SetID(u uint64) *EmployeeCreate {
 	ec.mutation.SetID(u)
@@ -465,6 +471,14 @@ func (ec *EmployeeCreate) check() error {
 			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Employee.organization_id": %w`, err)}
 		}
 	}
+	if _, ok := ec.mutation.CategoryID(); !ok {
+		return &ValidationError{Name: "category_id", err: errors.New(`ent: missing required field "Employee.category_id"`)}
+	}
+	if v, ok := ec.mutation.CategoryID(); ok {
+		if err := employee.CategoryIDValidator(v); err != nil {
+			return &ValidationError{Name: "category_id", err: fmt.Errorf(`ent: validator failed for field "Employee.category_id": %w`, err)}
+		}
+	}
 	return nil
 }
 
@@ -566,6 +580,10 @@ func (ec *EmployeeCreate) createSpec() (*Employee, *sqlgraph.CreateSpec) {
 		_spec.SetField(employee.FieldOrganizationID, field.TypeUint64, value)
 		_node.OrganizationID = value
 	}
+	if value, ok := ec.mutation.CategoryID(); ok {
+		_spec.SetField(employee.FieldCategoryID, field.TypeUint64, value)
+		_node.CategoryID = value
+	}
 	if nodes := ec.mutation.EmWorkExperiencesIDs(); len(nodes) > 0 {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,
@@ -872,6 +890,24 @@ func (u *EmployeeUpsert) AddOrganizationID(v uint64) *EmployeeUpsert {
 	return u
 }
 
+// SetCategoryID sets the "category_id" field.
+func (u *EmployeeUpsert) SetCategoryID(v uint64) *EmployeeUpsert {
+	u.Set(employee.FieldCategoryID, v)
+	return u
+}
+
+// UpdateCategoryID sets the "category_id" field to the value that was provided on create.
+func (u *EmployeeUpsert) UpdateCategoryID() *EmployeeUpsert {
+	u.SetExcluded(employee.FieldCategoryID)
+	return u
+}
+
+// AddCategoryID adds v to the "category_id" field.
+func (u *EmployeeUpsert) AddCategoryID(v uint64) *EmployeeUpsert {
+	u.Add(employee.FieldCategoryID, v)
+	return u
+}
+
 // UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
 // Using this option is equivalent to using:
 //
@@ -1182,6 +1218,27 @@ func (u *EmployeeUpsertOne) UpdateOrganizationID() *EmployeeUpsertOne {
 	})
 }
 
+// SetCategoryID sets the "category_id" field.
+func (u *EmployeeUpsertOne) SetCategoryID(v uint64) *EmployeeUpsertOne {
+	return u.Update(func(s *EmployeeUpsert) {
+		s.SetCategoryID(v)
+	})
+}
+
+// AddCategoryID adds v to the "category_id" field.
+func (u *EmployeeUpsertOne) AddCategoryID(v uint64) *EmployeeUpsertOne {
+	return u.Update(func(s *EmployeeUpsert) {
+		s.AddCategoryID(v)
+	})
+}
+
+// UpdateCategoryID sets the "category_id" field to the value that was provided on create.
+func (u *EmployeeUpsertOne) UpdateCategoryID() *EmployeeUpsertOne {
+	return u.Update(func(s *EmployeeUpsert) {
+		s.UpdateCategoryID()
+	})
+}
+
 // Exec executes the query.
 func (u *EmployeeUpsertOne) Exec(ctx context.Context) error {
 	if len(u.create.conflict) == 0 {
@@ -1658,6 +1715,27 @@ func (u *EmployeeUpsertBulk) UpdateOrganizationID() *EmployeeUpsertBulk {
 	})
 }
 
+// SetCategoryID sets the "category_id" field.
+func (u *EmployeeUpsertBulk) SetCategoryID(v uint64) *EmployeeUpsertBulk {
+	return u.Update(func(s *EmployeeUpsert) {
+		s.SetCategoryID(v)
+	})
+}
+
+// AddCategoryID adds v to the "category_id" field.
+func (u *EmployeeUpsertBulk) AddCategoryID(v uint64) *EmployeeUpsertBulk {
+	return u.Update(func(s *EmployeeUpsert) {
+		s.AddCategoryID(v)
+	})
+}
+
+// UpdateCategoryID sets the "category_id" field to the value that was provided on create.
+func (u *EmployeeUpsertBulk) UpdateCategoryID() *EmployeeUpsertBulk {
+	return u.Update(func(s *EmployeeUpsert) {
+		s.UpdateCategoryID()
+	})
+}
+
 // Exec executes the query.
 func (u *EmployeeUpsertBulk) Exec(ctx context.Context) error {
 	if u.create.err != nil {

ent/employee_update.go

@@ -280,6 +280,27 @@ func (eu *EmployeeUpdate) AddOrganizationID(u int64) *EmployeeUpdate {
 	return eu
 }
 
+// SetCategoryID sets the "category_id" field.
+func (eu *EmployeeUpdate) SetCategoryID(u uint64) *EmployeeUpdate {
+	eu.mutation.ResetCategoryID()
+	eu.mutation.SetCategoryID(u)
+	return eu
+}
+
+// SetNillableCategoryID sets the "category_id" field if the given value is not nil.
+func (eu *EmployeeUpdate) SetNillableCategoryID(u *uint64) *EmployeeUpdate {
+	if u != nil {
+		eu.SetCategoryID(*u)
+	}
+	return eu
+}
+
+// AddCategoryID adds u to the "category_id" field.
+func (eu *EmployeeUpdate) AddCategoryID(u int64) *EmployeeUpdate {
+	eu.mutation.AddCategoryID(u)
+	return eu
+}
+
 // AddEmWorkExperienceIDs adds the "em_work_experiences" edge to the WorkExperience entity by IDs.
 func (eu *EmployeeUpdate) AddEmWorkExperienceIDs(ids ...uint64) *EmployeeUpdate {
 	eu.mutation.AddEmWorkExperienceIDs(ids...)
@@ -456,6 +477,11 @@ func (eu *EmployeeUpdate) check() error {
 			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Employee.organization_id": %w`, err)}
 		}
 	}
+	if v, ok := eu.mutation.CategoryID(); ok {
+		if err := employee.CategoryIDValidator(v); err != nil {
+			return &ValidationError{Name: "category_id", err: fmt.Errorf(`ent: validator failed for field "Employee.category_id": %w`, err)}
+		}
+	}
 	return nil
 }
 
@@ -534,6 +560,12 @@ func (eu *EmployeeUpdate) sqlSave(ctx context.Context) (n int, err error) {
 	if value, ok := eu.mutation.AddedOrganizationID(); ok {
 		_spec.AddField(employee.FieldOrganizationID, field.TypeUint64, value)
 	}
+	if value, ok := eu.mutation.CategoryID(); ok {
+		_spec.SetField(employee.FieldCategoryID, field.TypeUint64, value)
+	}
+	if value, ok := eu.mutation.AddedCategoryID(); ok {
+		_spec.AddField(employee.FieldCategoryID, field.TypeUint64, value)
+	}
 	if eu.mutation.EmWorkExperiencesCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,
@@ -894,6 +926,27 @@ func (euo *EmployeeUpdateOne) AddOrganizationID(u int64) *EmployeeUpdateOne {
 	return euo
 }
 
+// SetCategoryID sets the "category_id" field.
+func (euo *EmployeeUpdateOne) SetCategoryID(u uint64) *EmployeeUpdateOne {
+	euo.mutation.ResetCategoryID()
+	euo.mutation.SetCategoryID(u)
+	return euo
+}
+
+// SetNillableCategoryID sets the "category_id" field if the given value is not nil.
+func (euo *EmployeeUpdateOne) SetNillableCategoryID(u *uint64) *EmployeeUpdateOne {
+	if u != nil {
+		euo.SetCategoryID(*u)
+	}
+	return euo
+}
+
+// AddCategoryID adds u to the "category_id" field.
+func (euo *EmployeeUpdateOne) AddCategoryID(u int64) *EmployeeUpdateOne {
+	euo.mutation.AddCategoryID(u)
+	return euo
+}
+
 // AddEmWorkExperienceIDs adds the "em_work_experiences" edge to the WorkExperience entity by IDs.
 func (euo *EmployeeUpdateOne) AddEmWorkExperienceIDs(ids ...uint64) *EmployeeUpdateOne {
 	euo.mutation.AddEmWorkExperienceIDs(ids...)
@@ -1083,6 +1136,11 @@ func (euo *EmployeeUpdateOne) check() error {
 			return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "Employee.organization_id": %w`, err)}
 		}
 	}
+	if v, ok := euo.mutation.CategoryID(); ok {
+		if err := employee.CategoryIDValidator(v); err != nil {
+			return &ValidationError{Name: "category_id", err: fmt.Errorf(`ent: validator failed for field "Employee.category_id": %w`, err)}
+		}
+	}
 	return nil
 }
 
@@ -1178,6 +1236,12 @@ func (euo *EmployeeUpdateOne) sqlSave(ctx context.Context) (_node *Employee, err
 	if value, ok := euo.mutation.AddedOrganizationID(); ok {
 		_spec.AddField(employee.FieldOrganizationID, field.TypeUint64, value)
 	}
+	if value, ok := euo.mutation.CategoryID(); ok {
+		_spec.SetField(employee.FieldCategoryID, field.TypeUint64, value)
+	}
+	if value, ok := euo.mutation.AddedCategoryID(); ok {
+		_spec.AddField(employee.FieldCategoryID, field.TypeUint64, value)
+	}
 	if euo.mutation.EmWorkExperiencesCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,

ent/employeeconfig.go

@@ -0,0 +1,172 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"fmt"
+	"strings"
+	"time"
+	"wechat-api/ent/employeeconfig"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/sql"
+)
+
+// EmployeeConfig is the model entity for the EmployeeConfig schema.
+type EmployeeConfig struct {
+	config `json:"-"`
+	// ID of the ent.
+	ID uint64 `json:"id,omitempty"`
+	// Create Time | 创建日期
+	CreatedAt time.Time `json:"created_at,omitempty"`
+	// Update Time | 修改日期
+	UpdatedAt time.Time `json:"updated_at,omitempty"`
+	// Delete Time | 删除日期
+	DeletedAt time.Time `json:"deleted_at,omitempty"`
+	// 类型：scene-场景 switch_in-接入方式
+	Stype string `json:"stype,omitempty"`
+	// 标题
+	Title string `json:"title,omitempty"`
+	// 图片地址
+	Photo string `json:"photo,omitempty"`
+	// 机构 ID
+	OrganizationID uint64 `json:"organization_id,omitempty"`
+	selectValues   sql.SelectValues
+}
+
+// scanValues returns the types for scanning values from sql.Rows.
+func (*EmployeeConfig) scanValues(columns []string) ([]any, error) {
+	values := make([]any, len(columns))
+	for i := range columns {
+		switch columns[i] {
+		case employeeconfig.FieldID, employeeconfig.FieldOrganizationID:
+			values[i] = new(sql.NullInt64)
+		case employeeconfig.FieldStype, employeeconfig.FieldTitle, employeeconfig.FieldPhoto:
+			values[i] = new(sql.NullString)
+		case employeeconfig.FieldCreatedAt, employeeconfig.FieldUpdatedAt, employeeconfig.FieldDeletedAt:
+			values[i] = new(sql.NullTime)
+		default:
+			values[i] = new(sql.UnknownType)
+		}
+	}
+	return values, nil
+}
+
+// assignValues assigns the values that were returned from sql.Rows (after scanning)
+// to the EmployeeConfig fields.
+func (ec *EmployeeConfig) assignValues(columns []string, values []any) error {
+	if m, n := len(values), len(columns); m < n {
+		return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
+	}
+	for i := range columns {
+		switch columns[i] {
+		case employeeconfig.FieldID:
+			value, ok := values[i].(*sql.NullInt64)
+			if !ok {
+				return fmt.Errorf("unexpected type %T for field id", value)
+			}
+			ec.ID = uint64(value.Int64)
+		case employeeconfig.FieldCreatedAt:
+			if value, ok := values[i].(*sql.NullTime); !ok {
+				return fmt.Errorf("unexpected type %T for field created_at", values[i])
+			} else if value.Valid {
+				ec.CreatedAt = value.Time
+			}
+		case employeeconfig.FieldUpdatedAt:
+			if value, ok := values[i].(*sql.NullTime); !ok {
+				return fmt.Errorf("unexpected type %T for field updated_at", values[i])
+			} else if value.Valid {
+				ec.UpdatedAt = value.Time
+			}
+		case employeeconfig.FieldDeletedAt:
+			if value, ok := values[i].(*sql.NullTime); !ok {
+				return fmt.Errorf("unexpected type %T for field deleted_at", values[i])
+			} else if value.Valid {
+				ec.DeletedAt = value.Time
+			}
+		case employeeconfig.FieldStype:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field stype", values[i])
+			} else if value.Valid {
+				ec.Stype = value.String
+			}
+		case employeeconfig.FieldTitle:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field title", values[i])
+			} else if value.Valid {
+				ec.Title = value.String
+			}
+		case employeeconfig.FieldPhoto:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field photo", values[i])
+			} else if value.Valid {
+				ec.Photo = value.String
+			}
+		case employeeconfig.FieldOrganizationID:
+			if value, ok := values[i].(*sql.NullInt64); !ok {
+				return fmt.Errorf("unexpected type %T for field organization_id", values[i])
+			} else if value.Valid {
+				ec.OrganizationID = uint64(value.Int64)
+			}
+		default:
+			ec.selectValues.Set(columns[i], values[i])
+		}
+	}
+	return nil
+}
+
+// Value returns the ent.Value that was dynamically selected and assigned to the EmployeeConfig.
+// This includes values selected through modifiers, order, etc.
+func (ec *EmployeeConfig) Value(name string) (ent.Value, error) {
+	return ec.selectValues.Get(name)
+}
+
+// Update returns a builder for updating this EmployeeConfig.
+// Note that you need to call EmployeeConfig.Unwrap() before calling this method if this EmployeeConfig
+// was returned from a transaction, and the transaction was committed or rolled back.
+func (ec *EmployeeConfig) Update() *EmployeeConfigUpdateOne {
+	return NewEmployeeConfigClient(ec.config).UpdateOne(ec)
+}
+
+// Unwrap unwraps the EmployeeConfig entity that was returned from a transaction after it was closed,
+// so that all future queries will be executed through the driver which created the transaction.
+func (ec *EmployeeConfig) Unwrap() *EmployeeConfig {
+	_tx, ok := ec.config.driver.(*txDriver)
+	if !ok {
+		panic("ent: EmployeeConfig is not a transactional entity")
+	}
+	ec.config.driver = _tx.drv
+	return ec
+}
+
+// String implements the fmt.Stringer.
+func (ec *EmployeeConfig) String() string {
+	var builder strings.Builder
+	builder.WriteString("EmployeeConfig(")
+	builder.WriteString(fmt.Sprintf("id=%v, ", ec.ID))
+	builder.WriteString("created_at=")
+	builder.WriteString(ec.CreatedAt.Format(time.ANSIC))
+	builder.WriteString(", ")
+	builder.WriteString("updated_at=")
+	builder.WriteString(ec.UpdatedAt.Format(time.ANSIC))
+	builder.WriteString(", ")
+	builder.WriteString("deleted_at=")
+	builder.WriteString(ec.DeletedAt.Format(time.ANSIC))
+	builder.WriteString(", ")
+	builder.WriteString("stype=")
+	builder.WriteString(ec.Stype)
+	builder.WriteString(", ")
+	builder.WriteString("title=")
+	builder.WriteString(ec.Title)
+	builder.WriteString(", ")
+	builder.WriteString("photo=")
+	builder.WriteString(ec.Photo)
+	builder.WriteString(", ")
+	builder.WriteString("organization_id=")
+	builder.WriteString(fmt.Sprintf("%v", ec.OrganizationID))
+	builder.WriteByte(')')
+	return builder.String()
+}
+
+// EmployeeConfigs is a parsable slice of EmployeeConfig.
+type EmployeeConfigs []*EmployeeConfig

ent/employeeconfig/employeeconfig.go

@@ -0,0 +1,122 @@
+// Code generated by ent, DO NOT EDIT.
+
+package employeeconfig
+
+import (
+	"time"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/sql"
+)
+
+const (
+	// Label holds the string label denoting the employeeconfig type in the database.
+	Label = "employee_config"
+	// FieldID holds the string denoting the id field in the database.
+	FieldID = "id"
+	// FieldCreatedAt holds the string denoting the created_at field in the database.
+	FieldCreatedAt = "created_at"
+	// FieldUpdatedAt holds the string denoting the updated_at field in the database.
+	FieldUpdatedAt = "updated_at"
+	// FieldDeletedAt holds the string denoting the deleted_at field in the database.
+	FieldDeletedAt = "deleted_at"
+	// FieldStype holds the string denoting the stype field in the database.
+	FieldStype = "stype"
+	// FieldTitle holds the string denoting the title field in the database.
+	FieldTitle = "title"
+	// FieldPhoto holds the string denoting the photo field in the database.
+	FieldPhoto = "photo"
+	// FieldOrganizationID holds the string denoting the organization_id field in the database.
+	FieldOrganizationID = "organization_id"
+	// Table holds the table name of the employeeconfig in the database.
+	Table = "employee_config"
+)
+
+// Columns holds all SQL columns for employeeconfig fields.
+var Columns = []string{
+	FieldID,
+	FieldCreatedAt,
+	FieldUpdatedAt,
+	FieldDeletedAt,
+	FieldStype,
+	FieldTitle,
+	FieldPhoto,
+	FieldOrganizationID,
+}
+
+// ValidColumn reports if the column name is valid (part of the table columns).
+func ValidColumn(column string) bool {
+	for i := range Columns {
+		if column == Columns[i] {
+			return true
+		}
+	}
+	return false
+}
+
+// Note that the variables below are initialized by the runtime
+// package on the initialization of the application. Therefore,
+// it should be imported in the main as follows:
+//
+//	import _ "wechat-api/ent/runtime"
+var (
+	Hooks        [1]ent.Hook
+	Interceptors [1]ent.Interceptor
+	// DefaultCreatedAt holds the default value on creation for the "created_at" field.
+	DefaultCreatedAt func() time.Time
+	// DefaultUpdatedAt holds the default value on creation for the "updated_at" field.
+	DefaultUpdatedAt func() time.Time
+	// UpdateDefaultUpdatedAt holds the default value on update for the "updated_at" field.
+	UpdateDefaultUpdatedAt func() time.Time
+	// DefaultStype holds the default value on creation for the "stype" field.
+	DefaultStype string
+	// DefaultTitle holds the default value on creation for the "title" field.
+	DefaultTitle string
+	// DefaultPhoto holds the default value on creation for the "photo" field.
+	DefaultPhoto string
+	// DefaultOrganizationID holds the default value on creation for the "organization_id" field.
+	DefaultOrganizationID uint64
+)
+
+// OrderOption defines the ordering options for the EmployeeConfig queries.
+type OrderOption func(*sql.Selector)
+
+// ByID orders the results by the id field.
+func ByID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldID, opts...).ToFunc()
+}
+
+// ByCreatedAt orders the results by the created_at field.
+func ByCreatedAt(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldCreatedAt, opts...).ToFunc()
+}
+
+// ByUpdatedAt orders the results by the updated_at field.
+func ByUpdatedAt(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldUpdatedAt, opts...).ToFunc()
+}
+
+// ByDeletedAt orders the results by the deleted_at field.
+func ByDeletedAt(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldDeletedAt, opts...).ToFunc()
+}
+
+// ByStype orders the results by the stype field.
+func ByStype(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldStype, opts...).ToFunc()
+}
+
+// ByTitle orders the results by the title field.
+func ByTitle(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldTitle, opts...).ToFunc()
+}
+
+// ByPhoto orders the results by the photo field.
+func ByPhoto(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldPhoto, opts...).ToFunc()
+}
+
+// ByOrganizationID orders the results by the organization_id field.
+func ByOrganizationID(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldOrganizationID, opts...).ToFunc()
+}

ent/employeeconfig/where.go

@@ -0,0 +1,480 @@
+// Code generated by ent, DO NOT EDIT.
+
+package employeeconfig
+
+import (
+	"time"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+)
+
+// ID filters vertices based on their ID field.
+func ID(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldID, id))
+}
+
+// IDEQ applies the EQ predicate on the ID field.
+func IDEQ(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldID, id))
+}
+
+// IDNEQ applies the NEQ predicate on the ID field.
+func IDNEQ(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldID, id))
+}
+
+// IDIn applies the In predicate on the ID field.
+func IDIn(ids ...uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldID, ids...))
+}
+
+// IDNotIn applies the NotIn predicate on the ID field.
+func IDNotIn(ids ...uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldID, ids...))
+}
+
+// IDGT applies the GT predicate on the ID field.
+func IDGT(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldID, id))
+}
+
+// IDGTE applies the GTE predicate on the ID field.
+func IDGTE(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldID, id))
+}
+
+// IDLT applies the LT predicate on the ID field.
+func IDLT(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldID, id))
+}
+
+// IDLTE applies the LTE predicate on the ID field.
+func IDLTE(id uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldID, id))
+}
+
+// CreatedAt applies equality check predicate on the "created_at" field. It's identical to CreatedAtEQ.
+func CreatedAt(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldCreatedAt, v))
+}
+
+// UpdatedAt applies equality check predicate on the "updated_at" field. It's identical to UpdatedAtEQ.
+func UpdatedAt(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldUpdatedAt, v))
+}
+
+// DeletedAt applies equality check predicate on the "deleted_at" field. It's identical to DeletedAtEQ.
+func DeletedAt(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldDeletedAt, v))
+}
+
+// Stype applies equality check predicate on the "stype" field. It's identical to StypeEQ.
+func Stype(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldStype, v))
+}
+
+// Title applies equality check predicate on the "title" field. It's identical to TitleEQ.
+func Title(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldTitle, v))
+}
+
+// Photo applies equality check predicate on the "photo" field. It's identical to PhotoEQ.
+func Photo(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldPhoto, v))
+}
+
+// OrganizationID applies equality check predicate on the "organization_id" field. It's identical to OrganizationIDEQ.
+func OrganizationID(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldOrganizationID, v))
+}
+
+// CreatedAtEQ applies the EQ predicate on the "created_at" field.
+func CreatedAtEQ(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldCreatedAt, v))
+}
+
+// CreatedAtNEQ applies the NEQ predicate on the "created_at" field.
+func CreatedAtNEQ(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldCreatedAt, v))
+}
+
+// CreatedAtIn applies the In predicate on the "created_at" field.
+func CreatedAtIn(vs ...time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldCreatedAt, vs...))
+}
+
+// CreatedAtNotIn applies the NotIn predicate on the "created_at" field.
+func CreatedAtNotIn(vs ...time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldCreatedAt, vs...))
+}
+
+// CreatedAtGT applies the GT predicate on the "created_at" field.
+func CreatedAtGT(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldCreatedAt, v))
+}
+
+// CreatedAtGTE applies the GTE predicate on the "created_at" field.
+func CreatedAtGTE(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldCreatedAt, v))
+}
+
+// CreatedAtLT applies the LT predicate on the "created_at" field.
+func CreatedAtLT(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldCreatedAt, v))
+}
+
+// CreatedAtLTE applies the LTE predicate on the "created_at" field.
+func CreatedAtLTE(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldCreatedAt, v))
+}
+
+// UpdatedAtEQ applies the EQ predicate on the "updated_at" field.
+func UpdatedAtEQ(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldUpdatedAt, v))
+}
+
+// UpdatedAtNEQ applies the NEQ predicate on the "updated_at" field.
+func UpdatedAtNEQ(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldUpdatedAt, v))
+}
+
+// UpdatedAtIn applies the In predicate on the "updated_at" field.
+func UpdatedAtIn(vs ...time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldUpdatedAt, vs...))
+}
+
+// UpdatedAtNotIn applies the NotIn predicate on the "updated_at" field.
+func UpdatedAtNotIn(vs ...time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldUpdatedAt, vs...))
+}
+
+// UpdatedAtGT applies the GT predicate on the "updated_at" field.
+func UpdatedAtGT(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldUpdatedAt, v))
+}
+
+// UpdatedAtGTE applies the GTE predicate on the "updated_at" field.
+func UpdatedAtGTE(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldUpdatedAt, v))
+}
+
+// UpdatedAtLT applies the LT predicate on the "updated_at" field.
+func UpdatedAtLT(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldUpdatedAt, v))
+}
+
+// UpdatedAtLTE applies the LTE predicate on the "updated_at" field.
+func UpdatedAtLTE(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldUpdatedAt, v))
+}
+
+// DeletedAtEQ applies the EQ predicate on the "deleted_at" field.
+func DeletedAtEQ(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldDeletedAt, v))
+}
+
+// DeletedAtNEQ applies the NEQ predicate on the "deleted_at" field.
+func DeletedAtNEQ(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldDeletedAt, v))
+}
+
+// DeletedAtIn applies the In predicate on the "deleted_at" field.
+func DeletedAtIn(vs ...time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldDeletedAt, vs...))
+}
+
+// DeletedAtNotIn applies the NotIn predicate on the "deleted_at" field.
+func DeletedAtNotIn(vs ...time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldDeletedAt, vs...))
+}
+
+// DeletedAtGT applies the GT predicate on the "deleted_at" field.
+func DeletedAtGT(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldDeletedAt, v))
+}
+
+// DeletedAtGTE applies the GTE predicate on the "deleted_at" field.
+func DeletedAtGTE(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldDeletedAt, v))
+}
+
+// DeletedAtLT applies the LT predicate on the "deleted_at" field.
+func DeletedAtLT(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldDeletedAt, v))
+}
+
+// DeletedAtLTE applies the LTE predicate on the "deleted_at" field.
+func DeletedAtLTE(v time.Time) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldDeletedAt, v))
+}
+
+// DeletedAtIsNil applies the IsNil predicate on the "deleted_at" field.
+func DeletedAtIsNil() predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIsNull(FieldDeletedAt))
+}
+
+// DeletedAtNotNil applies the NotNil predicate on the "deleted_at" field.
+func DeletedAtNotNil() predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotNull(FieldDeletedAt))
+}
+
+// StypeEQ applies the EQ predicate on the "stype" field.
+func StypeEQ(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldStype, v))
+}
+
+// StypeNEQ applies the NEQ predicate on the "stype" field.
+func StypeNEQ(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldStype, v))
+}
+
+// StypeIn applies the In predicate on the "stype" field.
+func StypeIn(vs ...string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldStype, vs...))
+}
+
+// StypeNotIn applies the NotIn predicate on the "stype" field.
+func StypeNotIn(vs ...string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldStype, vs...))
+}
+
+// StypeGT applies the GT predicate on the "stype" field.
+func StypeGT(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldStype, v))
+}
+
+// StypeGTE applies the GTE predicate on the "stype" field.
+func StypeGTE(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldStype, v))
+}
+
+// StypeLT applies the LT predicate on the "stype" field.
+func StypeLT(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldStype, v))
+}
+
+// StypeLTE applies the LTE predicate on the "stype" field.
+func StypeLTE(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldStype, v))
+}
+
+// StypeContains applies the Contains predicate on the "stype" field.
+func StypeContains(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldContains(FieldStype, v))
+}
+
+// StypeHasPrefix applies the HasPrefix predicate on the "stype" field.
+func StypeHasPrefix(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldHasPrefix(FieldStype, v))
+}
+
+// StypeHasSuffix applies the HasSuffix predicate on the "stype" field.
+func StypeHasSuffix(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldHasSuffix(FieldStype, v))
+}
+
+// StypeEqualFold applies the EqualFold predicate on the "stype" field.
+func StypeEqualFold(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEqualFold(FieldStype, v))
+}
+
+// StypeContainsFold applies the ContainsFold predicate on the "stype" field.
+func StypeContainsFold(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldContainsFold(FieldStype, v))
+}
+
+// TitleEQ applies the EQ predicate on the "title" field.
+func TitleEQ(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldTitle, v))
+}
+
+// TitleNEQ applies the NEQ predicate on the "title" field.
+func TitleNEQ(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldTitle, v))
+}
+
+// TitleIn applies the In predicate on the "title" field.
+func TitleIn(vs ...string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldTitle, vs...))
+}
+
+// TitleNotIn applies the NotIn predicate on the "title" field.
+func TitleNotIn(vs ...string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldTitle, vs...))
+}
+
+// TitleGT applies the GT predicate on the "title" field.
+func TitleGT(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldTitle, v))
+}
+
+// TitleGTE applies the GTE predicate on the "title" field.
+func TitleGTE(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldTitle, v))
+}
+
+// TitleLT applies the LT predicate on the "title" field.
+func TitleLT(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldTitle, v))
+}
+
+// TitleLTE applies the LTE predicate on the "title" field.
+func TitleLTE(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldTitle, v))
+}
+
+// TitleContains applies the Contains predicate on the "title" field.
+func TitleContains(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldContains(FieldTitle, v))
+}
+
+// TitleHasPrefix applies the HasPrefix predicate on the "title" field.
+func TitleHasPrefix(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldHasPrefix(FieldTitle, v))
+}
+
+// TitleHasSuffix applies the HasSuffix predicate on the "title" field.
+func TitleHasSuffix(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldHasSuffix(FieldTitle, v))
+}
+
+// TitleEqualFold applies the EqualFold predicate on the "title" field.
+func TitleEqualFold(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEqualFold(FieldTitle, v))
+}
+
+// TitleContainsFold applies the ContainsFold predicate on the "title" field.
+func TitleContainsFold(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldContainsFold(FieldTitle, v))
+}
+
+// PhotoEQ applies the EQ predicate on the "photo" field.
+func PhotoEQ(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldPhoto, v))
+}
+
+// PhotoNEQ applies the NEQ predicate on the "photo" field.
+func PhotoNEQ(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldPhoto, v))
+}
+
+// PhotoIn applies the In predicate on the "photo" field.
+func PhotoIn(vs ...string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldPhoto, vs...))
+}
+
+// PhotoNotIn applies the NotIn predicate on the "photo" field.
+func PhotoNotIn(vs ...string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldPhoto, vs...))
+}
+
+// PhotoGT applies the GT predicate on the "photo" field.
+func PhotoGT(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldPhoto, v))
+}
+
+// PhotoGTE applies the GTE predicate on the "photo" field.
+func PhotoGTE(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldPhoto, v))
+}
+
+// PhotoLT applies the LT predicate on the "photo" field.
+func PhotoLT(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldPhoto, v))
+}
+
+// PhotoLTE applies the LTE predicate on the "photo" field.
+func PhotoLTE(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldPhoto, v))
+}
+
+// PhotoContains applies the Contains predicate on the "photo" field.
+func PhotoContains(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldContains(FieldPhoto, v))
+}
+
+// PhotoHasPrefix applies the HasPrefix predicate on the "photo" field.
+func PhotoHasPrefix(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldHasPrefix(FieldPhoto, v))
+}
+
+// PhotoHasSuffix applies the HasSuffix predicate on the "photo" field.
+func PhotoHasSuffix(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldHasSuffix(FieldPhoto, v))
+}
+
+// PhotoEqualFold applies the EqualFold predicate on the "photo" field.
+func PhotoEqualFold(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEqualFold(FieldPhoto, v))
+}
+
+// PhotoContainsFold applies the ContainsFold predicate on the "photo" field.
+func PhotoContainsFold(v string) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldContainsFold(FieldPhoto, v))
+}
+
+// OrganizationIDEQ applies the EQ predicate on the "organization_id" field.
+func OrganizationIDEQ(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldEQ(FieldOrganizationID, v))
+}
+
+// OrganizationIDNEQ applies the NEQ predicate on the "organization_id" field.
+func OrganizationIDNEQ(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNEQ(FieldOrganizationID, v))
+}
+
+// OrganizationIDIn applies the In predicate on the "organization_id" field.
+func OrganizationIDIn(vs ...uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIn(FieldOrganizationID, vs...))
+}
+
+// OrganizationIDNotIn applies the NotIn predicate on the "organization_id" field.
+func OrganizationIDNotIn(vs ...uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotIn(FieldOrganizationID, vs...))
+}
+
+// OrganizationIDGT applies the GT predicate on the "organization_id" field.
+func OrganizationIDGT(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGT(FieldOrganizationID, v))
+}
+
+// OrganizationIDGTE applies the GTE predicate on the "organization_id" field.
+func OrganizationIDGTE(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldGTE(FieldOrganizationID, v))
+}
+
+// OrganizationIDLT applies the LT predicate on the "organization_id" field.
+func OrganizationIDLT(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLT(FieldOrganizationID, v))
+}
+
+// OrganizationIDLTE applies the LTE predicate on the "organization_id" field.
+func OrganizationIDLTE(v uint64) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldLTE(FieldOrganizationID, v))
+}
+
+// OrganizationIDIsNil applies the IsNil predicate on the "organization_id" field.
+func OrganizationIDIsNil() predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldIsNull(FieldOrganizationID))
+}
+
+// OrganizationIDNotNil applies the NotNil predicate on the "organization_id" field.
+func OrganizationIDNotNil() predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.FieldNotNull(FieldOrganizationID))
+}
+
+// And groups predicates with the AND operator between them.
+func And(predicates ...predicate.EmployeeConfig) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.AndPredicates(predicates...))
+}
+
+// Or groups predicates with the OR operator between them.
+func Or(predicates ...predicate.EmployeeConfig) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.OrPredicates(predicates...))
+}
+
+// Not applies the not operator on the given predicate.
+func Not(p predicate.EmployeeConfig) predicate.EmployeeConfig {
+	return predicate.EmployeeConfig(sql.NotPredicates(p))
+}

ent/employeeconfig_create.go

@@ -0,0 +1,918 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+	"wechat-api/ent/employeeconfig"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// EmployeeConfigCreate is the builder for creating a EmployeeConfig entity.
+type EmployeeConfigCreate struct {
+	config
+	mutation *EmployeeConfigMutation
+	hooks    []Hook
+	conflict []sql.ConflictOption
+}
+
+// SetCreatedAt sets the "created_at" field.
+func (ecc *EmployeeConfigCreate) SetCreatedAt(t time.Time) *EmployeeConfigCreate {
+	ecc.mutation.SetCreatedAt(t)
+	return ecc
+}
+
+// SetNillableCreatedAt sets the "created_at" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillableCreatedAt(t *time.Time) *EmployeeConfigCreate {
+	if t != nil {
+		ecc.SetCreatedAt(*t)
+	}
+	return ecc
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (ecc *EmployeeConfigCreate) SetUpdatedAt(t time.Time) *EmployeeConfigCreate {
+	ecc.mutation.SetUpdatedAt(t)
+	return ecc
+}
+
+// SetNillableUpdatedAt sets the "updated_at" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillableUpdatedAt(t *time.Time) *EmployeeConfigCreate {
+	if t != nil {
+		ecc.SetUpdatedAt(*t)
+	}
+	return ecc
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (ecc *EmployeeConfigCreate) SetDeletedAt(t time.Time) *EmployeeConfigCreate {
+	ecc.mutation.SetDeletedAt(t)
+	return ecc
+}
+
+// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillableDeletedAt(t *time.Time) *EmployeeConfigCreate {
+	if t != nil {
+		ecc.SetDeletedAt(*t)
+	}
+	return ecc
+}
+
+// SetStype sets the "stype" field.
+func (ecc *EmployeeConfigCreate) SetStype(s string) *EmployeeConfigCreate {
+	ecc.mutation.SetStype(s)
+	return ecc
+}
+
+// SetNillableStype sets the "stype" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillableStype(s *string) *EmployeeConfigCreate {
+	if s != nil {
+		ecc.SetStype(*s)
+	}
+	return ecc
+}
+
+// SetTitle sets the "title" field.
+func (ecc *EmployeeConfigCreate) SetTitle(s string) *EmployeeConfigCreate {
+	ecc.mutation.SetTitle(s)
+	return ecc
+}
+
+// SetNillableTitle sets the "title" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillableTitle(s *string) *EmployeeConfigCreate {
+	if s != nil {
+		ecc.SetTitle(*s)
+	}
+	return ecc
+}
+
+// SetPhoto sets the "photo" field.
+func (ecc *EmployeeConfigCreate) SetPhoto(s string) *EmployeeConfigCreate {
+	ecc.mutation.SetPhoto(s)
+	return ecc
+}
+
+// SetNillablePhoto sets the "photo" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillablePhoto(s *string) *EmployeeConfigCreate {
+	if s != nil {
+		ecc.SetPhoto(*s)
+	}
+	return ecc
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (ecc *EmployeeConfigCreate) SetOrganizationID(u uint64) *EmployeeConfigCreate {
+	ecc.mutation.SetOrganizationID(u)
+	return ecc
+}
+
+// SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
+func (ecc *EmployeeConfigCreate) SetNillableOrganizationID(u *uint64) *EmployeeConfigCreate {
+	if u != nil {
+		ecc.SetOrganizationID(*u)
+	}
+	return ecc
+}
+
+// SetID sets the "id" field.
+func (ecc *EmployeeConfigCreate) SetID(u uint64) *EmployeeConfigCreate {
+	ecc.mutation.SetID(u)
+	return ecc
+}
+
+// Mutation returns the EmployeeConfigMutation object of the builder.
+func (ecc *EmployeeConfigCreate) Mutation() *EmployeeConfigMutation {
+	return ecc.mutation
+}
+
+// Save creates the EmployeeConfig in the database.
+func (ecc *EmployeeConfigCreate) Save(ctx context.Context) (*EmployeeConfig, error) {
+	if err := ecc.defaults(); err != nil {
+		return nil, err
+	}
+	return withHooks(ctx, ecc.sqlSave, ecc.mutation, ecc.hooks)
+}
+
+// SaveX calls Save and panics if Save returns an error.
+func (ecc *EmployeeConfigCreate) SaveX(ctx context.Context) *EmployeeConfig {
+	v, err := ecc.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return v
+}
+
+// Exec executes the query.
+func (ecc *EmployeeConfigCreate) Exec(ctx context.Context) error {
+	_, err := ecc.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (ecc *EmployeeConfigCreate) ExecX(ctx context.Context) {
+	if err := ecc.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// defaults sets the default values of the builder before save.
+func (ecc *EmployeeConfigCreate) defaults() error {
+	if _, ok := ecc.mutation.CreatedAt(); !ok {
+		if employeeconfig.DefaultCreatedAt == nil {
+			return fmt.Errorf("ent: uninitialized employeeconfig.DefaultCreatedAt (forgotten import ent/runtime?)")
+		}
+		v := employeeconfig.DefaultCreatedAt()
+		ecc.mutation.SetCreatedAt(v)
+	}
+	if _, ok := ecc.mutation.UpdatedAt(); !ok {
+		if employeeconfig.DefaultUpdatedAt == nil {
+			return fmt.Errorf("ent: uninitialized employeeconfig.DefaultUpdatedAt (forgotten import ent/runtime?)")
+		}
+		v := employeeconfig.DefaultUpdatedAt()
+		ecc.mutation.SetUpdatedAt(v)
+	}
+	if _, ok := ecc.mutation.Stype(); !ok {
+		v := employeeconfig.DefaultStype
+		ecc.mutation.SetStype(v)
+	}
+	if _, ok := ecc.mutation.Title(); !ok {
+		v := employeeconfig.DefaultTitle
+		ecc.mutation.SetTitle(v)
+	}
+	if _, ok := ecc.mutation.Photo(); !ok {
+		v := employeeconfig.DefaultPhoto
+		ecc.mutation.SetPhoto(v)
+	}
+	if _, ok := ecc.mutation.OrganizationID(); !ok {
+		v := employeeconfig.DefaultOrganizationID
+		ecc.mutation.SetOrganizationID(v)
+	}
+	return nil
+}
+
+// check runs all checks and user-defined validators on the builder.
+func (ecc *EmployeeConfigCreate) check() error {
+	if _, ok := ecc.mutation.CreatedAt(); !ok {
+		return &ValidationError{Name: "created_at", err: errors.New(`ent: missing required field "EmployeeConfig.created_at"`)}
+	}
+	if _, ok := ecc.mutation.UpdatedAt(); !ok {
+		return &ValidationError{Name: "updated_at", err: errors.New(`ent: missing required field "EmployeeConfig.updated_at"`)}
+	}
+	if _, ok := ecc.mutation.Stype(); !ok {
+		return &ValidationError{Name: "stype", err: errors.New(`ent: missing required field "EmployeeConfig.stype"`)}
+	}
+	if _, ok := ecc.mutation.Title(); !ok {
+		return &ValidationError{Name: "title", err: errors.New(`ent: missing required field "EmployeeConfig.title"`)}
+	}
+	if _, ok := ecc.mutation.Photo(); !ok {
+		return &ValidationError{Name: "photo", err: errors.New(`ent: missing required field "EmployeeConfig.photo"`)}
+	}
+	return nil
+}
+
+func (ecc *EmployeeConfigCreate) sqlSave(ctx context.Context) (*EmployeeConfig, error) {
+	if err := ecc.check(); err != nil {
+		return nil, err
+	}
+	_node, _spec := ecc.createSpec()
+	if err := sqlgraph.CreateNode(ctx, ecc.driver, _spec); err != nil {
+		if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return nil, err
+	}
+	if _spec.ID.Value != _node.ID {
+		id := _spec.ID.Value.(int64)
+		_node.ID = uint64(id)
+	}
+	ecc.mutation.id = &_node.ID
+	ecc.mutation.done = true
+	return _node, nil
+}
+
+func (ecc *EmployeeConfigCreate) createSpec() (*EmployeeConfig, *sqlgraph.CreateSpec) {
+	var (
+		_node = &EmployeeConfig{config: ecc.config}
+		_spec = sqlgraph.NewCreateSpec(employeeconfig.Table, sqlgraph.NewFieldSpec(employeeconfig.FieldID, field.TypeUint64))
+	)
+	_spec.OnConflict = ecc.conflict
+	if id, ok := ecc.mutation.ID(); ok {
+		_node.ID = id
+		_spec.ID.Value = id
+	}
+	if value, ok := ecc.mutation.CreatedAt(); ok {
+		_spec.SetField(employeeconfig.FieldCreatedAt, field.TypeTime, value)
+		_node.CreatedAt = value
+	}
+	if value, ok := ecc.mutation.UpdatedAt(); ok {
+		_spec.SetField(employeeconfig.FieldUpdatedAt, field.TypeTime, value)
+		_node.UpdatedAt = value
+	}
+	if value, ok := ecc.mutation.DeletedAt(); ok {
+		_spec.SetField(employeeconfig.FieldDeletedAt, field.TypeTime, value)
+		_node.DeletedAt = value
+	}
+	if value, ok := ecc.mutation.Stype(); ok {
+		_spec.SetField(employeeconfig.FieldStype, field.TypeString, value)
+		_node.Stype = value
+	}
+	if value, ok := ecc.mutation.Title(); ok {
+		_spec.SetField(employeeconfig.FieldTitle, field.TypeString, value)
+		_node.Title = value
+	}
+	if value, ok := ecc.mutation.Photo(); ok {
+		_spec.SetField(employeeconfig.FieldPhoto, field.TypeString, value)
+		_node.Photo = value
+	}
+	if value, ok := ecc.mutation.OrganizationID(); ok {
+		_spec.SetField(employeeconfig.FieldOrganizationID, field.TypeUint64, value)
+		_node.OrganizationID = value
+	}
+	return _node, _spec
+}
+
+// OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
+// of the `INSERT` statement. For example:
+//
+//	client.EmployeeConfig.Create().
+//		SetCreatedAt(v).
+//		OnConflict(
+//			// Update the row with the new values
+//			// the was proposed for insertion.
+//			sql.ResolveWithNewValues(),
+//		).
+//		// Override some of the fields with custom
+//		// update values.
+//		Update(func(u *ent.EmployeeConfigUpsert) {
+//			SetCreatedAt(v+v).
+//		}).
+//		Exec(ctx)
+func (ecc *EmployeeConfigCreate) OnConflict(opts ...sql.ConflictOption) *EmployeeConfigUpsertOne {
+	ecc.conflict = opts
+	return &EmployeeConfigUpsertOne{
+		create: ecc,
+	}
+}
+
+// OnConflictColumns calls `OnConflict` and configures the columns
+// as conflict target. Using this option is equivalent to using:
+//
+//	client.EmployeeConfig.Create().
+//		OnConflict(sql.ConflictColumns(columns...)).
+//		Exec(ctx)
+func (ecc *EmployeeConfigCreate) OnConflictColumns(columns ...string) *EmployeeConfigUpsertOne {
+	ecc.conflict = append(ecc.conflict, sql.ConflictColumns(columns...))
+	return &EmployeeConfigUpsertOne{
+		create: ecc,
+	}
+}
+
+type (
+	// EmployeeConfigUpsertOne is the builder for "upsert"-ing
+	//  one EmployeeConfig node.
+	EmployeeConfigUpsertOne struct {
+		create *EmployeeConfigCreate
+	}
+
+	// EmployeeConfigUpsert is the "OnConflict" setter.
+	EmployeeConfigUpsert struct {
+		*sql.UpdateSet
+	}
+)
+
+// SetUpdatedAt sets the "updated_at" field.
+func (u *EmployeeConfigUpsert) SetUpdatedAt(v time.Time) *EmployeeConfigUpsert {
+	u.Set(employeeconfig.FieldUpdatedAt, v)
+	return u
+}
+
+// UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
+func (u *EmployeeConfigUpsert) UpdateUpdatedAt() *EmployeeConfigUpsert {
+	u.SetExcluded(employeeconfig.FieldUpdatedAt)
+	return u
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (u *EmployeeConfigUpsert) SetDeletedAt(v time.Time) *EmployeeConfigUpsert {
+	u.Set(employeeconfig.FieldDeletedAt, v)
+	return u
+}
+
+// UpdateDeletedAt sets the "deleted_at" field to the value that was provided on create.
+func (u *EmployeeConfigUpsert) UpdateDeletedAt() *EmployeeConfigUpsert {
+	u.SetExcluded(employeeconfig.FieldDeletedAt)
+	return u
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (u *EmployeeConfigUpsert) ClearDeletedAt() *EmployeeConfigUpsert {
+	u.SetNull(employeeconfig.FieldDeletedAt)
+	return u
+}
+
+// SetStype sets the "stype" field.
+func (u *EmployeeConfigUpsert) SetStype(v string) *EmployeeConfigUpsert {
+	u.Set(employeeconfig.FieldStype, v)
+	return u
+}
+
+// UpdateStype sets the "stype" field to the value that was provided on create.
+func (u *EmployeeConfigUpsert) UpdateStype() *EmployeeConfigUpsert {
+	u.SetExcluded(employeeconfig.FieldStype)
+	return u
+}
+
+// SetTitle sets the "title" field.
+func (u *EmployeeConfigUpsert) SetTitle(v string) *EmployeeConfigUpsert {
+	u.Set(employeeconfig.FieldTitle, v)
+	return u
+}
+
+// UpdateTitle sets the "title" field to the value that was provided on create.
+func (u *EmployeeConfigUpsert) UpdateTitle() *EmployeeConfigUpsert {
+	u.SetExcluded(employeeconfig.FieldTitle)
+	return u
+}
+
+// SetPhoto sets the "photo" field.
+func (u *EmployeeConfigUpsert) SetPhoto(v string) *EmployeeConfigUpsert {
+	u.Set(employeeconfig.FieldPhoto, v)
+	return u
+}
+
+// UpdatePhoto sets the "photo" field to the value that was provided on create.
+func (u *EmployeeConfigUpsert) UpdatePhoto() *EmployeeConfigUpsert {
+	u.SetExcluded(employeeconfig.FieldPhoto)
+	return u
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (u *EmployeeConfigUpsert) SetOrganizationID(v uint64) *EmployeeConfigUpsert {
+	u.Set(employeeconfig.FieldOrganizationID, v)
+	return u
+}
+
+// UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
+func (u *EmployeeConfigUpsert) UpdateOrganizationID() *EmployeeConfigUpsert {
+	u.SetExcluded(employeeconfig.FieldOrganizationID)
+	return u
+}
+
+// AddOrganizationID adds v to the "organization_id" field.
+func (u *EmployeeConfigUpsert) AddOrganizationID(v uint64) *EmployeeConfigUpsert {
+	u.Add(employeeconfig.FieldOrganizationID, v)
+	return u
+}
+
+// ClearOrganizationID clears the value of the "organization_id" field.
+func (u *EmployeeConfigUpsert) ClearOrganizationID() *EmployeeConfigUpsert {
+	u.SetNull(employeeconfig.FieldOrganizationID)
+	return u
+}
+
+// UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
+// Using this option is equivalent to using:
+//
+//	client.EmployeeConfig.Create().
+//		OnConflict(
+//			sql.ResolveWithNewValues(),
+//			sql.ResolveWith(func(u *sql.UpdateSet) {
+//				u.SetIgnore(employeeconfig.FieldID)
+//			}),
+//		).
+//		Exec(ctx)
+func (u *EmployeeConfigUpsertOne) UpdateNewValues() *EmployeeConfigUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
+		if _, exists := u.create.mutation.ID(); exists {
+			s.SetIgnore(employeeconfig.FieldID)
+		}
+		if _, exists := u.create.mutation.CreatedAt(); exists {
+			s.SetIgnore(employeeconfig.FieldCreatedAt)
+		}
+	}))
+	return u
+}
+
+// Ignore sets each column to itself in case of conflict.
+// Using this option is equivalent to using:
+//
+//	client.EmployeeConfig.Create().
+//	    OnConflict(sql.ResolveWithIgnore()).
+//	    Exec(ctx)
+func (u *EmployeeConfigUpsertOne) Ignore() *EmployeeConfigUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
+	return u
+}
+
+// DoNothing configures the conflict_action to `DO NOTHING`.
+// Supported only by SQLite and PostgreSQL.
+func (u *EmployeeConfigUpsertOne) DoNothing() *EmployeeConfigUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.DoNothing())
+	return u
+}
+
+// Update allows overriding fields `UPDATE` values. See the EmployeeConfigCreate.OnConflict
+// documentation for more info.
+func (u *EmployeeConfigUpsertOne) Update(set func(*EmployeeConfigUpsert)) *EmployeeConfigUpsertOne {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
+		set(&EmployeeConfigUpsert{UpdateSet: update})
+	}))
+	return u
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (u *EmployeeConfigUpsertOne) SetUpdatedAt(v time.Time) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetUpdatedAt(v)
+	})
+}
+
+// UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertOne) UpdateUpdatedAt() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateUpdatedAt()
+	})
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (u *EmployeeConfigUpsertOne) SetDeletedAt(v time.Time) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetDeletedAt(v)
+	})
+}
+
+// UpdateDeletedAt sets the "deleted_at" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertOne) UpdateDeletedAt() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateDeletedAt()
+	})
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (u *EmployeeConfigUpsertOne) ClearDeletedAt() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.ClearDeletedAt()
+	})
+}
+
+// SetStype sets the "stype" field.
+func (u *EmployeeConfigUpsertOne) SetStype(v string) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetStype(v)
+	})
+}
+
+// UpdateStype sets the "stype" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertOne) UpdateStype() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateStype()
+	})
+}
+
+// SetTitle sets the "title" field.
+func (u *EmployeeConfigUpsertOne) SetTitle(v string) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetTitle(v)
+	})
+}
+
+// UpdateTitle sets the "title" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertOne) UpdateTitle() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateTitle()
+	})
+}
+
+// SetPhoto sets the "photo" field.
+func (u *EmployeeConfigUpsertOne) SetPhoto(v string) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetPhoto(v)
+	})
+}
+
+// UpdatePhoto sets the "photo" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertOne) UpdatePhoto() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdatePhoto()
+	})
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (u *EmployeeConfigUpsertOne) SetOrganizationID(v uint64) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetOrganizationID(v)
+	})
+}
+
+// AddOrganizationID adds v to the "organization_id" field.
+func (u *EmployeeConfigUpsertOne) AddOrganizationID(v uint64) *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.AddOrganizationID(v)
+	})
+}
+
+// UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertOne) UpdateOrganizationID() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateOrganizationID()
+	})
+}
+
+// ClearOrganizationID clears the value of the "organization_id" field.
+func (u *EmployeeConfigUpsertOne) ClearOrganizationID() *EmployeeConfigUpsertOne {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.ClearOrganizationID()
+	})
+}
+
+// Exec executes the query.
+func (u *EmployeeConfigUpsertOne) Exec(ctx context.Context) error {
+	if len(u.create.conflict) == 0 {
+		return errors.New("ent: missing options for EmployeeConfigCreate.OnConflict")
+	}
+	return u.create.Exec(ctx)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (u *EmployeeConfigUpsertOne) ExecX(ctx context.Context) {
+	if err := u.create.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// Exec executes the UPSERT query and returns the inserted/updated ID.
+func (u *EmployeeConfigUpsertOne) ID(ctx context.Context) (id uint64, err error) {
+	node, err := u.create.Save(ctx)
+	if err != nil {
+		return id, err
+	}
+	return node.ID, nil
+}
+
+// IDX is like ID, but panics if an error occurs.
+func (u *EmployeeConfigUpsertOne) IDX(ctx context.Context) uint64 {
+	id, err := u.ID(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return id
+}
+
+// EmployeeConfigCreateBulk is the builder for creating many EmployeeConfig entities in bulk.
+type EmployeeConfigCreateBulk struct {
+	config
+	err      error
+	builders []*EmployeeConfigCreate
+	conflict []sql.ConflictOption
+}
+
+// Save creates the EmployeeConfig entities in the database.
+func (eccb *EmployeeConfigCreateBulk) Save(ctx context.Context) ([]*EmployeeConfig, error) {
+	if eccb.err != nil {
+		return nil, eccb.err
+	}
+	specs := make([]*sqlgraph.CreateSpec, len(eccb.builders))
+	nodes := make([]*EmployeeConfig, len(eccb.builders))
+	mutators := make([]Mutator, len(eccb.builders))
+	for i := range eccb.builders {
+		func(i int, root context.Context) {
+			builder := eccb.builders[i]
+			builder.defaults()
+			var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
+				mutation, ok := m.(*EmployeeConfigMutation)
+				if !ok {
+					return nil, fmt.Errorf("unexpected mutation type %T", m)
+				}
+				if err := builder.check(); err != nil {
+					return nil, err
+				}
+				builder.mutation = mutation
+				var err error
+				nodes[i], specs[i] = builder.createSpec()
+				if i < len(mutators)-1 {
+					_, err = mutators[i+1].Mutate(root, eccb.builders[i+1].mutation)
+				} else {
+					spec := &sqlgraph.BatchCreateSpec{Nodes: specs}
+					spec.OnConflict = eccb.conflict
+					// Invoke the actual operation on the latest mutation in the chain.
+					if err = sqlgraph.BatchCreate(ctx, eccb.driver, spec); err != nil {
+						if sqlgraph.IsConstraintError(err) {
+							err = &ConstraintError{msg: err.Error(), wrap: err}
+						}
+					}
+				}
+				if err != nil {
+					return nil, err
+				}
+				mutation.id = &nodes[i].ID
+				if specs[i].ID.Value != nil && nodes[i].ID == 0 {
+					id := specs[i].ID.Value.(int64)
+					nodes[i].ID = uint64(id)
+				}
+				mutation.done = true
+				return nodes[i], nil
+			})
+			for i := len(builder.hooks) - 1; i >= 0; i-- {
+				mut = builder.hooks[i](mut)
+			}
+			mutators[i] = mut
+		}(i, ctx)
+	}
+	if len(mutators) > 0 {
+		if _, err := mutators[0].Mutate(ctx, eccb.builders[0].mutation); err != nil {
+			return nil, err
+		}
+	}
+	return nodes, nil
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (eccb *EmployeeConfigCreateBulk) SaveX(ctx context.Context) []*EmployeeConfig {
+	v, err := eccb.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return v
+}
+
+// Exec executes the query.
+func (eccb *EmployeeConfigCreateBulk) Exec(ctx context.Context) error {
+	_, err := eccb.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (eccb *EmployeeConfigCreateBulk) ExecX(ctx context.Context) {
+	if err := eccb.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
+// of the `INSERT` statement. For example:
+//
+//	client.EmployeeConfig.CreateBulk(builders...).
+//		OnConflict(
+//			// Update the row with the new values
+//			// the was proposed for insertion.
+//			sql.ResolveWithNewValues(),
+//		).
+//		// Override some of the fields with custom
+//		// update values.
+//		Update(func(u *ent.EmployeeConfigUpsert) {
+//			SetCreatedAt(v+v).
+//		}).
+//		Exec(ctx)
+func (eccb *EmployeeConfigCreateBulk) OnConflict(opts ...sql.ConflictOption) *EmployeeConfigUpsertBulk {
+	eccb.conflict = opts
+	return &EmployeeConfigUpsertBulk{
+		create: eccb,
+	}
+}
+
+// OnConflictColumns calls `OnConflict` and configures the columns
+// as conflict target. Using this option is equivalent to using:
+//
+//	client.EmployeeConfig.Create().
+//		OnConflict(sql.ConflictColumns(columns...)).
+//		Exec(ctx)
+func (eccb *EmployeeConfigCreateBulk) OnConflictColumns(columns ...string) *EmployeeConfigUpsertBulk {
+	eccb.conflict = append(eccb.conflict, sql.ConflictColumns(columns...))
+	return &EmployeeConfigUpsertBulk{
+		create: eccb,
+	}
+}
+
+// EmployeeConfigUpsertBulk is the builder for "upsert"-ing
+// a bulk of EmployeeConfig nodes.
+type EmployeeConfigUpsertBulk struct {
+	create *EmployeeConfigCreateBulk
+}
+
+// UpdateNewValues updates the mutable fields using the new values that
+// were set on create. Using this option is equivalent to using:
+//
+//	client.EmployeeConfig.Create().
+//		OnConflict(
+//			sql.ResolveWithNewValues(),
+//			sql.ResolveWith(func(u *sql.UpdateSet) {
+//				u.SetIgnore(employeeconfig.FieldID)
+//			}),
+//		).
+//		Exec(ctx)
+func (u *EmployeeConfigUpsertBulk) UpdateNewValues() *EmployeeConfigUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
+		for _, b := range u.create.builders {
+			if _, exists := b.mutation.ID(); exists {
+				s.SetIgnore(employeeconfig.FieldID)
+			}
+			if _, exists := b.mutation.CreatedAt(); exists {
+				s.SetIgnore(employeeconfig.FieldCreatedAt)
+			}
+		}
+	}))
+	return u
+}
+
+// Ignore sets each column to itself in case of conflict.
+// Using this option is equivalent to using:
+//
+//	client.EmployeeConfig.Create().
+//		OnConflict(sql.ResolveWithIgnore()).
+//		Exec(ctx)
+func (u *EmployeeConfigUpsertBulk) Ignore() *EmployeeConfigUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
+	return u
+}
+
+// DoNothing configures the conflict_action to `DO NOTHING`.
+// Supported only by SQLite and PostgreSQL.
+func (u *EmployeeConfigUpsertBulk) DoNothing() *EmployeeConfigUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.DoNothing())
+	return u
+}
+
+// Update allows overriding fields `UPDATE` values. See the EmployeeConfigCreateBulk.OnConflict
+// documentation for more info.
+func (u *EmployeeConfigUpsertBulk) Update(set func(*EmployeeConfigUpsert)) *EmployeeConfigUpsertBulk {
+	u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
+		set(&EmployeeConfigUpsert{UpdateSet: update})
+	}))
+	return u
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (u *EmployeeConfigUpsertBulk) SetUpdatedAt(v time.Time) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetUpdatedAt(v)
+	})
+}
+
+// UpdateUpdatedAt sets the "updated_at" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertBulk) UpdateUpdatedAt() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateUpdatedAt()
+	})
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (u *EmployeeConfigUpsertBulk) SetDeletedAt(v time.Time) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetDeletedAt(v)
+	})
+}
+
+// UpdateDeletedAt sets the "deleted_at" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertBulk) UpdateDeletedAt() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateDeletedAt()
+	})
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (u *EmployeeConfigUpsertBulk) ClearDeletedAt() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.ClearDeletedAt()
+	})
+}
+
+// SetStype sets the "stype" field.
+func (u *EmployeeConfigUpsertBulk) SetStype(v string) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetStype(v)
+	})
+}
+
+// UpdateStype sets the "stype" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertBulk) UpdateStype() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateStype()
+	})
+}
+
+// SetTitle sets the "title" field.
+func (u *EmployeeConfigUpsertBulk) SetTitle(v string) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetTitle(v)
+	})
+}
+
+// UpdateTitle sets the "title" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertBulk) UpdateTitle() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateTitle()
+	})
+}
+
+// SetPhoto sets the "photo" field.
+func (u *EmployeeConfigUpsertBulk) SetPhoto(v string) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetPhoto(v)
+	})
+}
+
+// UpdatePhoto sets the "photo" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertBulk) UpdatePhoto() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdatePhoto()
+	})
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (u *EmployeeConfigUpsertBulk) SetOrganizationID(v uint64) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.SetOrganizationID(v)
+	})
+}
+
+// AddOrganizationID adds v to the "organization_id" field.
+func (u *EmployeeConfigUpsertBulk) AddOrganizationID(v uint64) *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.AddOrganizationID(v)
+	})
+}
+
+// UpdateOrganizationID sets the "organization_id" field to the value that was provided on create.
+func (u *EmployeeConfigUpsertBulk) UpdateOrganizationID() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.UpdateOrganizationID()
+	})
+}
+
+// ClearOrganizationID clears the value of the "organization_id" field.
+func (u *EmployeeConfigUpsertBulk) ClearOrganizationID() *EmployeeConfigUpsertBulk {
+	return u.Update(func(s *EmployeeConfigUpsert) {
+		s.ClearOrganizationID()
+	})
+}
+
+// Exec executes the query.
+func (u *EmployeeConfigUpsertBulk) Exec(ctx context.Context) error {
+	if u.create.err != nil {
+		return u.create.err
+	}
+	for i, b := range u.create.builders {
+		if len(b.conflict) != 0 {
+			return fmt.Errorf("ent: OnConflict was set for builder %d. Set it on the EmployeeConfigCreateBulk instead", i)
+		}
+	}
+	if len(u.create.conflict) == 0 {
+		return errors.New("ent: missing options for EmployeeConfigCreateBulk.OnConflict")
+	}
+	return u.create.Exec(ctx)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (u *EmployeeConfigUpsertBulk) ExecX(ctx context.Context) {
+	if err := u.create.Exec(ctx); err != nil {
+		panic(err)
+	}
+}

ent/employeeconfig_delete.go

@@ -0,0 +1,88 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"wechat-api/ent/employeeconfig"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// EmployeeConfigDelete is the builder for deleting a EmployeeConfig entity.
+type EmployeeConfigDelete struct {
+	config
+	hooks    []Hook
+	mutation *EmployeeConfigMutation
+}
+
+// Where appends a list predicates to the EmployeeConfigDelete builder.
+func (ecd *EmployeeConfigDelete) Where(ps ...predicate.EmployeeConfig) *EmployeeConfigDelete {
+	ecd.mutation.Where(ps...)
+	return ecd
+}
+
+// Exec executes the deletion query and returns how many vertices were deleted.
+func (ecd *EmployeeConfigDelete) Exec(ctx context.Context) (int, error) {
+	return withHooks(ctx, ecd.sqlExec, ecd.mutation, ecd.hooks)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (ecd *EmployeeConfigDelete) ExecX(ctx context.Context) int {
+	n, err := ecd.Exec(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return n
+}
+
+func (ecd *EmployeeConfigDelete) sqlExec(ctx context.Context) (int, error) {
+	_spec := sqlgraph.NewDeleteSpec(employeeconfig.Table, sqlgraph.NewFieldSpec(employeeconfig.FieldID, field.TypeUint64))
+	if ps := ecd.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	affected, err := sqlgraph.DeleteNodes(ctx, ecd.driver, _spec)
+	if err != nil && sqlgraph.IsConstraintError(err) {
+		err = &ConstraintError{msg: err.Error(), wrap: err}
+	}
+	ecd.mutation.done = true
+	return affected, err
+}
+
+// EmployeeConfigDeleteOne is the builder for deleting a single EmployeeConfig entity.
+type EmployeeConfigDeleteOne struct {
+	ecd *EmployeeConfigDelete
+}
+
+// Where appends a list predicates to the EmployeeConfigDelete builder.
+func (ecdo *EmployeeConfigDeleteOne) Where(ps ...predicate.EmployeeConfig) *EmployeeConfigDeleteOne {
+	ecdo.ecd.mutation.Where(ps...)
+	return ecdo
+}
+
+// Exec executes the deletion query.
+func (ecdo *EmployeeConfigDeleteOne) Exec(ctx context.Context) error {
+	n, err := ecdo.ecd.Exec(ctx)
+	switch {
+	case err != nil:
+		return err
+	case n == 0:
+		return &NotFoundError{employeeconfig.Label}
+	default:
+		return nil
+	}
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (ecdo *EmployeeConfigDeleteOne) ExecX(ctx context.Context) {
+	if err := ecdo.Exec(ctx); err != nil {
+		panic(err)
+	}
+}

ent/employeeconfig_query.go

@@ -0,0 +1,526 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"fmt"
+	"math"
+	"wechat-api/ent/employeeconfig"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// EmployeeConfigQuery is the builder for querying EmployeeConfig entities.
+type EmployeeConfigQuery struct {
+	config
+	ctx        *QueryContext
+	order      []employeeconfig.OrderOption
+	inters     []Interceptor
+	predicates []predicate.EmployeeConfig
+	// intermediate query (i.e. traversal path).
+	sql  *sql.Selector
+	path func(context.Context) (*sql.Selector, error)
+}
+
+// Where adds a new predicate for the EmployeeConfigQuery builder.
+func (ecq *EmployeeConfigQuery) Where(ps ...predicate.EmployeeConfig) *EmployeeConfigQuery {
+	ecq.predicates = append(ecq.predicates, ps...)
+	return ecq
+}
+
+// Limit the number of records to be returned by this query.
+func (ecq *EmployeeConfigQuery) Limit(limit int) *EmployeeConfigQuery {
+	ecq.ctx.Limit = &limit
+	return ecq
+}
+
+// Offset to start from.
+func (ecq *EmployeeConfigQuery) Offset(offset int) *EmployeeConfigQuery {
+	ecq.ctx.Offset = &offset
+	return ecq
+}
+
+// Unique configures the query builder to filter duplicate records on query.
+// By default, unique is set to true, and can be disabled using this method.
+func (ecq *EmployeeConfigQuery) Unique(unique bool) *EmployeeConfigQuery {
+	ecq.ctx.Unique = &unique
+	return ecq
+}
+
+// Order specifies how the records should be ordered.
+func (ecq *EmployeeConfigQuery) Order(o ...employeeconfig.OrderOption) *EmployeeConfigQuery {
+	ecq.order = append(ecq.order, o...)
+	return ecq
+}
+
+// First returns the first EmployeeConfig entity from the query.
+// Returns a *NotFoundError when no EmployeeConfig was found.
+func (ecq *EmployeeConfigQuery) First(ctx context.Context) (*EmployeeConfig, error) {
+	nodes, err := ecq.Limit(1).All(setContextOp(ctx, ecq.ctx, "First"))
+	if err != nil {
+		return nil, err
+	}
+	if len(nodes) == 0 {
+		return nil, &NotFoundError{employeeconfig.Label}
+	}
+	return nodes[0], nil
+}
+
+// FirstX is like First, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) FirstX(ctx context.Context) *EmployeeConfig {
+	node, err := ecq.First(ctx)
+	if err != nil && !IsNotFound(err) {
+		panic(err)
+	}
+	return node
+}
+
+// FirstID returns the first EmployeeConfig ID from the query.
+// Returns a *NotFoundError when no EmployeeConfig ID was found.
+func (ecq *EmployeeConfigQuery) FirstID(ctx context.Context) (id uint64, err error) {
+	var ids []uint64
+	if ids, err = ecq.Limit(1).IDs(setContextOp(ctx, ecq.ctx, "FirstID")); err != nil {
+		return
+	}
+	if len(ids) == 0 {
+		err = &NotFoundError{employeeconfig.Label}
+		return
+	}
+	return ids[0], nil
+}
+
+// FirstIDX is like FirstID, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) FirstIDX(ctx context.Context) uint64 {
+	id, err := ecq.FirstID(ctx)
+	if err != nil && !IsNotFound(err) {
+		panic(err)
+	}
+	return id
+}
+
+// Only returns a single EmployeeConfig entity found by the query, ensuring it only returns one.
+// Returns a *NotSingularError when more than one EmployeeConfig entity is found.
+// Returns a *NotFoundError when no EmployeeConfig entities are found.
+func (ecq *EmployeeConfigQuery) Only(ctx context.Context) (*EmployeeConfig, error) {
+	nodes, err := ecq.Limit(2).All(setContextOp(ctx, ecq.ctx, "Only"))
+	if err != nil {
+		return nil, err
+	}
+	switch len(nodes) {
+	case 1:
+		return nodes[0], nil
+	case 0:
+		return nil, &NotFoundError{employeeconfig.Label}
+	default:
+		return nil, &NotSingularError{employeeconfig.Label}
+	}
+}
+
+// OnlyX is like Only, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) OnlyX(ctx context.Context) *EmployeeConfig {
+	node, err := ecq.Only(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return node
+}
+
+// OnlyID is like Only, but returns the only EmployeeConfig ID in the query.
+// Returns a *NotSingularError when more than one EmployeeConfig ID is found.
+// Returns a *NotFoundError when no entities are found.
+func (ecq *EmployeeConfigQuery) OnlyID(ctx context.Context) (id uint64, err error) {
+	var ids []uint64
+	if ids, err = ecq.Limit(2).IDs(setContextOp(ctx, ecq.ctx, "OnlyID")); err != nil {
+		return
+	}
+	switch len(ids) {
+	case 1:
+		id = ids[0]
+	case 0:
+		err = &NotFoundError{employeeconfig.Label}
+	default:
+		err = &NotSingularError{employeeconfig.Label}
+	}
+	return
+}
+
+// OnlyIDX is like OnlyID, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) OnlyIDX(ctx context.Context) uint64 {
+	id, err := ecq.OnlyID(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return id
+}
+
+// All executes the query and returns a list of EmployeeConfigs.
+func (ecq *EmployeeConfigQuery) All(ctx context.Context) ([]*EmployeeConfig, error) {
+	ctx = setContextOp(ctx, ecq.ctx, "All")
+	if err := ecq.prepareQuery(ctx); err != nil {
+		return nil, err
+	}
+	qr := querierAll[[]*EmployeeConfig, *EmployeeConfigQuery]()
+	return withInterceptors[[]*EmployeeConfig](ctx, ecq, qr, ecq.inters)
+}
+
+// AllX is like All, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) AllX(ctx context.Context) []*EmployeeConfig {
+	nodes, err := ecq.All(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return nodes
+}
+
+// IDs executes the query and returns a list of EmployeeConfig IDs.
+func (ecq *EmployeeConfigQuery) IDs(ctx context.Context) (ids []uint64, err error) {
+	if ecq.ctx.Unique == nil && ecq.path != nil {
+		ecq.Unique(true)
+	}
+	ctx = setContextOp(ctx, ecq.ctx, "IDs")
+	if err = ecq.Select(employeeconfig.FieldID).Scan(ctx, &ids); err != nil {
+		return nil, err
+	}
+	return ids, nil
+}
+
+// IDsX is like IDs, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) IDsX(ctx context.Context) []uint64 {
+	ids, err := ecq.IDs(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return ids
+}
+
+// Count returns the count of the given query.
+func (ecq *EmployeeConfigQuery) Count(ctx context.Context) (int, error) {
+	ctx = setContextOp(ctx, ecq.ctx, "Count")
+	if err := ecq.prepareQuery(ctx); err != nil {
+		return 0, err
+	}
+	return withInterceptors[int](ctx, ecq, querierCount[*EmployeeConfigQuery](), ecq.inters)
+}
+
+// CountX is like Count, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) CountX(ctx context.Context) int {
+	count, err := ecq.Count(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return count
+}
+
+// Exist returns true if the query has elements in the graph.
+func (ecq *EmployeeConfigQuery) Exist(ctx context.Context) (bool, error) {
+	ctx = setContextOp(ctx, ecq.ctx, "Exist")
+	switch _, err := ecq.FirstID(ctx); {
+	case IsNotFound(err):
+		return false, nil
+	case err != nil:
+		return false, fmt.Errorf("ent: check existence: %w", err)
+	default:
+		return true, nil
+	}
+}
+
+// ExistX is like Exist, but panics if an error occurs.
+func (ecq *EmployeeConfigQuery) ExistX(ctx context.Context) bool {
+	exist, err := ecq.Exist(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return exist
+}
+
+// Clone returns a duplicate of the EmployeeConfigQuery builder, including all associated steps. It can be
+// used to prepare common query builders and use them differently after the clone is made.
+func (ecq *EmployeeConfigQuery) Clone() *EmployeeConfigQuery {
+	if ecq == nil {
+		return nil
+	}
+	return &EmployeeConfigQuery{
+		config:     ecq.config,
+		ctx:        ecq.ctx.Clone(),
+		order:      append([]employeeconfig.OrderOption{}, ecq.order...),
+		inters:     append([]Interceptor{}, ecq.inters...),
+		predicates: append([]predicate.EmployeeConfig{}, ecq.predicates...),
+		// clone intermediate query.
+		sql:  ecq.sql.Clone(),
+		path: ecq.path,
+	}
+}
+
+// GroupBy is used to group vertices by one or more fields/columns.
+// It is often used with aggregate functions, like: count, max, mean, min, sum.
+//
+// Example:
+//
+//	var v []struct {
+//		CreatedAt time.Time `json:"created_at,omitempty"`
+//		Count int `json:"count,omitempty"`
+//	}
+//
+//	client.EmployeeConfig.Query().
+//		GroupBy(employeeconfig.FieldCreatedAt).
+//		Aggregate(ent.Count()).
+//		Scan(ctx, &v)
+func (ecq *EmployeeConfigQuery) GroupBy(field string, fields ...string) *EmployeeConfigGroupBy {
+	ecq.ctx.Fields = append([]string{field}, fields...)
+	grbuild := &EmployeeConfigGroupBy{build: ecq}
+	grbuild.flds = &ecq.ctx.Fields
+	grbuild.label = employeeconfig.Label
+	grbuild.scan = grbuild.Scan
+	return grbuild
+}
+
+// Select allows the selection one or more fields/columns for the given query,
+// instead of selecting all fields in the entity.
+//
+// Example:
+//
+//	var v []struct {
+//		CreatedAt time.Time `json:"created_at,omitempty"`
+//	}
+//
+//	client.EmployeeConfig.Query().
+//		Select(employeeconfig.FieldCreatedAt).
+//		Scan(ctx, &v)
+func (ecq *EmployeeConfigQuery) Select(fields ...string) *EmployeeConfigSelect {
+	ecq.ctx.Fields = append(ecq.ctx.Fields, fields...)
+	sbuild := &EmployeeConfigSelect{EmployeeConfigQuery: ecq}
+	sbuild.label = employeeconfig.Label
+	sbuild.flds, sbuild.scan = &ecq.ctx.Fields, sbuild.Scan
+	return sbuild
+}
+
+// Aggregate returns a EmployeeConfigSelect configured with the given aggregations.
+func (ecq *EmployeeConfigQuery) Aggregate(fns ...AggregateFunc) *EmployeeConfigSelect {
+	return ecq.Select().Aggregate(fns...)
+}
+
+func (ecq *EmployeeConfigQuery) prepareQuery(ctx context.Context) error {
+	for _, inter := range ecq.inters {
+		if inter == nil {
+			return fmt.Errorf("ent: uninitialized interceptor (forgotten import ent/runtime?)")
+		}
+		if trv, ok := inter.(Traverser); ok {
+			if err := trv.Traverse(ctx, ecq); err != nil {
+				return err
+			}
+		}
+	}
+	for _, f := range ecq.ctx.Fields {
+		if !employeeconfig.ValidColumn(f) {
+			return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
+		}
+	}
+	if ecq.path != nil {
+		prev, err := ecq.path(ctx)
+		if err != nil {
+			return err
+		}
+		ecq.sql = prev
+	}
+	return nil
+}
+
+func (ecq *EmployeeConfigQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*EmployeeConfig, error) {
+	var (
+		nodes = []*EmployeeConfig{}
+		_spec = ecq.querySpec()
+	)
+	_spec.ScanValues = func(columns []string) ([]any, error) {
+		return (*EmployeeConfig).scanValues(nil, columns)
+	}
+	_spec.Assign = func(columns []string, values []any) error {
+		node := &EmployeeConfig{config: ecq.config}
+		nodes = append(nodes, node)
+		return node.assignValues(columns, values)
+	}
+	for i := range hooks {
+		hooks[i](ctx, _spec)
+	}
+	if err := sqlgraph.QueryNodes(ctx, ecq.driver, _spec); err != nil {
+		return nil, err
+	}
+	if len(nodes) == 0 {
+		return nodes, nil
+	}
+	return nodes, nil
+}
+
+func (ecq *EmployeeConfigQuery) sqlCount(ctx context.Context) (int, error) {
+	_spec := ecq.querySpec()
+	_spec.Node.Columns = ecq.ctx.Fields
+	if len(ecq.ctx.Fields) > 0 {
+		_spec.Unique = ecq.ctx.Unique != nil && *ecq.ctx.Unique
+	}
+	return sqlgraph.CountNodes(ctx, ecq.driver, _spec)
+}
+
+func (ecq *EmployeeConfigQuery) querySpec() *sqlgraph.QuerySpec {
+	_spec := sqlgraph.NewQuerySpec(employeeconfig.Table, employeeconfig.Columns, sqlgraph.NewFieldSpec(employeeconfig.FieldID, field.TypeUint64))
+	_spec.From = ecq.sql
+	if unique := ecq.ctx.Unique; unique != nil {
+		_spec.Unique = *unique
+	} else if ecq.path != nil {
+		_spec.Unique = true
+	}
+	if fields := ecq.ctx.Fields; len(fields) > 0 {
+		_spec.Node.Columns = make([]string, 0, len(fields))
+		_spec.Node.Columns = append(_spec.Node.Columns, employeeconfig.FieldID)
+		for i := range fields {
+			if fields[i] != employeeconfig.FieldID {
+				_spec.Node.Columns = append(_spec.Node.Columns, fields[i])
+			}
+		}
+	}
+	if ps := ecq.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if limit := ecq.ctx.Limit; limit != nil {
+		_spec.Limit = *limit
+	}
+	if offset := ecq.ctx.Offset; offset != nil {
+		_spec.Offset = *offset
+	}
+	if ps := ecq.order; len(ps) > 0 {
+		_spec.Order = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	return _spec
+}
+
+func (ecq *EmployeeConfigQuery) sqlQuery(ctx context.Context) *sql.Selector {
+	builder := sql.Dialect(ecq.driver.Dialect())
+	t1 := builder.Table(employeeconfig.Table)
+	columns := ecq.ctx.Fields
+	if len(columns) == 0 {
+		columns = employeeconfig.Columns
+	}
+	selector := builder.Select(t1.Columns(columns...)...).From(t1)
+	if ecq.sql != nil {
+		selector = ecq.sql
+		selector.Select(selector.Columns(columns...)...)
+	}
+	if ecq.ctx.Unique != nil && *ecq.ctx.Unique {
+		selector.Distinct()
+	}
+	for _, p := range ecq.predicates {
+		p(selector)
+	}
+	for _, p := range ecq.order {
+		p(selector)
+	}
+	if offset := ecq.ctx.Offset; offset != nil {
+		// limit is mandatory for offset clause. We start
+		// with default value, and override it below if needed.
+		selector.Offset(*offset).Limit(math.MaxInt32)
+	}
+	if limit := ecq.ctx.Limit; limit != nil {
+		selector.Limit(*limit)
+	}
+	return selector
+}
+
+// EmployeeConfigGroupBy is the group-by builder for EmployeeConfig entities.
+type EmployeeConfigGroupBy struct {
+	selector
+	build *EmployeeConfigQuery
+}
+
+// Aggregate adds the given aggregation functions to the group-by query.
+func (ecgb *EmployeeConfigGroupBy) Aggregate(fns ...AggregateFunc) *EmployeeConfigGroupBy {
+	ecgb.fns = append(ecgb.fns, fns...)
+	return ecgb
+}
+
+// Scan applies the selector query and scans the result into the given value.
+func (ecgb *EmployeeConfigGroupBy) Scan(ctx context.Context, v any) error {
+	ctx = setContextOp(ctx, ecgb.build.ctx, "GroupBy")
+	if err := ecgb.build.prepareQuery(ctx); err != nil {
+		return err
+	}
+	return scanWithInterceptors[*EmployeeConfigQuery, *EmployeeConfigGroupBy](ctx, ecgb.build, ecgb, ecgb.build.inters, v)
+}
+
+func (ecgb *EmployeeConfigGroupBy) sqlScan(ctx context.Context, root *EmployeeConfigQuery, v any) error {
+	selector := root.sqlQuery(ctx).Select()
+	aggregation := make([]string, 0, len(ecgb.fns))
+	for _, fn := range ecgb.fns {
+		aggregation = append(aggregation, fn(selector))
+	}
+	if len(selector.SelectedColumns()) == 0 {
+		columns := make([]string, 0, len(*ecgb.flds)+len(ecgb.fns))
+		for _, f := range *ecgb.flds {
+			columns = append(columns, selector.C(f))
+		}
+		columns = append(columns, aggregation...)
+		selector.Select(columns...)
+	}
+	selector.GroupBy(selector.Columns(*ecgb.flds...)...)
+	if err := selector.Err(); err != nil {
+		return err
+	}
+	rows := &sql.Rows{}
+	query, args := selector.Query()
+	if err := ecgb.build.driver.Query(ctx, query, args, rows); err != nil {
+		return err
+	}
+	defer rows.Close()
+	return sql.ScanSlice(rows, v)
+}
+
+// EmployeeConfigSelect is the builder for selecting fields of EmployeeConfig entities.
+type EmployeeConfigSelect struct {
+	*EmployeeConfigQuery
+	selector
+}
+
+// Aggregate adds the given aggregation functions to the selector query.
+func (ecs *EmployeeConfigSelect) Aggregate(fns ...AggregateFunc) *EmployeeConfigSelect {
+	ecs.fns = append(ecs.fns, fns...)
+	return ecs
+}
+
+// Scan applies the selector query and scans the result into the given value.
+func (ecs *EmployeeConfigSelect) Scan(ctx context.Context, v any) error {
+	ctx = setContextOp(ctx, ecs.ctx, "Select")
+	if err := ecs.prepareQuery(ctx); err != nil {
+		return err
+	}
+	return scanWithInterceptors[*EmployeeConfigQuery, *EmployeeConfigSelect](ctx, ecs.EmployeeConfigQuery, ecs, ecs.inters, v)
+}
+
+func (ecs *EmployeeConfigSelect) sqlScan(ctx context.Context, root *EmployeeConfigQuery, v any) error {
+	selector := root.sqlQuery(ctx)
+	aggregation := make([]string, 0, len(ecs.fns))
+	for _, fn := range ecs.fns {
+		aggregation = append(aggregation, fn(selector))
+	}
+	switch n := len(*ecs.selector.flds); {
+	case n == 0 && len(aggregation) > 0:
+		selector.Select(aggregation...)
+	case n != 0 && len(aggregation) > 0:
+		selector.AppendSelect(aggregation...)
+	}
+	rows := &sql.Rows{}
+	query, args := selector.Query()
+	if err := ecs.driver.Query(ctx, query, args, rows); err != nil {
+		return err
+	}
+	defer rows.Close()
+	return sql.ScanSlice(rows, v)
+}

ent/employeeconfig_update.go

@@ -0,0 +1,450 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+	"wechat-api/ent/employeeconfig"
+	"wechat-api/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// EmployeeConfigUpdate is the builder for updating EmployeeConfig entities.
+type EmployeeConfigUpdate struct {
+	config
+	hooks    []Hook
+	mutation *EmployeeConfigMutation
+}
+
+// Where appends a list predicates to the EmployeeConfigUpdate builder.
+func (ecu *EmployeeConfigUpdate) Where(ps ...predicate.EmployeeConfig) *EmployeeConfigUpdate {
+	ecu.mutation.Where(ps...)
+	return ecu
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (ecu *EmployeeConfigUpdate) SetUpdatedAt(t time.Time) *EmployeeConfigUpdate {
+	ecu.mutation.SetUpdatedAt(t)
+	return ecu
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (ecu *EmployeeConfigUpdate) SetDeletedAt(t time.Time) *EmployeeConfigUpdate {
+	ecu.mutation.SetDeletedAt(t)
+	return ecu
+}
+
+// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
+func (ecu *EmployeeConfigUpdate) SetNillableDeletedAt(t *time.Time) *EmployeeConfigUpdate {
+	if t != nil {
+		ecu.SetDeletedAt(*t)
+	}
+	return ecu
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (ecu *EmployeeConfigUpdate) ClearDeletedAt() *EmployeeConfigUpdate {
+	ecu.mutation.ClearDeletedAt()
+	return ecu
+}
+
+// SetStype sets the "stype" field.
+func (ecu *EmployeeConfigUpdate) SetStype(s string) *EmployeeConfigUpdate {
+	ecu.mutation.SetStype(s)
+	return ecu
+}
+
+// SetNillableStype sets the "stype" field if the given value is not nil.
+func (ecu *EmployeeConfigUpdate) SetNillableStype(s *string) *EmployeeConfigUpdate {
+	if s != nil {
+		ecu.SetStype(*s)
+	}
+	return ecu
+}
+
+// SetTitle sets the "title" field.
+func (ecu *EmployeeConfigUpdate) SetTitle(s string) *EmployeeConfigUpdate {
+	ecu.mutation.SetTitle(s)
+	return ecu
+}
+
+// SetNillableTitle sets the "title" field if the given value is not nil.
+func (ecu *EmployeeConfigUpdate) SetNillableTitle(s *string) *EmployeeConfigUpdate {
+	if s != nil {
+		ecu.SetTitle(*s)
+	}
+	return ecu
+}
+
+// SetPhoto sets the "photo" field.
+func (ecu *EmployeeConfigUpdate) SetPhoto(s string) *EmployeeConfigUpdate {
+	ecu.mutation.SetPhoto(s)
+	return ecu
+}
+
+// SetNillablePhoto sets the "photo" field if the given value is not nil.
+func (ecu *EmployeeConfigUpdate) SetNillablePhoto(s *string) *EmployeeConfigUpdate {
+	if s != nil {
+		ecu.SetPhoto(*s)
+	}
+	return ecu
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (ecu *EmployeeConfigUpdate) SetOrganizationID(u uint64) *EmployeeConfigUpdate {
+	ecu.mutation.ResetOrganizationID()
+	ecu.mutation.SetOrganizationID(u)
+	return ecu
+}
+
+// SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
+func (ecu *EmployeeConfigUpdate) SetNillableOrganizationID(u *uint64) *EmployeeConfigUpdate {
+	if u != nil {
+		ecu.SetOrganizationID(*u)
+	}
+	return ecu
+}
+
+// AddOrganizationID adds u to the "organization_id" field.
+func (ecu *EmployeeConfigUpdate) AddOrganizationID(u int64) *EmployeeConfigUpdate {
+	ecu.mutation.AddOrganizationID(u)
+	return ecu
+}
+
+// ClearOrganizationID clears the value of the "organization_id" field.
+func (ecu *EmployeeConfigUpdate) ClearOrganizationID() *EmployeeConfigUpdate {
+	ecu.mutation.ClearOrganizationID()
+	return ecu
+}
+
+// Mutation returns the EmployeeConfigMutation object of the builder.
+func (ecu *EmployeeConfigUpdate) Mutation() *EmployeeConfigMutation {
+	return ecu.mutation
+}
+
+// Save executes the query and returns the number of nodes affected by the update operation.
+func (ecu *EmployeeConfigUpdate) Save(ctx context.Context) (int, error) {
+	if err := ecu.defaults(); err != nil {
+		return 0, err
+	}
+	return withHooks(ctx, ecu.sqlSave, ecu.mutation, ecu.hooks)
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (ecu *EmployeeConfigUpdate) SaveX(ctx context.Context) int {
+	affected, err := ecu.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return affected
+}
+
+// Exec executes the query.
+func (ecu *EmployeeConfigUpdate) Exec(ctx context.Context) error {
+	_, err := ecu.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (ecu *EmployeeConfigUpdate) ExecX(ctx context.Context) {
+	if err := ecu.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// defaults sets the default values of the builder before save.
+func (ecu *EmployeeConfigUpdate) defaults() error {
+	if _, ok := ecu.mutation.UpdatedAt(); !ok {
+		if employeeconfig.UpdateDefaultUpdatedAt == nil {
+			return fmt.Errorf("ent: uninitialized employeeconfig.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
+		}
+		v := employeeconfig.UpdateDefaultUpdatedAt()
+		ecu.mutation.SetUpdatedAt(v)
+	}
+	return nil
+}
+
+func (ecu *EmployeeConfigUpdate) sqlSave(ctx context.Context) (n int, err error) {
+	_spec := sqlgraph.NewUpdateSpec(employeeconfig.Table, employeeconfig.Columns, sqlgraph.NewFieldSpec(employeeconfig.FieldID, field.TypeUint64))
+	if ps := ecu.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if value, ok := ecu.mutation.UpdatedAt(); ok {
+		_spec.SetField(employeeconfig.FieldUpdatedAt, field.TypeTime, value)
+	}
+	if value, ok := ecu.mutation.DeletedAt(); ok {
+		_spec.SetField(employeeconfig.FieldDeletedAt, field.TypeTime, value)
+	}
+	if ecu.mutation.DeletedAtCleared() {
+		_spec.ClearField(employeeconfig.FieldDeletedAt, field.TypeTime)
+	}
+	if value, ok := ecu.mutation.Stype(); ok {
+		_spec.SetField(employeeconfig.FieldStype, field.TypeString, value)
+	}
+	if value, ok := ecu.mutation.Title(); ok {
+		_spec.SetField(employeeconfig.FieldTitle, field.TypeString, value)
+	}
+	if value, ok := ecu.mutation.Photo(); ok {
+		_spec.SetField(employeeconfig.FieldPhoto, field.TypeString, value)
+	}
+	if value, ok := ecu.mutation.OrganizationID(); ok {
+		_spec.SetField(employeeconfig.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if value, ok := ecu.mutation.AddedOrganizationID(); ok {
+		_spec.AddField(employeeconfig.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if ecu.mutation.OrganizationIDCleared() {
+		_spec.ClearField(employeeconfig.FieldOrganizationID, field.TypeUint64)
+	}
+	if n, err = sqlgraph.UpdateNodes(ctx, ecu.driver, _spec); err != nil {
+		if _, ok := err.(*sqlgraph.NotFoundError); ok {
+			err = &NotFoundError{employeeconfig.Label}
+		} else if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return 0, err
+	}
+	ecu.mutation.done = true
+	return n, nil
+}
+
+// EmployeeConfigUpdateOne is the builder for updating a single EmployeeConfig entity.
+type EmployeeConfigUpdateOne struct {
+	config
+	fields   []string
+	hooks    []Hook
+	mutation *EmployeeConfigMutation
+}
+
+// SetUpdatedAt sets the "updated_at" field.
+func (ecuo *EmployeeConfigUpdateOne) SetUpdatedAt(t time.Time) *EmployeeConfigUpdateOne {
+	ecuo.mutation.SetUpdatedAt(t)
+	return ecuo
+}
+
+// SetDeletedAt sets the "deleted_at" field.
+func (ecuo *EmployeeConfigUpdateOne) SetDeletedAt(t time.Time) *EmployeeConfigUpdateOne {
+	ecuo.mutation.SetDeletedAt(t)
+	return ecuo
+}
+
+// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
+func (ecuo *EmployeeConfigUpdateOne) SetNillableDeletedAt(t *time.Time) *EmployeeConfigUpdateOne {
+	if t != nil {
+		ecuo.SetDeletedAt(*t)
+	}
+	return ecuo
+}
+
+// ClearDeletedAt clears the value of the "deleted_at" field.
+func (ecuo *EmployeeConfigUpdateOne) ClearDeletedAt() *EmployeeConfigUpdateOne {
+	ecuo.mutation.ClearDeletedAt()
+	return ecuo
+}
+
+// SetStype sets the "stype" field.
+func (ecuo *EmployeeConfigUpdateOne) SetStype(s string) *EmployeeConfigUpdateOne {
+	ecuo.mutation.SetStype(s)
+	return ecuo
+}
+
+// SetNillableStype sets the "stype" field if the given value is not nil.
+func (ecuo *EmployeeConfigUpdateOne) SetNillableStype(s *string) *EmployeeConfigUpdateOne {
+	if s != nil {
+		ecuo.SetStype(*s)
+	}
+	return ecuo
+}
+
+// SetTitle sets the "title" field.
+func (ecuo *EmployeeConfigUpdateOne) SetTitle(s string) *EmployeeConfigUpdateOne {
+	ecuo.mutation.SetTitle(s)
+	return ecuo
+}
+
+// SetNillableTitle sets the "title" field if the given value is not nil.
+func (ecuo *EmployeeConfigUpdateOne) SetNillableTitle(s *string) *EmployeeConfigUpdateOne {
+	if s != nil {
+		ecuo.SetTitle(*s)
+	}
+	return ecuo
+}
+
+// SetPhoto sets the "photo" field.
+func (ecuo *EmployeeConfigUpdateOne) SetPhoto(s string) *EmployeeConfigUpdateOne {
+	ecuo.mutation.SetPhoto(s)
+	return ecuo
+}
+
+// SetNillablePhoto sets the "photo" field if the given value is not nil.
+func (ecuo *EmployeeConfigUpdateOne) SetNillablePhoto(s *string) *EmployeeConfigUpdateOne {
+	if s != nil {
+		ecuo.SetPhoto(*s)
+	}
+	return ecuo
+}
+
+// SetOrganizationID sets the "organization_id" field.
+func (ecuo *EmployeeConfigUpdateOne) SetOrganizationID(u uint64) *EmployeeConfigUpdateOne {
+	ecuo.mutation.ResetOrganizationID()
+	ecuo.mutation.SetOrganizationID(u)
+	return ecuo
+}
+
+// SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
+func (ecuo *EmployeeConfigUpdateOne) SetNillableOrganizationID(u *uint64) *EmployeeConfigUpdateOne {
+	if u != nil {
+		ecuo.SetOrganizationID(*u)
+	}
+	return ecuo
+}
+
+// AddOrganizationID adds u to the "organization_id" field.
+func (ecuo *EmployeeConfigUpdateOne) AddOrganizationID(u int64) *EmployeeConfigUpdateOne {
+	ecuo.mutation.AddOrganizationID(u)
+	return ecuo
+}
+
+// ClearOrganizationID clears the value of the "organization_id" field.
+func (ecuo *EmployeeConfigUpdateOne) ClearOrganizationID() *EmployeeConfigUpdateOne {
+	ecuo.mutation.ClearOrganizationID()
+	return ecuo
+}
+
+// Mutation returns the EmployeeConfigMutation object of the builder.
+func (ecuo *EmployeeConfigUpdateOne) Mutation() *EmployeeConfigMutation {
+	return ecuo.mutation
+}
+
+// Where appends a list predicates to the EmployeeConfigUpdate builder.
+func (ecuo *EmployeeConfigUpdateOne) Where(ps ...predicate.EmployeeConfig) *EmployeeConfigUpdateOne {
+	ecuo.mutation.Where(ps...)
+	return ecuo
+}
+
+// Select allows selecting one or more fields (columns) of the returned entity.
+// The default is selecting all fields defined in the entity schema.
+func (ecuo *EmployeeConfigUpdateOne) Select(field string, fields ...string) *EmployeeConfigUpdateOne {
+	ecuo.fields = append([]string{field}, fields...)
+	return ecuo
+}
+
+// Save executes the query and returns the updated EmployeeConfig entity.
+func (ecuo *EmployeeConfigUpdateOne) Save(ctx context.Context) (*EmployeeConfig, error) {
+	if err := ecuo.defaults(); err != nil {
+		return nil, err
+	}
+	return withHooks(ctx, ecuo.sqlSave, ecuo.mutation, ecuo.hooks)
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (ecuo *EmployeeConfigUpdateOne) SaveX(ctx context.Context) *EmployeeConfig {
+	node, err := ecuo.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return node
+}
+
+// Exec executes the query on the entity.
+func (ecuo *EmployeeConfigUpdateOne) Exec(ctx context.Context) error {
+	_, err := ecuo.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (ecuo *EmployeeConfigUpdateOne) ExecX(ctx context.Context) {
+	if err := ecuo.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// defaults sets the default values of the builder before save.
+func (ecuo *EmployeeConfigUpdateOne) defaults() error {
+	if _, ok := ecuo.mutation.UpdatedAt(); !ok {
+		if employeeconfig.UpdateDefaultUpdatedAt == nil {
+			return fmt.Errorf("ent: uninitialized employeeconfig.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
+		}
+		v := employeeconfig.UpdateDefaultUpdatedAt()
+		ecuo.mutation.SetUpdatedAt(v)
+	}
+	return nil
+}
+
+func (ecuo *EmployeeConfigUpdateOne) sqlSave(ctx context.Context) (_node *EmployeeConfig, err error) {
+	_spec := sqlgraph.NewUpdateSpec(employeeconfig.Table, employeeconfig.Columns, sqlgraph.NewFieldSpec(employeeconfig.FieldID, field.TypeUint64))
+	id, ok := ecuo.mutation.ID()
+	if !ok {
+		return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "EmployeeConfig.id" for update`)}
+	}
+	_spec.Node.ID.Value = id
+	if fields := ecuo.fields; len(fields) > 0 {
+		_spec.Node.Columns = make([]string, 0, len(fields))
+		_spec.Node.Columns = append(_spec.Node.Columns, employeeconfig.FieldID)
+		for _, f := range fields {
+			if !employeeconfig.ValidColumn(f) {
+				return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
+			}
+			if f != employeeconfig.FieldID {
+				_spec.Node.Columns = append(_spec.Node.Columns, f)
+			}
+		}
+	}
+	if ps := ecuo.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if value, ok := ecuo.mutation.UpdatedAt(); ok {
+		_spec.SetField(employeeconfig.FieldUpdatedAt, field.TypeTime, value)
+	}
+	if value, ok := ecuo.mutation.DeletedAt(); ok {
+		_spec.SetField(employeeconfig.FieldDeletedAt, field.TypeTime, value)
+	}
+	if ecuo.mutation.DeletedAtCleared() {
+		_spec.ClearField(employeeconfig.FieldDeletedAt, field.TypeTime)
+	}
+	if value, ok := ecuo.mutation.Stype(); ok {
+		_spec.SetField(employeeconfig.FieldStype, field.TypeString, value)
+	}
+	if value, ok := ecuo.mutation.Title(); ok {
+		_spec.SetField(employeeconfig.FieldTitle, field.TypeString, value)
+	}
+	if value, ok := ecuo.mutation.Photo(); ok {
+		_spec.SetField(employeeconfig.FieldPhoto, field.TypeString, value)
+	}
+	if value, ok := ecuo.mutation.OrganizationID(); ok {
+		_spec.SetField(employeeconfig.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if value, ok := ecuo.mutation.AddedOrganizationID(); ok {
+		_spec.AddField(employeeconfig.FieldOrganizationID, field.TypeUint64, value)
+	}
+	if ecuo.mutation.OrganizationIDCleared() {
+		_spec.ClearField(employeeconfig.FieldOrganizationID, field.TypeUint64)
+	}
+	_node = &EmployeeConfig{config: ecuo.config}
+	_spec.Assign = _node.assignValues
+	_spec.ScanValues = _node.scanValues
+	if err = sqlgraph.UpdateNode(ctx, ecuo.driver, _spec); err != nil {
+		if _, ok := err.(*sqlgraph.NotFoundError); ok {
+			err = &NotFoundError{employeeconfig.Label}
+		} else if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return nil, err
+	}
+	ecuo.mutation.done = true
+	return _node, nil
+}

ent/ent.go

@@ -10,8 +10,10 @@ import (
 	"sync"
 	"wechat-api/ent/agent"
 	"wechat-api/ent/batchmsg"
+	"wechat-api/ent/category"
 	"wechat-api/ent/contact"
 	"wechat-api/ent/employee"
+	"wechat-api/ent/employeeconfig"
 	"wechat-api/ent/label"
 	"wechat-api/ent/labelrelationship"
 	"wechat-api/ent/message"
@@ -91,8 +93,10 @@ func checkColumn(table, column string) error {
 		columnCheck = sql.NewColumnCheck(map[string]func(string) bool{
 			agent.Table:             agent.ValidColumn,
 			batchmsg.Table:          batchmsg.ValidColumn,
+			category.Table:          category.ValidColumn,
 			contact.Table:           contact.ValidColumn,
 			employee.Table:          employee.ValidColumn,
+			employeeconfig.Table:    employeeconfig.ValidColumn,
 			label.Table:             label.ValidColumn,
 			labelrelationship.Table: labelrelationship.ValidColumn,
 			message.Table:           message.ValidColumn,

ent/hook/hook.go

@@ -32,6 +32,18 @@ func (f BatchMsgFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, er
 	return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.BatchMsgMutation", m)
 }
 
+// The CategoryFunc type is an adapter to allow the use of ordinary
+// function as Category mutator.
+type CategoryFunc func(context.Context, *ent.CategoryMutation) (ent.Value, error)
+
+// Mutate calls f(ctx, m).
+func (f CategoryFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, error) {
+	if mv, ok := m.(*ent.CategoryMutation); ok {
+		return f(ctx, mv)
+	}
+	return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.CategoryMutation", m)
+}
+
 // The ContactFunc type is an adapter to allow the use of ordinary
 // function as Contact mutator.
 type ContactFunc func(context.Context, *ent.ContactMutation) (ent.Value, error)
@@ -56,6 +68,18 @@ func (f EmployeeFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, er
 	return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.EmployeeMutation", m)
 }
 
+// The EmployeeConfigFunc type is an adapter to allow the use of ordinary
+// function as EmployeeConfig mutator.
+type EmployeeConfigFunc func(context.Context, *ent.EmployeeConfigMutation) (ent.Value, error)
+
+// Mutate calls f(ctx, m).
+func (f EmployeeConfigFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, error) {
+	if mv, ok := m.(*ent.EmployeeConfigMutation); ok {
+		return f(ctx, mv)
+	}
+	return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.EmployeeConfigMutation", m)
+}
+
 // The LabelFunc type is an adapter to allow the use of ordinary
 // function as Label mutator.
 type LabelFunc func(context.Context, *ent.LabelMutation) (ent.Value, error)

ent/intercept/intercept.go

@@ -8,8 +8,10 @@ import (
 	"wechat-api/ent"
 	"wechat-api/ent/agent"
 	"wechat-api/ent/batchmsg"
+	"wechat-api/ent/category"
 	"wechat-api/ent/contact"
 	"wechat-api/ent/employee"
+	"wechat-api/ent/employeeconfig"
 	"wechat-api/ent/label"
 	"wechat-api/ent/labelrelationship"
 	"wechat-api/ent/message"
@@ -138,6 +140,33 @@ func (f TraverseBatchMsg) Traverse(ctx context.Context, q ent.Query) error {
 	return fmt.Errorf("unexpected query type %T. expect *ent.BatchMsgQuery", q)
 }
 
+// The CategoryFunc type is an adapter to allow the use of ordinary function as a Querier.
+type CategoryFunc func(context.Context, *ent.CategoryQuery) (ent.Value, error)
+
+// Query calls f(ctx, q).
+func (f CategoryFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
+	if q, ok := q.(*ent.CategoryQuery); ok {
+		return f(ctx, q)
+	}
+	return nil, fmt.Errorf("unexpected query type %T. expect *ent.CategoryQuery", q)
+}
+
+// The TraverseCategory type is an adapter to allow the use of ordinary function as Traverser.
+type TraverseCategory func(context.Context, *ent.CategoryQuery) error
+
+// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
+func (f TraverseCategory) Intercept(next ent.Querier) ent.Querier {
+	return next
+}
+
+// Traverse calls f(ctx, q).
+func (f TraverseCategory) Traverse(ctx context.Context, q ent.Query) error {
+	if q, ok := q.(*ent.CategoryQuery); ok {
+		return f(ctx, q)
+	}
+	return fmt.Errorf("unexpected query type %T. expect *ent.CategoryQuery", q)
+}
+
 // The ContactFunc type is an adapter to allow the use of ordinary function as a Querier.
 type ContactFunc func(context.Context, *ent.ContactQuery) (ent.Value, error)
 
@@ -192,6 +221,33 @@ func (f TraverseEmployee) Traverse(ctx context.Context, q ent.Query) error {
 	return fmt.Errorf("unexpected query type %T. expect *ent.EmployeeQuery", q)
 }
 
+// The EmployeeConfigFunc type is an adapter to allow the use of ordinary function as a Querier.
+type EmployeeConfigFunc func(context.Context, *ent.EmployeeConfigQuery) (ent.Value, error)
+
+// Query calls f(ctx, q).
+func (f EmployeeConfigFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
+	if q, ok := q.(*ent.EmployeeConfigQuery); ok {
+		return f(ctx, q)
+	}
+	return nil, fmt.Errorf("unexpected query type %T. expect *ent.EmployeeConfigQuery", q)
+}
+
+// The TraverseEmployeeConfig type is an adapter to allow the use of ordinary function as Traverser.
+type TraverseEmployeeConfig func(context.Context, *ent.EmployeeConfigQuery) error
+
+// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
+func (f TraverseEmployeeConfig) Intercept(next ent.Querier) ent.Querier {
+	return next
+}
+
+// Traverse calls f(ctx, q).
+func (f TraverseEmployeeConfig) Traverse(ctx context.Context, q ent.Query) error {
+	if q, ok := q.(*ent.EmployeeConfigQuery); ok {
+		return f(ctx, q)
+	}
+	return fmt.Errorf("unexpected query type %T. expect *ent.EmployeeConfigQuery", q)
+}
+
 // The LabelFunc type is an adapter to allow the use of ordinary function as a Querier.
 type LabelFunc func(context.Context, *ent.LabelQuery) (ent.Value, error)
 
@@ -550,10 +606,14 @@ func NewQuery(q ent.Query) (Query, error) {
 		return &query[*ent.AgentQuery, predicate.Agent, agent.OrderOption]{typ: ent.TypeAgent, tq: q}, nil
 	case *ent.BatchMsgQuery:
 		return &query[*ent.BatchMsgQuery, predicate.BatchMsg, batchmsg.OrderOption]{typ: ent.TypeBatchMsg, tq: q}, nil
+	case *ent.CategoryQuery:
+		return &query[*ent.CategoryQuery, predicate.Category, category.OrderOption]{typ: ent.TypeCategory, tq: q}, nil
 	case *ent.ContactQuery:
 		return &query[*ent.ContactQuery, predicate.Contact, contact.OrderOption]{typ: ent.TypeContact, tq: q}, nil
 	case *ent.EmployeeQuery:
 		return &query[*ent.EmployeeQuery, predicate.Employee, employee.OrderOption]{typ: ent.TypeEmployee, tq: q}, nil
+	case *ent.EmployeeConfigQuery:
+		return &query[*ent.EmployeeConfigQuery, predicate.EmployeeConfig, employeeconfig.OrderOption]{typ: ent.TypeEmployeeConfig, tq: q}, nil
 	case *ent.LabelQuery:
 		return &query[*ent.LabelQuery, predicate.Label, label.OrderOption]{typ: ent.TypeLabel, tq: q}, nil
 	case *ent.LabelRelationshipQuery:

ent/migrate/schema.go

@@ -21,6 +21,8 @@ var (
 		{Name: "background", Type: field.TypeString, Nullable: true, Size: 1000, Comment: "background | 背景介绍", Default: ""},
 		{Name: "examples", Type: field.TypeString, Nullable: true, Size: 5000, Comment: "examples | 对话案例", Default: ""},
 		{Name: "organization_id", Type: field.TypeUint64, Comment: "organization_id | 租户ID"},
+		{Name: "dataset_id", Type: field.TypeString, Size: 255, Comment: "dataset_id | 知识库ID", Default: ""},
+		{Name: "collection_id", Type: field.TypeString, Size: 255, Comment: "collection_id | 集合ID", Default: ""},
 	}
 	// AgentTable holds the schema information for the "agent" table.
 	AgentTable = &schema.Table{
@@ -73,6 +75,28 @@ var (
 			},
 		},
 	}
+	// CategoryColumns holds the columns for the "category" table.
+	CategoryColumns = []*schema.Column{
+		{Name: "id", Type: field.TypeUint64, Increment: true},
+		{Name: "created_at", Type: field.TypeTime, Comment: "Create Time | 创建日期"},
+		{Name: "updated_at", Type: field.TypeTime, Comment: "Update Time | 修改日期"},
+		{Name: "deleted_at", Type: field.TypeTime, Nullable: true, Comment: "Delete Time | 删除日期"},
+		{Name: "name", Type: field.TypeString, Size: 255, Comment: "name | 角色名称"},
+		{Name: "organization_id", Type: field.TypeUint64, Comment: "organization_id | 租户ID"},
+	}
+	// CategoryTable holds the schema information for the "category" table.
+	CategoryTable = &schema.Table{
+		Name:       "category",
+		Columns:    CategoryColumns,
+		PrimaryKey: []*schema.Column{CategoryColumns[0]},
+		Indexes: []*schema.Index{
+			{
+				Name:    "category_organization_id",
+				Unique:  false,
+				Columns: []*schema.Column{CategoryColumns[5]},
+			},
+		},
+	}
 	// ContactColumns holds the columns for the "contact" table.
 	ContactColumns = []*schema.Column{
 		{Name: "id", Type: field.TypeUint64, Increment: true},
@@ -145,6 +169,7 @@ var (
 		{Name: "switch_in", Type: field.TypeString, Comment: "switch_in | 支持介入", Default: ""},
 		{Name: "video_url", Type: field.TypeString, Size: 1000, Comment: "video_url | 视频地址", Default: ""},
 		{Name: "organization_id", Type: field.TypeUint64, Comment: "organization_id | 租户ID"},
+		{Name: "category_id", Type: field.TypeUint64, Comment: "category_id | 分类ID"},
 	}
 	// EmployeeTable holds the schema information for the "employee" table.
 	EmployeeTable = &schema.Table{
@@ -159,6 +184,30 @@ var (
 			},
 		},
 	}
+	// EmployeeConfigColumns holds the columns for the "employee_config" table.
+	EmployeeConfigColumns = []*schema.Column{
+		{Name: "id", Type: field.TypeUint64, Increment: true},
+		{Name: "created_at", Type: field.TypeTime, Comment: "Create Time | 创建日期"},
+		{Name: "updated_at", Type: field.TypeTime, Comment: "Update Time | 修改日期"},
+		{Name: "deleted_at", Type: field.TypeTime, Nullable: true, Comment: "Delete Time | 删除日期"},
+		{Name: "stype", Type: field.TypeString, Comment: "类型：scene-场景 switch_in-接入方式", Default: ""},
+		{Name: "title", Type: field.TypeString, Comment: "标题", Default: ""},
+		{Name: "photo", Type: field.TypeString, Comment: "图片地址", Default: ""},
+		{Name: "organization_id", Type: field.TypeUint64, Nullable: true, Comment: "机构 ID", Default: 1},
+	}
+	// EmployeeConfigTable holds the schema information for the "employee_config" table.
+	EmployeeConfigTable = &schema.Table{
+		Name:       "employee_config",
+		Columns:    EmployeeConfigColumns,
+		PrimaryKey: []*schema.Column{EmployeeConfigColumns[0]},
+		Indexes: []*schema.Index{
+			{
+				Name:    "employeeconfig_stype",
+				Unique:  false,
+				Columns: []*schema.Column{EmployeeConfigColumns[4]},
+			},
+		},
+	}
 	// LabelColumns holds the columns for the "label" table.
 	LabelColumns = []*schema.Column{
 		{Name: "id", Type: field.TypeUint64, Increment: true},
@@ -574,6 +623,8 @@ var (
 		{Name: "tel", Type: field.TypeString, Comment: "手机号", Default: ""},
 		{Name: "head_big", Type: field.TypeString, Comment: "微信头像", Default: ""},
 		{Name: "organization_id", Type: field.TypeUint64, Nullable: true, Comment: "机构 ID", Default: 1},
+		{Name: "api_base", Type: field.TypeString, Nullable: true, Comment: "大模型服务地址", Default: ""},
+		{Name: "api_key", Type: field.TypeString, Nullable: true, Comment: "大模型服务密钥", Default: ""},
 		{Name: "agent_id", Type: field.TypeUint64, Comment: "模式ID", Default: 0},
 		{Name: "server_id", Type: field.TypeUint64, Nullable: true, Comment: "服务器id", Default: 0},
 	}
@@ -585,13 +636,13 @@ var (
 		ForeignKeys: []*schema.ForeignKey{
 			{
 				Symbol:     "wx_agent_wx_agent",
-				Columns:    []*schema.Column{WxColumns[14]},
+				Columns:    []*schema.Column{WxColumns[16]},
 				RefColumns: []*schema.Column{AgentColumns[0]},
 				OnDelete:   schema.NoAction,
 			},
 			{
 				Symbol:     "wx_server_wxs",
-				Columns:    []*schema.Column{WxColumns[15]},
+				Columns:    []*schema.Column{WxColumns[17]},
 				RefColumns: []*schema.Column{ServerColumns[0]},
 				OnDelete:   schema.SetNull,
 			},
@@ -600,7 +651,7 @@ var (
 			{
 				Name:    "wx_server_id_port",
 				Unique:  true,
-				Columns: []*schema.Column{WxColumns[15], WxColumns[5]},
+				Columns: []*schema.Column{WxColumns[17], WxColumns[5]},
 			},
 			{
 				Name:    "wx_wxid",
@@ -628,8 +679,10 @@ var (
 	Tables = []*schema.Table{
 		AgentTable,
 		BatchMsgTable,
+		CategoryTable,
 		ContactTable,
 		EmployeeTable,
+		EmployeeConfigTable,
 		LabelTable,
 		LabelRelationshipTable,
 		MessagesTable,
@@ -653,12 +706,18 @@ func init() {
 	BatchMsgTable.Annotation = &entsql.Annotation{
 		Table: "batch_msg",
 	}
+	CategoryTable.Annotation = &entsql.Annotation{
+		Table: "category",
+	}
 	ContactTable.Annotation = &entsql.Annotation{
 		Table: "contact",
 	}
 	EmployeeTable.Annotation = &entsql.Annotation{
 		Table: "employee",
 	}
+	EmployeeConfigTable.Annotation = &entsql.Annotation{
+		Table: "employee_config",
+	}
 	LabelTable.Annotation = &entsql.Annotation{
 		Table: "label",
 	}

ent/pagination.go

@@ -7,8 +7,10 @@ import (
 	"fmt"
 	"wechat-api/ent/agent"
 	"wechat-api/ent/batchmsg"
+	"wechat-api/ent/category"
 	"wechat-api/ent/contact"
 	"wechat-api/ent/employee"
+	"wechat-api/ent/employeeconfig"
 	"wechat-api/ent/label"
 	"wechat-api/ent/labelrelationship"
 	"wechat-api/ent/message"
@@ -232,6 +234,87 @@ func (bm *BatchMsgQuery) Page(
 	return ret, nil
 }
 
+type CategoryPager struct {
+	Order  category.OrderOption
+	Filter func(*CategoryQuery) (*CategoryQuery, error)
+}
+
+// CategoryPaginateOption enables pagination customization.
+type CategoryPaginateOption func(*CategoryPager)
+
+// DefaultCategoryOrder is the default ordering of Category.
+var DefaultCategoryOrder = Desc(category.FieldID)
+
+func newCategoryPager(opts []CategoryPaginateOption) (*CategoryPager, error) {
+	pager := &CategoryPager{}
+	for _, opt := range opts {
+		opt(pager)
+	}
+	if pager.Order == nil {
+		pager.Order = DefaultCategoryOrder
+	}
+	return pager, nil
+}
+
+func (p *CategoryPager) ApplyFilter(query *CategoryQuery) (*CategoryQuery, error) {
+	if p.Filter != nil {
+		return p.Filter(query)
+	}
+	return query, nil
+}
+
+// CategoryPageList is Category PageList result.
+type CategoryPageList struct {
+	List        []*Category  `json:"list"`
+	PageDetails *PageDetails `json:"pageDetails"`
+}
+
+func (c *CategoryQuery) Page(
+	ctx context.Context, pageNum uint64, pageSize uint64, opts ...CategoryPaginateOption,
+) (*CategoryPageList, error) {
+
+	pager, err := newCategoryPager(opts)
+	if err != nil {
+		return nil, err
+	}
+
+	if c, err = pager.ApplyFilter(c); err != nil {
+		return nil, err
+	}
+
+	ret := &CategoryPageList{}
+
+	ret.PageDetails = &PageDetails{
+		Page: pageNum,
+		Size: pageSize,
+	}
+
+	query := c.Clone()
+	query.ctx.Fields = nil
+	count, err := query.Count(ctx)
+
+	if err != nil {
+		return nil, err
+	}
+
+	ret.PageDetails.Total = uint64(count)
+
+	if pager.Order != nil {
+		c = c.Order(pager.Order)
+	} else {
+		c = c.Order(DefaultCategoryOrder)
+	}
+
+	c = c.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
+	list, err := c.All(ctx)
+	if err != nil {
+		return nil, err
+	}
+	ret.List = list
+
+	return ret, nil
+}
+
 type ContactPager struct {
 	Order  contact.OrderOption
 	Filter func(*ContactQuery) (*ContactQuery, error)
@@ -394,6 +477,87 @@ func (e *EmployeeQuery) Page(
 	return ret, nil
 }
 
+type EmployeeConfigPager struct {
+	Order  employeeconfig.OrderOption
+	Filter func(*EmployeeConfigQuery) (*EmployeeConfigQuery, error)
+}
+
+// EmployeeConfigPaginateOption enables pagination customization.
+type EmployeeConfigPaginateOption func(*EmployeeConfigPager)
+
+// DefaultEmployeeConfigOrder is the default ordering of EmployeeConfig.
+var DefaultEmployeeConfigOrder = Desc(employeeconfig.FieldID)
+
+func newEmployeeConfigPager(opts []EmployeeConfigPaginateOption) (*EmployeeConfigPager, error) {
+	pager := &EmployeeConfigPager{}
+	for _, opt := range opts {
+		opt(pager)
+	}
+	if pager.Order == nil {
+		pager.Order = DefaultEmployeeConfigOrder
+	}
+	return pager, nil
+}
+
+func (p *EmployeeConfigPager) ApplyFilter(query *EmployeeConfigQuery) (*EmployeeConfigQuery, error) {
+	if p.Filter != nil {
+		return p.Filter(query)
+	}
+	return query, nil
+}
+
+// EmployeeConfigPageList is EmployeeConfig PageList result.
+type EmployeeConfigPageList struct {
+	List        []*EmployeeConfig `json:"list"`
+	PageDetails *PageDetails      `json:"pageDetails"`
+}
+
+func (ec *EmployeeConfigQuery) Page(
+	ctx context.Context, pageNum uint64, pageSize uint64, opts ...EmployeeConfigPaginateOption,
+) (*EmployeeConfigPageList, error) {
+
+	pager, err := newEmployeeConfigPager(opts)
+	if err != nil {
+		return nil, err
+	}
+
+	if ec, err = pager.ApplyFilter(ec); err != nil {
+		return nil, err
+	}
+
+	ret := &EmployeeConfigPageList{}
+
+	ret.PageDetails = &PageDetails{
+		Page: pageNum,
+		Size: pageSize,
+	}
+
+	query := ec.Clone()
+	query.ctx.Fields = nil
+	count, err := query.Count(ctx)
+
+	if err != nil {
+		return nil, err
+	}
+
+	ret.PageDetails.Total = uint64(count)
+
+	if pager.Order != nil {
+		ec = ec.Order(pager.Order)
+	} else {
+		ec = ec.Order(DefaultEmployeeConfigOrder)
+	}
+
+	ec = ec.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
+	list, err := ec.All(ctx)
+	if err != nil {
+		return nil, err
+	}
+	ret.List = list
+
+	return ret, nil
+}
+
 type LabelPager struct {
 	Order  label.OrderOption
 	Filter func(*LabelQuery) (*LabelQuery, error)

ent/predicate/predicate.go

@@ -12,12 +12,18 @@ type Agent func(*sql.Selector)
 // BatchMsg is the predicate function for batchmsg builders.
 type BatchMsg func(*sql.Selector)
 
+// Category is the predicate function for category builders.
+type Category func(*sql.Selector)
+
 // Contact is the predicate function for contact builders.
 type Contact func(*sql.Selector)
 
 // Employee is the predicate function for employee builders.
 type Employee func(*sql.Selector)
 
+// EmployeeConfig is the predicate function for employeeconfig builders.
+type EmployeeConfig func(*sql.Selector)
+
 // Label is the predicate function for label builders.
 type Label func(*sql.Selector)
 

ent/runtime/runtime.go

@@ -6,8 +6,10 @@ import (
 	"time"
 	"wechat-api/ent/agent"
 	"wechat-api/ent/batchmsg"
+	"wechat-api/ent/category"
 	"wechat-api/ent/contact"
 	"wechat-api/ent/employee"
+	"wechat-api/ent/employeeconfig"
 	"wechat-api/ent/label"
 	"wechat-api/ent/labelrelationship"
 	"wechat-api/ent/message"
@@ -77,6 +79,18 @@ func init() {
 	agentDescOrganizationID := agentFields[5].Descriptor()
 	// agent.OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
 	agent.OrganizationIDValidator = agentDescOrganizationID.Validators[0].(func(uint64) error)
+	// agentDescDatasetID is the schema descriptor for dataset_id field.
+	agentDescDatasetID := agentFields[6].Descriptor()
+	// agent.DefaultDatasetID holds the default value on creation for the dataset_id field.
+	agent.DefaultDatasetID = agentDescDatasetID.Default.(string)
+	// agent.DatasetIDValidator is a validator for the "dataset_id" field. It is called by the builders before save.
+	agent.DatasetIDValidator = agentDescDatasetID.Validators[0].(func(string) error)
+	// agentDescCollectionID is the schema descriptor for collection_id field.
+	agentDescCollectionID := agentFields[7].Descriptor()
+	// agent.DefaultCollectionID holds the default value on creation for the collection_id field.
+	agent.DefaultCollectionID = agentDescCollectionID.Default.(string)
+	// agent.CollectionIDValidator is a validator for the "collection_id" field. It is called by the builders before save.
+	agent.CollectionIDValidator = agentDescCollectionID.Validators[0].(func(string) error)
 	batchmsgMixin := schema.BatchMsg{}.Mixin()
 	batchmsgMixinHooks1 := batchmsgMixin[1].Hooks()
 	batchmsg.Hooks[0] = batchmsgMixinHooks1[0]
@@ -104,6 +118,33 @@ func init() {
 	batchmsgDescOrganizationID := batchmsgFields[12].Descriptor()
 	// batchmsg.OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
 	batchmsg.OrganizationIDValidator = batchmsgDescOrganizationID.Validators[0].(func(uint64) error)
+	categoryMixin := schema.Category{}.Mixin()
+	categoryMixinHooks1 := categoryMixin[1].Hooks()
+	category.Hooks[0] = categoryMixinHooks1[0]
+	categoryMixinInters1 := categoryMixin[1].Interceptors()
+	category.Interceptors[0] = categoryMixinInters1[0]
+	categoryMixinFields0 := categoryMixin[0].Fields()
+	_ = categoryMixinFields0
+	categoryFields := schema.Category{}.Fields()
+	_ = categoryFields
+	// categoryDescCreatedAt is the schema descriptor for created_at field.
+	categoryDescCreatedAt := categoryMixinFields0[1].Descriptor()
+	// category.DefaultCreatedAt holds the default value on creation for the created_at field.
+	category.DefaultCreatedAt = categoryDescCreatedAt.Default.(func() time.Time)
+	// categoryDescUpdatedAt is the schema descriptor for updated_at field.
+	categoryDescUpdatedAt := categoryMixinFields0[2].Descriptor()
+	// category.DefaultUpdatedAt holds the default value on creation for the updated_at field.
+	category.DefaultUpdatedAt = categoryDescUpdatedAt.Default.(func() time.Time)
+	// category.UpdateDefaultUpdatedAt holds the default value on update for the updated_at field.
+	category.UpdateDefaultUpdatedAt = categoryDescUpdatedAt.UpdateDefault.(func() time.Time)
+	// categoryDescName is the schema descriptor for name field.
+	categoryDescName := categoryFields[0].Descriptor()
+	// category.NameValidator is a validator for the "name" field. It is called by the builders before save.
+	category.NameValidator = categoryDescName.Validators[0].(func(string) error)
+	// categoryDescOrganizationID is the schema descriptor for organization_id field.
+	categoryDescOrganizationID := categoryFields[1].Descriptor()
+	// category.OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
+	category.OrganizationIDValidator = categoryDescOrganizationID.Validators[0].(func(uint64) error)
 	contactMixin := schema.Contact{}.Mixin()
 	contactMixinHooks2 := contactMixin[2].Hooks()
 	contact.Hooks[0] = contactMixinHooks2[0]
@@ -282,6 +323,45 @@ func init() {
 	employeeDescOrganizationID := employeeFields[13].Descriptor()
 	// employee.OrganizationIDValidator is a validator for the "organization_id" field. It is called by the builders before save.
 	employee.OrganizationIDValidator = employeeDescOrganizationID.Validators[0].(func(uint64) error)
+	// employeeDescCategoryID is the schema descriptor for category_id field.
+	employeeDescCategoryID := employeeFields[14].Descriptor()
+	// employee.CategoryIDValidator is a validator for the "category_id" field. It is called by the builders before save.
+	employee.CategoryIDValidator = employeeDescCategoryID.Validators[0].(func(uint64) error)
+	employeeconfigMixin := schema.EmployeeConfig{}.Mixin()
+	employeeconfigMixinHooks1 := employeeconfigMixin[1].Hooks()
+	employeeconfig.Hooks[0] = employeeconfigMixinHooks1[0]
+	employeeconfigMixinInters1 := employeeconfigMixin[1].Interceptors()
+	employeeconfig.Interceptors[0] = employeeconfigMixinInters1[0]
+	employeeconfigMixinFields0 := employeeconfigMixin[0].Fields()
+	_ = employeeconfigMixinFields0
+	employeeconfigFields := schema.EmployeeConfig{}.Fields()
+	_ = employeeconfigFields
+	// employeeconfigDescCreatedAt is the schema descriptor for created_at field.
+	employeeconfigDescCreatedAt := employeeconfigMixinFields0[1].Descriptor()
+	// employeeconfig.DefaultCreatedAt holds the default value on creation for the created_at field.
+	employeeconfig.DefaultCreatedAt = employeeconfigDescCreatedAt.Default.(func() time.Time)
+	// employeeconfigDescUpdatedAt is the schema descriptor for updated_at field.
+	employeeconfigDescUpdatedAt := employeeconfigMixinFields0[2].Descriptor()
+	// employeeconfig.DefaultUpdatedAt holds the default value on creation for the updated_at field.
+	employeeconfig.DefaultUpdatedAt = employeeconfigDescUpdatedAt.Default.(func() time.Time)
+	// employeeconfig.UpdateDefaultUpdatedAt holds the default value on update for the updated_at field.
+	employeeconfig.UpdateDefaultUpdatedAt = employeeconfigDescUpdatedAt.UpdateDefault.(func() time.Time)
+	// employeeconfigDescStype is the schema descriptor for stype field.
+	employeeconfigDescStype := employeeconfigFields[0].Descriptor()
+	// employeeconfig.DefaultStype holds the default value on creation for the stype field.
+	employeeconfig.DefaultStype = employeeconfigDescStype.Default.(string)
+	// employeeconfigDescTitle is the schema descriptor for title field.
+	employeeconfigDescTitle := employeeconfigFields[1].Descriptor()
+	// employeeconfig.DefaultTitle holds the default value on creation for the title field.
+	employeeconfig.DefaultTitle = employeeconfigDescTitle.Default.(string)
+	// employeeconfigDescPhoto is the schema descriptor for photo field.
+	employeeconfigDescPhoto := employeeconfigFields[2].Descriptor()
+	// employeeconfig.DefaultPhoto holds the default value on creation for the photo field.
+	employeeconfig.DefaultPhoto = employeeconfigDescPhoto.Default.(string)
+	// employeeconfigDescOrganizationID is the schema descriptor for organization_id field.
+	employeeconfigDescOrganizationID := employeeconfigFields[3].Descriptor()
+	// employeeconfig.DefaultOrganizationID holds the default value on creation for the organization_id field.
+	employeeconfig.DefaultOrganizationID = employeeconfigDescOrganizationID.Default.(uint64)
 	labelMixin := schema.Label{}.Mixin()
 	labelMixinFields0 := labelMixin[0].Fields()
 	_ = labelMixinFields0
@@ -740,6 +820,14 @@ func init() {
 	wxDescAgentID := wxFields[10].Descriptor()
 	// wx.DefaultAgentID holds the default value on creation for the agent_id field.
 	wx.DefaultAgentID = wxDescAgentID.Default.(uint64)
+	// wxDescAPIBase is the schema descriptor for api_base field.
+	wxDescAPIBase := wxFields[11].Descriptor()
+	// wx.DefaultAPIBase holds the default value on creation for the api_base field.
+	wx.DefaultAPIBase = wxDescAPIBase.Default.(string)
+	// wxDescAPIKey is the schema descriptor for api_key field.
+	wxDescAPIKey := wxFields[12].Descriptor()
+	// wx.DefaultAPIKey holds the default value on creation for the api_key field.
+	wx.DefaultAPIKey = wxDescAPIKey.Default.(string)
 }
 
 const (

ent/schema/agent.go

@@ -24,6 +24,8 @@ func (Agent) Fields() []ent.Field {
 		field.String("background").Optional().Default("").MaxLen(1000).Comment("background | 背景介绍"),
 		field.String("examples").Optional().Default("").MaxLen(5000).Comment("examples | 对话案例"),
 		field.Uint64("organization_id").Positive().Comment("organization_id | 租户ID"),
+		field.String("dataset_id").Default("").MaxLen(255).Comment("dataset_id | 知识库ID"),
+		field.String("collection_id").Default("").MaxLen(255).Comment("collection_id | 集合ID"),
 	}
 }
 

ent/schema/category.go

@@ -0,0 +1,47 @@
+package schema
+
+import (
+	"entgo.io/ent/schema/index"
+	"wechat-api/ent/schema/localmixin"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/entsql"
+	"entgo.io/ent/schema"
+	"entgo.io/ent/schema/field"
+	"github.com/suyuan32/simple-admin-common/orm/ent/mixins"
+)
+
+type Category struct {
+	ent.Schema
+}
+
+func (Category) Fields() []ent.Field {
+	return []ent.Field{
+		field.String("name").MaxLen(255).Comment("name | 角色名称"),
+		field.Uint64("organization_id").Positive().Comment("organization_id | 租户ID"),
+	}
+}
+
+func (Category) Mixin() []ent.Mixin {
+	return []ent.Mixin{
+		mixins.IDMixin{},
+		localmixin.SoftDeleteMixin{},
+	}
+}
+
+func (Category) Edges() []ent.Edge {
+	return []ent.Edge{}
+}
+
+func (Category) Indexes() []ent.Index {
+	return []ent.Index{
+		index.Fields("organization_id"),
+	}
+}
+
+func (Category) Annotations() []schema.Annotation {
+	return []schema.Annotation{
+		entsql.WithComments(true),
+		entsql.Annotation{Table: "category"},
+	}
+}

ent/schema/employee.go

@@ -32,6 +32,7 @@ func (Employee) Fields() []ent.Field {
 		field.String("switch_in").Default("").Comment("switch_in | 支持介入"),
 		field.String("video_url").MaxLen(1000).Default("").Comment("video_url | 视频地址"),
 		field.Uint64("organization_id").Positive().Comment("organization_id | 租户ID"),
+		field.Uint64("category_id").Positive().Comment("category_id | 分类ID"),
 	}
 }
 

ent/schema/employee_config.go

@@ -0,0 +1,48 @@
+package schema
+
+import (
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/entsql"
+	"entgo.io/ent/schema"
+	"entgo.io/ent/schema/field"
+	"entgo.io/ent/schema/index"
+	"github.com/suyuan32/simple-admin-common/orm/ent/mixins"
+	"wechat-api/ent/schema/localmixin"
+)
+
+type EmployeeConfig struct {
+	ent.Schema
+}
+
+func (EmployeeConfig) Fields() []ent.Field {
+	return []ent.Field{
+		field.String("stype").Default("").Comment("类型：scene-场景 switch_in-接入方式"),
+		field.String("title").Default("").Comment("标题"),
+		field.String("photo").Default("").Comment("图片地址"),
+		field.Uint64("organization_id").Optional().Default(1).Comment("机构 ID"),
+	}
+}
+
+func (EmployeeConfig) Mixin() []ent.Mixin {
+	return []ent.Mixin{
+		mixins.IDMixin{},
+		localmixin.SoftDeleteMixin{},
+	}
+}
+
+func (EmployeeConfig) Indexes() []ent.Index {
+	return []ent.Index{
+		index.Fields("stype"),
+	}
+}
+
+func (EmployeeConfig) Edges() []ent.Edge {
+	return nil
+}
+
+func (EmployeeConfig) Annotations() []schema.Annotation {
+	return []schema.Annotation{
+		entsql.WithComments(true),
+		entsql.Annotation{Table: "employee_config"},
+	}
+}

ent/schema/wx.go

@@ -52,6 +52,12 @@ func (Wx) Fields() []ent.Field {
 		field.Uint64("agent_id").Default(0).
 			Comment("模式ID").
 			Annotations(entsql.WithComments(true)),
+		field.String("api_base").Optional().Default("").
+			Annotations(entsql.WithComments(true)).
+			Comment("大模型服务地址"),
+		field.String("api_key").Optional().Default("").
+			Annotations(entsql.WithComments(true)).
+			Comment("大模型服务密钥"),
 	}
 }
 

ent/set_not_nil.go

@@ -200,6 +200,54 @@ func (a *AgentCreate) SetNotNilOrganizationID(value *uint64) *AgentCreate {
 }
 
 // set field if value's pointer is not nil.
+func (a *AgentUpdate) SetNotNilDatasetID(value *string) *AgentUpdate {
+	if value != nil {
+		return a.SetDatasetID(*value)
+	}
+	return a
+}
+
+// set field if value's pointer is not nil.
+func (a *AgentUpdateOne) SetNotNilDatasetID(value *string) *AgentUpdateOne {
+	if value != nil {
+		return a.SetDatasetID(*value)
+	}
+	return a
+}
+
+// set field if value's pointer is not nil.
+func (a *AgentCreate) SetNotNilDatasetID(value *string) *AgentCreate {
+	if value != nil {
+		return a.SetDatasetID(*value)
+	}
+	return a
+}
+
+// set field if value's pointer is not nil.
+func (a *AgentUpdate) SetNotNilCollectionID(value *string) *AgentUpdate {
+	if value != nil {
+		return a.SetCollectionID(*value)
+	}
+	return a
+}
+
+// set field if value's pointer is not nil.
+func (a *AgentUpdateOne) SetNotNilCollectionID(value *string) *AgentUpdateOne {
+	if value != nil {
+		return a.SetCollectionID(*value)
+	}
+	return a
+}
+
+// set field if value's pointer is not nil.
+func (a *AgentCreate) SetNotNilCollectionID(value *string) *AgentCreate {
+	if value != nil {
+		return a.SetCollectionID(*value)
+	}
+	return a
+}
+
+// set field if value's pointer is not nil.
 func (bm *BatchMsgUpdate) SetNotNilUpdatedAt(value *time.Time) *BatchMsgUpdate {
 	if value != nil {
 		return bm.SetUpdatedAt(*value)
@@ -560,6 +608,102 @@ func (bm *BatchMsgCreate) SetNotNilOrganizationID(value *uint64) *BatchMsgCreate
 }
 
 // set field if value's pointer is not nil.
+func (c *CategoryUpdate) SetNotNilUpdatedAt(value *time.Time) *CategoryUpdate {
+	if value != nil {
+		return c.SetUpdatedAt(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdateOne) SetNotNilUpdatedAt(value *time.Time) *CategoryUpdateOne {
+	if value != nil {
+		return c.SetUpdatedAt(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryCreate) SetNotNilUpdatedAt(value *time.Time) *CategoryCreate {
+	if value != nil {
+		return c.SetUpdatedAt(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdate) SetNotNilDeletedAt(value *time.Time) *CategoryUpdate {
+	if value != nil {
+		return c.SetDeletedAt(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdateOne) SetNotNilDeletedAt(value *time.Time) *CategoryUpdateOne {
+	if value != nil {
+		return c.SetDeletedAt(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryCreate) SetNotNilDeletedAt(value *time.Time) *CategoryCreate {
+	if value != nil {
+		return c.SetDeletedAt(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdate) SetNotNilName(value *string) *CategoryUpdate {
+	if value != nil {
+		return c.SetName(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdateOne) SetNotNilName(value *string) *CategoryUpdateOne {
+	if value != nil {
+		return c.SetName(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryCreate) SetNotNilName(value *string) *CategoryCreate {
+	if value != nil {
+		return c.SetName(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdate) SetNotNilOrganizationID(value *uint64) *CategoryUpdate {
+	if value != nil {
+		return c.SetOrganizationID(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryUpdateOne) SetNotNilOrganizationID(value *uint64) *CategoryUpdateOne {
+	if value != nil {
+		return c.SetOrganizationID(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
+func (c *CategoryCreate) SetNotNilOrganizationID(value *uint64) *CategoryCreate {
+	if value != nil {
+		return c.SetOrganizationID(*value)
+	}
+	return c
+}
+
+// set field if value's pointer is not nil.
 func (c *ContactUpdate) SetNotNilUpdatedAt(value *time.Time) *ContactUpdate {
 	if value != nil {
 		return c.SetUpdatedAt(*value)
@@ -1400,6 +1544,174 @@ func (e *EmployeeCreate) SetNotNilOrganizationID(value *uint64) *EmployeeCreate
 }
 
 // set field if value's pointer is not nil.
+func (e *EmployeeUpdate) SetNotNilCategoryID(value *uint64) *EmployeeUpdate {
+	if value != nil {
+		return e.SetCategoryID(*value)
+	}
+	return e
+}
+
+// set field if value's pointer is not nil.
+func (e *EmployeeUpdateOne) SetNotNilCategoryID(value *uint64) *EmployeeUpdateOne {
+	if value != nil {
+		return e.SetCategoryID(*value)
+	}
+	return e
+}
+
+// set field if value's pointer is not nil.
+func (e *EmployeeCreate) SetNotNilCategoryID(value *uint64) *EmployeeCreate {
+	if value != nil {
+		return e.SetCategoryID(*value)
+	}
+	return e
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdate) SetNotNilUpdatedAt(value *time.Time) *EmployeeConfigUpdate {
+	if value != nil {
+		return ec.SetUpdatedAt(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdateOne) SetNotNilUpdatedAt(value *time.Time) *EmployeeConfigUpdateOne {
+	if value != nil {
+		return ec.SetUpdatedAt(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigCreate) SetNotNilUpdatedAt(value *time.Time) *EmployeeConfigCreate {
+	if value != nil {
+		return ec.SetUpdatedAt(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdate) SetNotNilDeletedAt(value *time.Time) *EmployeeConfigUpdate {
+	if value != nil {
+		return ec.SetDeletedAt(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdateOne) SetNotNilDeletedAt(value *time.Time) *EmployeeConfigUpdateOne {
+	if value != nil {
+		return ec.SetDeletedAt(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigCreate) SetNotNilDeletedAt(value *time.Time) *EmployeeConfigCreate {
+	if value != nil {
+		return ec.SetDeletedAt(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdate) SetNotNilStype(value *string) *EmployeeConfigUpdate {
+	if value != nil {
+		return ec.SetStype(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdateOne) SetNotNilStype(value *string) *EmployeeConfigUpdateOne {
+	if value != nil {
+		return ec.SetStype(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigCreate) SetNotNilStype(value *string) *EmployeeConfigCreate {
+	if value != nil {
+		return ec.SetStype(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdate) SetNotNilTitle(value *string) *EmployeeConfigUpdate {
+	if value != nil {
+		return ec.SetTitle(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdateOne) SetNotNilTitle(value *string) *EmployeeConfigUpdateOne {
+	if value != nil {
+		return ec.SetTitle(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigCreate) SetNotNilTitle(value *string) *EmployeeConfigCreate {
+	if value != nil {
+		return ec.SetTitle(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdate) SetNotNilPhoto(value *string) *EmployeeConfigUpdate {
+	if value != nil {
+		return ec.SetPhoto(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdateOne) SetNotNilPhoto(value *string) *EmployeeConfigUpdateOne {
+	if value != nil {
+		return ec.SetPhoto(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigCreate) SetNotNilPhoto(value *string) *EmployeeConfigCreate {
+	if value != nil {
+		return ec.SetPhoto(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdate) SetNotNilOrganizationID(value *uint64) *EmployeeConfigUpdate {
+	if value != nil {
+		return ec.SetOrganizationID(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigUpdateOne) SetNotNilOrganizationID(value *uint64) *EmployeeConfigUpdateOne {
+	if value != nil {
+		return ec.SetOrganizationID(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
+func (ec *EmployeeConfigCreate) SetNotNilOrganizationID(value *uint64) *EmployeeConfigCreate {
+	if value != nil {
+		return ec.SetOrganizationID(*value)
+	}
+	return ec
+}
+
+// set field if value's pointer is not nil.
 func (l *LabelUpdate) SetNotNilUpdatedAt(value *time.Time) *LabelUpdate {
 	if value != nil {
 		return l.SetUpdatedAt(*value)
@@ -4134,3 +4446,51 @@ func (w *WxCreate) SetNotNilAgentID(value *uint64) *WxCreate {
 	}
 	return w
 }
+
+// set field if value's pointer is not nil.
+func (w *WxUpdate) SetNotNilAPIBase(value *string) *WxUpdate {
+	if value != nil {
+		return w.SetAPIBase(*value)
+	}
+	return w
+}
+
+// set field if value's pointer is not nil.
+func (w *WxUpdateOne) SetNotNilAPIBase(value *string) *WxUpdateOne {
+	if value != nil {
+		return w.SetAPIBase(*value)
+	}
+	return w
+}
+
+// set field if value's pointer is not nil.
+func (w *WxCreate) SetNotNilAPIBase(value *string) *WxCreate {
+	if value != nil {
+		return w.SetAPIBase(*value)
+	}
+	return w
+}
+
+// set field if value's pointer is not nil.
+func (w *WxUpdate) SetNotNilAPIKey(value *string) *WxUpdate {
+	if value != nil {
+		return w.SetAPIKey(*value)
+	}
+	return w
+}
+
+// set field if value's pointer is not nil.
+func (w *WxUpdateOne) SetNotNilAPIKey(value *string) *WxUpdateOne {
+	if value != nil {
+		return w.SetAPIKey(*value)
+	}
+	return w
+}
+
+// set field if value's pointer is not nil.
+func (w *WxCreate) SetNotNilAPIKey(value *string) *WxCreate {
+	if value != nil {
+		return w.SetAPIKey(*value)
+	}
+	return w
+}

ent/tx.go

@@ -18,10 +18,14 @@ type Tx struct {
 	Agent *AgentClient
 	// BatchMsg is the client for interacting with the BatchMsg builders.
 	BatchMsg *BatchMsgClient
+	// Category is the client for interacting with the Category builders.
+	Category *CategoryClient
 	// Contact is the client for interacting with the Contact builders.
 	Contact *ContactClient
 	// Employee is the client for interacting with the Employee builders.
 	Employee *EmployeeClient
+	// EmployeeConfig is the client for interacting with the EmployeeConfig builders.
+	EmployeeConfig *EmployeeConfigClient
 	// Label is the client for interacting with the Label builders.
 	Label *LabelClient
 	// LabelRelationship is the client for interacting with the LabelRelationship builders.
@@ -181,8 +185,10 @@ func (tx *Tx) Client() *Client {
 func (tx *Tx) init() {
 	tx.Agent = NewAgentClient(tx.config)
 	tx.BatchMsg = NewBatchMsgClient(tx.config)
+	tx.Category = NewCategoryClient(tx.config)
 	tx.Contact = NewContactClient(tx.config)
 	tx.Employee = NewEmployeeClient(tx.config)
+	tx.EmployeeConfig = NewEmployeeConfigClient(tx.config)
 	tx.Label = NewLabelClient(tx.config)
 	tx.LabelRelationship = NewLabelRelationshipClient(tx.config)
 	tx.Message = NewMessageClient(tx.config)

ent/wx.go

@@ -49,6 +49,10 @@ type Wx struct {
 	OrganizationID uint64 `json:"organization_id,omitempty"`
 	// 模式ID
 	AgentID uint64 `json:"agent_id,omitempty"`
+	// 大模型服务地址
+	APIBase string `json:"api_base,omitempty"`
+	// 大模型服务密钥
+	APIKey string `json:"api_key,omitempty"`
 	// Edges holds the relations/edges for other nodes in the graph.
 	// The values are being populated by the WxQuery when eager-loading is set.
 	Edges        WxEdges `json:"edges"`
@@ -95,7 +99,7 @@ func (*Wx) scanValues(columns []string) ([]any, error) {
 		switch columns[i] {
 		case wx.FieldID, wx.FieldStatus, wx.FieldServerID, wx.FieldOrganizationID, wx.FieldAgentID:
 			values[i] = new(sql.NullInt64)
-		case wx.FieldPort, wx.FieldProcessID, wx.FieldCallback, wx.FieldWxid, wx.FieldAccount, wx.FieldNickname, wx.FieldTel, wx.FieldHeadBig:
+		case wx.FieldPort, wx.FieldProcessID, wx.FieldCallback, wx.FieldWxid, wx.FieldAccount, wx.FieldNickname, wx.FieldTel, wx.FieldHeadBig, wx.FieldAPIBase, wx.FieldAPIKey:
 			values[i] = new(sql.NullString)
 		case wx.FieldCreatedAt, wx.FieldUpdatedAt, wx.FieldDeletedAt:
 			values[i] = new(sql.NullTime)
@@ -210,6 +214,18 @@ func (w *Wx) assignValues(columns []string, values []any) error {
 			} else if value.Valid {
 				w.AgentID = uint64(value.Int64)
 			}
+		case wx.FieldAPIBase:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field api_base", values[i])
+			} else if value.Valid {
+				w.APIBase = value.String
+			}
+		case wx.FieldAPIKey:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field api_key", values[i])
+			} else if value.Valid {
+				w.APIKey = value.String
+			}
 		default:
 			w.selectValues.Set(columns[i], values[i])
 		}
@@ -300,6 +316,12 @@ func (w *Wx) String() string {
 	builder.WriteString(", ")
 	builder.WriteString("agent_id=")
 	builder.WriteString(fmt.Sprintf("%v", w.AgentID))
+	builder.WriteString(", ")
+	builder.WriteString("api_base=")
+	builder.WriteString(w.APIBase)
+	builder.WriteString(", ")
+	builder.WriteString("api_key=")
+	builder.WriteString(w.APIKey)
 	builder.WriteByte(')')
 	return builder.String()
 }

ent/wx/where.go

@@ -130,6 +130,16 @@ func AgentID(v uint64) predicate.Wx {
 	return predicate.Wx(sql.FieldEQ(FieldAgentID, v))
 }
 
+// APIBase applies equality check predicate on the "api_base" field. It's identical to APIBaseEQ.
+func APIBase(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldEQ(FieldAPIBase, v))
+}
+
+// APIKey applies equality check predicate on the "api_key" field. It's identical to APIKeyEQ.
+func APIKey(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldEQ(FieldAPIKey, v))
+}
+
 // CreatedAtEQ applies the EQ predicate on the "created_at" field.
 func CreatedAtEQ(v time.Time) predicate.Wx {
 	return predicate.Wx(sql.FieldEQ(FieldCreatedAt, v))
@@ -930,6 +940,156 @@ func AgentIDNotIn(vs ...uint64) predicate.Wx {
 	return predicate.Wx(sql.FieldNotIn(FieldAgentID, vs...))
 }
 
+// APIBaseEQ applies the EQ predicate on the "api_base" field.
+func APIBaseEQ(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldEQ(FieldAPIBase, v))
+}
+
+// APIBaseNEQ applies the NEQ predicate on the "api_base" field.
+func APIBaseNEQ(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldNEQ(FieldAPIBase, v))
+}
+
+// APIBaseIn applies the In predicate on the "api_base" field.
+func APIBaseIn(vs ...string) predicate.Wx {
+	return predicate.Wx(sql.FieldIn(FieldAPIBase, vs...))
+}
+
+// APIBaseNotIn applies the NotIn predicate on the "api_base" field.
+func APIBaseNotIn(vs ...string) predicate.Wx {
+	return predicate.Wx(sql.FieldNotIn(FieldAPIBase, vs...))
+}
+
+// APIBaseGT applies the GT predicate on the "api_base" field.
+func APIBaseGT(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldGT(FieldAPIBase, v))
+}
+
+// APIBaseGTE applies the GTE predicate on the "api_base" field.
+func APIBaseGTE(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldGTE(FieldAPIBase, v))
+}
+
+// APIBaseLT applies the LT predicate on the "api_base" field.
+func APIBaseLT(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldLT(FieldAPIBase, v))
+}
+
+// APIBaseLTE applies the LTE predicate on the "api_base" field.
+func APIBaseLTE(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldLTE(FieldAPIBase, v))
+}
+
+// APIBaseContains applies the Contains predicate on the "api_base" field.
+func APIBaseContains(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldContains(FieldAPIBase, v))
+}
+
+// APIBaseHasPrefix applies the HasPrefix predicate on the "api_base" field.
+func APIBaseHasPrefix(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldHasPrefix(FieldAPIBase, v))
+}
+
+// APIBaseHasSuffix applies the HasSuffix predicate on the "api_base" field.
+func APIBaseHasSuffix(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldHasSuffix(FieldAPIBase, v))
+}
+
+// APIBaseIsNil applies the IsNil predicate on the "api_base" field.
+func APIBaseIsNil() predicate.Wx {
+	return predicate.Wx(sql.FieldIsNull(FieldAPIBase))
+}
+
+// APIBaseNotNil applies the NotNil predicate on the "api_base" field.
+func APIBaseNotNil() predicate.Wx {
+	return predicate.Wx(sql.FieldNotNull(FieldAPIBase))
+}
+
+// APIBaseEqualFold applies the EqualFold predicate on the "api_base" field.
+func APIBaseEqualFold(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldEqualFold(FieldAPIBase, v))
+}
+
+// APIBaseContainsFold applies the ContainsFold predicate on the "api_base" field.
+func APIBaseContainsFold(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldContainsFold(FieldAPIBase, v))
+}
+
+// APIKeyEQ applies the EQ predicate on the "api_key" field.
+func APIKeyEQ(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldEQ(FieldAPIKey, v))
+}
+
+// APIKeyNEQ applies the NEQ predicate on the "api_key" field.
+func APIKeyNEQ(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldNEQ(FieldAPIKey, v))
+}
+
+// APIKeyIn applies the In predicate on the "api_key" field.
+func APIKeyIn(vs ...string) predicate.Wx {
+	return predicate.Wx(sql.FieldIn(FieldAPIKey, vs...))
+}
+
+// APIKeyNotIn applies the NotIn predicate on the "api_key" field.
+func APIKeyNotIn(vs ...string) predicate.Wx {
+	return predicate.Wx(sql.FieldNotIn(FieldAPIKey, vs...))
+}
+
+// APIKeyGT applies the GT predicate on the "api_key" field.
+func APIKeyGT(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldGT(FieldAPIKey, v))
+}
+
+// APIKeyGTE applies the GTE predicate on the "api_key" field.
+func APIKeyGTE(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldGTE(FieldAPIKey, v))
+}
+
+// APIKeyLT applies the LT predicate on the "api_key" field.
+func APIKeyLT(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldLT(FieldAPIKey, v))
+}
+
+// APIKeyLTE applies the LTE predicate on the "api_key" field.
+func APIKeyLTE(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldLTE(FieldAPIKey, v))
+}
+
+// APIKeyContains applies the Contains predicate on the "api_key" field.
+func APIKeyContains(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldContains(FieldAPIKey, v))
+}
+
+// APIKeyHasPrefix applies the HasPrefix predicate on the "api_key" field.
+func APIKeyHasPrefix(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldHasPrefix(FieldAPIKey, v))
+}
+
+// APIKeyHasSuffix applies the HasSuffix predicate on the "api_key" field.
+func APIKeyHasSuffix(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldHasSuffix(FieldAPIKey, v))
+}
+
+// APIKeyIsNil applies the IsNil predicate on the "api_key" field.
+func APIKeyIsNil() predicate.Wx {
+	return predicate.Wx(sql.FieldIsNull(FieldAPIKey))
+}
+
+// APIKeyNotNil applies the NotNil predicate on the "api_key" field.
+func APIKeyNotNil() predicate.Wx {
+	return predicate.Wx(sql.FieldNotNull(FieldAPIKey))
+}
+
+// APIKeyEqualFold applies the EqualFold predicate on the "api_key" field.
+func APIKeyEqualFold(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldEqualFold(FieldAPIKey, v))
+}
+
+// APIKeyContainsFold applies the ContainsFold predicate on the "api_key" field.
+func APIKeyContainsFold(v string) predicate.Wx {
+	return predicate.Wx(sql.FieldContainsFold(FieldAPIKey, v))
+}
+
 // HasServer applies the HasEdge predicate on the "server" edge.
 func HasServer() predicate.Wx {
 	return predicate.Wx(func(s *sql.Selector) {

ent/wx/wx.go

@@ -45,6 +45,10 @@ const (
 	FieldOrganizationID = "organization_id"
 	// FieldAgentID holds the string denoting the agent_id field in the database.
 	FieldAgentID = "agent_id"
+	// FieldAPIBase holds the string denoting the api_base field in the database.
+	FieldAPIBase = "api_base"
+	// FieldAPIKey holds the string denoting the api_key field in the database.
+	FieldAPIKey = "api_key"
 	// EdgeServer holds the string denoting the server edge name in mutations.
 	EdgeServer = "server"
 	// EdgeAgent holds the string denoting the agent edge name in mutations.
@@ -85,6 +89,8 @@ var Columns = []string{
 	FieldHeadBig,
 	FieldOrganizationID,
 	FieldAgentID,
+	FieldAPIBase,
+	FieldAPIKey,
 }
 
 // ValidColumn reports if the column name is valid (part of the table columns).
@@ -135,6 +141,10 @@ var (
 	DefaultOrganizationID uint64
 	// DefaultAgentID holds the default value on creation for the "agent_id" field.
 	DefaultAgentID uint64
+	// DefaultAPIBase holds the default value on creation for the "api_base" field.
+	DefaultAPIBase string
+	// DefaultAPIKey holds the default value on creation for the "api_key" field.
+	DefaultAPIKey string
 )
 
 // OrderOption defines the ordering options for the Wx queries.
@@ -220,6 +230,16 @@ func ByAgentID(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldAgentID, opts...).ToFunc()
 }
 
+// ByAPIBase orders the results by the api_base field.
+func ByAPIBase(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldAPIBase, opts...).ToFunc()
+}
+
+// ByAPIKey orders the results by the api_key field.
+func ByAPIKey(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldAPIKey, opts...).ToFunc()
+}
+
 // ByServerField orders the results by server field.
 func ByServerField(field string, opts ...sql.OrderTermOption) OrderOption {
 	return func(s *sql.Selector) {

ent/wx_create.go

@@ -234,6 +234,34 @@ func (wc *WxCreate) SetNillableAgentID(u *uint64) *WxCreate {
 	return wc
 }
 
+// SetAPIBase sets the "api_base" field.
+func (wc *WxCreate) SetAPIBase(s string) *WxCreate {
+	wc.mutation.SetAPIBase(s)
+	return wc
+}
+
+// SetNillableAPIBase sets the "api_base" field if the given value is not nil.
+func (wc *WxCreate) SetNillableAPIBase(s *string) *WxCreate {
+	if s != nil {
+		wc.SetAPIBase(*s)
+	}
+	return wc
+}
+
+// SetAPIKey sets the "api_key" field.
+func (wc *WxCreate) SetAPIKey(s string) *WxCreate {
+	wc.mutation.SetAPIKey(s)
+	return wc
+}
+
+// SetNillableAPIKey sets the "api_key" field if the given value is not nil.
+func (wc *WxCreate) SetNillableAPIKey(s *string) *WxCreate {
+	if s != nil {
+		wc.SetAPIKey(*s)
+	}
+	return wc
+}
+
 // SetID sets the "id" field.
 func (wc *WxCreate) SetID(u uint64) *WxCreate {
 	wc.mutation.SetID(u)
@@ -349,6 +377,14 @@ func (wc *WxCreate) defaults() error {
 		v := wx.DefaultAgentID
 		wc.mutation.SetAgentID(v)
 	}
+	if _, ok := wc.mutation.APIBase(); !ok {
+		v := wx.DefaultAPIBase
+		wc.mutation.SetAPIBase(v)
+	}
+	if _, ok := wc.mutation.APIKey(); !ok {
+		v := wx.DefaultAPIKey
+		wc.mutation.SetAPIKey(v)
+	}
 	return nil
 }
 
@@ -475,6 +511,14 @@ func (wc *WxCreate) createSpec() (*Wx, *sqlgraph.CreateSpec) {
 		_spec.SetField(wx.FieldOrganizationID, field.TypeUint64, value)
 		_node.OrganizationID = value
 	}
+	if value, ok := wc.mutation.APIBase(); ok {
+		_spec.SetField(wx.FieldAPIBase, field.TypeString, value)
+		_node.APIBase = value
+	}
+	if value, ok := wc.mutation.APIKey(); ok {
+		_spec.SetField(wx.FieldAPIKey, field.TypeString, value)
+		_node.APIKey = value
+	}
 	if nodes := wc.mutation.ServerIDs(); len(nodes) > 0 {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.M2O,
@@ -765,6 +809,42 @@ func (u *WxUpsert) UpdateAgentID() *WxUpsert {
 	return u
 }
 
+// SetAPIBase sets the "api_base" field.
+func (u *WxUpsert) SetAPIBase(v string) *WxUpsert {
+	u.Set(wx.FieldAPIBase, v)
+	return u
+}
+
+// UpdateAPIBase sets the "api_base" field to the value that was provided on create.
+func (u *WxUpsert) UpdateAPIBase() *WxUpsert {
+	u.SetExcluded(wx.FieldAPIBase)
+	return u
+}
+
+// ClearAPIBase clears the value of the "api_base" field.
+func (u *WxUpsert) ClearAPIBase() *WxUpsert {
+	u.SetNull(wx.FieldAPIBase)
+	return u
+}
+
+// SetAPIKey sets the "api_key" field.
+func (u *WxUpsert) SetAPIKey(v string) *WxUpsert {
+	u.Set(wx.FieldAPIKey, v)
+	return u
+}
+
+// UpdateAPIKey sets the "api_key" field to the value that was provided on create.
+func (u *WxUpsert) UpdateAPIKey() *WxUpsert {
+	u.SetExcluded(wx.FieldAPIKey)
+	return u
+}
+
+// ClearAPIKey clears the value of the "api_key" field.
+func (u *WxUpsert) ClearAPIKey() *WxUpsert {
+	u.SetNull(wx.FieldAPIKey)
+	return u
+}
+
 // UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
 // Using this option is equivalent to using:
 //
@@ -1054,6 +1134,48 @@ func (u *WxUpsertOne) UpdateAgentID() *WxUpsertOne {
 	})
 }
 
+// SetAPIBase sets the "api_base" field.
+func (u *WxUpsertOne) SetAPIBase(v string) *WxUpsertOne {
+	return u.Update(func(s *WxUpsert) {
+		s.SetAPIBase(v)
+	})
+}
+
+// UpdateAPIBase sets the "api_base" field to the value that was provided on create.
+func (u *WxUpsertOne) UpdateAPIBase() *WxUpsertOne {
+	return u.Update(func(s *WxUpsert) {
+		s.UpdateAPIBase()
+	})
+}
+
+// ClearAPIBase clears the value of the "api_base" field.
+func (u *WxUpsertOne) ClearAPIBase() *WxUpsertOne {
+	return u.Update(func(s *WxUpsert) {
+		s.ClearAPIBase()
+	})
+}
+
+// SetAPIKey sets the "api_key" field.
+func (u *WxUpsertOne) SetAPIKey(v string) *WxUpsertOne {
+	return u.Update(func(s *WxUpsert) {
+		s.SetAPIKey(v)
+	})
+}
+
+// UpdateAPIKey sets the "api_key" field to the value that was provided on create.
+func (u *WxUpsertOne) UpdateAPIKey() *WxUpsertOne {
+	return u.Update(func(s *WxUpsert) {
+		s.UpdateAPIKey()
+	})
+}
+
+// ClearAPIKey clears the value of the "api_key" field.
+func (u *WxUpsertOne) ClearAPIKey() *WxUpsertOne {
+	return u.Update(func(s *WxUpsert) {
+		s.ClearAPIKey()
+	})
+}
+
 // Exec executes the query.
 func (u *WxUpsertOne) Exec(ctx context.Context) error {
 	if len(u.create.conflict) == 0 {
@@ -1509,6 +1631,48 @@ func (u *WxUpsertBulk) UpdateAgentID() *WxUpsertBulk {
 	})
 }
 
+// SetAPIBase sets the "api_base" field.
+func (u *WxUpsertBulk) SetAPIBase(v string) *WxUpsertBulk {
+	return u.Update(func(s *WxUpsert) {
+		s.SetAPIBase(v)
+	})
+}
+
+// UpdateAPIBase sets the "api_base" field to the value that was provided on create.
+func (u *WxUpsertBulk) UpdateAPIBase() *WxUpsertBulk {
+	return u.Update(func(s *WxUpsert) {
+		s.UpdateAPIBase()
+	})
+}
+
+// ClearAPIBase clears the value of the "api_base" field.
+func (u *WxUpsertBulk) ClearAPIBase() *WxUpsertBulk {
+	return u.Update(func(s *WxUpsert) {
+		s.ClearAPIBase()
+	})
+}
+
+// SetAPIKey sets the "api_key" field.
+func (u *WxUpsertBulk) SetAPIKey(v string) *WxUpsertBulk {
+	return u.Update(func(s *WxUpsert) {
+		s.SetAPIKey(v)
+	})
+}
+
+// UpdateAPIKey sets the "api_key" field to the value that was provided on create.
+func (u *WxUpsertBulk) UpdateAPIKey() *WxUpsertBulk {
+	return u.Update(func(s *WxUpsert) {
+		s.UpdateAPIKey()
+	})
+}
+
+// ClearAPIKey clears the value of the "api_key" field.
+func (u *WxUpsertBulk) ClearAPIKey() *WxUpsertBulk {
+	return u.Update(func(s *WxUpsert) {
+		s.ClearAPIKey()
+	})
+}
+
 // Exec executes the query.
 func (u *WxUpsertBulk) Exec(ctx context.Context) error {
 	if u.create.err != nil {

ent/wx_update.go

@@ -256,6 +256,46 @@ func (wu *WxUpdate) SetNillableAgentID(u *uint64) *WxUpdate {
 	return wu
 }
 
+// SetAPIBase sets the "api_base" field.
+func (wu *WxUpdate) SetAPIBase(s string) *WxUpdate {
+	wu.mutation.SetAPIBase(s)
+	return wu
+}
+
+// SetNillableAPIBase sets the "api_base" field if the given value is not nil.
+func (wu *WxUpdate) SetNillableAPIBase(s *string) *WxUpdate {
+	if s != nil {
+		wu.SetAPIBase(*s)
+	}
+	return wu
+}
+
+// ClearAPIBase clears the value of the "api_base" field.
+func (wu *WxUpdate) ClearAPIBase() *WxUpdate {
+	wu.mutation.ClearAPIBase()
+	return wu
+}
+
+// SetAPIKey sets the "api_key" field.
+func (wu *WxUpdate) SetAPIKey(s string) *WxUpdate {
+	wu.mutation.SetAPIKey(s)
+	return wu
+}
+
+// SetNillableAPIKey sets the "api_key" field if the given value is not nil.
+func (wu *WxUpdate) SetNillableAPIKey(s *string) *WxUpdate {
+	if s != nil {
+		wu.SetAPIKey(*s)
+	}
+	return wu
+}
+
+// ClearAPIKey clears the value of the "api_key" field.
+func (wu *WxUpdate) ClearAPIKey() *WxUpdate {
+	wu.mutation.ClearAPIKey()
+	return wu
+}
+
 // SetServer sets the "server" edge to the Server entity.
 func (wu *WxUpdate) SetServer(s *Server) *WxUpdate {
 	return wu.SetServerID(s.ID)
@@ -396,6 +436,18 @@ func (wu *WxUpdate) sqlSave(ctx context.Context) (n int, err error) {
 	if wu.mutation.OrganizationIDCleared() {
 		_spec.ClearField(wx.FieldOrganizationID, field.TypeUint64)
 	}
+	if value, ok := wu.mutation.APIBase(); ok {
+		_spec.SetField(wx.FieldAPIBase, field.TypeString, value)
+	}
+	if wu.mutation.APIBaseCleared() {
+		_spec.ClearField(wx.FieldAPIBase, field.TypeString)
+	}
+	if value, ok := wu.mutation.APIKey(); ok {
+		_spec.SetField(wx.FieldAPIKey, field.TypeString, value)
+	}
+	if wu.mutation.APIKeyCleared() {
+		_spec.ClearField(wx.FieldAPIKey, field.TypeString)
+	}
 	if wu.mutation.ServerCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.M2O,
@@ -700,6 +752,46 @@ func (wuo *WxUpdateOne) SetNillableAgentID(u *uint64) *WxUpdateOne {
 	return wuo
 }
 
+// SetAPIBase sets the "api_base" field.
+func (wuo *WxUpdateOne) SetAPIBase(s string) *WxUpdateOne {
+	wuo.mutation.SetAPIBase(s)
+	return wuo
+}
+
+// SetNillableAPIBase sets the "api_base" field if the given value is not nil.
+func (wuo *WxUpdateOne) SetNillableAPIBase(s *string) *WxUpdateOne {
+	if s != nil {
+		wuo.SetAPIBase(*s)
+	}
+	return wuo
+}
+
+// ClearAPIBase clears the value of the "api_base" field.
+func (wuo *WxUpdateOne) ClearAPIBase() *WxUpdateOne {
+	wuo.mutation.ClearAPIBase()
+	return wuo
+}
+
+// SetAPIKey sets the "api_key" field.
+func (wuo *WxUpdateOne) SetAPIKey(s string) *WxUpdateOne {
+	wuo.mutation.SetAPIKey(s)
+	return wuo
+}
+
+// SetNillableAPIKey sets the "api_key" field if the given value is not nil.
+func (wuo *WxUpdateOne) SetNillableAPIKey(s *string) *WxUpdateOne {
+	if s != nil {
+		wuo.SetAPIKey(*s)
+	}
+	return wuo
+}
+
+// ClearAPIKey clears the value of the "api_key" field.
+func (wuo *WxUpdateOne) ClearAPIKey() *WxUpdateOne {
+	wuo.mutation.ClearAPIKey()
+	return wuo
+}
+
 // SetServer sets the "server" edge to the Server entity.
 func (wuo *WxUpdateOne) SetServer(s *Server) *WxUpdateOne {
 	return wuo.SetServerID(s.ID)
@@ -870,6 +962,18 @@ func (wuo *WxUpdateOne) sqlSave(ctx context.Context) (_node *Wx, err error) {
 	if wuo.mutation.OrganizationIDCleared() {
 		_spec.ClearField(wx.FieldOrganizationID, field.TypeUint64)
 	}
+	if value, ok := wuo.mutation.APIBase(); ok {
+		_spec.SetField(wx.FieldAPIBase, field.TypeString, value)
+	}
+	if wuo.mutation.APIBaseCleared() {
+		_spec.ClearField(wx.FieldAPIBase, field.TypeString)
+	}
+	if value, ok := wuo.mutation.APIKey(); ok {
+		_spec.SetField(wx.FieldAPIKey, field.TypeString, value)
+	}
+	if wuo.mutation.APIKeyCleared() {
+		_spec.ClearField(wx.FieldAPIKey, field.TypeString)
+	}
 	if wuo.mutation.ServerCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.M2O,

go.mod

@@ -5,8 +5,10 @@ go 1.22.1
 require (
 	entgo.io/ent v0.13.1
 	github.com/casbin/casbin/v2 v2.85.0
+	github.com/go-resty/resty/v2 v2.14.0
 	github.com/imroc/req/v3 v3.43.1
 	github.com/redis/go-redis/v9 v9.5.1
+	github.com/robfig/cron/v3 v3.0.1
 	github.com/spf13/cast v1.6.0
 	github.com/suyuan32/simple-admin-common v1.3.11
 	github.com/suyuan32/simple-admin-core v1.3.11
@@ -91,7 +93,6 @@ require (
 	github.com/quic-go/qpack v0.4.0 // indirect
 	github.com/quic-go/quic-go v0.42.0 // indirect
 	github.com/refraction-networking/utls v1.6.3 // indirect
-	github.com/robfig/cron/v3 v3.0.1 // indirect
 	github.com/spaolacci/murmur3 v1.1.0 // indirect
 	github.com/zclconf/go-cty v1.14.3 // indirect
 	go.etcd.io/etcd/api/v3 v3.5.12 // indirect
@@ -112,16 +113,17 @@ require (
 	go.uber.org/mock v0.4.0 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
 	go.uber.org/zap v1.24.0 // indirect
-	golang.org/x/crypto v0.21.0 // indirect
+	golang.org/x/crypto v0.26.0 // indirect
 	golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8 // indirect
-	golang.org/x/mod v0.16.0 // indirect
-	golang.org/x/net v0.22.0 // indirect
+	golang.org/x/mod v0.17.0 // indirect
+	golang.org/x/net v0.28.0 // indirect
 	golang.org/x/oauth2 v0.18.0 // indirect
-	golang.org/x/sys v0.18.0 // indirect
-	golang.org/x/term v0.18.0 // indirect
-	golang.org/x/text v0.14.0 // indirect
-	golang.org/x/time v0.5.0 // indirect
-	golang.org/x/tools v0.19.0 // indirect
+	golang.org/x/sync v0.8.0 // indirect
+	golang.org/x/sys v0.23.0 // indirect
+	golang.org/x/term v0.23.0 // indirect
+	golang.org/x/text v0.17.0 // indirect
+	golang.org/x/time v0.6.0 // indirect
+	golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d // indirect
 	google.golang.org/appengine v1.6.8 // indirect
 	google.golang.org/genproto/googleapis/api v0.0.0-20240318140521-94a12d6c2237 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240318140521-94a12d6c2237 // indirect

go.sum

@@ -145,6 +145,8 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
 github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
 github.com/go-playground/validator/v10 v10.19.0 h1:ol+5Fu+cSq9JD7SoSqe04GMI92cbn0+wvQ3bZ8b/AU4=
 github.com/go-playground/validator/v10 v10.19.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
+github.com/go-resty/resty/v2 v2.14.0 h1:/rhkzsAqGQkozwfKS5aFAbb6TyKd3zyFRWcdRXLPCAU=
+github.com/go-resty/resty/v2 v2.14.0/go.mod h1:IW6mekUOsElt9C7oWr0XRt9BNSD6D5rr9mhk6NjmNHg=
 github.com/go-sql-driver/mysql v1.5.1-0.20200311113236-681ffa848bae/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
 github.com/go-sql-driver/mysql v1.8.0 h1:UtktXaU2Nb64z/pLiGIxY4431SJ4/dR5cjMmlVHgnT4=
 github.com/go-sql-driver/mysql v1.8.0/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
@@ -254,8 +256,6 @@ github.com/hashicorp/serf v0.8.2/go.mod h1:6hOLApaqBFA1NXqRQAsxw9QxuDEvNxSQRwA/J
 github.com/imroc/req/v3 v3.43.1 h1:tsWAhvxik4egtHAvMlxcjaWJtHlJL8EpBqJMOm5rmyQ=
 github.com/imroc/req/v3 v3.43.1/go.mod h1:SQIz5iYop16MJxbo8ib+4LnostGCok8NQf8ToyQc2xA=
 github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
-github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
-github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 github.com/jackc/chunkreader v1.0.0/go.mod h1:RT6O25fNZIuasFJRyZ4R/Y2BbhasbmZXF9QQ7T3kePo=
 github.com/jackc/chunkreader/v2 v2.0.0/go.mod h1:odVSm741yZoC3dpHEUXIqA9tQRhFrgOHwnPIn9lDKlk=
 github.com/jackc/chunkreader/v2 v2.0.1 h1:i+RDz65UE+mmpjTfyz0MoVTnzeYxroil2G82ki7MGG8=
@@ -585,8 +585,12 @@ golang.org/x/crypto v0.0.0-20210616213533-5ff15b29337e/go.mod h1:GvvjBRRGRdwPK5y
 golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
 golang.org/x/crypto v0.6.0/go.mod h1:OFC/31mSvZgRz0V1QTNCzfAI1aIRzbiufJtkMIlEp58=
-golang.org/x/crypto v0.21.0 h1:X31++rzVUdKhX5sWmSOFZxx8UW/ldWx55cbf08iNAMA=
-golang.org/x/crypto v0.21.0/go.mod h1:0BP7YvVV9gBbVKyeTG0Gyn+gZm94bibOW5BjDEYAOMs=
+golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc=
+golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
+golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
+golang.org/x/crypto v0.25.0/go.mod h1:T+wALwcMOSE0kXgUAnPAHqTLW+XHgcELELW8VaDgm/M=
+golang.org/x/crypto v0.26.0 h1:RrRspgV4mU+YwB4FYnuBoKsUapNIL5cohGAmSH3azsw=
+golang.org/x/crypto v0.26.0/go.mod h1:GY7jblb9wI+FOo5y8/S2oY4zWP07AkOJ4+jxCqdqn54=
 golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
 golang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
 golang.org/x/exp v0.0.0-20190510132918-efd6b22b2522/go.mod h1:ZjyILWgesfNpC6sMxTJOJm9Kp84zZh5NQWvqDGG3Qr8=
@@ -612,8 +616,11 @@ golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
-golang.org/x/mod v0.16.0 h1:QX4fJ0Rr5cPQCF7O9lh9Se4pmwfwskqZfq5moyldzic=
-golang.org/x/mod v0.16.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
+golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
+golang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
+golang.org/x/mod v0.15.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
+golang.org/x/mod v0.17.0 h1:zY54UmvipHiNd+pm+m0x9KhZ9hl1/7QNMyxXbc6ICqA=
+golang.org/x/mod v0.17.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
 golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20181023162649-9b4f9f5ad519/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
@@ -636,8 +643,13 @@ golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v
 golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=
 golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
 golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
-golang.org/x/net v0.22.0 h1:9sGLhx7iRIHEiX0oAJ3MRZMUCElJgy7Br1nO+AMN3Tc=
-golang.org/x/net v0.22.0/go.mod h1:JKghWKKOSdJwpW2GEx0Ja7fmaKnMsbu+MWVZTokSYmg=
+golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
+golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
+golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=
+golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
+golang.org/x/net v0.27.0/go.mod h1:dDi0PyhWNoiUOrAS8uXv/vnScO4wnHQO4mj9fn/RytE=
+golang.org/x/net v0.28.0 h1:a9JDOJc5GMUJ0+UDqmLT86WiEy7iWyIhz8gz8E4e5hE=
+golang.org/x/net v0.28.0/go.mod h1:yqtgsTWOOnlGLG9GFRrK3++bGOUEkNBoHZc8MEDWPNg=
 golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
 golang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
 golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
@@ -652,8 +664,12 @@ golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJ
 golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
+golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y=
 golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
+golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sys v0.0.0-20180823144017-11551d06cbcc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
@@ -686,14 +702,25 @@ golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBc
 golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
-golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.23.0 h1:YfKFowiIMvtgl1UERQoTPPToxltDeZfbj4H7dVUCwmM=
+golang.org/x/sys v0.23.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/telemetry v0.0.0-20240228155512-f48c80bd79b2/go.mod h1:TeRTkGYfJXctD9OcfyVLyj2J3IxLnKwHJR8f4D8a3YE=
 golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
 golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
 golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
-golang.org/x/term v0.18.0 h1:FcHjZXDMxI8mM3nwhX9HlKop4C0YQvCVCdwYl2wOtE8=
-golang.org/x/term v0.18.0/go.mod h1:ILwASektA3OnRv7amZ1xhE/KTR+u50pbXfZ03+6Nx58=
+golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
+golang.org/x/term v0.12.0/go.mod h1:owVbMEjm3cBLCHdkQu9b1opXd4ETQWc3BhuQGKgXgvU=
+golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk=
+golang.org/x/term v0.20.0/go.mod h1:8UkIAJTvZgivsXaD6/pH6U9ecQzZ45awqEOzuCvwpFY=
+golang.org/x/term v0.22.0/go.mod h1:F3qCibpT5AMpCRfhfT53vVJwhLtIVHhB9XDjfFvnMI4=
+golang.org/x/term v0.23.0 h1:F6D4vR+EHoL9/sWAWgAR1H2DcHr4PareCbAaCo1RpuU=
+golang.org/x/term v0.23.0/go.mod h1:DgV24QBUrK6jhZXl+20l6UWznPlwAHm1Q1mGHtydmSk=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.1-0.20180807135948-17ff2d5776d2/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
@@ -703,12 +730,17 @@ golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
 golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
 golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
-golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
+golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
+golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
 golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=
+golang.org/x/text v0.17.0 h1:XtiM5bkSOt+ewxlOE/aE/AKEHibwj/6gvWMl9Rsh0Qc=
+golang.org/x/text v0.17.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
 golang.org/x/time v0.0.0-20181108054448-85acf8d2951c/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
 golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
-golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
-golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
+golang.org/x/time v0.6.0 h1:eTDhh4ZXt5Qf0augr54TN6suAUudPcawVZeIAPU7D4U=
+golang.org/x/time v0.6.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
 golang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20181030221726-6c7e314b6563/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
@@ -739,8 +771,10 @@ golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4f
 golang.org/x/tools v0.1.0/go.mod h1:xkSsbof2nBLbhDlRMhhhyNLN/zl3eTqcnHD5viDpcZ0=
 golang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=
 golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
-golang.org/x/tools v0.19.0 h1:tfGCXNR1OsFG+sVdLAitlpjAvD/I6dHDKnYrpEZUHkw=
-golang.org/x/tools v0.19.0/go.mod h1:qoJWxmGSIBmAeriMx19ogtrEPrGtDbPK634QFIcLAhc=
+golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
+golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=
+golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d h1:vU5i/LfpvrRCpgM/VPfJLg5KjxD3E+hfT1SH+d9zLwg=
+golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk=
 golang.org/x/xerrors v0.0.0-20190410155217-1f06c39b4373/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20190513163551-3ee3066db522/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=

hook/fastgpt/collection.go

@@ -0,0 +1,333 @@
+package fastgpt
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"time"
+)
+
+type CollectionResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       string `json:"data"`
+}
+
+type CreateCollectionReq struct {
+	DatasetId string      `json:"datasetId"`
+	ParentId  string      `json:"parentId,optional"`
+	Name      string      `json:"name"`
+	Type      string      `json:"type"`
+	Metadata  interface{} `json:"metadata,optional"`
+}
+
+type CreateTextCollectionReq struct {
+	DatasetId     string      `json:"datasetId"`
+	ParentId      string      `json:"parentId,optional"`
+	Name          string      `json:"name"`
+	Text          string      `json:"text"`
+	Metadata      interface{} `json:"metadata,optional"`
+	TrainingType  string      `json:"trainingType,optional"`
+	ChunkSize     uint64      `json:"chunkSize,optional"`
+	ChunkSplitter string      `json:"chunkSplitter,optional"`
+	QaPrompt      string      `json:"qaPrompt,optional"`
+}
+
+type CreateTextCollectionResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       struct {
+		CollectionID string `json:"collectionId"`
+		Results      struct {
+			InsertLen int   `json:"insertLen"`
+			OverToken []any `json:"overToken"`
+			Repeat    []any `json:"repeat"`
+			Error     []any `json:"error"`
+		} `json:"results"`
+	} `json:"data"`
+}
+
+type CreateLinkCollectionReq struct {
+	DatasetId string `json:"datasetId"`
+	ParentId  string `json:"parentId,optional"`
+	Name      string `json:"name"`
+	Link      string `json:"link"`
+
+	Metadata      interface{} `json:"metadata,optional"`
+	TrainingType  string      `json:"trainingType,optional"`
+	ChunkSize     uint64      `json:"chunkSize,optional"`
+	ChunkSplitter string      `json:"chunkSplitter,optional"`
+	QaPrompt      string      `json:"qaPrompt,optional"`
+}
+
+type CreateLinkCollectionResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       struct {
+		CollectionID string `json:"collectionId"`
+	} `json:"data"`
+}
+
+type CreateFileCollectionReq struct {
+	DatasetId string `json:"datasetId"`
+	ParentId  string `json:"parentId,optional"`
+
+	Metadata      interface{} `json:"metadata,optional"`
+	TrainingType  string      `json:"trainingType,optional"`
+	ChunkSize     uint64      `json:"chunkSize,optional"`
+	ChunkSplitter string      `json:"chunkSplitter,optional"`
+	QaPrompt      string      `json:"qaPrompt,optional"`
+}
+
+type CreateFileCollectionResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText,optional"`
+	Message    string `json:"message,optional"`
+	Data       struct {
+		CollectionID string `json:"collectionId,optional"`
+		Results      struct {
+			InsertLen int   `json:"insertLen"`
+			OverToken []any `json:"overToken"`
+			Repeat    []any `json:"repeat"`
+			Error     []any `json:"error"`
+		} `json:"results,optional"`
+	} `json:"data,optional"`
+}
+
+type GetCollectionListReq struct {
+	PageNum    int    `json:"pageNum,optional"`
+	PageSize   int    `json:"pageSize,optional"`
+	DatasetId  string `json:"datasetId"`
+	ParentId   string `json:"parentId,optional"`
+	SearchText string `json:"searchText,optional"`
+}
+
+type CollectionInfo struct {
+	ID        string `json:"_id"`
+	ParentID  string `json:"parentId"`
+	TeamID    string `json:"teamId"`
+	TmbID     string `json:"tmbId"`
+	DatasetID struct {
+		ID          string    `json:"_id"`
+		ParentID    string    `json:"parentId"`
+		TeamID      string    `json:"teamId"`
+		TmbID       string    `json:"tmbId"`
+		Type        string    `json:"type"`
+		Status      string    `json:"status"`
+		Avatar      string    `json:"avatar"`
+		Name        string    `json:"name"`
+		VectorModel string    `json:"vectorModel"`
+		AgentModel  string    `json:"agentModel"`
+		Intro       string    `json:"intro"`
+		Permission  string    `json:"permission"`
+		UpdateTime  time.Time `json:"updateTime"`
+	} `json:"datasetId"`
+	Type          string    `json:"type"`
+	Name          string    `json:"name"`
+	TrainingType  string    `json:"trainingType"`
+	ChunkSize     uint64    `json:"chunkSize"`
+	ChunkSplitter string    `json:"chunkSplitter"`
+	QaPrompt      string    `json:"qaPrompt"`
+	RawTextLength uint64    `json:"rawTextLength"`
+	HashRawText   string    `json:"hashRawText"`
+	CreateTime    time.Time `json:"createTime"`
+	UpdateTime    time.Time `json:"updateTime"`
+	CanWrite      bool      `json:"canWrite"`
+	SourceName    string    `json:"sourceName"`
+}
+
+type CollectionListResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       struct {
+		PageNum  int                    `json:"pageNum"`
+		PageSize int                    `json:"pageSize"`
+		Data     []CollectionSimpleInfo `json:"data"`
+		Total    int                    `json:"total"`
+	} `json:"data"`
+}
+
+type CollectionSimpleInfo struct {
+	ID             string    `json:"_id"`
+	ParentID       string    `json:"parentId"`
+	TmbID          string    `json:"tmbId"`
+	Type           string    `json:"type"`
+	Name           string    `json:"name"`
+	UpdateTime     time.Time `json:"updateTime"`
+	DataAmount     uint64    `json:"dataAmount"`
+	TrainingAmount uint64    `json:"trainingAmount"`
+}
+
+type CollectionDetailResp struct {
+	Code       int            `json:"code"`
+	StatusText string         `json:"statusText"`
+	Message    string         `json:"message"`
+	Data       CollectionInfo `json:"data"`
+}
+
+type UpdateCollectionReq struct {
+	ID       string `json:"id"`
+	ParentId string `json:"parentId"`
+	Name     string `json:"name"`
+}
+
+// GetCollectionDetail 获取合集详情
+func GetCollectionDetail(id string) (collection *CollectionDetailResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetQueryParam("id", id).
+		Get("core/dataset/collection/detail")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// GetCollectionList 获取合集列表
+func GetCollectionList(params *GetCollectionListReq) (collectionList *CollectionListResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collectionList).
+		SetBody(params).
+		Post("core/dataset/collection/list")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateEmptyCollection 创建空合集
+func CreateEmptyCollection(data *CreateCollectionReq) (collection *CollectionResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetBody(data).
+		Post("core/dataset/collection/create")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateTextCollection 创建文本合集
+func CreateTextCollection(data *CreateTextCollectionReq) (collection *CreateTextCollectionResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetBody(*data).
+		Post("core/dataset/collection/create/text")
+
+	if err != nil {
+		fmt.Printf(err.Error())
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateLinkCollection 创建链接合集
+func CreateLinkCollection(data *CreateLinkCollectionReq) (collection *CreateLinkCollectionResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetBody(*data).
+		Post("core/dataset/collection/create/link")
+
+	if err != nil {
+		fmt.Printf(err.Error())
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateFileCollection 创建文件合集
+func CreateFileCollection(data *map[string]string, filePath string) (collection *CreateFileCollectionResp, err error) {
+	dataString, _ := json.Marshal(*data)
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetFormData(map[string]string{"data": string(dataString)}).
+		SetFile("file", filePath).
+		Post("core/dataset/collection/create/localFile")
+
+	fmt.Printf("%v, %v", err, resp.String())
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// UpdateCollection 修改合集
+func UpdateCollection(data *UpdateCollectionReq) (collection *CollectionResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetBody(*data).
+		Put("core/dataset/collection/update")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// DeleteCollection 删除合集
+func DeleteCollection(id string) (collection *CollectionResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&collection).
+		SetQueryParam("id", id).
+		Delete("core/dataset/collection/delete")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}

hook/fastgpt/data.go

@@ -0,0 +1,198 @@
+package fastgpt
+
+import (
+	"errors"
+)
+
+type DataResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       string `json:"data,optional"`
+}
+
+type CreateBulkDataReq struct {
+	CollectionID string         `json:"collectionId"`
+	TrainingMode string         `json:"trainingMode"`
+	Prompt       string         `json:"prompt,optional"`
+	BillID       string         `json:"billId,optional"`
+	Data         []DataQuestion `json:"data"`
+}
+
+type DataQuestion struct {
+	Q       string  `json:"q"`
+	A       string  `json:"a,optional"`
+	Indexes []Index `json:"indexes,omitempty,optional"`
+}
+
+type Index struct {
+	DataId       string `json:"dataId,optional"`
+	DefaultIndex bool   `json:"defaultIndex,optional"`
+	Text         string `json:"text"`
+}
+
+type CreateBulkDataResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Data       struct {
+		InsertLen uint64 `json:"insertLen"`
+		OverToken []any  `json:"overToken"`
+		Repeat    []any  `json:"repeat"`
+		Error     []any  `json:"error"`
+	} `json:"data"`
+}
+
+type GetDataListReq struct {
+	PageNum      int    `json:"pageNum,optional"`
+	PageSize     int    `json:"pageSize,optional"`
+	CollectionId string `json:"collectionId,optional"`
+	SearchText   string `json:"searchText,optional"`
+}
+
+type DataInfo struct {
+	ID         string `json:"id"`
+	TeamID     string `json:"teamId"`
+	Q          string `json:"q"`
+	A          string `json:"a"`
+	ChunkIndex uint64 `json:"chunkIndex"`
+	Indexes    []struct {
+		DefaultIndex bool   `json:"defaultIndex"`
+		DataID       string `json:"dataId"`
+		Text         string `json:"text"`
+		ID           string `json:"_id"`
+	} `json:"indexes"`
+	DatasetID    string `json:"datasetId"`
+	CollectionID string `json:"collectionId"`
+	SourceName   string `json:"sourceName"`
+	SourceID     string `json:"sourceId"`
+	IsOwner      bool   `json:"isOwner"`
+	CanWrite     bool   `json:"canWrite"`
+}
+
+type DataListResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       struct {
+		PageNum  int `json:"pageNum"`
+		PageSize int `json:"pageSize"`
+		Data     []struct {
+			ID           string `json:"_id"`
+			DatasetID    string `json:"datasetId"`
+			CollectionID string `json:"collectionId"`
+			Q            string `json:"q"`
+			A            string `json:"a"`
+			ChunkIndex   uint64 `json:"chunkIndex"`
+		} `json:"data,optional"`
+		Total int `json:"total"`
+	} `json:"data,optional"`
+}
+
+type DataDetail struct {
+	Code       int      `json:"code"`
+	StatusText string   `json:"statusText"`
+	Message    string   `json:"message"`
+	Data       DataInfo `json:"data,optional"`
+}
+
+type UpdateDataReq struct {
+	DataId  string  `json:"dataId" validate:"required"`
+	Q       string  `json:"q" validate:"required"`
+	A       string  `json:"a" validate:"required"`
+	Indexes []Index `json:"indexes,omitempty,optional"`
+}
+
+// GetDataDetail 获取数据详情
+func GetDataDetail(id string) (data *DataDetail, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&data).
+		SetQueryParam("id", id).
+		Get("core/dataset/data/detail")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// GetDataList 获取数据列表
+func GetDataList(params *GetDataListReq) (dataList *DataListResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&dataList).
+		SetBody(params).
+		Post("core/dataset/data/list")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateBulkData 批量创建数据
+func CreateBulkData(data *CreateBulkDataReq) (response *CreateBulkDataResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&response).
+		SetBody(*data).
+		Post("core/dataset/data/pushData")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// UpdateData 修改数据
+func UpdateData(data *UpdateDataReq) (response *DataResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&response).
+		SetBody(*data).
+		Put("core/dataset/data/update")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// DeleteData 删除数据
+func DeleteData(id string) (response *DataResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&response).
+		SetQueryParam("id", id).
+		Delete("core/dataset/data/delete")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}

hook/fastgpt/dataset.go

@@ -0,0 +1,185 @@
+package fastgpt
+
+import (
+	"errors"
+	"time"
+)
+
+type DatasetResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       string `json:"data"`
+}
+
+type DatasetReq struct {
+	ParentID    string `json:"parentId,optional"`
+	Type        string `json:"type,optional"`
+	Avatar      string `json:"avatar,optional"`
+	Name        string `json:"name"`
+	Intro       string `json:"intro,optional"`
+	VectorModel string `json:"vectorModel"`
+	AgentModel  string `json:"agentModel"`
+}
+
+type DatasetInfo struct {
+	ID          string `json:"_id"`
+	ParentID    string `json:"parentId,optional"`
+	TeamID      string `json:"teamId,optional"`
+	TmbID       string `json:"tmbId,optional"`
+	Type        string `json:"type,optional"`
+	Status      string `json:"status,optional"`
+	Avatar      string `json:"avatar,optional"`
+	Name        string `json:"name,optional"`
+	VectorModel struct {
+		Model            string `json:"model"`
+		Name             string `json:"name"`
+		Avatar           string `json:"avatar"`
+		CharsPointsPrice uint64 `json:"charsPointsPrice"`
+		DefaultToken     uint64 `json:"defaultToken"`
+		MaxToken         uint64 `json:"maxToken"`
+		Weight           uint64 `json:"weight"`
+		DefaultConfig    struct {
+		} `json:"defaultConfig"`
+		DbConfig struct {
+		} `json:"dbConfig"`
+		QueryConfig struct {
+		} `json:"queryConfig"`
+	} `json:"vectorModel,optional"`
+	AgentModel struct {
+		Model                   string  `json:"model"`
+		Name                    string  `json:"name"`
+		Avatar                  string  `json:"avatar"`
+		MaxContext              uint64  `json:"maxContext"`
+		MaxResponse             uint64  `json:"maxResponse"`
+		QuoteMaxToken           uint64  `json:"quoteMaxToken"`
+		MaxTemperature          float64 `json:"maxTemperature"`
+		CharsPointsPrice        uint64  `json:"charsPointsPrice"`
+		Censor                  bool    `json:"censor"`
+		Vision                  bool    `json:"vision"`
+		DatasetProcess          bool    `json:"datasetProcess"`
+		UsedInClassify          bool    `json:"usedInClassify"`
+		UsedInExtractFields     bool    `json:"usedInExtractFields"`
+		UsedInToolCall          bool    `json:"usedInToolCall"`
+		UsedInQueryExtension    bool    `json:"usedInQueryExtension"`
+		ToolChoice              bool    `json:"toolChoice"`
+		FunctionCall            bool    `json:"functionCall"`
+		CustomCQPrompt          string  `json:"customCQPrompt"`
+		CustomExtractPrompt     string  `json:"customExtractPrompt"`
+		DefaultSystemChatPrompt string  `json:"defaultSystemChatPrompt"`
+		DefaultConfig           struct {
+		} `json:"defaultConfig"`
+	} `json:"agentModel,optional"`
+	Intro      string `json:"intro,optional"`
+	Permission struct {
+		Value        int64 `json:"value"`
+		IsOwner      bool  `json:"isOwner"`
+		HasManagePer bool  `json:"hasManagePer"`
+		HasWritePer  bool  `json:"hasWritePer"`
+		HasReadPer   bool  `json:"hasReadPer"`
+	} `json:"permission,optional"`
+	DefaultPermission uint64    `json:"defaultPermission.optional"`
+	UpdateTime        time.Time `json:"updateTime,optional"`
+	V                 uint64    `json:"__v,optional"`
+}
+
+type DatasetList struct {
+	Code       int           `json:"code"`
+	StatusText string        `json:"statusText"`
+	Message    string        `json:"message"`
+	Data       []DatasetInfo `json:"data"`
+}
+
+type DatasetDetail struct {
+	Code       int         `json:"code"`
+	StatusText string      `json:"statusText"`
+	Message    string      `json:"message"`
+	Data       DatasetInfo `json:"data"`
+}
+
+// GetDatasetDetail 获取数据集详情
+func GetDatasetDetail(id string) (dataset *DatasetDetail, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&dataset).
+		SetQueryParam("id", id).
+		Get("core/dataset/detail")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+	return
+}
+
+// GetDatasetList 获取数据集列表
+func GetDatasetList(parentId string) (datasetList *DatasetList, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&datasetList).
+		SetBody(map[string]string{"parentId": parentId}).
+		Post("core/dataset/list")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateDataset 创建数据集
+func CreateDataset(data *DatasetReq) (dataset *DatasetResp, err error) {
+	data.Type = "dataset"
+	if data.Avatar == "" {
+		data.Avatar = "/icon/logo.svg"
+	}
+
+	resp, err := NewResty().
+		R().
+		SetResult(&dataset).
+		SetBody(*data).
+		Post("core/dataset/create")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}
+
+// CreateDatasetFolder 创建数据集目录
+func CreateDatasetFolder(data *DatasetReq) (*DatasetResp, error) {
+	data.Type = "folder"
+	data.Avatar = "/imgs/files/folder.svg"
+	return CreateDataset(data)
+}
+
+// DeleteDataset 删除数据集
+func DeleteDataset(id string) (dataset *DatasetResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&dataset).
+		SetQueryParam("id", id).
+		Delete("core/dataset/delete")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}

hook/fastgpt/resty.go

@@ -0,0 +1,55 @@
+package fastgpt
+
+import (
+	"github.com/go-resty/resty/v2"
+	"log"
+	"os"
+)
+
+// NewResty new resty client
+func NewResty() *resty.Client {
+	client := resty.New()
+	logger := newLogger()
+	client.SetRetryCount(2).
+		SetLogger(logger).
+		SetHeader("Content-Type", "application/json").
+		SetBaseURL("https://fastgpt.gkscrm.com/api").
+		SetHeader("Authorization", "Bearer "+getToken())
+
+	return client
+}
+
+// getToken 获取配置的可用的Token
+func getToken() string {
+	return "fastgpt-czbAiqKKse65hwwviZhwkgvyDEKE3aeB31Fx18oUsAGojyWQ672HdsWZi1E1C"
+}
+
+type logger struct {
+	l *log.Logger
+}
+
+func (l *logger) Errorf(format string, v ...interface{}) {
+	l.output("ERROR RESTY "+format, v...)
+}
+
+func (l *logger) Warnf(format string, v ...interface{}) {
+	l.output("WARN RESTY "+format, v...)
+}
+
+func (l *logger) Debugf(format string, v ...interface{}) {
+	l.output("DEBUG RESTY "+format, v...)
+}
+
+func (l *logger) output(format string, v ...interface{}) {
+	if len(v) == 0 {
+		l.l.Print(format)
+		return
+	}
+	l.l.Printf(format, v...)
+}
+
+// newLogger return a new logger
+func newLogger() *logger {
+	l := &logger{l: log.New(os.Stdout, "", log.Ldate|log.Lmicroseconds)}
+	return l
+}

hook/fastgpt/search.go

@@ -0,0 +1,61 @@
+package fastgpt
+
+import (
+	"errors"
+)
+
+type SearchReq struct {
+	DatasetID   string  `json:"datasetId"`           //知识库ID
+	Text        string  `json:"text"`                //需要测试的文本
+	Limit       uint64  `json:"limit"`               //最大 tokens 数量
+	Similarity  float64 `json:"similarity,optional"` //最低相关度（0~1，可选）
+	SearchMode  string  `json:"searchMode"`          //搜索模式：embedding | fullTextRecall | mixedRecall
+	UsingReRank bool    `json:"usingReRank"`         //使用重排
+}
+
+type SearchResp struct {
+	Code       int    `json:"code"`
+	StatusText string `json:"statusText"`
+	Message    string `json:"message"`
+	Data       struct {
+		List []struct {
+			ID           string `json:"id"`
+			DatasetID    string `json:"datasetId"`
+			CollectionID string `json:"collectionId"`
+			SourceName   string `json:"sourceName"`
+			Q            string `json:"q"`
+			A            string `json:"a"`
+			ChunkIndex   int    `json:"chunkIndex"`
+			Score        []struct {
+				Type  string  `json:"type"`
+				Value float64 `json:"value"`
+				Index uint64  `json:"index"`
+			} `json:"score"`
+		} `json:"list,optional"`
+		Duration              string `json:"duration"`
+		SearchMode            string `json:"searchMode"`
+		Limit                 uint64 `json:"limit"`
+		Similarity            uint64 `json:"similarity"`
+		UsingReRank           bool   `json:"usingReRank"`
+		UsingSimilarityFilter bool   `json:"usingSimilarityFilter"`
+	} `json:"data,optional"`
+}
+
+// SearchText 搜索数据
+func SearchText(params *SearchReq) (dataList *SearchResp, err error) {
+	resp, err := NewResty().
+		R().
+		SetResult(&dataList).
+		SetBody(params).
+		Post("core/dataset/searchTest")
+
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.IsError() {
+		return nil, errors.New(resp.String())
+	}
+
+	return
+}

hook/fastgpt/usage.go

@@ -0,0 +1,264 @@
+package fastgpt
+
+/**
+以下为知识库相关
+================================================================
+*/
+
+/**
+// 创建数据集
+var info DatasetReq
+info.ParentID = "66cc44d725aea7116ded5281"
+info.Name = "dataset-测试数据(huaguo)"
+info.Intro = "测试数据(huaguo)"
+info.AgentModel = "gpt-3.5-turbo"
+info.VectorModel = "text-embedding-ada-002"
+
+var dataset *DatasetResp
+dataset, err := CreateDataset(&info)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+	return
+}
+fmt.Printf("%v", dataset)
+fmt.Println(dataset.Code, dataset.StatusText, dataset.Message, dataset.Data)
+*/
+
+/**
+// 删除数据集
+resp, err := DeleteDataset("66cc43e125aea7116ded514f")
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("code=%v, data=%v, message=%v, statusText=%v", resp.Code, resp.Data, resp.Message, resp.StatusText)
+*/
+
+/**
+// 获取数据集详情
+resp, err := GetDatasetDetail("66cc44d725aea7116ded5281")
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("code=%v, data=%v, message=%v, statusText=%v", resp.Code, resp.Data, resp.Message, resp.StatusText)
+*/
+
+/**
+// 获取数据集列表
+resp, err := GetDatasetList("66cc44d725aea7116ded5281")
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("%v", resp)
+*/
+
+/**
+以下为集合相关
+================================================================
+*/
+
+/**
+// 获取集合详情
+resp, err := GetCollectionDetail("66cc458925aea7116ded53bc")
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("code=%v, data=%v", resp.Code, resp.Data)
+*/
+
+/**
+// 获取集合列表
+var params GetCollectionListReq
+params.DatasetId = "66cc458925aea7116ded53ba"
+params.PageNum = 1
+params.PageSize = 20
+resp, err := GetCollectionList(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("code=%v, data=%v", resp.Code, resp.Data)
+*/
+
+/**
+// 创建空集合
+var params CreateCollectionReq
+params.DatasetId = "66cc458925aea7116ded53ba"
+params.Name = "空集合v3"
+params.Type = "folder"
+resp, err := CreateEmptyCollection(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("%v", resp)
+*/
+
+/**
+// 创建文本集合
+var params CreateTextCollectionReq
+params.DatasetId = "66cc458925aea7116ded53ba"
+params.Name = "文本测试训练v5"
+params.Text = " 《毒战》改编自杜琪峰执导的同名电影，讲述为了抓捕韩国最大规模贩毒组织的神秘大BOSS李先生，警察元浩与李先生手下成员乐联手破案的故事。 "
+params.TrainingType = "qa"
+params.ChunkSize = 8000
+params.ChunkSplitter = ""
+params.QaPrompt = "11"
+resp, err := CreateTextCollection(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("%v", resp)
+*/
+
+/**
+// 创建链接集合
+var params CreateLinkCollectionReq
+params.DatasetId = "66cc458925aea7116ded53ba"
+params.Name = "链接测试训练v5"
+params.Link = "https://doc.in/docs/course/quick-start/"
+params.TrainingType = "chunk"
+params.ChunkSize = 512
+params.ChunkSplitter = ""
+params.QaPrompt = ""
+metadata := make(map[string]string)
+metadata["webPageSelector"] = ".docs-content"
+params.Metadata = metadata
+resp, err := CreateLinkCollection(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("collectionId=%v", resp.Data.CollectionID)
+*/
+
+/**
+// 创建文件集合
+data := make(map[string]string, 8)
+filePath := "/tmp/movie.txt"
+data["datasetId"] = "66cc458925aea7116ded53ba"
+data["trainingType"] = "chunk"
+data["chunkSize"] = "512"
+data["chunkSplitter"] = ""
+data["qaPrompt"] = ""
+resp, err := CreateFileCollection(&data, filePath)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("%v", resp)
+*/
+
+/**
+// 更新集合
+var params UpdateCollectionReq
+params.ID = "66cd6c0025aea7116dedb313"
+params.Name = "链接集合V6"
+resp, err := UpdateCollection(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+// 删除集合
+resp, err := DeleteCollection("66cd6c0025aea7116dedb313")
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+以下为数据相关
+================================================================
+*/
+
+/**
+// 获取数据详情
+resp, err := GetDataDetail("66cd773b25aea7116dedc017")
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+// 获取数据列表
+var params GetDataListReq
+params.PageSize = 10
+params.PageNum = 1
+params.CollectionId = "66cd773b25aea7116dedc01a"
+resp, err := GetDataList(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+// 批量创建数据
+var params CreateBulkDataReq
+params.CollectionID = "66cd773b25aea7116dedc01a"
+params.TrainingMode = "chunk"
+params.Data = []DataQuestion{
+	{
+		Q: "你是谁？",
+		A: "我是FastGPT助手",
+	},
+	{
+		Q: "你会什么？",
+		A: "我什么都会",
+		Indexes: []Index{
+			{Text: "自定义索引1"},
+			{Text: "自定义索引2"},
+		},
+	},
+}
+resp, err := CreateBulkData(&params)
+if err != nil {
+	fmt.Printf("%v", err.Error())
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+// 更新数据
+var params UpdateDataReq
+params.ID = "66cd7e1925aea7116dedc64b"
+params.Q = "你是谁3？"
+params.A = "我是FastGPT助手3"
+params.Indexes = []Index{
+	{Text: "自定义索引1", DataId: "66cd7e1925aea7116dedc64b", DefaultIndex: true},
+	{Text: "自定义索引2", DataId: "66cd7e1925aea7116dedc64b", DefaultIndex: false},
+}
+resp, err := UpdateData(&params)
+if err != nil {
+	fmt.Printf("%v", err)
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+// 删除数据
+resp, err := DeleteData("66cd7e1925aea7116dedc651")
+if err != nil {
+	fmt.Printf("%v", err)
+}
+fmt.Printf("resp=%v", resp)
+*/
+
+/**
+以下为搜索相关
+*/
+
+/**
+// 搜索数据
+var params SearchReq
+params.DatasetID = "66cc458925aea7116ded53ba"
+params.Text = "毒战"
+params.Limit = 20
+params.SearchMode = "fullTextRecall"
+params.UsingReRank = true
+resp, err := SearchText(&params)
+if err != nil {
+	fmt.Printf("%v", err)
+}
+fmt.Printf("resp=%v", resp)
+*/

hook/sys.go

@@ -18,6 +18,21 @@ func (h *Hook) Logout() (result LogoutResp, err error) {
 	return
 }
 
+// 结束微信
+func (h *Hook) TerminateThisWeChat(pid string) (result LogoutResp, err error) {
+	resp, err := h.Client.R().SetBody(&TerminateThisWeChatReq{
+		PID: pid,
+	}).SetSuccessResult(&result).Post("http://" + h.ServerIp + ":" + h.AdminPort + "/TerminateThisWeChat")
+	if err != nil {
+		return
+	}
+	if !resp.IsSuccessState() {
+		err = fmt.Errorf("TerminateThisWeChatReq failed with status code %d", resp.StatusCode)
+		return
+	}
+	return
+}
+
 // 获取微信总数
 func (h *Hook) GetWeChatProcessNumber() (result GetWeChatProcessNumberResp, err error) {
 	resp, err := h.Client.R().SetSuccessResult(&result).Post("http://" + h.ServerIp + ":" + h.AdminPort + "/Get_WeChatProcessNumber")

hook/type.go

@@ -203,6 +203,10 @@ type StartWechatReq struct {
 	StartPort string `json:"StartPort"`
 }
 
+type TerminateThisWeChatReq struct {
+	PID string `json:"PID"`
+}
+
 type StartWechatResp struct {
 	StartPort string `json:"StartPort"`
 	Success   string `json:"success"`

internal/handler/Msg/create_msg_handler.go

@@ -5,7 +5,7 @@ import (
 
 	"github.com/zeromicro/go-zero/rest/httpx"
 
-	"wechat-api/internal/logic/Msg"
+	"wechat-api/internal/logic/msg"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 )

internal/handler/Msg/delete_msg_handler.go

@@ -5,7 +5,7 @@ import (
 
 	"github.com/zeromicro/go-zero/rest/httpx"
 
-	"wechat-api/internal/logic/Msg"
+	"wechat-api/internal/logic/msg"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 )

internal/handler/Msg/get_msg_by_id_handler.go

@@ -5,7 +5,7 @@ import (
 
 	"github.com/zeromicro/go-zero/rest/httpx"
 
-	"wechat-api/internal/logic/Msg"
+	"wechat-api/internal/logic/msg"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 )

internal/handler/Msg/get_msg_list_handler.go

@@ -5,7 +5,7 @@ import (
 
 	"github.com/zeromicro/go-zero/rest/httpx"
 
-	"wechat-api/internal/logic/Msg"
+	"wechat-api/internal/logic/msg"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 )

internal/handler/Msg/update_msg_handler.go

@@ -5,7 +5,7 @@ import (
 
 	"github.com/zeromicro/go-zero/rest/httpx"
 
-	"wechat-api/internal/logic/Msg"
+	"wechat-api/internal/logic/msg"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 )

internal/handler/Wx/check_wx_handler.go

@@ -0,0 +1,44 @@
+package Wx
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/Wx"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /wx/check Wx CheckWx
+//
+// Check wx information | 检查并更新Wx状态
+//
+// Check wx information | 检查并更新Wx状态
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: WxInfo
+//
+// Responses:
+//  200: BaseMsgResp
+
+func CheckWxHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.WxInfo
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := Wx.NewCheckWxLogic(r.Context(), svcCtx)
+		resp, err := l.CheckWx(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/Wx/delete_wx_handler.go

@@ -27,7 +27,7 @@ import (
 
 func DeleteWxHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
-		var req types.IDsReq
+		var req types.IDReq
 		if err := httpx.Parse(r, &req, true); err != nil {
 			httpx.ErrorCtx(r.Context(), w, err)
 			return

internal/handler/Wxhook/terminate_this_we_chat_handler.go

@@ -0,0 +1,44 @@
+package Wxhook
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/Wxhook"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /wxhook/terminateThisWeChat Wxhook TerminateThisWeChat
+//
+// 结束微信
+//
+// 结束微信
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: IDReq
+//
+// Responses:
+//  200: BaseMsgResp
+
+func TerminateThisWeChatHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.IDReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := Wxhook.NewTerminateThisWeChatLogic(r.Context(), svcCtx)
+		resp, err := l.TerminateThisWeChat(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/create_agent_data_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/data/create agent CreateAgentData
+//
+// Create data | 添加data
+//
+// Create data | 添加data
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: CreateDataInfoReq
+//
+// Responses:
+//  200: BaseDataInfo
+
+func CreateAgentDataHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.CreateDataInfoReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewCreateAgentDataLogic(r.Context(), svcCtx)
+		resp, err := l.CreateAgentData(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/delete_agent_data_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/data/delete agent DeleteAgentData
+//
+// Delete data | 删除data
+//
+// Delete data | 删除data
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: DeleteDataReq
+//
+// Responses:
+//  200: BaseDataInfo
+
+func DeleteAgentDataHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.DeleteDataReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewDeleteAgentDataLogic(r.Context(), svcCtx)
+		resp, err := l.DeleteAgentData(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/get_agent_collection_info_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/collection/detail agent GetAgentCollectionInfo
+//
+// Get collect detail | 获取collection详情
+//
+// Get collect detail | 获取collection详情
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: CollectionDetailReq
+//
+// Responses:
+//  200: CollectionInfoResp
+
+func GetAgentCollectionInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.CollectionDetailReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewGetAgentCollectionInfoLogic(r.Context(), svcCtx)
+		resp, err := l.GetAgentCollectionInfo(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/get_agent_collection_list_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/collection/list agent GetAgentCollectionList
+//
+// Get collect list | 获取collection列表
+//
+// Get collect list | 获取collection列表
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: CollectionListReq
+//
+// Responses:
+//  200: CollectionListResp
+
+func GetAgentCollectionListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.CollectionListReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewGetAgentCollectionListLogic(r.Context(), svcCtx)
+		resp, err := l.GetAgentCollectionList(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/get_agent_data_detail_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/data/detail agent GetAgentDataDetail
+//
+// Get data detail | 获取data详情
+//
+// Get data detail | 获取data详情
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: DataDetailReq
+//
+// Responses:
+//  200: DataDetailResp
+
+func GetAgentDataDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.DataDetailReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewGetAgentDataDetailLogic(r.Context(), svcCtx)
+		resp, err := l.GetAgentDataDetail(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/get_agent_data_list_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/data/list agent GetAgentDataList
+//
+// Get data list | 获取data列表
+//
+// Get data list | 获取data列表
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: DataListReq
+//
+// Responses:
+//  200: DataListResp
+
+func GetAgentDataListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.DataListReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewGetAgentDataListLogic(r.Context(), svcCtx)
+		resp, err := l.GetAgentDataList(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/agent/update_agent_data_handler.go

@@ -0,0 +1,44 @@
+package agent
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/agent"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /agent/data/update agent UpdateAgentData
+//
+// Update data | 修改data
+//
+// Update data | 修改data
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: UpdateDataInfoReq
+//
+// Responses:
+//  200: BaseDataInfo
+
+func UpdateAgentDataHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.UpdateDataInfoReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := agent.NewUpdateAgentDataLogic(r.Context(), svcCtx)
+		resp, err := l.UpdateAgentData(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/category/create_category_handler.go

@@ -0,0 +1,44 @@
+package category
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/category"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /category/create category CreateCategory
+//
+// Create category information | 创建Category
+//
+// Create category information | 创建Category
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: CategoryInfo
+//
+// Responses:
+//  200: BaseMsgResp
+
+func CreateCategoryHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.CategoryInfo
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := category.NewCreateCategoryLogic(r.Context(), svcCtx)
+		resp, err := l.CreateCategory(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/category/delete_category_handler.go

@@ -0,0 +1,44 @@
+package category
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/category"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /category/delete category DeleteCategory
+//
+// Delete category information | 删除Category信息
+//
+// Delete category information | 删除Category信息
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: IDsReq
+//
+// Responses:
+//  200: BaseMsgResp
+
+func DeleteCategoryHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.IDsReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := category.NewDeleteCategoryLogic(r.Context(), svcCtx)
+		resp, err := l.DeleteCategory(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/category/get_category_by_id_handler.go

@@ -0,0 +1,44 @@
+package category
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/category"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /category category GetCategoryById
+//
+// Get category by ID | 通过ID获取Category
+//
+// Get category by ID | 通过ID获取Category
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: IDReq
+//
+// Responses:
+//  200: CategoryInfoResp
+
+func GetCategoryByIdHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.IDReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := category.NewGetCategoryByIdLogic(r.Context(), svcCtx)
+		resp, err := l.GetCategoryById(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/category/get_category_list_handler.go

@@ -0,0 +1,44 @@
+package category
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/category"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /category/list category GetCategoryList
+//
+// Get category list | 获取Category列表
+//
+// Get category list | 获取Category列表
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: CategoryListReq
+//
+// Responses:
+//  200: CategoryListResp
+
+func GetCategoryListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.CategoryListReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := category.NewGetCategoryListLogic(r.Context(), svcCtx)
+		resp, err := l.GetCategoryList(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/category/update_category_handler.go

@@ -0,0 +1,44 @@
+package category
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/category"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route post /category/update category UpdateCategory
+//
+// Update category information | 更新Category
+//
+// Update category information | 更新Category
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: CategoryInfo
+//
+// Responses:
+//  200: BaseMsgResp
+
+func UpdateCategoryHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.CategoryInfo
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := category.NewUpdateCategoryLogic(r.Context(), svcCtx)
+		resp, err := l.UpdateCategory(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/employee_config/get_employee_list_config_handler.go

@@ -0,0 +1,44 @@
+package employee_config
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"wechat-api/internal/logic/employee_config"
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+)
+
+// swagger:route get /employee_config/list employee_config GetEmployeeListConfig
+//
+// Get employee config list | 获取EmployeeConfig列表
+//
+// Get employee config list | 获取EmployeeConfig列表
+//
+// Parameters:
+//  + name: body
+//    require: true
+//    in: body
+//    type: EmployeeConfigListReq
+//
+// Responses:
+//  200: EmployeeConfigListResp
+
+func GetEmployeeListConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.EmployeeConfigListReq
+		if err := httpx.Parse(r, &req, true); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := employee_config.NewGetEmployeeListConfigLogic(r.Context(), svcCtx)
+		resp, err := l.GetEmployeeListConfig(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

internal/handler/routes.go

@@ -14,8 +14,10 @@ import (
 	agent "wechat-api/internal/handler/agent"
 	base "wechat-api/internal/handler/base"
 	batch_msg "wechat-api/internal/handler/batch_msg"
+	category "wechat-api/internal/handler/category"
 	contact "wechat-api/internal/handler/contact"
 	employee "wechat-api/internal/handler/employee"
+	employee_config "wechat-api/internal/handler/employee_config"
 	label "wechat-api/internal/handler/label"
 	label_relationship "wechat-api/internal/handler/label_relationship"
 	message_records "wechat-api/internal/handler/message_records"
@@ -87,6 +89,11 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 				},
 				{
 					Method:  http.MethodPost,
+					Path:    "/wx/check",
+					Handler: Wx.CheckWxHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
 					Path:    "/wx/update",
 					Handler: Wx.UpdateWxHandler(serverCtx),
 				},
@@ -139,6 +146,41 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 					Path:    "/agent",
 					Handler: agent.GetAgentByIdHandler(serverCtx),
 				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/collection/list",
+					Handler: agent.GetAgentCollectionListHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/collection/detail",
+					Handler: agent.GetAgentCollectionInfoHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/data/list",
+					Handler: agent.GetAgentDataListHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/data/detail",
+					Handler: agent.GetAgentDataDetailHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/data/create",
+					Handler: agent.CreateAgentDataHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/data/update",
+					Handler: agent.UpdateAgentDataHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/agent/data/delete",
+					Handler: agent.DeleteAgentDataHandler(serverCtx),
+				},
 			}...,
 		),
 		rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
@@ -170,6 +212,11 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 				},
 				{
 					Method:  http.MethodPost,
+					Path:    "/wxhook/terminateThisWeChat",
+					Handler: Wxhook.TerminateThisWeChatHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
 					Path:    "/wxhook/getFriendsAndGroups",
 					Handler: Wxhook.GetFriendsAndGroupsHandler(serverCtx),
 				},
@@ -740,6 +787,20 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 	)
 
 	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.Authority},
+			[]rest.Route{
+				{
+					Method:  http.MethodGet,
+					Path:    "/employee_config/list",
+					Handler: employee_config.GetEmployeeListConfigHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
+	)
+
+	server.AddRoutes(
 		[]rest.Route{
 			{
 				Method:  http.MethodPost,
@@ -773,4 +834,38 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 			},
 		},
 	)
+
+	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.Authority},
+			[]rest.Route{
+				{
+					Method:  http.MethodPost,
+					Path:    "/category/create",
+					Handler: category.CreateCategoryHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/category/update",
+					Handler: category.UpdateCategoryHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/category/delete",
+					Handler: category.DeleteCategoryHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/category/list",
+					Handler: category.GetCategoryListHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/category",
+					Handler: category.GetCategoryByIdHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
+	)
 }

internal/logic/WechatServer/create_server_logic.go

@@ -2,7 +2,6 @@ package WechatServer
 
 import (
 	"context"
-
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 	"wechat-api/internal/utils/dberrorhandler"
@@ -27,7 +26,7 @@ func NewCreateServerLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Crea
 }
 
 func (l *CreateServerLogic) CreateServer(req *types.ServerInfo) (*types.BaseMsgResp, error) {
-	_, err := l.svcCtx.DB.Server.Create().
+	serverInfo, err := l.svcCtx.DB.Server.Create().
 		SetNotNilStatus(req.Status).
 		SetNotNilName(req.Name).
 		SetNotNilPublicIP(req.PublicIp).
@@ -38,6 +37,10 @@ func (l *CreateServerLogic) CreateServer(req *types.ServerInfo) (*types.BaseMsgR
 	if err != nil {
 		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
 	}
-
+	data := map[string]interface{}{
+		"private_ip": serverInfo.PrivateIP,
+		"admin_port": serverInfo.AdminPort,
+	}
+	l.svcCtx.Rds.HSet(l.ctx, "server_info", serverInfo.ID, data)
 	return &types.BaseMsgResp{Msg: errormsg.CreateSuccess}, nil
 }

internal/logic/WechatServer/update_server_logic.go

@@ -37,6 +37,10 @@ func (l *UpdateServerLogic) UpdateServer(req *types.ServerInfo) (*types.BaseMsgR
 	if err != nil {
 		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
 	}
-
+	data := map[string]interface{}{
+		"private_ip": req.PrivateIp,
+		"admin_port": req.AdminPort,
+	}
+	l.svcCtx.Rds.HSet(l.ctx, "server_info", *req.Id, data)
 	return &types.BaseMsgResp{Msg: errormsg.UpdateSuccess}, nil
 }

internal/logic/Wx/check_wx_logic.go

@@ -0,0 +1,125 @@
+package Wx
+
+import (
+	"context"
+	"errors"
+	"github.com/suyuan32/simple-admin-common/msg/errormsg"
+	"github.com/zeromicro/go-zero/core/errorx"
+	"wechat-api/ent"
+	agentModel "wechat-api/ent/agent"
+	"wechat-api/ent/wx"
+	"wechat-api/hook"
+	"wechat-api/internal/utils/dberrorhandler"
+
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type CheckWxLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewCheckWxLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CheckWxLogic {
+	return &CheckWxLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx}
+}
+
+func (l *CheckWxLogic) CheckWx(req *types.WxInfo) (resp *types.BaseMsgResp, err error) {
+	isAdmin := l.ctx.Value("isAdmin").(bool)
+	if !isAdmin {
+		l.Error("没有权限，请联系管理员", nil)
+		return nil, nil
+	}
+
+	// 获取服务器信息
+	server, err := l.svcCtx.DB.Server.Get(l.ctx, *req.ServerId)
+	if err != nil {
+		l.Error("获取服务器信息失败", err)
+		return nil, err
+	}
+
+	hook := hook.NewHook(server.PrivateIP, server.AdminPort, *req.Port)
+
+	//  判断端口是否被占用
+	portOccupied, err := hook.GetPortOccupiedInfo(*req.Port)
+	if err != nil {
+		l.Error("获取端口占用信息失败", err)
+		return nil, err
+	}
+
+	if portOccupied.Occupied != "1" {
+		l.Error("端口未使用，请重新扫码登陆", err)
+		return nil, err
+	}
+
+	// 判断所选模式是否可用
+	if req.AgentId != nil {
+		agent, err := l.svcCtx.DB.Agent.Query().Where(agentModel.Status(1), agentModel.ID(*req.AgentId)).First(l.ctx)
+		if err != nil && ent.IsNotFound(err) || agent == nil {
+			return nil, errors.New("所选模式不存在或者已被禁用，请选择其他模式")
+		}
+	}
+
+	loginStatus, err := hook.IsLoginStatus()
+	if err != nil {
+		return nil, err
+	}
+
+	if loginStatus.Onlinestatus == "3" {
+		selfInfo, err := hook.GetSelfLoginInfo()
+		if err != nil {
+			return nil, err
+		}
+
+		wxInfo, err := l.svcCtx.DB.Wx.Query().Where(wx.ServerIDEQ(server.ID)).Where(wx.PortEQ(*req.Port)).Limit(1).Only(l.ctx)
+		if err != nil && !ent.IsNotFound(err) {
+			l.Error("查询微信信息失败", err)
+			return nil, err
+		}
+
+		if req.Callback != nil && *req.Callback != "" {
+			_ = hook.ConfigureMsgRecive("1", *req.Callback)
+		}
+
+		var dbErr error
+		if ent.IsNotFound(err) {
+			l.Error("账号不存在，请联系管理员", err)
+			return nil, err
+		} else {
+			_, dbErr = l.svcCtx.DB.Wx.UpdateOneID(wxInfo.ID).
+				SetNotNilProcessID(&selfInfo.ProcessID).
+				SetNotNilCallback(req.Callback).
+				SetNotNilWxid(&selfInfo.Wxid).
+				SetNotNilAccount(&selfInfo.Account).
+				SetNotNilNickname(&selfInfo.Nickname).
+				SetNotNilTel(&selfInfo.Tel).
+				SetNotNilHeadBig(&selfInfo.HeadBig).
+				SetStatus(1).
+				Save(l.ctx)
+
+			data := map[string]interface{}{
+				"nickname":  selfInfo.Nickname,
+				"server_id": wxInfo.ServerID,
+				"port":      wxInfo.Port,
+				"api_base":  wxInfo.APIBase,
+				"api_key":   wxInfo.APIKey,
+				"agent_id":  wxInfo.AgentID,
+			}
+			l.svcCtx.Rds.HSet(l.ctx, "wx_info", selfInfo.Wxid, data)
+		}
+		if dbErr != nil {
+			return nil, dberrorhandler.DefaultEntError(l.Logger, dbErr, req)
+		}
+	} else {
+		l.Error("微信未登录", "loginStatus", loginStatus)
+		return nil, errorx.NewDefaultError("微信未登陆，当前登陆状态：" + loginStatus.Onlinestatus + "，" + loginStatus.Msg)
+	}
+
+	return &types.BaseMsgResp{Msg: errormsg.Success}, nil
+}

internal/logic/Wx/create_wx_logic.go

@@ -2,10 +2,11 @@ package Wx
 
 import (
 	"context"
-	"errors"
 	"github.com/zeromicro/go-zero/core/errorx"
+	"math/rand"
+	"strconv"
+	"time"
 	"wechat-api/ent"
-	agentModel "wechat-api/ent/agent"
 	"wechat-api/ent/wx"
 	"wechat-api/hook"
 	"wechat-api/internal/utils/dberrorhandler"
@@ -32,7 +33,12 @@ func NewCreateWxLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateWx
 }
 
 func (l *CreateWxLogic) CreateWx(req *types.WxInfo) (*types.BaseMsgResp, error) {
-	organizationId := l.ctx.Value("organizationId").(uint64)
+	isAdmin := l.ctx.Value("isAdmin").(bool)
+	if !isAdmin {
+		l.Error("没有权限，请联系管理员", nil)
+		return nil, errorx.NewDefaultError("没有权限，请联系管理员")
+	}
+	// 获取服务器信息
 	server, err := l.svcCtx.DB.Server.Get(l.ctx, *req.ServerId)
 	if err != nil {
 		l.Error("获取服务器信息失败", err)
@@ -41,82 +47,42 @@ func (l *CreateWxLogic) CreateWx(req *types.WxInfo) (*types.BaseMsgResp, error)
 
 	hook := hook.NewHook(server.PrivateIP, server.AdminPort, *req.Port)
 
+	// 判断端口是否被占用
 	portOccupied, err := hook.GetPortOccupiedInfo(*req.Port)
 	if err != nil {
 		l.Error("获取端口占用信息失败", err)
 		return nil, err
 	}
 
-	if portOccupied.Occupied != "1" {
-		l.Error("端口未使用，请重新扫码登陆", err)
+	if portOccupied.Occupied == "1" {
+		l.Error("端口已占用，请改用其他端口", err)
 		return nil, err
 	}
 
-	// 判断所选模式是否可用
-	if req.AgentId != nil {
-		agent, err := l.svcCtx.DB.Agent.Query().Where(agentModel.Status(1), agentModel.ID(*req.AgentId)).First(l.ctx)
-		if err != nil && ent.IsNotFound(err) || agent == nil {
-			return nil, errors.New("所选模式不存在或者已被禁用，请选择其他模式")
-		}
-	}
-
-	loginStatus, err := hook.IsLoginStatus()
-	if err != nil {
+	_, err = l.svcCtx.DB.Wx.Query().Where(wx.ServerIDEQ(server.ID)).Where(wx.PortEQ(*req.Port)).Limit(1).Only(l.ctx)
+	if err != nil && !ent.IsNotFound(err) {
+		l.Error("查询微信信息失败", err)
 		return nil, err
 	}
 
-	if loginStatus.Onlinestatus == "3" {
-		selfInfo, err := hook.GetSelfLoginInfo()
+	if ent.IsNotFound(err) {
+		wxid := strconv.Itoa(rand.New(rand.NewSource(time.Now().UnixNano())).Intn(1000000))
+		_, err := l.svcCtx.DB.Wx.Create().
+			SetServerID(*req.ServerId).
+			SetPort(*req.Port).
+			SetCallback(*req.Callback).
+			SetOrganizationID(*req.OrganizationId).
+			SetNotNilAPIBase(req.ApiBase).
+			SetNotNilAPIKey(req.ApiKey).
+			SetWxid(wxid).
+			SetStatus(0).
+			Save(l.ctx)
 		if err != nil {
-			return nil, err
-		}
-
-		wxInfo, err := l.svcCtx.DB.Wx.Query().Where(wx.ServerIDEQ(server.ID)).Where(wx.PortEQ(*req.Port)).Limit(1).Only(l.ctx)
-		if err != nil && !ent.IsNotFound(err) {
-			l.Error("查询微信信息失败", err)
-			return nil, err
-		}
-
-		var dbErr error
-		if ent.IsNotFound(err) {
-			if req.Callback != nil && *req.Callback != "" {
-				_ = hook.ConfigureMsgRecive("1", *req.Callback)
-			}
-			_, dbErr = l.svcCtx.DB.Wx.Create().
-				SetNotNilServerID(req.ServerId).
-				SetNotNilPort(req.Port).
-				SetNotNilProcessID(&selfInfo.ProcessID).
-				SetNotNilCallback(req.Callback).
-				SetNotNilWxid(&selfInfo.Wxid).
-				SetNotNilAccount(&selfInfo.Account).
-				SetNotNilNickname(&selfInfo.Nickname).
-				SetNotNilTel(&selfInfo.Tel).
-				SetNotNilHeadBig(&selfInfo.HeadBig).
-				SetStatus(1).
-				SetOrganizationID(organizationId).
-				SetNotNilAgentID(req.AgentId).
-				Save(l.ctx)
-		} else {
-			if req.Callback != nil && *req.Callback != "" {
-				_ = hook.ConfigureMsgRecive("1", *req.Callback)
-			}
-			_, dbErr = l.svcCtx.DB.Wx.UpdateOneID(wxInfo.ID).
-				SetNotNilProcessID(&selfInfo.ProcessID).
-				SetNotNilCallback(req.Callback).
-				SetNotNilWxid(&selfInfo.Wxid).
-				SetNotNilAccount(&selfInfo.Account).
-				SetNotNilNickname(&selfInfo.Nickname).
-				SetNotNilTel(&selfInfo.Tel).
-				SetNotNilHeadBig(&selfInfo.HeadBig).
-				SetStatus(1).
-				Save(l.ctx)
-		}
-		if dbErr != nil {
-			return nil, dberrorhandler.DefaultEntError(l.Logger, dbErr, req)
+			return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
 		}
 	} else {
-		l.Error("微信未登录", "loginStatus", loginStatus)
-		return nil, errorx.NewDefaultError("微信未登陆，当前登陆状态：" + loginStatus.Onlinestatus + "，" + loginStatus.Msg)
+		l.Error("端口已使用，请使用其他端口", err)
+		return nil, err
 	}
 
 	return &types.BaseMsgResp{Msg: errormsg.Success}, nil

internal/logic/Wx/delete_wx_logic.go

@@ -2,6 +2,10 @@ package Wx
 
 import (
 	"context"
+	"math/rand"
+	"strconv"
+	"time"
+	"wechat-api/hook"
 
 	"wechat-api/ent/wx"
 	"wechat-api/internal/svc"
@@ -26,14 +30,64 @@ func NewDeleteWxLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteWx
 	}
 }
 
-func (l *DeleteWxLogic) DeleteWx(req *types.IDsReq) (resp *types.BaseMsgResp, err error) {
+func (l *DeleteWxLogic) DeleteWx(req *types.IDReq) (resp *types.BaseMsgResp, err error) {
 	isAdmin := l.ctx.Value("isAdmin").(bool)
+
+	// 获取账号信息
+	wxInfo, err := l.svcCtx.DB.Wx.Query().
+		Where(
+			wx.IDEQ(req.Id),
+		).
+		Only(l.ctx)
+	if err != nil {
+		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
+	}
+
+	// 获取服务器信息
+	serverInfo, err := l.svcCtx.DB.Server.Get(l.ctx, wxInfo.ServerID)
+	if err != nil {
+		l.Error("获取服务器信息失败", err)
+		return nil, err
+	}
+
+	hook := hook.NewHook(serverInfo.PrivateIP, serverInfo.AdminPort, wxInfo.Port)
+
+	//  判断端口是否被占用
+	portOccupied, err := hook.GetPortOccupiedInfo(wxInfo.Port)
+	if err != nil {
+		l.Error("获取端口占用信息失败", err)
+		return nil, err
+	}
+
+	if portOccupied.Occupied == "1" {
+		// 判断账号是否正在登录
+		loginStatus, err := hook.IsLoginStatus()
+		if err != nil {
+			return nil, err
+		}
+
+		if loginStatus.Onlinestatus == "3" {
+			l.Error("请先退出账号再删除", nil)
+			return nil, nil
+		}
+	}
+
+	randNum := strconv.Itoa(rand.New(rand.NewSource(time.Now().UnixNano())).Intn(1000000))
+
 	if isAdmin {
-		_, err = l.svcCtx.DB.Wx.Delete().Where(wx.IDIn(req.Ids...)).Exec(l.ctx)
+		err = l.svcCtx.DB.Wx.UpdateOneID(req.Id).
+			SetWxid(randNum).
+			SetPort(randNum).
+			Exec(l.ctx)
+		_, err = l.svcCtx.DB.Wx.Delete().Where(wx.IDEQ(req.Id)).Exec(l.ctx)
 	} else {
 		organizationId := l.ctx.Value("organizationId").(uint64)
-
-		_, err = l.svcCtx.DB.Wx.Delete().Where(wx.IDIn(req.Ids...), wx.OrganizationIDEQ(organizationId)).Exec(l.ctx)
+		err = l.svcCtx.DB.Wx.UpdateOneID(req.Id).
+			Where(wx.OrganizationID(organizationId)).
+			SetWxid(randNum).
+			SetPort(randNum).
+			Exec(l.ctx)
+		_, err = l.svcCtx.DB.Wx.Delete().Where(wx.IDEQ(req.Id), wx.OrganizationIDEQ(organizationId)).Exec(l.ctx)
 	}
 
 	if err != nil {

internal/logic/Wx/get_wx_list_logic.go

@@ -50,6 +50,14 @@ func (l *GetWxListLogic) GetWxList(req *types.WxListReq) (*types.WxListResp, err
 		if req.OrganizationId != nil {
 			predicates = append(predicates, wx.OrganizationIDEQ(*req.OrganizationId))
 		}
+		if req.OrganizationName != nil {
+			departmentList, _ := l.svcCtx.CoreRpc.GetDepartmentList(l.ctx, &core.DepartmentListReq{Name: req.OrganizationName})
+			organizationIds := make([]uint64, 0)
+			for _, department := range departmentList.Data {
+				organizationIds = append(organizationIds, *department.Id)
+			}
+			predicates = append(predicates, wx.OrganizationIDIn(organizationIds...))
+		}
 	}
 	if req.ServerId != nil {
 		predicates = append(predicates, wx.ServerIDEQ(*req.ServerId))
@@ -170,6 +178,8 @@ func (l *GetWxListLogic) GetWxList(req *types.WxListReq) (*types.WxListResp, err
 				OrganizationName: departmentInfo.Name,
 				AgentId:          &v.AgentID,
 				AgentInfo:        &agent,
+				ApiBase:          &v.APIBase,
+				ApiKey:           &v.APIKey,
 			})
 	}
 

internal/logic/Wx/update_wx_logic.go

@@ -3,6 +3,7 @@ package Wx
 import (
 	"context"
 	"wechat-api/ent/wx"
+	"wechat-api/hook"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 	"wechat-api/internal/utils/dberrorhandler"
@@ -28,41 +29,72 @@ func NewUpdateWxLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateWx
 func (l *UpdateWxLogic) UpdateWx(req *types.WxInfo) (resp *types.BaseMsgResp, err error) {
 	isAdmin := l.ctx.Value("isAdmin").(bool)
 
+	wxinfo, err := l.svcCtx.DB.Wx.Query().
+		Where(
+			wx.IDEQ(*req.Id),
+		).
+		WithAgent().
+		Only(l.ctx)
+	if err != nil {
+		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
+	}
+
 	if isAdmin {
-		err = l.svcCtx.DB.Wx.UpdateOneID(*req.Id).
-			SetNotNilStatus(req.Status).
-			SetNotNilServerID(req.ServerId).
-			SetNotNilPort(req.Port).
-			SetNotNilProcessID(req.ProcessId).
+		tx, err := l.svcCtx.DB.Tx(context.Background())
+
+		if err != nil {
+			return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
+		}
+
+		err = tx.Wx.UpdateOneID(*req.Id).
 			SetNotNilCallback(req.Callback).
-			SetNotNilWxid(req.Wxid).
-			SetNotNilAccount(req.Account).
-			SetNotNilNickname(req.Nickname).
-			SetNotNilTel(req.Tel).
-			SetNotNilHeadBig(req.HeadBig).
 			SetNotNilAgentID(req.AgentId).
+			SetNotNilOrganizationID(req.OrganizationId).
+			SetNotNilAPIBase(req.ApiBase).
+			SetNotNilAPIKey(req.ApiKey).
 			Exec(l.ctx)
+
+		if req.Callback != nil && *req.Callback != "" {
+			serverinfo, serverErr := tx.Server.Get(l.ctx, wxinfo.ServerID)
+			if serverErr != nil {
+				_ = tx.Rollback()
+				return nil, dberrorhandler.DefaultEntError(l.Logger, serverErr, req)
+			}
+			hook := hook.NewHook(serverinfo.PrivateIP, serverinfo.AdminPort, wxinfo.Port)
+			loginStatus, _ := hook.IsLoginStatus()
+			if err != nil {
+				_ = tx.Rollback()
+				return nil, err
+			}
+			if loginStatus.Onlinestatus == "3" {
+				err = hook.ConfigureMsgRecive("1", *req.Callback)
+				if err != nil {
+					_ = tx.Rollback()
+					return nil, err
+				}
+			}
+		}
+
+		if err != nil {
+			_ = tx.Rollback()
+			l.Error("更新失败", err)
+			return nil, err
+		}
+		// 所有操作成功，提交事务
+		err = tx.Commit()
+		if err != nil {
+			return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
+		}
 	} else {
 		organizationId := l.ctx.Value("organizationId").(uint64)
 		err = l.svcCtx.DB.Wx.UpdateOneID(*req.Id).
 			Where(wx.OrganizationID(organizationId)).
-			SetNotNilStatus(req.Status).
-			SetNotNilServerID(req.ServerId).
-			SetNotNilPort(req.Port).
-			SetNotNilProcessID(req.ProcessId).
-			SetNotNilCallback(req.Callback).
-			SetNotNilWxid(req.Wxid).
-			SetNotNilAccount(req.Account).
-			SetNotNilNickname(req.Nickname).
-			SetNotNilTel(req.Tel).
-			SetNotNilHeadBig(req.HeadBig).
 			SetNotNilAgentID(req.AgentId).
 			Exec(l.ctx)
+		if err != nil {
+			return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
+		}
 	}
-
-	if err != nil {
-		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
-	}
-
+	l.svcCtx.Rds.HDel(l.ctx, "wx_info", wxinfo.Wxid)
 	return &types.BaseMsgResp{Msg: errormsg.UpdateSuccess}, nil
 }

internal/logic/Wxhook/logout_logic.go

@@ -59,6 +59,6 @@ func (l *LogoutLogic) Logout(req *types.IDReq) (resp *types.BaseMsgResp, err err
 	} else {
 		resp.Code = errorcode.Unknown
 	}
-
+	_, _ = hookClient.TerminateThisWeChat(wxInfo.ProcessID)
 	return
 }

internal/logic/Wxhook/refresh_login_q_r_logic.go

@@ -3,7 +3,6 @@ package Wxhook
 import (
 	"context"
 	"github.com/zeromicro/go-zero/core/errorx"
-	"time"
 	"wechat-api/hook"
 
 	"wechat-api/internal/svc"
@@ -27,14 +26,21 @@ func NewRefreshLoginQRLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Re
 }
 
 func (l *RefreshLoginQRLogic) RefreshLoginQR(req *types.RefreshLoginQRReq) (resp *types.RefreshLoginQRResp, err error) {
+	//wxInfo, err := l.svcCtx.DB.Wx.Query().Where(wx.ServerIDEQ(*req.ServerId)).Where(wx.PortEQ(*req.Port)).Limit(1).Only(l.ctx)
+	//if err != nil {
+	//	l.Error("查询微信信息失败", err)
+	//	return
+	//}
 
+	// 获取服务器信息
 	server, err := l.svcCtx.DB.Server.Get(l.ctx, *req.ServerId)
 	if err != nil {
 		l.Error("获取服务器信息失败", err)
 		return
 	}
 	hookClient := hook.NewHook(server.PrivateIP, server.AdminPort, *req.Port)
-
+	//_, _ = hookClient.TerminateThisWeChat(wxInfo.ProcessID)
+	// 如端口未被占用则启动微信
 	portOccupied, err := hookClient.GetPortOccupiedInfo(*req.Port)
 	if err != nil {
 		l.Error("获取端口占用信息失败", err)
@@ -53,9 +59,6 @@ func (l *RefreshLoginQRLogic) RefreshLoginQR(req *types.RefreshLoginQRReq) (resp
 		}
 	}
 
-	// 间隔一秒钟
-	time.Sleep(1 * time.Second)
-
 	loginStatus, err := hookClient.IsLoginStatus()
 	if err != nil {
 		l.Error("查询登录状态失败", err)

internal/logic/Wxhook/terminate_this_we_chat_logic.go

@@ -0,0 +1,29 @@
+package Wxhook
+
+import (
+	"context"
+
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type TerminateThisWeChatLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewTerminateThisWeChatLogic(ctx context.Context, svcCtx *svc.ServiceContext) *TerminateThisWeChatLogic {
+	return &TerminateThisWeChatLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx}
+}
+
+func (l *TerminateThisWeChatLogic) TerminateThisWeChat(req *types.IDReq) (resp *types.BaseMsgResp, err error) {
+	// todo: add your logic here and delete this line
+
+	return
+}

internal/logic/agent/create_agent_data_logic.go

@@ -0,0 +1,54 @@
+package agent
+
+import (
+	"context"
+	"fmt"
+	"github.com/suyuan32/simple-admin-common/msg/errormsg"
+	"github.com/zeromicro/go-zero/core/errorx"
+	"wechat-api/hook/fastgpt"
+
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type CreateAgentDataLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewCreateAgentDataLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateAgentDataLogic {
+	return &CreateAgentDataLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx}
+}
+
+func (l *CreateAgentDataLogic) CreateAgentData(req *types.CreateDataInfoReq) (*types.BaseDataInfo, error) {
+	params := fastgpt.CreateBulkDataReq{}
+	params.CollectionID = *req.CollectionId
+	params.TrainingMode = "chunk"
+	pair := make([]fastgpt.DataQuestion, 0, 1)
+	pair = append(pair, fastgpt.DataQuestion{
+		Q: *req.Q,
+		A: *req.A,
+	})
+	params.Data = append(params.Data, pair...)
+
+	resp, err := fastgpt.CreateBulkData(&params)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("fastgpt create data failed " + err.Error())
+	}
+
+	if resp.Code == 200 {
+		return &types.BaseDataInfo{
+			Code: 0,
+			Msg:  errormsg.Success,
+			Data: fmt.Sprintf("Insert %d rows", resp.Data.InsertLen),
+		}, nil
+	}
+
+	return nil, errorx.NewInvalidArgumentError(resp.StatusText)
+}

internal/logic/agent/create_agent_logic.go

@@ -2,6 +2,9 @@ package agent
 
 import (
 	"context"
+	"errors"
+	"github.com/zeromicro/go-zero/core/errorx"
+	"wechat-api/hook/fastgpt"
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 	"wechat-api/internal/utils/dberrorhandler"
@@ -28,7 +31,7 @@ func NewCreateAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Creat
 func (l *CreateAgentLogic) CreateAgent(req *types.AgentInfo) (*types.BaseMsgResp, error) {
 	organizationId := l.ctx.Value("organizationId").(uint64)
 
-	_, err := l.svcCtx.DB.Agent.Create().
+	agent, err := l.svcCtx.DB.Agent.Create().
 		SetNotNilName(req.Name).
 		SetNotNilRole(req.Role).
 		//SetNotNilStatus(req.Status).
@@ -41,5 +44,44 @@ func (l *CreateAgentLogic) CreateAgent(req *types.AgentInfo) (*types.BaseMsgResp
 		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
 	}
 
+	/**
+	1. 创建知识库
+	2. 创建默认集合
+	3. 保存到Agent信息里
+	*/
+	var datasetReq fastgpt.DatasetReq
+	datasetReq.Name = *req.Name
+	datasetReq.Intro = *req.Name
+	datasetReq.AgentModel = "gpt-3.5-turbo"
+	datasetReq.VectorModel = "text-embedding-ada-002"
+	datasetResp, err := fastgpt.CreateDataset(&datasetReq)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("fastgpt create dataset failed " + err.Error())
+	}
+	if datasetResp.Code != 200 {
+		return nil, errors.New(datasetResp.Message)
+	}
+
+	var collectionReq fastgpt.CreateCollectionReq
+	collectionReq.DatasetId = datasetResp.Data
+	collectionReq.Name = "手动录入"
+	collectionReq.Type = "folder"
+	collectionResp, err := fastgpt.CreateEmptyCollection(&collectionReq)
+	if err != nil {
+		return nil, errors.New("create dataset failed")
+	}
+	if collectionResp.Code != 200 {
+		return nil, errorx.NewInvalidArgumentError("fastgpt create collection failed " + err.Error())
+	}
+
+	_, err = l.svcCtx.DB.Agent.UpdateOneID(agent.ID).
+		SetNotNilDatasetID(&datasetResp.Data).
+		SetNotNilCollectionID(&collectionResp.Data).
+		Save(l.ctx)
+
+	if err != nil {
+		return nil, errors.New("update dataset and collection failed")
+	}
+
 	return &types.BaseMsgResp{Msg: errormsg.CreateSuccess}, nil
 }

internal/logic/agent/delete_agent_data_logic.go

@@ -0,0 +1,43 @@
+package agent
+
+import (
+	"context"
+	"github.com/suyuan32/simple-admin-common/msg/errormsg"
+	"github.com/zeromicro/go-zero/core/errorx"
+	"wechat-api/hook/fastgpt"
+
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type DeleteAgentDataLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewDeleteAgentDataLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteAgentDataLogic {
+	return &DeleteAgentDataLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx}
+}
+
+func (l *DeleteAgentDataLogic) DeleteAgentData(req *types.DeleteDataReq) (*types.BaseDataInfo, error) {
+	resp, err := fastgpt.DeleteData(*req.ID)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("fastgpt create data failed " + err.Error())
+	}
+
+	if resp.Code == 200 {
+		return &types.BaseDataInfo{
+			Code: 0,
+			Msg:  errormsg.Success,
+			Data: "",
+		}, nil
+	}
+
+	return nil, errorx.NewInvalidArgumentError(resp.StatusText)
+}

internal/logic/agent/get_agent_by_id_logic.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"wechat-api/ent"
 	"wechat-api/ent/agent"
-
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 	"wechat-api/internal/utils/dberrorhandler"
@@ -47,11 +46,13 @@ func (l *GetAgentByIdLogic) GetAgentById(req *types.IDReq) (*types.AgentInfoResp
 				CreatedAt: pointy.GetPointer(data.CreatedAt.UnixMilli()),
 				UpdatedAt: pointy.GetPointer(data.UpdatedAt.UnixMilli()),
 			},
-			Name:       &data.Name,
-			Role:       &data.Role,
-			Status:     &data.Status,
-			Background: &data.Background,
-			Examples:   &data.Examples,
+			Name:         &data.Name,
+			Role:         &data.Role,
+			Status:       &data.Status,
+			Background:   &data.Background,
+			Examples:     &data.Examples,
+			DatasetId:    &data.DatasetID,
+			CollectionId: &data.CollectionID,
 		},
 	}, nil
 }

internal/logic/agent/get_agent_collection_info_logic.go

@@ -0,0 +1,71 @@
+package agent
+
+import (
+	"context"
+	"github.com/suyuan32/simple-admin-common/msg/errormsg"
+	"github.com/zeromicro/go-zero/core/errorx"
+	"wechat-api/hook/fastgpt"
+
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetAgentCollectionInfoLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetAgentCollectionInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAgentCollectionInfoLogic {
+	return &GetAgentCollectionInfoLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx}
+}
+
+func (l *GetAgentCollectionInfoLogic) GetAgentCollectionInfo(req *types.CollectionDetailReq) (*types.CollectionInfoResp, error) {
+	resp, err := fastgpt.GetCollectionDetail(*req.ID)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("fastgpt get collection failed " + err.Error())
+	}
+
+	if resp.Code == 200 {
+		return &types.CollectionInfoResp{
+			BaseDataInfo: types.BaseDataInfo{
+				Code: 0,
+				Msg:  errormsg.Success,
+			},
+			Data: types.CollectionInfo{
+				ID:            &resp.Data.ID,
+				ParentID:      &resp.Data.ParentID,
+				TmbId:         &resp.Data.TmbID,
+				Type:          &resp.Data.Type,
+				Name:          &resp.Data.Name,
+				TrainingType:  &resp.Data.TrainingType,
+				ChunkSize:     &resp.Data.ChunkSize,
+				ChunkSplitter: &resp.Data.ChunkSplitter,
+				QaPrompt:      &resp.Data.QaPrompt,
+				RawTextLength: &resp.Data.RawTextLength,
+				CanWrite:      &resp.Data.CanWrite,
+				SourceName:    &resp.Data.SourceName,
+				DatasetId: types.DatasetId{
+					ID:          &resp.Data.DatasetID.ID,
+					ParentID:    &resp.Data.DatasetID.ParentID,
+					TeamId:      &resp.Data.DatasetID.TeamID,
+					TmbId:       &resp.Data.DatasetID.TmbID,
+					Type:        &resp.Data.DatasetID.Type,
+					Status:      &resp.Data.DatasetID.Status,
+					Avatar:      &resp.Data.DatasetID.Avatar,
+					Name:        &resp.Data.DatasetID.Name,
+					VectorModel: &resp.Data.DatasetID.VectorModel,
+					AgentModel:  &resp.Data.DatasetID.AgentModel,
+					Intro:       &resp.Data.DatasetID.Intro,
+				},
+			},
+		}, nil
+	}
+
+	return nil, errorx.NewInvalidArgumentError(resp.StatusText)
+}

internal/logic/agent/get_agent_collection_list_logic.go

@@ -0,0 +1,60 @@
+package agent
+
+import (
+	"context"
+	"github.com/zeromicro/go-zero/core/errorx"
+	"wechat-api/hook/fastgpt"
+
+	"wechat-api/internal/svc"
+	"wechat-api/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetAgentCollectionListLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetAgentCollectionListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAgentCollectionListLogic {
+	return &GetAgentCollectionListLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx}
+}
+
+func (l *GetAgentCollectionListLogic) GetAgentCollectionList(req *types.CollectionListReq) (*types.CollectionListResp, error) {
+	collectionResp := types.CollectionListResp{}
+	collectionResp.PageNum = req.PageNum
+	collectionResp.PageSize = req.PageSize
+	var total int
+
+	params := fastgpt.GetCollectionListReq{}
+	params.DatasetId = *req.DatasetId
+	params.PageNum = *req.PageNum
+	params.PageSize = *req.PageSize
+	resp, err := fastgpt.GetCollectionList(&params)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("fastgpt error" + err.Error())
+	}
+
+	collectionResp.Data = make([]types.CollectionSimpleInfo, 0, 1)
+	collectionResp.Total = &total
+	if resp.Code == 200 && resp.Data.Total > 0 {
+		for _, val := range resp.Data.Data {
+			collectionResp.Data = append(collectionResp.Data, types.CollectionSimpleInfo{
+				ID:             &val.ID,
+				ParentID:       &val.ParentID,
+				TmbId:          &val.TmbID,
+				Type:           &val.Type,
+				Name:           &val.Name,
+				DataAmount:     &val.DataAmount,
+				TrainingAmount: &val.TrainingAmount,
+			})
+		}
+		collectionResp.Total = &resp.Data.Total
+	}
+
+	return &collectionResp, nil
+}