fix: short term solution for exhaustion

Signed-off-by: AlexsJones <alexsimonjones@gmail.com>
This commit is contained in:
AlexsJones 2023-03-28 14:00:21 +01:00
parent d0e4684e82
commit 5890e3a79c
2 changed files with 6 additions and 1 deletions

View File

@ -104,6 +104,11 @@ var AnalyzeCmd = &cobra.Command{
if explain {
parsedText, err := analyzer.ParseViaAI(ctx, aiClient, analysis.Error, nocache)
if err != nil {
// Check for exhaustion
if strings.Contains(err.Error(), "status code: 429") {
color.Red("Exhausted API quota. Please try again later")
os.Exit(1)
}
color.Red("Error: %v", err)
continue
}

View File

@ -61,7 +61,7 @@ func ParseViaAI(ctx context.Context, aiClient ai.IAI, prompt []string,
response, err := aiClient.GetCompletion(ctx, inputKey)
if err != nil {
color.Red("error getting completion: %v", err)
return "", nil
return "", err
}
if !viper.IsSet(sEnc) {