32 lines
588 B
Go
32 lines
588 B
Go
package main
|
|
|
|
import "github.com/ollama/ollama/api"
|
|
|
|
type WebModel struct {
|
|
Conversation []WebMessage
|
|
}
|
|
|
|
type WebMessage struct {
|
|
Id int
|
|
Replace bool
|
|
New bool
|
|
Content string
|
|
}
|
|
|
|
func ConvertMessage(message api.Message, id int) WebMessage {
|
|
return WebMessage{
|
|
Id: id,
|
|
Content: message.Content,
|
|
}
|
|
}
|
|
|
|
func (wm *WebModel) AddMessage(message api.Message, id int) {
|
|
wm.Conversation = append(wm.Conversation, ConvertMessage(message, id))
|
|
}
|
|
|
|
func (wm *WebModel) AddMessages(messages []api.Message) {
|
|
for id, message := range messages {
|
|
wm.AddMessage(message, id)
|
|
}
|
|
}
|