Add extra claims into token creation (#55)

GetStream · Feb 26, 2021 · f21fbe5 · f21fbe5
1 parent fdad850
commit f21fbe5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stream_chat/base/client.py b/stream_chat/base/client.py
@@ -38,8 +38,8 @@ def normalize_sort(self, sort=None):
                         sort_fields.append({"field": k, "direction": v})
         return sort_fields
 
-    def create_token(self, user_id, exp=None):
-        payload = {"user_id": user_id}
+    def create_token(self, user_id, exp=None, **claims):
+        payload = {**claims, "user_id": user_id}
         if exp is not None:
             payload["exp"] = exp
         return jwt.encode(payload, self.api_secret, algorithm="HS256")