mirror of
https://github.com/2930134478/AI-CS.git
synced 2026-06-15 00:44:30 +08:00
537yaha
297 lines
11 KiB
Go
297 lines
11 KiB
Go
package service
|
||
|
||
import "time"
|
||
|
||
// BroadcastHub 描述 WebSocket Hub 的广播能力。
|
||
type BroadcastHub interface {
|
||
BroadcastMessage(conversationID uint, messageType string, data interface{})
|
||
BroadcastToAllAgents(messageType string, data interface{})
|
||
}
|
||
|
||
// InitConversationInput 对话初始化需要的输入数据。
|
||
type InitConversationInput struct {
|
||
VisitorID uint
|
||
Website string
|
||
Referrer string
|
||
Browser string
|
||
OS string
|
||
Language string
|
||
IPAddress string
|
||
ChatMode string // 对话模式：human（人工客服）、ai（AI客服）
|
||
AIConfigID *uint // AI 配置 ID（访客选择的模型配置，AI 模式时必需）
|
||
}
|
||
|
||
// InitConversationResult 对话初始化后的返回结果。
|
||
type InitConversationResult struct {
|
||
ConversationID uint
|
||
Status string
|
||
}
|
||
|
||
// UpdateConversationContactInput 更新访客联系信息时需要的参数。
|
||
type UpdateConversationContactInput struct {
|
||
ConversationID uint
|
||
Email *string
|
||
Phone *string
|
||
Notes *string
|
||
}
|
||
|
||
// ConversationSummary 用于会话列表展示的概要信息。
|
||
type ConversationSummary struct {
|
||
ID uint
|
||
ConversationType string // visitor | internal
|
||
VisitorID uint
|
||
AgentID uint
|
||
Status string
|
||
ChatMode string // human | ai
|
||
CreatedAt time.Time
|
||
UpdatedAt time.Time
|
||
LastMessage *LastMessageSummary
|
||
UnreadCount int64
|
||
LastSeenAt *time.Time // 最后活跃时间，用于判断在线状态
|
||
HasParticipated bool // 当前用户是否参与过该会话（是否发送过消息）
|
||
}
|
||
|
||
// LastMessageSummary 会话最后一条消息的摘要信息。
|
||
type LastMessageSummary struct {
|
||
ID uint
|
||
Content string
|
||
SenderIsAgent bool
|
||
MessageType string
|
||
IsRead bool
|
||
ReadAt *time.Time
|
||
CreatedAt time.Time
|
||
}
|
||
|
||
// ConversationDetail 在会话概要基础上附加访客信息。
|
||
type ConversationDetail struct {
|
||
ConversationSummary
|
||
Website string
|
||
Referrer string
|
||
Browser string
|
||
OS string
|
||
Language string
|
||
IPAddress string
|
||
Location string
|
||
Email string
|
||
Phone string
|
||
Notes string
|
||
LastSeen *time.Time
|
||
}
|
||
|
||
// CreateMessageInput 创建消息时需要的参数。
|
||
type CreateMessageInput struct {
|
||
ConversationID uint
|
||
Content string
|
||
SenderID uint
|
||
SenderIsAgent bool
|
||
// 文件相关字段（可选）
|
||
FileURL *string // 文件URL
|
||
FileType *string // 文件类型：image, document
|
||
FileName *string // 原始文件名
|
||
FileSize *int64 // 文件大小（字节）
|
||
MimeType *string // MIME类型
|
||
// 回复数据源开关（仅 AI 模式有效）：不传或 nil 时使用默认（知识库+大模型开，联网关）
|
||
UseKnowledgeBase *bool // 是否使用知识库检索，默认 true
|
||
UseLLM *bool // 无知识库匹配时是否用大模型回复，默认 true
|
||
UseWebSearch *bool // 是否允许联网搜索（需本回合 NeedWebSearch 或策略触发），默认 false
|
||
NeedWebSearch bool // 本回合是否请求联网搜索（如用户点击「联网搜索」），默认 false
|
||
}
|
||
|
||
// CreateAgentInput 创建客服或管理员账号需要的参数。
|
||
type CreateAgentInput struct {
|
||
Username string
|
||
Password string
|
||
Role string
|
||
}
|
||
|
||
// MarkMessagesReadResult 消息标记已读后的返回信息。
|
||
type MarkMessagesReadResult struct {
|
||
ConversationID uint
|
||
MessageIDs []uint
|
||
UnreadCount int64
|
||
ReadAt time.Time
|
||
}
|
||
|
||
// UpdateProfileInput 更新个人资料时需要的参数。
|
||
type UpdateProfileInput struct {
|
||
UserID uint
|
||
Nickname *string
|
||
Email *string
|
||
ReceiveAIConversations *bool // 是否接收 AI 对话（可选）
|
||
}
|
||
|
||
// ProfileResult 个人资料信息。
|
||
type ProfileResult struct {
|
||
ID uint `json:"id"`
|
||
Username string `json:"username"`
|
||
Role string `json:"role"`
|
||
AvatarURL string `json:"avatar_url"`
|
||
Nickname string `json:"nickname"`
|
||
Email string `json:"email"`
|
||
ReceiveAIConversations bool `json:"receive_ai_conversations"` // 是否接收 AI 对话
|
||
}
|
||
|
||
// UserSummary 用户列表摘要信息（不包含密码）。
|
||
type UserSummary struct {
|
||
ID uint `json:"id"`
|
||
Username string `json:"username"`
|
||
Role string `json:"role"`
|
||
Nickname string `json:"nickname"`
|
||
Email string `json:"email"`
|
||
AvatarURL string `json:"avatar_url"`
|
||
ReceiveAIConversations bool `json:"receive_ai_conversations"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
UpdatedAt time.Time `json:"updated_at"`
|
||
}
|
||
|
||
// CreateUserInput 创建用户输入。
|
||
type CreateUserInput struct {
|
||
Username string // 用户名（必需）
|
||
Password string // 密码（必需）
|
||
Role string // 角色："admin" 或 "agent"（必需）
|
||
Nickname *string // 昵称（可选）
|
||
Email *string // 邮箱（可选）
|
||
}
|
||
|
||
// UpdateUserInput 更新用户输入。
|
||
type UpdateUserInput struct {
|
||
UserID uint // 用户ID（必需）
|
||
Role *string // 角色（可选）
|
||
Nickname *string // 昵称（可选）
|
||
Email *string // 邮箱（可选）
|
||
ReceiveAIConversations *bool // 是否接收 AI 对话（可选）
|
||
}
|
||
|
||
// UpdatePasswordInput 更新密码输入。
|
||
type UpdatePasswordInput struct {
|
||
UserID uint // 用户ID（必需）
|
||
OldPassword *string // 旧密码（可选，管理员修改其他用户密码时不需要）
|
||
NewPassword string // 新密码（必需）
|
||
IsAdmin bool // 是否是管理员操作（必需）
|
||
}
|
||
|
||
// FAQSummary FAQ（常见问题）摘要信息。
|
||
type FAQSummary struct {
|
||
ID uint `json:"id"`
|
||
Question string `json:"question"` // 问题
|
||
Answer string `json:"answer"` // 答案
|
||
Keywords string `json:"keywords"` // 关键词（用于搜索）
|
||
CreatedAt time.Time `json:"created_at"` // 创建时间
|
||
UpdatedAt time.Time `json:"updated_at"` // 更新时间
|
||
}
|
||
|
||
// CreateFAQInput 创建 FAQ 输入。
|
||
type CreateFAQInput struct {
|
||
Question string // 问题（必需）
|
||
Answer string // 答案（必需）
|
||
Keywords string // 关键词（可选，用逗号或空格分隔）
|
||
}
|
||
|
||
// UpdateFAQInput 更新 FAQ 输入。
|
||
type UpdateFAQInput struct {
|
||
Question *string // 问题（可选）
|
||
Answer *string // 答案（可选）
|
||
Keywords *string // 关键词（可选）
|
||
}
|
||
|
||
// OnlineAgent 在线客服信息（供访客查看）。
|
||
type OnlineAgent struct {
|
||
ID uint `json:"id"` // 客服ID
|
||
Nickname string `json:"nickname"` // 昵称
|
||
AvatarURL string `json:"avatar_url"` // 头像URL
|
||
}
|
||
|
||
// DocumentSummary 文档摘要信息。
|
||
type DocumentSummary struct {
|
||
ID uint `json:"id"`
|
||
KnowledgeBaseID uint `json:"knowledge_base_id"`
|
||
Title string `json:"title"`
|
||
Content string `json:"content"`
|
||
Summary string `json:"summary"`
|
||
Type string `json:"type"`
|
||
Status string `json:"status"`
|
||
EmbeddingStatus string `json:"embedding_status"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
UpdatedAt time.Time `json:"updated_at"`
|
||
}
|
||
|
||
// CreateDocumentInput 创建文档输入。
|
||
type CreateDocumentInput struct {
|
||
KnowledgeBaseID uint // 知识库 ID（必需）
|
||
Title string // 文档标题（必需）
|
||
Content string // 文档内容（必需）
|
||
Summary string // 文档摘要（可选）
|
||
Type string // 文档类型（可选，默认：document）
|
||
Status string // 文档状态（可选，默认：draft）
|
||
Metadata map[string]interface{} // 元数据（可选）
|
||
}
|
||
|
||
// UpdateDocumentInput 更新文档输入。
|
||
type UpdateDocumentInput struct {
|
||
Title *string // 文档标题（可选）
|
||
Content *string // 文档内容（可选）
|
||
Summary *string // 文档摘要（可选）
|
||
Type *string // 文档类型（可选）
|
||
Status *string // 文档状态（可选）
|
||
Metadata *map[string]interface{} // 元数据（可选）
|
||
}
|
||
|
||
// DocumentListResult 文档列表查询结果。
|
||
type DocumentListResult struct {
|
||
Documents []DocumentSummary `json:"documents"` // 文档列表
|
||
Total int64 `json:"total"` // 总记录数
|
||
Page int `json:"page"` // 当前页码
|
||
PageSize int `json:"page_size"` // 每页大小
|
||
TotalPage int `json:"total_page"` // 总页数
|
||
}
|
||
|
||
// KnowledgeBaseSummary 知识库摘要信息。
|
||
type KnowledgeBaseSummary struct {
|
||
ID uint `json:"id"`
|
||
Name string `json:"name"`
|
||
Description string `json:"description"`
|
||
DocumentCount int64 `json:"document_count"` // 文档数量（统计信息）
|
||
RAGEnabled bool `json:"rag_enabled"` // 是否参与 RAG（对 AI 开放）
|
||
CreatedAt time.Time `json:"created_at"`
|
||
UpdatedAt time.Time `json:"updated_at"`
|
||
}
|
||
|
||
// CreateKnowledgeBaseInput 创建知识库输入。
|
||
type CreateKnowledgeBaseInput struct {
|
||
Name string // 知识库名称（必需）
|
||
Description string // 知识库描述（可选）
|
||
}
|
||
|
||
// UpdateKnowledgeBaseInput 更新知识库输入。
|
||
type UpdateKnowledgeBaseInput struct {
|
||
Name *string // 知识库名称（可选）
|
||
Description *string // 知识库描述（可选）
|
||
RAGEnabled *bool // 是否参与 RAG（可选）
|
||
}
|
||
|
||
// MessageAttachment 当前用户消息的附件（用于多模态：识图等）
|
||
type MessageAttachment struct {
|
||
FileURL string // 文件 URL（创建消息时返回的 file_url）
|
||
FileType string // image / document
|
||
MimeType string // 如 image/jpeg
|
||
}
|
||
|
||
// GenerateAIResponseInput 生成 AI 回复时的选项（数据源开关等）。
|
||
type GenerateAIResponseInput struct {
|
||
UseKnowledgeBase *bool // 是否使用知识库，默认 true
|
||
UseLLM *bool // 无知识库时是否用大模型回复，默认 true
|
||
UseWebSearch *bool // 是否允许联网，默认 false
|
||
NeedWebSearch bool // 本回合是否请求联网（如用户点击按钮），默认 false
|
||
Attachment *MessageAttachment // 当前条消息的附件（如图片），用于多模态识图
|
||
}
|
||
|
||
// GenerateAIResponseResult 生成 AI 回复的结果（内容 + 使用的数据源标记）。
|
||
type GenerateAIResponseResult struct {
|
||
Content string // 合成的一条回复
|
||
SourcesUsed string // 逗号分隔，如 "knowledge_base" / "knowledge_base,llm" / "llm,web"，供前端展示
|
||
// 生图时返回生成图片的 URL，写入 AI 消息的 file_url
|
||
GeneratedFileURL *string
|
||
// GenerationFailed 为 true 表示大模型调用失败，内容为兜底话术（仍返回 err==nil 时由 message 层写入 is_ai_generation_failed）
|
||
GenerationFailed bool
|
||
}
|