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

Better error handling

Browse source
This commit is contained in:
Atridad Lahiji 2024-01-24 22:33:58 -07:00
parent e56724a61e
commit 6783d82bf2
No known key found for this signature in database
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lib/replicate.go
View file

@ -33,10 +33,14 @@ func ReplicateTextGeneration(prompt string) (string, error) {
return "", predictionError return "", predictionError
} }
if prediction == nil {
return "", errors.New("there was an error generating a response based on this prompt... please reach out to @himbothyswaggins to fix this issue")
}
test, ok := prediction.([]interface{}) test, ok := prediction.([]interface{})
if !ok { if !ok {
return "", errors.New("there was an error generating the image based on this prompt... this usually happens when the generated image violates safety requirements") return "", errors.New("there was an error generating a response based on this prompt... please reach out to @himbothyswaggins to fix this issue")
} }
strs := make([]string, len(test)) strs := make([]string, len(test))
@ -76,16 +80,20 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
return nil, predictionError return nil, predictionError
} }
if prediction == nil {
return nil, errors.New("there was an error generating the image based on this prompt... please reach out to @himbothyswaggins to fix this issue")
}
test, ok := prediction.([]interface{}) test, ok := prediction.([]interface{})
if !ok { if !ok {
return nil, errors.New("there was an error generating the image based on this prompt... this usually happens when the generated image violates safety requirements") return nil, errors.New("there was an error generating the image based on this prompt... please reach out to @himbothyswaggins to fix this issue")
} }
imgUrl, ok := test[0].(string) imgUrl, ok := test[0].(string)
if !ok { if !ok {
return nil, errors.New("there was an error generating the image based on this prompt... this usually happens when the generated image violates safety requirements") return nil, errors.New("there was an error generating the image based on this prompt... please reach out to @himbothyswaggins to fix this issue")
} }
imageRes, imageGetErr := http.Get(imgUrl) imageRes, imageGetErr := http.Get(imgUrl)