From 5923bff5ab734633ae06b6ab4493838014794d06 Mon Sep 17 00:00:00 2001 From: Charles Frye Date: Fri, 3 May 2024 10:36:41 -0700 Subject: [PATCH] adds rate limit handler from slack SDK (#731) --- 10_integrations/webscraper.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/10_integrations/webscraper.py b/10_integrations/webscraper.py index e85135b08..d817b323e 100644 --- a/10_integrations/webscraper.py +++ b/10_integrations/webscraper.py @@ -39,7 +39,9 @@ async def get_links(url: str) -> set[str]: return set(links) -slack_sdk_image = modal.Image.debian_slim().pip_install("slack-sdk") +slack_sdk_image = modal.Image.debian_slim(python_version="3.10").pip_install( + "slack-sdk==3.27.1" +) @app.function( @@ -48,9 +50,13 @@ async def get_links(url: str) -> set[str]: ) def bot_token_msg(channel, message): import slack_sdk + from slack_sdk.http_retry.builtin_handlers import RateLimitErrorRetryHandler - print(f"Posting {message} to #{channel}") client = slack_sdk.WebClient(token=os.environ["SLACK_BOT_TOKEN"]) + rate_limit_handler = RateLimitErrorRetryHandler(max_retry_count=3) + client.retry_handlers.append(rate_limit_handler) + + print(f"Posting {message} to #{channel}") client.chat_postMessage(channel=channel, text=message)