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{ req := openai.ChatCompletionRequest{
Model: openai.GPT4, Model: openai.GPT3Dot5Turbo,
Messages: msg, 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) { m.client.Log = zerolog.New(zerolog.NewConsoleWriter(func(w *zerolog.ConsoleWriter) {
w.TimeFormat = time.Stamp 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) m.cryptoHelper, err = cryptohelper.NewCryptoHelper(client, []byte(m.config.Pickle), m.config.DBPath)
if err != nil { if err != nil {
@ -117,22 +117,32 @@ func (m *Matrix) InviteHandler() (event.Type, mautrix.EventHandler) {
func (m *Matrix) RespondHandler() (event.Type, mautrix.EventHandler) { func (m *Matrix) RespondHandler() (event.Type, mautrix.EventHandler) {
return event.EventMessage, func(source mautrix.EventSource, evt *event.Event) { return event.EventMessage, func(source mautrix.EventSource, evt *event.Event) {
content := evt.Content.AsMessage()
eventID := evt.ID
m.client.Log.Info(). m.client.Log.Info().
Str("body", evt.Content.AsMessage().Body). Str("content", content.Body).
Msg("Received message") Msg("Received message")
if evt.Sender != id.UserID(m.config.UserID) { if evt.Sender != id.UserID(m.config.UserID) {
msgBody := evt.Content.AsMessage().Body resp, err := m.gptClient.Complete(NewConversation(content.Body))
resp, err := m.gptClient.Complete(NewConversation(msgBody))
if err != nil { if err != nil {
fmt.Println("OpenAI API returned with ", err) m.client.Log.Error().Err(err).Msg("OpenAI API returned with ")
return return
} }
// Send the OpenAI response back to the chat
formattedResp := format.RenderMarkdown(resp, true, false) 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")
} }
} }
} }