fix: await context runtime

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.0.13"
+version = "0.0.14"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_llamaindex/_cli/_runtime/_runtime.py

@@ -12,6 +12,7 @@
     InputRequiredEvent,
     JsonPickleSerializer,
 )
+from llama_index.core.workflow.handler import WorkflowHandler
 from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
@@ -69,9 +70,9 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
             start_event_class = self.context.workflow._start_event_class
             ev = start_event_class(**self.context.input_json)
 
-            ctx: Context = self._get_context()
+            ctx: Context = await self._get_context()
 
-            handler = self.context.workflow.run(start_event=ev, ctx=ctx)
+            handler: WorkflowHandler = self.context.workflow.run(start_event=ev, ctx=ctx, **self.context.input_json)
 
             resume_trigger: UiPathResumeTrigger = None