69 lines
1.3 KiB
Go
69 lines
1.3 KiB
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)
|
|
}
|