Safety check before generating a prompt

novosadkry · Jul 10, 2024 · d15312c · d15312c
1 parent 6d0179b
commit d15312c
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/NOVAxis/Modules/Chat/ChatModule.cs b/NOVAxis/Modules/Chat/ChatModule.cs
@@ -51,6 +51,12 @@ public async Task CmdAiReply(IUserMessage message)
 
             var reply = await RequestAiReply(message);
 
+            if (reply is null)
+            {
+                await FollowupAsync("Pro tuhle zprávu nelze vytvořit odpověď.", ephemeral: true);
+                return;
+            }
+
             var id = InteractionCache.Store(message);
             var components = AiReplyComponents(id).Build();
 
@@ -89,6 +95,19 @@ await interaction.UpdateAsync(x =>
 
                 var reply = await RequestAiReply(message);
 
+                if (reply is null)
+                {
+                    InteractionCache.Remove(id);
+
+                    await interaction.ModifyOriginalResponseAsync(x =>
+                    {
+                        x.Content = "Pro tuhle zprávu nelze vytvořit odpověď.";
+                        x.Components = AiReplyComponents(id, disabled: true).Build();
+                    });
+
+                    return;
+                }
+
                 await interaction.ModifyOriginalResponseAsync(x =>
                 {
                     x.Content = reply;
@@ -137,6 +156,9 @@ where AllowedContentTypes.Contains(attachment.ContentType)
                 });
             }
 
+            if (promptMessage.Content.Count == 0)
+                return null;
+
             var request = new MessageParameters
             {
                 MaxTokens = 128,