Back to OpenAI for cost LOL

command/ask.go

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

command/code.go

@@ -1,64 +0,0 @@
-package command
-
-import (
-	"bytes"
-	"context"
-	"errors"
-	"fmt"
-	"himbot/lib"
-	"time"
-
-	"github.com/diamondburned/arikawa/v3/api"
-	"github.com/diamondburned/arikawa/v3/api/cmdroute"
-	"github.com/diamondburned/arikawa/v3/utils/json/option"
-	"github.com/diamondburned/arikawa/v3/utils/sendpart"
-)
-
-func Code(ctx context.Context, data cmdroute.CommandData) *api.InteractionResponseData {
-	// Cooldown Logic
-	allowed, cooldownString := lib.CooldownHandler(*data.Event, "code", time.Minute)
-
-	if !allowed {
-		return lib.ErrorResponse(errors.New(cooldownString))
-	}
-
-	// Command Logic
-	var options struct {
-		Prompt string `discord:"prompt"`
-	}
-
-	if err := data.Options.Unmarshal(&options); err != nil {
-		lib.CancelCooldown(data.Event.User.ID.String(), "ask")
-		return lib.ErrorResponse(err)
-	}
-
-	respString, err := lib.ReplicateCodeGeneration(options.Prompt)
-
-	if err != nil {
-		fmt.Printf("ChatCompletion error: %v\n", err)
-		lib.CancelCooldown(data.Event.User.ID.String(), "ask")
-		return &api.InteractionResponseData{
-			Content:         option.NewNullableString("ChatCompletion Error!"),
-			AllowedMentions: &api.AllowedMentions{},
-		}
-	}
-
-	if len(respString) > 1800 {
-		textFile := bytes.NewBuffer([]byte(respString))
-
-		file := sendpart.File{
-			Name:   "himbot_response.md",
-			Reader: textFile,
-		}
-
-		return &api.InteractionResponseData{
-			Content:         option.NewNullableString("Prompt: " + options.Prompt + "\n"),
-			AllowedMentions: &api.AllowedMentions{},
-			Files:           []sendpart.File{file},
-		}
-	}
-	return &api.InteractionResponseData{
-		Content:         option.NewNullableString("Prompt: " + options.Prompt + "\n--------------------\n" + respString),
-		AllowedMentions: &api.AllowedMentions{},
-	}
-}

command/pic.go

@@ -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.ReplicateImageGeneration(options.Prompt, filename)
+	imageFile, err := lib.OpenAIImageGeneration(options.Prompt, filename)
 
 	if err != nil {
 		lib.CancelCooldown(data.Event.User.ID.String(), "pic")