Fix AI follow up message handling

teknologi-umum · Jan 1, 2024 · 1bf7dff · 1bf7dff
1 parent d39d65c
commit 1bf7dff
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/BotNet.CommandHandlers/BotUpdate/Message/AICallCommandHandler.cs b/BotNet.CommandHandlers/BotUpdate/Message/AICallCommandHandler.cs
@@ -1,19 +1,21 @@
 using BotNet.Commands;
 using BotNet.Commands.AI.OpenAI;
 using BotNet.Commands.BotUpdate.Message;
+using BotNet.Services.OpenAI;
 
 namespace BotNet.CommandHandlers.BotUpdate.Message {
 	public sealed class AICallCommandHandler(
 		ICommandQueue commandQueue,
-		ITelegramMessageCache telegramMessageCache
+		ITelegramMessageCache telegramMessageCache,
+		IntentDetector intentDetector
 	) : ICommandHandler<AICallCommand> {
 		private readonly ICommandQueue _commandQueue = commandQueue;
 		private readonly ITelegramMessageCache _telegramMessageCache = telegramMessageCache;
+		private readonly IntentDetector _intentDetector = intentDetector;
 
 		public async Task Handle(AICallCommand command, CancellationToken cancellationToken) {
 			switch (command.CallSign) {
-				// OpenAI GPT-4 Chat
-				case "AI" or "Bot" or "GPT":
+				case "AI" or "Bot" or "GPT" when command.ImageFileId is null:
 					await _commandQueue.DispatchAsync(
 						command: OpenAITextPrompt.FromAICallCommand(
 							aiCallCommand: command,
@@ -26,6 +28,8 @@ await _commandQueue.DispatchAsync(
 						)
 					);
 					break;
+				case "AI" or "Bot" or "GPT" when command.ImageFileId is { } imageFileId:
+					break;
 			}
 		}
 	}

diff --git a/BotNet.CommandHandlers/BotUpdate/Message/AIFollowUpMessageHandler.cs b/BotNet.CommandHandlers/BotUpdate/Message/AIFollowUpMessageHandler.cs
@@ -17,12 +17,11 @@ public async Task Handle(AIFollowUpMessage command, CancellationToken cancellati
 					await _commandQueue.DispatchAsync(
 						command: OpenAITextPrompt.FromAIFollowUpMessage(
 							aiFollowUpMessage: command,
-							thread: command.ReplyToMessageId.HasValue
-								? _telegramMessageCache.GetThread(
-									messageId: command.ReplyToMessageId.Value,
-									chatId: command.ChatId
+							thread: command.ReplyToMessage is null
+								? Enumerable.Empty<MessageBase>()
+								: _telegramMessageCache.GetThread(
+									firstMessage: command.ReplyToMessage
 								)
-								: Enumerable.Empty<MessageBase>()
 						)
 					);
 					break;