This repository was archived by the owner on Aug 2, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
src/ai/backend/manager/api Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -511,7 +511,8 @@ async def _create(request: web.Request, params: Any) -> web.Response:
511511 resp ['sessionId' ] = str (kernel_id ) # changed since API v5
512512
513513 async def kernelpullprogress (reporter : ProgressReporter ) -> None :
514- progress = [0 ,0 ]
514+ progress = [0 , 0 ]
515+
515516 async def _get_status (kernel_id ):
516517 async with root_ctx .db .begin () as conn :
517518 query = (
@@ -523,13 +524,15 @@ async def _get_status(kernel_id):
523524 .where (kernels .c .id == kernel_id )
524525 )
525526 result = await conn .execute (query )
527+
526528 return result .first ()
527529
528530 async def _update_progress (
529531 app : web .Application ,
530532 source : AgentId ,
531533 event : KernelPullProgressEvent
532534 ) -> None :
535+ # update both current and total
533536 progress [0 ] = int (event .current_progress )
534537 progress [1 ] = int (event .total_progress )
535538
@@ -548,7 +551,10 @@ async def _update_progress(
548551 await reporter .update (0 )
549552 await asyncio .sleep (0.5 )
550553
551- task_id = await root_ctx .background_task_manager .start (kernelpullprogress , name = 'kernel_pull_progress' )
554+ task_id = await root_ctx .background_task_manager .start (
555+ kernelpullprogress ,
556+ name = 'kernel-pull-progress'
557+ )
552558 resp ['background_task' ] = str (task_id )
553559 resp ['sessionName' ] = str (params ['session_name' ])
554560 resp ['status' ] = 'PENDING'
You can’t perform that action at this time.
0 commit comments