From c27fd5f1b20b540c1b37085947b0dce660ff9453 Mon Sep 17 00:00:00 2001 From: CahidArda Date: Thu, 23 Jan 2025 09:29:15 +0300 Subject: [PATCH 1/2] fix: set awaitHandlers to true in upstash ratelimit --- .../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..ca81c48bd364 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 } /** From a953e23927943f6dfceda7d1fe4d8ca340dce74d Mon Sep 17 00:00:00 2001 From: Jacob Lee Date: Fri, 24 Jan 2025 16:16:37 -0800 Subject: [PATCH 2/2] Format --- .../src/callbacks/handlers/upstash_ratelimit.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts b/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts index ca81c48bd364..d09a90cb17a2 100644 --- a/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts +++ b/libs/langchain-community/src/callbacks/handlers/upstash_ratelimit.ts @@ -101,7 +101,7 @@ class UpstashRatelimitHandler extends BaseCallbackHandler { this.llmOutputPromptTokenField = options.llmOutputPromptTokenField ?? "promptTokens"; - this.awaitHandlers = true + this.awaitHandlers = true; } /**