package proto

import "github.com/ollama/ollama/api"

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"`     //模型名称
	Url       string `json:"url"`       //模型地址
	APPID     string `json:"appid"`     //应用id
	APISecret string `json:"apiSecret"` //应用密钥
	APIKey    string `json:"apiKey"`    //应用key
}