21 lines
465 B
Go
21 lines
465 B
Go
package chat_context
|
|
|
|
import "github.com/sashabaranov/go-openai"
|
|
|
|
type ChatMessage struct {
|
|
//当前记录ID
|
|
ID string `json:"id,omitempty"`
|
|
//上一条记录ID
|
|
PID string `json:"pid,omitempty"`
|
|
//消息内容
|
|
Message openai.ChatCompletionMessage `json:"message"`
|
|
//该消息tokens数
|
|
Tokens int `json:"tokens,omitempty"`
|
|
}
|
|
|
|
type ContextCache interface {
|
|
Get(key string) (*ChatMessage, error)
|
|
Set(key string, value *ChatMessage, ttl int) error
|
|
Close()
|
|
}
|