Updated auth_callback to be handled based on async/blocking implement…

…ation
ably · Oct 1, 2023 · 0f777c9 · 0f777c9
1 parent 274e956
commit 0f777c9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ably/rest/auth.py b/ably/rest/auth.py
@@ -189,8 +189,13 @@ async def request_token(self, token_params: Optional[dict] = None,
         log.debug("Token Params: %s" % token_params)
         if auth_callback:
             log.debug("using token auth with authCallback")
+            # handle auth_callback in accordance with async/blocking nature of the implementation
             try:
-                token_request = await auth_callback(token_params)
+                res = auth_callback(token_params)
+                if asyncio.iscoroutine(res):
+                    token_request = await res
+                else:
+                    token_request = res
             except Exception as e:
                 raise AblyException("auth_callback raised an exception", 401, 40170, cause=e)
         elif auth_url: