atridad/himbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Replicate

Browse Source
This commit is contained in:
Atridad Lahiji
2024-02-24 01:52:59 -07:00
parent 870f024420
commit f4aab9470f
6 changed files with 155 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
command/ask.go
View File

@ -32,7 +32,7 @@ func Ask(ctx context.Context, data cmdroute.CommandData) *api.InteractionRespons
return lib.ErrorResponse(err)
}
respString, err := lib.OpenAITextGeneration(options.Prompt)
respString, err := lib.ReplicateTextGeneration(options.Prompt)
if err != nil {
fmt.Printf("ChatCompletion error: %v\n", err)

2
command/pic.go
View File

@ -37,7 +37,7 @@ func Pic(ctx context.Context, data cmdroute.CommandData) *api.InteractionRespons
// Concatenate clean username and timestamp to form filename
filename := data.Event.Sender().Username + "_" + timestamp + ".jpg"
imageFile, err := lib.OpenAIImageGeneration(options.Prompt, filename)
imageFile, err := lib.ReplicateImageGeneration(options.Prompt, filename)
if err != nil {
lib.CancelCooldown(data.Event.Member.User.ID.String(), "pic")