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

Better error handling

Browse Source
This commit is contained in:
Atridad Lahiji
2024-01-24 22:33:58 -07:00
parent e56724a61e
commit 6783d82bf2
1 changed files with 11 additions and 3 deletions
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)