diff --git a/pyproject.toml b/pyproject.toml index 3f0f70a..6ac6c3c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "uipath-mcp" -version = "0.0.100" +version = "0.0.101" description = "UiPath MCP SDK" readme = { file = "README.md", content-type = "text/markdown" } requires-python = ">=3.11" diff --git a/src/uipath_mcp/_cli/_runtime/_runtime.py b/src/uipath_mcp/_cli/_runtime/_runtime.py index 104690e..10a7afd 100644 --- a/src/uipath_mcp/_cli/_runtime/_runtime.py +++ b/src/uipath_mcp/_cli/_runtime/_runtime.py @@ -175,17 +175,14 @@ async def execute(self) -> Optional[UiPathRuntimeResult]: root_span.set_attribute("command", str(self._server.command)) root_span.set_attribute("args", json.dumps(self._server.args)) root_span.set_attribute("span_type", "MCP Server") - - def token_factory() -> str: - return os.environ.get("UIPATH_ACCESS_TOKEN", "") - + bearer_token = os.environ.get("UIPATH_ACCESS_TOKEN") self._signalr_client = SignalRClient( signalr_url, - access_token_factory=token_factory, headers={ "X-UiPath-Internal-TenantId": self.context.trace_context.tenant_id, # type: ignore "X-UiPath-Internal-AccountId": self.context.trace_context.org_id, # type: ignore "X-UIPATH-FolderKey": self.context.folder_key, + "Authorization": f"Bearer {bearer_token}", }, ) self._signalr_client.on("MessageReceived", self._handle_signalr_message)