skobkin/telegram-ollama-reply-bot
1
0
Fork 0
Code Issues 2 Pull requests 1 Releases Activity

Chat history and context improvements #35

Merged
skobkin merged 3 commits from fix_quoted_messages_usernames_context into main 2024-11-03 22:28:40 +00:00
Conversation 0 Commits 3 Files changed 5 +11 -7
2 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 541c1d3bbf - Show all commits

6
bot/bot.go
View file

@ -233,9 +233,11 @@ func (b *Bot) summarizeHandler(bot *telego.Bot, update telego.Update) {
slog.Debug("Got completion. Going to send.", "llm-completion", llmReply)
replyMarkdown := b.escapeMarkdownV1Symbols(llmReply)
message := tu.Message(
chatID,
b.escapeMarkdownV1Symbols(llmReply),
replyMarkdown,
).WithParseMode("Markdown")
_, err = bot.SendMessage(b.reply(update.Message, message))
@ -245,6 +247,8 @@ func (b *Bot) summarizeHandler(bot *telego.Bot, update telego.Update) {
b.trySendReplyError(update.Message)
}
b.saveBotReplyToHistory(update.Message, replyMarkdown)
}
func (b *Bot) helpHandler(bot *telego.Bot, update telego.Update) {

12
bot/message_history.go
View file

@ -61,14 +61,14 @@ func (b *Bot) saveChatMessageToHistory(message *telego.Message) {
b.history[chatId].Push(msgData)
}
func (b *Bot) saveBotReplyToHistory(message *telego.Message, text string) {
chatId := message.Chat.ID
func (b *Bot) saveBotReplyToHistory(replyTo *telego.Message, text string) {
chatId := replyTo.Chat.ID
slog.Info(
"history-reply-save",
"chat", chatId,
"to_id", message.From.ID,
"to_name", message.From.FirstName,
"to_id", replyTo.From.ID,
"to_name", replyTo.From.FirstName,
"text", text,
)
@ -84,8 +84,8 @@ func (b *Bot) saveBotReplyToHistory(message *telego.Message, text string) {
IsMe: true,
}
if message.ReplyToMessage != nil {
replyMessage := message.ReplyToMessage
if replyTo.ReplyToMessage != nil {
replyMessage := replyTo.ReplyToMessage
msgData.ReplyTo = &MessageData{
Name: replyMessage.From.FirstName,