45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
package proto
|
||
|
||
import (
|
||
"github.com/ollama/ollama/api"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type AIQueueMessage struct {
|
||
Type string `json:"type"` //声明不同消息类型
|
||
ID uint `json:"id"` //消息id
|
||
Msg api.GenerateResponse `json:"msg"` //原始消息
|
||
}
|
||
|
||
type WSMessage struct {
|
||
Type string `json:"type"` //接收及发送消息类型
|
||
Msg AIQueueMessage `json:"msg"` //消息内容,只进行转发，不做处理
|
||
SessionID int `json:"session_id"` //应用层会话id
|
||
ToID int `json:"to_id"` //接收者id
|
||
}
|
||
|
||
type WSMessageReq struct {
|
||
Type string `json:"type"` //接收及发送消息类型
|
||
Function string `json:"function"` //功能名称
|
||
Msg string `json:"msg"` //消息内容,只进行转发，不做处理
|
||
SessionID int `json:"session_id"` //应用层会话id
|
||
ToID int `json:"to_id"` //接收者id
|
||
}
|
||
|
||
type ModelParam struct {
|
||
Model string `json:"model"` //模型名称
|
||
Domain string `json:"domain"` //domain参数，spark需要
|
||
Url string `json:"url"` //模型地址
|
||
APPID string `json:"appid"` //应用id
|
||
APISecret string `json:"apiSecret"` //应用密钥
|
||
APIKey string `json:"apiKey"` //应用key
|
||
}
|
||
|
||
type SessionResponse struct {
|
||
gorm.Model
|
||
UserID int `gorm:"column:user_id"` //只能由用户创建
|
||
Name string `gorm:"column:name"`
|
||
Context []int `gorm:"column:context;type:json"` //会话上下文
|
||
MsgCount int `gorm:"column:msg_count"` //消息数量
|
||
}
|