af52dd35a1
- Updated SaveHistory method across Mongo, Redis, and Weaviate to include a context parameter, improving flexibility in message storage. - Modified the Assistant struct in types.go to add new fields: Type, Avatar, and Flows, enhancing the representation of assistant attributes. - Adjusted the Xun implementation to support the new context handling in SaveHistory, ensuring messages can now include contextual information. - Updated tests to reflect changes in message structure and ensure proper functionality with the new context parameter.
50 lines
1.9 KiB
Go
50 lines
1.9 KiB
Go
package conversation
|
||
|
||
// Setting the conversation config
|
||
type Setting struct {
|
||
Connector string `json:"connector,omitempty"`
|
||
UserField string `json:"user_field,omitempty"` // the user id field name, default is user_id
|
||
Table string `json:"table,omitempty"`
|
||
MaxSize int `json:"max_size,omitempty" yaml:"max_size,omitempty"`
|
||
TTL int `json:"ttl,omitempty" yaml:"ttl,omitempty"`
|
||
}
|
||
|
||
// ChatInfo represents the chat information and its history
|
||
type ChatInfo struct {
|
||
Chat map[string]interface{} `json:"chat"`
|
||
History []map[string]interface{} `json:"history"`
|
||
}
|
||
|
||
// ChatFilter represents the filter parameters for GetChats
|
||
type ChatFilter struct {
|
||
Keywords string `json:"keywords,omitempty"`
|
||
Page int `json:"page,omitempty"` // 页码，从1开始
|
||
PageSize int `json:"pagesize,omitempty"` // 每页数量
|
||
Order string `json:"order,omitempty"` // desc/asc
|
||
}
|
||
|
||
// ChatGroup represents a group of chats by date
|
||
type ChatGroup struct {
|
||
Label string `json:"label"`
|
||
Chats []map[string]interface{} `json:"chats"`
|
||
}
|
||
|
||
// ChatGroupResponse represents paginated chat groups
|
||
type ChatGroupResponse struct {
|
||
Groups []ChatGroup `json:"groups"`
|
||
Page int `json:"page"` // 当前页码
|
||
PageSize int `json:"pagesize"` // 每页数量
|
||
Total int64 `json:"total"` // 总记录数
|
||
LastPage int `json:"last_page"` // 最后一页页码
|
||
}
|
||
|
||
// Conversation the store interface
|
||
type Conversation interface {
|
||
GetChats(sid string, filter ChatFilter) (*ChatGroupResponse, error)
|
||
GetChat(sid string, cid string) (*ChatInfo, error)
|
||
GetHistory(sid string, cid string) ([]map[string]interface{}, error)
|
||
SaveHistory(sid string, messages []map[string]interface{}, cid string, context map[string]interface{}) error
|
||
DeleteChat(sid string, cid string) error
|
||
DeleteAllChats(sid string) error
|
||
UpdateChatTitle(sid string, cid string, title string) error
|
||
}
|