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; } /**