Skip to content

Commit 5264b05

Browse files
committed
fix for job collection status
1 parent f046169 commit 5264b05

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

services/api-server/src/simcore_service_api_server/api/routes/function_job_collections_routes.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@
1616
from servicelib.utils import limited_gather
1717

1818
from ..._service_function_jobs import FunctionJobService
19+
from ..._service_function_jobs_task_client import (
20+
FunctionJobTaskClientService,
21+
)
1922
from ...models.pagination import Page, PaginationParams
2023
from ...models.schemas.errors import ErrorGet
2124
from ...services_rpc.wb_api_server import WbApiRpcClient
@@ -30,6 +33,7 @@
3033
)
3134
from ..dependencies.services import (
3235
get_function_job_service,
36+
get_function_job_task_client_service,
3337
)
3438
from ..dependencies.webserver_rpc import get_wb_api_rpc_client
3539
from ._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

Comments
 (0)