chore: merge

Author: radugheo

pyproject.toml

@@ -1,15 +1,16 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.1.4"
+version = "0.1.5"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
+    "aiosqlite>=0.21.0",
     "llama-index>=0.14.8",
     "llama-index-embeddings-azure-openai>=0.4.1",
     "llama-index-llms-azure-openai>=0.4.2",
     "openinference-instrumentation-llama-index>=4.3.9",
-    "uipath>=2.2.16, <2.3.0",
+    "uipath>=2.2.26, <2.3.0",
 ]
 classifiers = [
     "Intended Audience :: Developers",
@@ -29,6 +30,11 @@ bedrock = [
     "llama-index-llms-bedrock-converse>=0.3.0",
     "boto3>=1.28.0",
 ]
+vertex = [
+    "llama-index-llms-vertex>=0.4.0",
+    "google-cloud-aiplatform>=1.38.0",
+    "httpx>=0.27.0",
+]
 
 [project.entry-points."uipath.middlewares"]
 register = "uipath_llamaindex.middlewares:register_middleware"

src/uipath_llamaindex/llms/__init__.py

@@ -3,7 +3,19 @@
     UiPathOpenAI,
 )
 
+from .vertex import UiPathVertex
+
+from .supported_models import (
+    BedrockModels,
+    GeminiModels,
+    OpenAIModels,
+)
+
 __all__ = [
     "UiPathOpenAI",
+    "UiPathVertex",
     "OpenAIModel",
+    "OpenAIModels",
+    "GeminiModels",
+    "BedrockModels",
 ]