Merge branch 'main' of https://github.com/SlimeNull/OpenChat

Author: SlimeNull

OpenAI-DotNet/Chat/ChatError.cs

@@ -0,0 +1,37 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Text.Json.Serialization;
+using System.Threading.Tasks;
+
+namespace OpenAI.Chat
+{
+    public class ChatError
+    {
+        [JsonConstructor]
+        public ChatError(string message, string type, string param, string code)
+        {
+            Message = message;
+            Type = type;
+            Param = param;
+            Code = code;
+        }
+
+        [JsonInclude]
+        [JsonPropertyName("message")]
+        public string Message { get; set; }
+
+        [JsonInclude]
+        [JsonPropertyName("type")]
+        public string Type { get; set; }
+
+        [JsonInclude]
+        [JsonPropertyName("param")]
+        public string Param { get; set; }
+
+        [JsonInclude]
+        [JsonPropertyName("code")]
+        public string Code { get; set; }
+    }
+}

OpenAI-DotNet/Chat/ChatResponse.cs

@@ -42,6 +42,11 @@ public ChatResponse(string id, string @object, int created, string model, Usage
         [JsonPropertyName("choices")]
         public IReadOnlyList<Choice> Choices { get; private set; }
 
+        [JsonInclude]
+        [JsonPropertyName("error")]
+        public ChatError? Error { get; set; }
+
+
         [JsonIgnore]
         public Choice? FirstChoice => Choices.FirstOrDefault();
     }

OpenAI-DotNet/ResponseExtensions.cs

@@ -1,4 +1,5 @@
-using System;
+using OpenAI.Chat;
+using System;
 using System.Linq;
 using System.Net.Http;
 using System.Net.Http.Headers;
@@ -43,7 +44,16 @@ internal static async Task CheckResponseAsync(this HttpResponseMessage response,
             if (!response.IsSuccessStatusCode)
             {
                 var responseAsString = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
-                throw new HttpRequestException(message: $"{methodName} Failed! HTTP status code: {response.StatusCode} | Response body: {responseAsString}", null, statusCode: response.StatusCode);
+                var responseJson = JsonSerializer.Deserialize<ChatResponse>(responseAsString);
+
+                var e = new HttpRequestException(message: $"{methodName} Failed! HTTP status code: {response.StatusCode} | Response body: {responseAsString}", null, statusCode: response.StatusCode);
+
+                if (responseJson?.Error is { } error)
+                {
+                    e.Data.Add("Error", error);
+                }
+
+                throw e;
             }
         }
 

OpenGptChat/Views/Pages/ChatPage.xaml.cs

@@ -1,9 +1,13 @@
     using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Linq;
 using System.Threading.Tasks;
 using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Input;
 using CommunityToolkit.Mvvm.Input;
+using OpenAI.Chat;
 using OpenGptChat.Models;
 using OpenGptChat.Services;
 using OpenGptChat.Utilities;
@@ -118,6 +122,22 @@ await ChatService.ChatAsync(SessionId, input, content =>
             {
                 _ = NoteService.ShowAndWaitAsync($"{ex.GetType().Name}: {ex.Message}", 3000);
 
+                var chatError = ex.Data.Cast<DictionaryEntry>()
+                    .Where(v => v.Key as string is "Error")
+                    .Select(v => v.Value as ChatError)
+                    .FirstOrDefault();
+
+                if (chatError != null)
+                {
+                    if (chatError.Code == "context_length_exceeded")
+                    {
+                        if (ViewModel.Messages.Count > 0)
+                        {
+                            Console.WriteLine("token reaches the upper limit");
+                        }
+                    }
+                }
+
                 Rollback(requestMessageModel, responseMessageModel, input);
             }