diff --git a/packages/uipath-llamaindex/pyproject.toml b/packages/uipath-llamaindex/pyproject.toml index a8a62294..1c6718f5 100644 --- a/packages/uipath-llamaindex/pyproject.toml +++ b/packages/uipath-llamaindex/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "uipath-llamaindex" -version = "0.5.12" +version = "0.5.13" description = "Python SDK that enables developers to build and deploy LlamaIndex agents to the UiPath Cloud Platform" readme = { file = "README.md", content-type = "text/markdown" } requires-python = ">=3.11" diff --git a/packages/uipath-llamaindex/src/uipath_llamaindex/llms/_openai.py b/packages/uipath-llamaindex/src/uipath_llamaindex/llms/_openai.py index c8b62adf..71e841e6 100644 --- a/packages/uipath-llamaindex/src/uipath_llamaindex/llms/_openai.py +++ b/packages/uipath-llamaindex/src/uipath_llamaindex/llms/_openai.py @@ -50,7 +50,9 @@ def __init__( api_version: str = "2024-10-21", **kwargs: Any, ): + token = os.environ.get("UIPATH_ACCESS_TOKEN") default_headers_dict = { + "Authorization": f"Bearer {token}", "X-UiPath-LlmGateway-ApiFlavor": "auto", "X-UiPath-LlmGateway-RequestingProduct": "uipath-python-sdk", "X-UiPath-LlmGateway-RequestingFeature": "llama-index-agent", @@ -74,7 +76,7 @@ def __init__( "model": model_value, "deployment_name": model_value, "azure_endpoint": f"{base_url}/{vendor_endpoint}", - "api_key": os.environ.get("UIPATH_ACCESS_TOKEN"), + "api_key": "uipath-gateway", "api_version": api_version, "is_chat_model": True, "default_headers": default_headers_dict, diff --git a/packages/uipath-llamaindex/uv.lock b/packages/uipath-llamaindex/uv.lock index 11236140..9c79c6b7 100644 --- a/packages/uipath-llamaindex/uv.lock +++ b/packages/uipath-llamaindex/uv.lock @@ -3496,7 +3496,7 @@ wheels = [ [[package]] name = "uipath-llamaindex" -version = "0.5.12" +version = "0.5.13" source = { editable = "." } dependencies = [ { name = "aiosqlite" },