139 lines
3.2 KiB
Go
139 lines
3.2 KiB
Go
package chat
|
|
|
|
import (
|
|
"time"
|
|
|
|
"git.mlow.ca/mlow/lmcli/pkg/api"
|
|
"git.mlow.ca/mlow/lmcli/pkg/tui/model"
|
|
"git.mlow.ca/mlow/lmcli/pkg/tui/shared"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
func (m *Model) waitForResponseChunk() tea.Cmd {
|
|
return func() tea.Msg {
|
|
return msgChatResponseChunk(<-m.chatReplyChunks)
|
|
}
|
|
}
|
|
|
|
func (m *Model) loadConversationMessages() tea.Cmd {
|
|
return func() tea.Msg {
|
|
messages, err := m.App.LoadConversationMessages()
|
|
if err != nil {
|
|
return shared.AsMsgError(err)
|
|
}
|
|
rootMessages, err := m.App.LoadConversationRootMessages()
|
|
if err != nil {
|
|
return shared.AsMsgError(err)
|
|
}
|
|
return msgConversationMessagesLoaded{
|
|
messages, rootMessages,
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *Model) generateConversationTitle() tea.Cmd {
|
|
return func() tea.Msg {
|
|
title, err := m.App.GenerateConversationTitle(m.App.Messages)
|
|
if err != nil {
|
|
return shared.AsMsgError(err)
|
|
}
|
|
return msgConversationTitleGenerated(title)
|
|
}
|
|
}
|
|
|
|
func (m *Model) updateConversationTitle(conversation *api.Conversation) tea.Cmd {
|
|
return func() tea.Msg {
|
|
err := m.App.UpdateConversationTitle(conversation)
|
|
if err != nil {
|
|
return shared.WrapError(err)
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (m *Model) cloneMessage(message api.Message, selected bool) tea.Cmd {
|
|
return func() tea.Msg {
|
|
msg, err := m.App.CloneMessage(message, selected)
|
|
if err != nil {
|
|
return shared.WrapError(err)
|
|
}
|
|
return msgMessageCloned(msg)
|
|
}
|
|
}
|
|
|
|
func (m *Model) updateMessageContent(message *api.Message) tea.Cmd {
|
|
return func() tea.Msg {
|
|
err := m.App.UpdateMessageContent(message)
|
|
if err != nil {
|
|
return shared.WrapError(err)
|
|
}
|
|
return msgMessageUpdated(message)
|
|
}
|
|
}
|
|
|
|
func (m *Model) cycleSelectedRoot(conv *api.Conversation, dir model.MessageCycleDirection) tea.Cmd {
|
|
if len(m.App.RootMessages) < 2 {
|
|
|
|
return nil
|
|
}
|
|
|
|
return func() tea.Msg {
|
|
nextRoot, err := m.App.CycleSelectedRoot(conv, m.App.RootMessages, dir)
|
|
if err != nil {
|
|
return shared.WrapError(err)
|
|
}
|
|
return msgSelectedRootCycled(nextRoot)
|
|
}
|
|
}
|
|
|
|
func (m *Model) cycleSelectedReply(message *api.Message, dir model.MessageCycleDirection) tea.Cmd {
|
|
if len(message.Replies) < 2 {
|
|
return nil
|
|
}
|
|
|
|
return func() tea.Msg {
|
|
nextReply, err := m.App.CycleSelectedReply(message, dir)
|
|
if err != nil {
|
|
return shared.WrapError(err)
|
|
}
|
|
return msgSelectedReplyCycled(nextReply)
|
|
}
|
|
}
|
|
|
|
func (m *Model) persistConversation() tea.Cmd {
|
|
return func() tea.Msg {
|
|
conversation, messages, err := m.App.PersistConversation(m.App.Conversation, m.App.Messages)
|
|
if err != nil {
|
|
return shared.AsMsgError(err)
|
|
}
|
|
return msgConversationPersisted{conversation.ID == 0, conversation, messages}
|
|
}
|
|
}
|
|
|
|
func (m *Model) executeToolCalls(toolCalls []api.ToolCall) tea.Cmd {
|
|
return func() tea.Msg {
|
|
results, err := m.App.ExecuteToolCalls(toolCalls)
|
|
if err != nil {
|
|
return shared.AsMsgError(err)
|
|
}
|
|
return msgToolResults(results)
|
|
}
|
|
}
|
|
|
|
func (m *Model) promptLLM() tea.Cmd {
|
|
m.state = pendingResponse
|
|
m.replyCursor.Blink = false
|
|
|
|
m.startTime = time.Now()
|
|
m.elapsed = 0
|
|
m.tokenCount = 0
|
|
|
|
return func() tea.Msg {
|
|
resp, err := m.App.PromptLLM(m.App.Messages, m.chatReplyChunks, m.stopSignal)
|
|
if err != nil {
|
|
return msgChatResponseError{ Err: err }
|
|
}
|
|
return msgChatResponse(resp)
|
|
}
|
|
}
|