Fixed error handling!

This commit is contained in:
2024-02-23 18:42:51 -07:00
parent e90e4b6a0f
commit 0c236f1da3
4 changed files with 12 additions and 8 deletions

View File

@ -60,8 +60,13 @@ func (m *CooldownManager) IsOnCooldown(userID string, key string) (bool, time.Du
func CancelCooldown(userID string, key string) {
manager := GetInstance()
// Handle non-existent keys gracefully
if _, exists := manager.cooldowns[userID+":"+key]; !exists {
return
}
manager.mu.Lock()
defer manager.mu.Unlock()
delete(manager.cooldowns, userID+":"+key)
}

View File

@ -37,8 +37,7 @@ func OpenAITextGeneration(prompt string) (string, error) {
if err != nil {
fmt.Printf("Ask command error: %v\n", err)
return "", errors.New("there was an error generating the response based on this prompt... please reach out to @himbothyswaggins to fix this issue")
return "", errors.New("https://fly.storage.tigris.dev/atridad/himbot/no.gif")
}
return resp.Choices[0].Message.Content, nil
@ -64,7 +63,7 @@ func OpenAIImageGeneration(prompt string, filename string) (*bytes.Buffer, error
if err != nil {
fmt.Printf("Pic command error: %v\n", err)
return nil, errors.New("there was an error generating the image based on this prompt... please reach out to @himbothyswaggins to fix this issue")
return nil, errors.New("https://fly.storage.tigris.dev/atridad/himbot/hornypolice.gif")
}
imgUrl := imageResponse.Data[0].URL