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
var options struct {
Arg string `discord:"prompt"`
Prompt string `discord:"prompt"`
}
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{
{
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{
Content: option.NewNullableString(resp.Choices[0].Message.Content),
Content: option.NewNullableString("Prompt: " + options.Prompt + "\n" + "Response: " + respString),
AllowedMentions: &api.AllowedMentions{}, // don't mention anyone
}
}