Fixed error messages for provider requests (#778)

app/MindWork AI Studio/Provider/OpenAI/ProviderOpenAI.cs

@@ -1,3 +1,4 @@
+using System.Net;
 using System.Net.Http.Headers;
 using System.Runtime.CompilerServices;
 using System.Text;
@@ -5,6 +6,7 @@ using System.Text.Json;
 
 using AIStudio.Chat;
 using AIStudio.Settings;
+using AIStudio.Tools.PluginSystem;
 
 namespace AIStudio.Provider.OpenAI;
 
@@ -15,6 +17,8 @@ public sealed class ProviderOpenAI() : BaseProvider(LLMProviders.OPEN_AI, "https
 {
     private static readonly ILogger<ProviderOpenAI> LOGGER = Program.LOGGER_FACTORY.CreateLogger<ProviderOpenAI>();
     
+    private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(ProviderOpenAI).Namespace, nameof(ProviderOpenAI));
+    
     #region Implementation of IProvider
 
     /// <inheritdoc />
@@ -26,6 +30,28 @@ public sealed class ProviderOpenAI() : BaseProvider(LLMProviders.OPEN_AI, "https
     /// <inheritdoc />
     public override bool HasModelLoadingCapability => true;
 
+    protected override ProviderRequestFailureReason ClassifyProviderRequestFailure(HttpStatusCode statusCode, string responseBody)
+    {
+        if (statusCode is HttpStatusCode.TooManyRequests && HasInsufficientQuotaError(responseBody))
+            return ProviderRequestFailureReason.INSUFFICIENT_QUOTA;
+
+        return base.ClassifyProviderRequestFailure(statusCode, responseBody);
+    }
+
+    protected override ProviderRequestFailureReason ClassifyProviderRequestFailure(string? errorCode, string? errorType, string? errorMessage, string responseBody)
+    {
+        if (IsInsufficientQuota(errorCode) || IsInsufficientQuota(errorType) || HasInsufficientQuotaError(responseBody))
+            return ProviderRequestFailureReason.INSUFFICIENT_QUOTA;
+
+        return base.ClassifyProviderRequestFailure(errorCode, errorType, errorMessage, responseBody);
+    }
+
+    protected override string GetProviderRequestFailureUserMessage(ProviderRequestFailureReason failureReason) => failureReason switch
+    {
+        ProviderRequestFailureReason.INSUFFICIENT_QUOTA => TB("It looks like you do not have any API credits left with OpenAI. Please add credits to your account and try again."),
+        _ => base.GetProviderRequestFailureUserMessage(failureReason),
+    };
+
     /// <inheritdoc />
     public override async IAsyncEnumerable<ContentStreamChunk> StreamChatCompletion(Model chatModel, ChatThread chatThread, SettingsManager settingsManager, [EnumeratorCancellation] CancellationToken token = default)
     {
@@ -289,4 +315,59 @@ public sealed class ProviderOpenAI() : BaseProvider(LLMProviders.OPEN_AI, "https
             token,
             apiKeyProvisional);
     }
+
+    private static bool HasInsufficientQuotaError(string responseBody)
+    {
+        if (string.IsNullOrWhiteSpace(responseBody))
+            return false;
+
+        try
+        {
+            using var document = JsonDocument.Parse(responseBody);
+            return HasInsufficientQuotaError(document.RootElement);
+        }
+        catch (JsonException)
+        {
+            return false;
+        }
+    }
+
+    private static bool HasInsufficientQuotaError(JsonElement element)
+    {
+        switch (element.ValueKind)
+        {
+            case JsonValueKind.Object:
+                if (HasJsonStringValue(element, "type", "insufficient_quota") ||
+                    HasJsonStringValue(element, "code", "insufficient_quota"))
+                    return true;
+
+                foreach (var property in element.EnumerateObject())
+                    if (HasInsufficientQuotaError(property.Value))
+                        return true;
+
+                return false;
+
+            case JsonValueKind.Array:
+                foreach (var item in element.EnumerateArray())
+                    if (HasInsufficientQuotaError(item))
+                        return true;
+
+                return false;
+
+            default:
+                return false;
+        }
+    }
+
+    private static bool IsInsufficientQuota(string? value)
+    {
+        return value is not null && value.Equals("insufficient_quota", StringComparison.OrdinalIgnoreCase);
+    }
+
+    private static bool HasJsonStringValue(JsonElement element, string propertyName, string expectedValue)
+    {
+        return element.TryGetProperty(propertyName, out var propertyElement) &&
+               propertyElement.ValueKind is JsonValueKind.String &&
+               string.Equals(propertyElement.GetString(), expectedValue, StringComparison.OrdinalIgnoreCase);
+    }
 }