yx-dataset-server/app/schema/s_chat_assistant.go

package schema

import (
	"time"
)

// ChatAssistant ChatAssistant对象
type ChatAssistant struct {
	RecordID    string       `json:"record_id"`    // 记录id
	OrgId       string       `json:"org_id"`       // 组织id
	UserId      string       `json:"user_id"`      // 所属用户id
	Name        string       `json:"name"`         // 助手名称
	Desc        string       `json:"desc"`         // 描述
	RagChatId   string       `json:"rag_chat_id"`  // rag_chat_id
	Datasets    Datasets     `json:"datasets"`     // 绑定的知识库id
	CreatedAt   time.Time    `json:"created_at"`   // 创建时间
	CreatorId   string       `json:"creator_id"`   // 创建人id
	CreatorName string       `json:"creator_name"` // 创建人名称
	Sessions    ChatSessions `json:"sessions"`     // 会话列表
}

// ChatAssistantQueryParam 查询条件
type ChatAssistantQueryParam struct {
	OrgId     string
	LikeName  string
	RecordIDs []string
}

// ChatAssistantQueryOptions ChatAssistant对象查询可选参数项
type ChatAssistantQueryOptions struct {
	PageParam *PaginationParam // 分页参数
}

type ChatAssistants []*ChatAssistant

// ChatAssistantQueryResult ChatAssistant对象查询结果
type ChatAssistantQueryResult struct {
	Data       ChatAssistants
	PageResult *PaginationResult
}

// FillCreator 填充创建者信息
func (a ChatAssistants) FillCreator(users Users) {
	for _, v := range a {
		for _, u := range users {
			if v.CreatorId == u.RecordID {
				v.CreatorName = u.RealName
				continue
			}
		}
	}
}

// FillDatasetId 填充知识库id
func (a ChatAssistants) FillDatasetId(data ChatDatasets) {
	for _, v := range a {
		for _, d := range data {
			if v.RecordID == d.ChatAssistantId {
				v.Datasets = append(v.Datasets, &Dataset{RecordID: d.DatasetId})
				continue
			}
		}
	}
}

// FillDatasetId 填充知识库id
func (a ChatAssistants) FillDataset(data Datasets) {
	for _, v := range a {
		for _, d := range v.Datasets {
			for _, d2 := range data {
				if d.RecordID == d2.RecordID {
					d.Name = d2.Name
					d.OrgId = d2.OrgId
					d.RagDataId = d2.RagDataId
					d.FileCount = d2.FileCount
					continue
				}
			}
		}
	}
}