1- from typing import Callable , Optional , Any , List , Tuple , Union
1+ from typing import Callable , Optional , Any , List , Tuple , Union , Dict
22from functools import wraps
33import collections
44import hashlib
55import inspect
66
7- from functools import wraps
8-
9- from typing import Callable , Optional , Any , List , Tuple , Union , Dict
10-
11- import asyncio
12-
137from .dependencies import (
148 handle_callback_args ,
159 handle_grouped_callback_args ,
@@ -363,7 +357,7 @@ def _initialize_context(args, kwargs, inputs_state_indices, has_output, insert_o
363357
364358def _get_callback_manager (
365359 kwargs : dict , background : dict
366- ) -> Union [ BaseBackgroundCallbackManager , None ] :
360+ ) -> BaseBackgroundCallbackManager :
367361 """Set up the background callback and manage jobs."""
368362 callback_manager = background .get (
369363 "manager" , kwargs .get ("background_callback_manager" , None )
@@ -379,7 +373,7 @@ def _get_callback_manager(
379373 " and store results on redis.\n "
380374 )
381375
382- adapter = backends .request_adapter ()
376+ adapter = backends .backend . request_adapter ()
383377 old_job = adapter .args .getlist ("oldJob" ) if hasattr (adapter .args , "getlist" ) else []
384378
385379 if old_job :
@@ -439,7 +433,7 @@ def _setup_background_callback(
439433
440434def _progress_background_callback (response , callback_manager , background ):
441435 progress_outputs = background .get ("progress" )
442- adapter = backends .request_adapter ()
436+ adapter = backends .backend . request_adapter ()
443437 cache_key = adapter .args .get ("cacheKey" )
444438
445439 if progress_outputs :
@@ -457,7 +451,7 @@ def _update_background_callback(
457451 """Set up the background callback and manage jobs."""
458452 callback_manager = _get_callback_manager (kwargs , background )
459453
460- adapter = backends .request_adapter ()
454+ adapter = backends .backend . request_adapter ()
461455 cache_key = adapter .args .get ("cacheKey" ) if adapter else None
462456 job_id = adapter .args .get ("job" ) if adapter else None
463457
@@ -479,7 +473,7 @@ def _handle_rest_background_callback(
479473 multi ,
480474 has_update = False ,
481475):
482- adapter = backends .request_adapter ()
476+ adapter = backends .backend . request_adapter ()
483477 cache_key = adapter .args .get ("cacheKey" ) if adapter else None
484478 job_id = adapter .args .get ("job" ) if adapter else None
485479 # Must get job_running after get_result since get_results terminates it.
@@ -697,7 +691,7 @@ def add_context(*args, **kwargs):
697691 jsonResponse : Optional [str ] = None
698692 try :
699693 if background is not None :
700- adapter = backends .request_adapter ()
694+ adapter = backends .backend . request_adapter ()
701695 if not (adapter and adapter .args .get ("cacheKey" )):
702696 return _setup_background_callback (
703697 kwargs ,
@@ -769,7 +763,7 @@ async def async_add_context(*args, **kwargs):
769763
770764 try :
771765 if background is not None :
772- adapter = backends .request_adapter ()
766+ adapter = backends .backend . request_adapter ()
773767 if not (adapter and adapter .args .get ("cacheKey" )):
774768 return _setup_background_callback (
775769 kwargs ,
0 commit comments