Merge pull request #48 from UiPath/fix/add_agenthub

feat(llm): add agenthub

Author: ionmincu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.0.28"
+version = "0.0.29"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
@@ -9,7 +9,7 @@ dependencies = [
     "llama-index-embeddings-azure-openai>=0.3.8",
     "llama-index-llms-azure-openai>=0.3.2",
     "openinference-instrumentation-llama-index>=4.3.0",
-    "uipath>=2.0.71, <2.1.0",
+    "uipath>=2.0.75, <2.1.0",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",

src/uipath_llamaindex/_cli/_runtime/_context.py

@@ -11,5 +11,5 @@ class UiPathLlamaIndexRuntimeContext(UiPathRuntimeContext):
 
     config: Optional[LlamaIndexConfig] = None
     workflow: Optional[Workflow] = None
-    workflow_context: Optional[Context] = None
+    workflow_context: Optional[Context] = None  # type: ignore[type-arg]
     resumed_trigger: Optional[UiPathResumeTrigger] = None

src/uipath_llamaindex/_cli/_runtime/_runtime.py

@@ -10,9 +10,9 @@
     HumanResponseEvent,
     InputRequiredEvent,
     JsonPickleSerializer,
+    WorkflowTimeoutError,
 )
-from llama_index.core.workflow.errors import WorkflowTimeoutError
-from llama_index.core.workflow.handler import WorkflowHandler
+from llama_index.core.workflow.handler import WorkflowHandler  # type: ignore
 from openinference.instrumentation.llama_index import (
     LlamaIndexInstrumentor,
     get_current_span,

src/uipath_llamaindex/_cli/cli_init.py

@@ -4,13 +4,14 @@
 import uuid
 from typing import Any, Callable, Dict, overload
 
-from llama_index.core.workflow import StopEvent, Workflow
-from llama_index.core.workflow.drawing import StepConfig  # type: ignore
-from llama_index.core.workflow.events import (
+from llama_index.core.workflow import (
     HumanResponseEvent,
     InputRequiredEvent,
+    StopEvent,
+    Workflow,
 )
-from llama_index.core.workflow.utils import (
+from llama_index.core.workflow.drawing import StepConfig  # type: ignore
+from llama_index.core.workflow.utils import (  # type: ignore
     get_steps_from_class,
     get_steps_from_instance,
 )

src/uipath_llamaindex/llms/_openai_llms.py

@@ -3,6 +3,7 @@
 from typing import Any, Union
 
 from llama_index.llms.azure_openai import AzureOpenAI  # type: ignore
+from uipath.utils import EndpointManager
 
 
 class OpenAIModel(Enum):
@@ -42,7 +43,7 @@ def __init__(
         defaults = {
             "model": model_value,
             "deployment_name": model_value,
-            "azure_endpoint": f"{base_url}/llmgateway_/",
+            "azure_endpoint": f"{base_url}/{EndpointManager.get_passthrough_endpoint().format(model=model, api_version=api_version)}",
             "api_key": os.environ.get("UIPATH_ACCESS_TOKEN"),
             "api_version": api_version,
             "is_chat_model": True,