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 []Message) []llm.ChatMessage {
	length := len(history)

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

		for _, msg := range history {
			result = append(result, llm.ChatMessage{
				Name: msg.Name,
				Text: msg.Text,
			})
		}

		return result
	}

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