From bc9afaeb3c176408e2984b4f869363e510fdb7b7 Mon Sep 17 00:00:00 2001 From: Saurabh Misra Date: Sun, 16 Mar 2025 16:04:30 -0700 Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20Speed=20up=20method=20`Asy?= =?UTF-8?q?ncOperationPool.=5Fget=5Foperation`=20by=2010%=20(#1914)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com> --- skyvern/forge/async_operations.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/async_operations.py b/skyvern/forge/async_operations.py index db825887..f440c0ac 100644 --- a/skyvern/forge/async_operations.py +++ b/skyvern/forge/async_operations.py @@ -83,7 +83,11 @@ class AsyncOperationPool: self._add_operation(task_id, operation) def _get_operation(self, task_id: str, agent_phase: AgentPhase) -> AsyncOperation | None: - return self._operations.get(task_id, {}).get(agent_phase, None) + # Direct dictionary access and exception handling to minimize overhead + try: + return self._operations[task_id][agent_phase] + except KeyError: + return None def _remove_operations(self, task_id: str) -> None: if task_id in self._operations: