|
21 | 21 | UiPathBaseRuntime, |
22 | 22 | UiPathErrorCategory, |
23 | 23 | UiPathRuntimeResult, |
24 | | - UiPathTraceContext, |
25 | 24 | ) |
26 | 25 | from uipath.tracing import LlmOpsHttpExporter |
27 | 26 |
|
@@ -180,12 +179,11 @@ async def execute(self) -> Optional[UiPathRuntimeResult]: |
180 | 179 | self._signalr_client = SignalRClient( |
181 | 180 | signalr_url, |
182 | 181 | headers={ |
183 | | - "X-UiPath-Internal-TenantId": self.context.trace_context.tenant_id, |
184 | | - "X-UiPath-Internal-AccountId": self.context.trace_context.org_id, |
| 182 | + "X-UiPath-Internal-TenantId": self.context.trace_context.tenant_id, # type: ignore |
| 183 | + "X-UiPath-Internal-AccountId": self.context.trace_context.org_id, # type: ignore |
185 | 184 | "X-UIPATH-FolderKey": self.context.folder_key, |
186 | 185 | }, |
187 | 186 | ) |
188 | | - |
189 | 187 | self._signalr_client.on("MessageReceived", self._handle_signalr_message) |
190 | 188 | self._signalr_client.on( |
191 | 189 | "SessionClosed", self._handle_signalr_session_closed |
@@ -276,7 +274,7 @@ async def cleanup(self) -> None: |
276 | 274 | if sys.platform == "win32": |
277 | 275 | await asyncio.sleep(0.5) |
278 | 276 |
|
279 | | - async def _handle_signalr_session_closed(self, args: list[Any]) -> None: |
| 277 | + async def _handle_signalr_session_closed(self, args: list[str]) -> None: |
280 | 278 | """ |
281 | 279 | Handle session closed by server. |
282 | 280 | """ |
@@ -306,7 +304,7 @@ async def _handle_signalr_session_closed(self, args: list[Any]) -> None: |
306 | 304 | except Exception as e: |
307 | 305 | logger.error(f"Error terminating session {session_id}: {str(e)}") |
308 | 306 |
|
309 | | - async def _handle_signalr_message(self, args: list[Any]) -> None: |
| 307 | + async def _handle_signalr_message(self, args: list[str]) -> None: |
310 | 308 | """ |
311 | 309 | Handle incoming SignalR messages. |
312 | 310 | """ |
|
0 commit comments