atridad/himbot
1
0
Fork 0
Code Projects Releases Packages 1 Activity Actions

Bot update

Browse source
This commit is contained in:
Atridad Lahiji 2023-12-29 14:12:11 -07:00
parent aa10576330
commit b136365477
No known key found for this signature in database
1 changed files with 19 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
main.go
View file

@ -140,7 +140,7 @@ func (h *handler) cmdAsk(ctx context.Context, data cmdroute.CommandData) *api.In
// Command Logic // Command Logic
var options struct { var options struct {
Arg string `discord:"prompt"` Prompt string `discord:"prompt"`
} }
if err := data.Options.Unmarshal(&options); err != nil { if err := data.Options.Unmarshal(&options); err != nil {
@ -156,7 +156,7 @@ func (h *handler) cmdAsk(ctx context.Context, data cmdroute.CommandData) *api.In
Messages: []openai.ChatCompletionMessage{ Messages: []openai.ChatCompletionMessage{
{ {
Role: openai.ChatMessageRoleUser, Role: openai.ChatMessageRoleUser,
Content: options.Arg, Content: options.Prompt,
}, },
}, },
}, },
@ -170,8 +170,24 @@ func (h *handler) cmdAsk(ctx context.Context, data cmdroute.CommandData) *api.In
} }
} }
respString := resp.Choices[0].Message.Content
if len(respString) > 1800 {
textFile := bytes.NewBuffer([]byte(respString))
file := sendpart.File{
Name: "himbot_response.txt",
Reader: textFile,
}
return &api.InteractionResponseData{
Content: option.NewNullableString("Prompt: " + options.Prompt + "\n" + "Response:\n"),
AllowedMentions: &api.AllowedMentions{}, // don't mention anyone
Files: []sendpart.File{file},
}
}
return &api.InteractionResponseData{ return &api.InteractionResponseData{
Content: option.NewNullableString(resp.Choices[0].Message.Content), Content: option.NewNullableString("Prompt: " + options.Prompt + "\n" + "Response: " + respString),
AllowedMentions: &api.AllowedMentions{}, // don't mention anyone AllowedMentions: &api.AllowedMentions{}, // don't mention anyone
} }
} }