Merge pull request #4 from UiPath/fix/initial

infer input/output from start/stop events

Author: cristipufu

pyproject.toml

@@ -1,10 +1,11 @@
 [project]
-name = "uipath-llama"
+name = "uipath-llamaindex"
 version = "0.0.1"
-description = "UiPath Llama SDK"
+description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
+    "llama-index>=0.12.36",
     "uipath>=2.0.56",
 ]
 classifiers = [
@@ -21,11 +22,11 @@ maintainers = [
 ]
 
 [project.entry-points."uipath.middlewares"]
-register = "uipath_llama.middlewares:register_middleware"
+register = "uipath_llamaindex.middlewares:register_middleware"
 
 [project.urls]
 Homepage = "https://uipath.com"
-Repository = "https://github.com/UiPath/uipath-llama-python"
+Repository = "https://github.com/UiPath/uipath-llamaindex-python"
 
 [build-system]
 requires = ["hatchling"]

src/uipath_llama/_cli/_runtime/_context.py

@@ -0,0 +1,13 @@
+from typing import Optional
+
+from llama_index.core.workflow import Workflow
+from uipath._cli._runtime._contracts import UiPathRuntimeContext
+
+from .._utils._config import LlamaIndexConfig
+
+
+class UiPathLlamaIndexRuntimeContext(UiPathRuntimeContext):
+    """Context information passed throughout the runtime execution."""
+
+    config: Optional[LlamaIndexConfig] = None
+    workflow: Optional[Workflow] = None

src/uipath_llama/_cli/_runtime/_runtime.py

@@ -3,8 +3,8 @@
 from uipath._cli._runtime._contracts import UiPathErrorCategory, UiPathRuntimeError
 
 
-class UiPathLlamaRuntimeError(UiPathRuntimeError):
-    """Custom exception for Llama runtime errors with structured error information."""
+class UiPathLlamaIndexRuntimeError(UiPathRuntimeError):
+    """Custom exception for LlamaIndex runtime errors with structured error information."""
 
     def __init__(
         self,
@@ -14,4 +14,4 @@ def __init__(
         category: UiPathErrorCategory = UiPathErrorCategory.UNKNOWN,
         status: Optional[int] = None,
     ):
-        super().__init__(code, title, detail, category, status, prefix="Llama")
+        super().__init__(code, title, detail, category, status, prefix="LlamaIndex")