1616from servicelib .utils import limited_gather
1717
1818from ..._service_function_jobs import FunctionJobService
19+ from ..._service_function_jobs_task_client import (
20+ FunctionJobTaskClientService ,
21+ )
1922from ...models .pagination import Page , PaginationParams
2023from ...models .schemas .errors import ErrorGet
2124from ...services_rpc .wb_api_server import WbApiRpcClient
3033)
3134from ..dependencies .services import (
3235 get_function_job_service ,
36+ get_function_job_task_client_service ,
3337)
3438from ..dependencies .webserver_rpc import get_wb_api_rpc_client
3539from ._constants import (
@@ -261,8 +265,8 @@ async def function_job_collection_status(
261265 wb_api_rpc : Annotated [WbApiRpcClient , Depends (get_wb_api_rpc_client )],
262266 user_id : Annotated [UserID , Depends (get_current_user_id )], # Updated type
263267 product_name : Annotated [ProductName , Depends (get_product_name )],
264- function_job_service : Annotated [
265- FunctionJobService , Depends (get_function_job_service )
268+ function_job_task_client_service : Annotated [
269+ FunctionJobTaskClientService , Depends (get_function_job_task_client_service )
266270 ],
267271) -> FunctionJobCollectionStatus :
268272 task_manager = get_task_manager (app )
@@ -275,7 +279,7 @@ async def function_job_collection_status(
275279
276280 job_statuses = await limited_gather (
277281 * [
278- function_job_service .inspect_function_job (
282+ function_job_task_client_service .inspect_function_job (
279283 function_job = await get_function_job (
280284 function_job_id = function_job_id ,
281285 wb_api_rpc = wb_api_rpc ,
@@ -288,7 +292,6 @@ async def function_job_collection_status(
288292 user_id = user_id ,
289293 product_name = product_name ,
290294 ),
291- task_manager = task_manager ,
292295 )
293296 for function_job_id in function_job_collection .job_ids
294297 ]
0 commit comments