From a71067d96491a2318998d85848cdd5d91493bd74 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cahid=20Arda=20=C3=96z?= <cahidardaooz@hotmail.com>
Date: Sat, 25 Jan 2025 04:17:47 +0300
Subject: [PATCH] fix(community): Set awaitHandlers to true in upstash
 ratelimit (#7571)

Co-authored-by: Jacob Lee <jacoblee93@gmail.com>
---
 .../src/callbacks/handlers/upstash_ratelimit.ts                 | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts b/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts
index 255c474adbd0..d09a90cb17a2 100644
--- a/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts
+++ b/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts
@@ -100,6 +100,8 @@ class UpstashRatelimitHandler extends BaseCallbackHandler {
       options.llmOutputTotalTokenField ?? "totalTokens";
     this.llmOutputPromptTokenField =
       options.llmOutputPromptTokenField ?? "promptTokens";
+
+    this.awaitHandlers = true;
   }
 
   /**