Method Complete

api/client.go:264–322

(
	ctx context.Context,
	systemPrompt string,
	messages []map[string]any,
	opts CompleteOptions,
)

				if !sleepBackoff(ctx, attempt, cfg) {
					return nil, ctx.Err()
				}
				continue
			}
			break
		}

		bodyBytes, readErr := io.ReadAll(resp.Body)
		_ = resp.Body.Close()
		cancel()
		if readErr != nil {
			lastErr = readErr
			if attempt < cfg.MaxRetries {
				if !sleepBackoff(ctx, attempt, cfg) {
					return nil, ctx.Err()
				}
				continue
			}
			break
		}

		if resp.StatusCode == http.StatusOK {
			var data map[string]any
			if err := json.Unmarshal(bodyBytes, &data); err != nil {
				return nil, fmt.Errorf("invalid JSON in 200 response: %.200s", string(bodyBytes))
			}
			noteProviderSuccess(cooldownKey)
			return data, nil
		}

		statusErr := NewAPIStatusError(c.ProviderName, resp.StatusCode, resp.Status, bodyBytes)
		if resp.StatusCode == http.StatusTooManyRequests {
			noteProviderRateLimit(cooldownKey)
		}
		if isRetryableHTTPError(resp.StatusCode, statusErr.Body, cfg) && attempt < cfg.MaxRetries {
			if !sleepBackoff(ctx, attempt, cfg) {
				return nil, ctx.Err()
			}
			continue
		}

		return nil, statusErr
	}

	return nil, fmt.Errorf("request failed after %d attempts: %w", cfg.MaxRetries+1, lastErr)
}

func isRetryableHTTPError(statusCode int, errorText string, cfg RetryConfig) bool {
	if _, ok := cfg.RetryableStatuses[statusCode]; ok {
		return true
	}

	if _, ok := cfg.ConditionalRetryStatuses[statusCode]; ok {
		for keyword := range cfg.NonRetryableKeywords {
			if strings.Contains(errorText, keyword) {
				return false
			}
		}
		return true