From 6443236915a1a730300ba7422b025e4b2232c283 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Thu, 30 May 2024 12:35:03 -0700 Subject: [PATCH] refac --- pipelines/examples/rate_limit_filter_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipelines/examples/rate_limit_filter_pipeline.py b/pipelines/examples/rate_limit_filter_pipeline.py index f875d23..e25dc20 100644 --- a/pipelines/examples/rate_limit_filter_pipeline.py +++ b/pipelines/examples/rate_limit_filter_pipeline.py @@ -108,7 +108,7 @@ class Pipeline: print(body) print(user) - if user["role"] == "user": + if user.get("role", "admin") == "user": user_id = user["id"] if user and "id" in user else "default_user" if self.rate_limited(user_id): raise Exception("Rate limit exceeded. Please try again later.")