Merge pull request #12 from UiPath/fix/samples

feat: set signalr internal headers

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-mcp"
-version = "0.0.7"
+version = "0.0.8"
 description = "UiPath MCP SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -58,7 +58,10 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
 
             self.cancel_event = asyncio.Event()
 
-            self.signalr_client = SignalRClient(signalr_url)
+            self.signalr_client = SignalRClient(signalr_url, headers={
+                "X-UiPath-Internal-TenantId": self.context.trace_context.tenant_id,
+                "X-UiPath-Internal-AccountId": self.context.trace_context.org_id,
+            })
             self.signalr_client.on("MessageReceived", self.handle_signalr_message)
             self.signalr_client.on("SessionClosed", self.handle_signalr_session_closed)
             self.signalr_client.on_error(self.handle_signalr_error)