Skip to content

Commit 4587ece

Browse files
Fix variable name typo
1 parent d7f3ffd commit 4587ece

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/a2a/server/tasks/task_manager.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)