ewintr
/
matrix-gptzoo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add reply ids

This commit is contained in:
Erik Winter 2023-05-18 09:29:23 +02:00
parent 7453868901
commit 51464046ed
2 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bot/gpt.go
View File

@ -26,7 +26,7 @@ func (g GPT) Complete(conv *Conversation) (string, error) {
})
}
req := openai.ChatCompletionRequest{
Model: openai.GPT4,
Model: openai.GPT3Dot5Turbo,
Messages: msg,
}

26
bot/matrix.go
View File

@ -49,7 +49,7 @@ func (m *Matrix) Init() error {
m.client.Log = zerolog.New(zerolog.NewConsoleWriter(func(w *zerolog.ConsoleWriter) {
w.TimeFormat = time.Stamp
})).With().Timestamp().Logger()
})).With().Timestamp().Logger().Level(zerolog.InfoLevel)
m.cryptoHelper, err = cryptohelper.NewCryptoHelper(client, []byte(m.config.Pickle), m.config.DBPath)
if err != nil {
@ -117,22 +117,32 @@ func (m *Matrix) InviteHandler() (event.Type, mautrix.EventHandler) {
func (m *Matrix) RespondHandler() (event.Type, mautrix.EventHandler) {
return event.EventMessage, func(source mautrix.EventSource, evt *event.Event) {
content := evt.Content.AsMessage()
eventID := evt.ID
m.client.Log.Info().
Str("body", evt.Content.AsMessage().Body).
Str("content", content.Body).
Msg("Received message")
if evt.Sender != id.UserID(m.config.UserID) {
msgBody := evt.Content.AsMessage().Body
resp, err := m.gptClient.Complete(NewConversation(msgBody))
resp, err := m.gptClient.Complete(NewConversation(content.Body))
if err != nil {
fmt.Println("OpenAI API returned with ", err)
m.client.Log.Error().Err(err).Msg("OpenAI API returned with ")
return
}
// Send the OpenAI response back to the chat
formattedResp := format.RenderMarkdown(resp, true, false)
m.client.SendMessageEvent(evt.RoomID, event.EventMessage, &formattedResp)
formattedResp.RelatesTo = &event.RelatesTo{
InReplyTo: &event.InReplyTo{
EventID: eventID,
},
}
if _, err := m.client.SendMessageEvent(evt.RoomID, event.EventMessage, &formattedResp); err != nil {
m.client.Log.Err(err).Msg("failed to send message")
return
}
m.client.Log.Info().Str("message", fmt.Sprintf("%+v", formattedResp.Body)).Msg("Sent message")
}
}
}