@@ -117,25 +117,25 @@ async def get_async_jobs(request: web.Request) -> web.Response:
117117async def get_async_job_status (request : web .Request ) -> web .Response :
118118
119119 _req_ctx = AuthenticatedRequestContext .model_validate (request )
120- rabbitmq_rpc_client = get_rabbitmq_rpc_client (request .app )
121120
122121 path_params = parse_request_path_parameters_as (_PathParams , request )
123- async_job_rpc_status = await async_jobs .status (
124- rabbitmq_rpc_client = rabbitmq_rpc_client ,
125- rpc_namespace = STORAGE_RPC_NAMESPACE ,
126- job_id = path_params .task_id ,
127- job_filter = get_job_filter (
128- user_id = _req_ctx .user_id ,
129- product_name = _req_ctx .product_name ,
130- ),
122+ task_manager = get_task_manager (request .app )
123+ task_filter = get_job_filter (
124+ user_id = _req_ctx .user_id ,
125+ product_name = _req_ctx .product_name ,
131126 )
132- _task_id = f"{ async_job_rpc_status .job_id } "
127+ task_status = await task_manager .get_task_status (
128+ task_filter = TaskFilter .model_validate (task_filter .model_dump ()),
129+ task_uuid = path_params .task_id ,
130+ )
131+
132+ _task_id = f"{ task_status .task_uuid } "
133133 return create_data_response (
134134 TaskStatus (
135135 task_progress = TaskProgress (
136- task_id = _task_id , percent = async_job_rpc_status . progress .percent_value
136+ task_id = _task_id , percent = task_status . progress_report .percent_value
137137 ),
138- done = async_job_rpc_status . done ,
138+ done = task_status . is_done ,
139139 started = None ,
140140 ),
141141 status = status .HTTP_200_OK ,
0 commit comments