File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -78,7 +78,7 @@ async def get_task(self) -> Task | None:
7878 'Attempting to get task from store with id: %s' , self .task_id
7979 )
8080 self ._current_task = await self .task_store .get (
81- self .task_id , self ._context
81+ self .task_id , self ._call_context
8282 )
8383 if self ._current_task :
8484 logger .debug ('Task %s retrieved successfully.' , self .task_id )
@@ -172,7 +172,7 @@ async def ensure_task(
172172 logger .debug (
173173 'Attempting to retrieve existing task with id: %s' , self .task_id
174174 )
175- task = await self .task_store .get (self .task_id , self ._context )
175+ task = await self .task_store .get (self .task_id , self ._call_context )
176176
177177 if not task :
178178 logger .info (
@@ -236,7 +236,7 @@ async def _save_task(self, task: Task) -> None:
236236 task: The `Task` object to save.
237237 """
238238 logger .debug ('Saving task with id: %s' , task .id )
239- await self .task_store .save (task , self ._context )
239+ await self .task_store .save (task , self ._call_context )
240240 self ._current_task = task
241241 if not self .task_id :
242242 logger .info ('New task created with id: %s' , task .id )
You can’t perform that action at this time.
0 commit comments