From e335a6cf10679fee4959e2e640b6707a022fc3e3 Mon Sep 17 00:00:00 2001
From: Alina Sireneva <alina@tei.su>
Date: Sat, 2 Dec 2023 21:01:17 +0300
Subject: [PATCH] fix: accept sync function in `editMessageWith`

---
 packages/dispatcher/src/context/callback-query.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/dispatcher/src/context/callback-query.ts b/packages/dispatcher/src/context/callback-query.ts
index 7c4651a7..210681a7 100644
--- a/packages/dispatcher/src/context/callback-query.ts
+++ b/packages/dispatcher/src/context/callback-query.ts
@@ -1,6 +1,7 @@
 import {
     CallbackQuery,
     getMarkedPeerId,
+    MaybeAsync,
     Message,
     MtArgumentError,
     MtMessageNotFoundError,
@@ -72,7 +73,7 @@ export class CallbackQueryContext extends CallbackQuery implements UpdateContext
     /**
      * Shortcut for getting the message and editing it.
      */
-    async editMessageWith(handler: (msg: Message) => Promise<Parameters<CallbackQueryContext['editMessage']>[0]>) {
+    async editMessageWith(handler: (msg: Message) => MaybeAsync<Parameters<CallbackQueryContext['editMessage']>[0]>) {
         const msg = await this.getMessage()
         if (!msg) return