fix(community): Set awaitHandlers to true in upstash ratelimit (#7571)

Co-authored-by: Jacob Lee <jacoblee93@gmail.com>
langchain-ai · Jan 25, 2025 · a71067d · a71067d
1 parent 7326a63
commit a71067d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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;
   }
 
   /**