telegram-ollama-reply-bot/bot/request_context.go

package bot

import (
	"github.com/mymmrac/telego"
	"log/slog"
	"telegram-ollama-reply-bot/llm"
)

func (b *Bot) createLlmRequestContextFromMessage(message *telego.Message) llm.RequestContext {
	rc := llm.RequestContext{
		Empty: true,
	}

	if message == nil {
		slog.Debug("request context creation problem: no message provided. returning empty context.", "request-context", rc)

		return rc
	}

	rc.Empty = false

	user := message.From

	if user != nil {
		rc.User = llm.UserContext{
			Username:  user.Username,
			FirstName: user.FirstName,
			LastName:  user.LastName,
			IsPremium: user.IsPremium,
		}
	}

	// TODO: implement retrieval of chat description
	chat := message.Chat

	history := b.getChatHistory(chat.ID)

	rc.Chat = llm.ChatContext{
		Title: chat.Title,
		// TODO: fill when ChatFullInfo retrieved
		//Description: chat.Description,
		Type:    chat.Type,
		History: historyToLlmMessages(history),
	}

	slog.Debug("request context created", "request-context", rc)

	return rc
}

func historyToLlmMessages(history []MessageData) []llm.ChatMessage {
	length := len(history)

	if length > 0 {
		result := make([]llm.ChatMessage, 0, length)

		for _, msg := range history {
			result = append(result, messageDataToLlmMessage(msg))
		}

		return result
	}

	return make([]llm.ChatMessage, 0)
}

func messageDataToLlmMessage(data MessageData) llm.ChatMessage {
	llmMessage := llm.ChatMessage{
		Name:          data.Name,
		Username:      data.Username,
		Text:          data.Text,
		IsMe:          data.IsMe,
		IsUserRequest: data.IsUserRequest,
	}

	if data.ReplyTo != nil {
		replyMessage := messageDataToLlmMessage(*data.ReplyTo)
		llmMessage.ReplyTo = &replyMessage
	}

	return llmMessage
}