Commit 7133dc6
feat(mcp): suspend/resume the agent on long-running UiPath jobs (uipath.com/job), self-contained
Implements Option 6 of the long-running-jobs design entirely within uipath-langchain,
with NO change to base uipath-python. A UiPath MCP server that backs a tool with an
Orchestrator job advertises `uipath.com/job` on initialize; the client opts in per call
and suspends the LangGraph agent while the job runs, resuming with the result — all over
MCP `_meta`, so deployed agents gain the behavior on package upgrade with no agent.json
change.
Why self-contained (vs the base-SDK core in uipath-python #1717): base uipath-platform
has zero MCP-protocol code today (only WaitJob*/CreateTask* wait-models and McpService
CRUD). All MCP protocol code and every framework tool (process_tool, escalation_tool)
already live here; an McpJobExecutor is the same kind of glue. This keeps base generic,
removes the cross-package release coupling, and the base already provides the generic
durable primitive we build on (WaitJobRaw + jobs service). See the §3b placement debate
in mcp-longrunning-jobs-design.md.
- `jobs.py` (new): the `uipath.com/job` `_meta` helpers + `UiPathJobHandle`, `JobStart`,
`McpJobExecutor` Protocol, `BlockingJobExecutor` — framework- and MCP-SDK-neutral
(plain dicts; imports only the already-published `Job`/`JobState`/jobs service).
- `mcp_client.py`: read the initialize advertisement (`is_job_aware`/`job_version`),
thread request `_meta` through `call_tool(..., meta=)`.
- `mcp_tool.py`: `_invoke_job_aware` sends START `_meta`, parses the handle, delegates to
the injected `McpJobExecutor` with neutral `start`/`fetch` closures.
- `job_executor.py`: `LangGraphJobExecutor` (default) — START inside `@durable_interrupt`,
interrupt with `WaitJobRaw`, on resume re-derive the handle and FETCH. `interrupt` is
confined here.
No `uipath-platform` floor bump, no #1717 dependency → CI is green standalone. This is an
alternative to #906 (which stacks on #1717); pick one.
Tests: 9 new (advertisement, START/FETCH `_meta`, executor suspend/resume/faulted); full
test_mcp suite green (233 incl. circular-import guard); ruff + mypy clean.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>1 parent ccb95d3 commit 7133dc6
9 files changed
Lines changed: 765 additions & 16 deletions
File tree
- src/uipath_langchain/agent/tools/mcp
- tests/agent/tools/test_mcp
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
3 | 10 | | |
4 | 11 | | |
5 | 12 | | |
| |||
9 | 16 | | |
10 | 17 | | |
11 | 18 | | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
12 | 22 | | |
| 23 | + | |
13 | 24 | | |
14 | 25 | | |
| 26 | + | |
15 | 27 | | |
16 | 28 | | |
17 | 29 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
24 | 24 | | |
25 | 25 | | |
26 | 26 | | |
| 27 | + | |
| 28 | + | |
27 | 29 | | |
28 | 30 | | |
29 | 31 | | |
30 | 32 | | |
31 | 33 | | |
32 | 34 | | |
| 35 | + | |
33 | 36 | | |
34 | 37 | | |
35 | 38 | | |
| |||
502 | 505 | | |
503 | 506 | | |
504 | 507 | | |
| 508 | + | |
| 509 | + | |
| 510 | + | |
| 511 | + | |
| 512 | + | |
| 513 | + | |
| 514 | + | |
| 515 | + | |
| 516 | + | |
| 517 | + | |
| 518 | + | |
| 519 | + | |
| 520 | + | |
| 521 | + | |
| 522 | + | |
| 523 | + | |
| 524 | + | |
| 525 | + | |
| 526 | + | |
| 527 | + | |
| 528 | + | |
| 529 | + | |
| 530 | + | |
| 531 | + | |
| 532 | + | |
| 533 | + | |
| 534 | + | |
| 535 | + | |
| 536 | + | |
| 537 | + | |
| 538 | + | |
| 539 | + | |
| 540 | + | |
| 541 | + | |
| 542 | + | |
| 543 | + | |
| 544 | + | |
| 545 | + | |
| 546 | + | |
| 547 | + | |
| 548 | + | |
| 549 | + | |
| 550 | + | |
| 551 | + | |
| 552 | + | |
| 553 | + | |
| 554 | + | |
| 555 | + | |
| 556 | + | |
| 557 | + | |
| 558 | + | |
| 559 | + | |
505 | 560 | | |
506 | 561 | | |
507 | 562 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
| 69 | + | |
| 70 | + | |
| 71 | + | |
| 72 | + | |
| 73 | + | |
| 74 | + | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
| 90 | + | |
| 91 | + | |
| 92 | + | |
| 93 | + | |
| 94 | + | |
| 95 | + | |
| 96 | + | |
| 97 | + | |
| 98 | + | |
| 99 | + | |
| 100 | + | |
| 101 | + | |
| 102 | + | |
| 103 | + | |
| 104 | + | |
| 105 | + | |
| 106 | + | |
| 107 | + | |
| 108 | + | |
| 109 | + | |
| 110 | + | |
| 111 | + | |
| 112 | + | |
| 113 | + | |
| 114 | + | |
| 115 | + | |
| 116 | + | |
0 commit comments