missing await

mitre · Dec 19, 2024 · 755f8f0 · 755f8f0
1 parent 4bca515
commit 755f8f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/api/v2/managers/operation_api_manager.py b/app/api/v2/managers/operation_api_manager.py
@@ -100,7 +100,7 @@ async def create_potential_link(self, operation_id: str, data: dict, access: Bas
             raise JsonHttpBadRequest(f'Agent {agent.paw} missing specified executor')
         executor = self.build_executor(data=data.pop('executor', {}), agent=agent)
         ability = self.build_ability(data=data.pop('ability', {}), executor=executor)
-        self._call_ability_plugin_hooks(ability, executor)
+        await self._call_ability_plugin_hooks(ability, executor)
         encoded_command = self._encode_string(agent.replace(self._encode_string(data['executor']['command']),
                                               file_svc=self.services['file_svc']))
         link = Link.load(dict(command=encoded_command, plaintext_command=encoded_command, paw=agent.paw, ability=ability, executor=executor,

diff --git a/app/service/planning_svc.py b/app/service/planning_svc.py
@@ -351,7 +351,7 @@ async def _generate_new_links(self, operation, agent, abilities, link_status):
             executor = await agent.get_preferred_executor(ability)
             if not executor:
                 continue
-            self._call_ability_plugin_hooks(ability, executor)
+            await self._call_ability_plugin_hooks(ability, executor)
             if executor.command:
                 link = Link.load(dict(command=self.encode_string(executor.test), paw=agent.paw, score=0,
                                       ability=ability, executor=executor, status=link_status,