@@ -35,17 +35,20 @@ def __init__(
3535 delegate : UiPathRuntimeProtocol ,
3636 storage : UiPathResumableStorageProtocol ,
3737 trigger_manager : UiPathResumeTriggerProtocol ,
38+ runtime_id : str
3839 ):
3940 """Initialize the resumable runtime wrapper.
4041
4142 Args:
4243 delegate: The underlying runtime to wrap
4344 storage: Storage for persisting/retrieving resume triggers
4445 trigger_manager: Manager for creating and reading resume triggers
46+ runtime_id: Id used for runtime orchestration
4547 """
4648 self .delegate = delegate
4749 self .storage = storage
4850 self .trigger_manager = trigger_manager
51+ self .runtime_id = runtime_id
4952
5053 async def execute (
5154 self ,
@@ -115,7 +118,7 @@ async def _restore_resume_input(
115118 return input
116119
117120 # Otherwise, fetch from storage
118- trigger = await self .storage .get_latest_trigger ()
121+ trigger = await self .storage .get_latest_trigger (self . runtime_id )
119122 if not trigger :
120123 return None
121124
@@ -141,7 +144,7 @@ async def _handle_suspension(
141144
142145 # Check if trigger already exists in result
143146 if result .trigger :
144- await self .storage .save_trigger (result .trigger )
147+ await self .storage .save_trigger (self . runtime_id , result .trigger )
145148 return result
146149
147150 suspended_result = UiPathRuntimeResult (
@@ -154,7 +157,7 @@ async def _handle_suspension(
154157 result .output
155158 )
156159
157- await self .storage .save_trigger (suspended_result .trigger )
160+ await self .storage .save_trigger (self . runtime_id , suspended_result .trigger )
158161
159162 return suspended_result
160163
0 commit comments