@herbertgao/pi-extensions 2026.9.11 → 2026.9.13

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (192) hide show
  1. package/README.md +4 -4
  2. package/THIRD_PARTY_NOTICES.md +1 -1
  3. package/node_modules/@herbertgao/pi-cc-extensions/package.json +1 -1
  4. package/node_modules/@herbertgao/pi-subagents/CHANGELOG.md +8 -0
  5. package/node_modules/@herbertgao/pi-subagents/package.json +1 -1
  6. package/node_modules/@herbertgao/pi-subagents/src/agent-manager.ts +5 -4
  7. package/node_modules/@herbertgao/pi-subagents/src/mention-clone.ts +39 -16
  8. package/node_modules/@juicesharp/rpiv-ask-user-question/package.json +2 -2
  9. package/node_modules/@narumitw/pi-btw/README.md +21 -4
  10. package/node_modules/@narumitw/pi-btw/dist/index.ts +1668 -958
  11. package/node_modules/@narumitw/pi-btw/dist/index.ts.map +4 -4
  12. package/node_modules/@narumitw/pi-btw/docs/workflows.md +10 -2
  13. package/node_modules/@narumitw/pi-btw/package.json +1 -1
  14. package/node_modules/@narumitw/pi-btw/src/btw.ts +17 -79
  15. package/node_modules/@narumitw/pi-btw/src/conversation-context.ts +74 -0
  16. package/node_modules/@narumitw/pi-btw/src/fullscreen-ui.ts +196 -7
  17. package/node_modules/@narumitw/pi-btw/src/main-thread-updates.ts +40 -0
  18. package/node_modules/@narumitw/pi-btw/src/menu.ts +34 -2
  19. package/node_modules/@narumitw/pi-btw/src/settings.ts +49 -0
  20. package/node_modules/@narumitw/pi-btw/src/transcript-pager.ts +9 -1
  21. package/node_modules/@narumitw/pi-btw/src/workspace-layout.ts +559 -0
  22. package/node_modules/pi-mcp-adapter/CHANGELOG.md +62 -0
  23. package/node_modules/pi-mcp-adapter/README.md +75 -7
  24. package/node_modules/pi-mcp-adapter/bearer-command-resolver.ts +193 -0
  25. package/node_modules/pi-mcp-adapter/cli.js +56 -5
  26. package/node_modules/pi-mcp-adapter/commands.ts +96 -5
  27. package/node_modules/pi-mcp-adapter/config.ts +73 -6
  28. package/node_modules/pi-mcp-adapter/direct-tool-surface.ts +22 -28
  29. package/node_modules/pi-mcp-adapter/direct-tools.ts +13 -5
  30. package/node_modules/pi-mcp-adapter/dist/abort.d.ts +2 -0
  31. package/node_modules/pi-mcp-adapter/dist/abort.js +36 -0
  32. package/node_modules/pi-mcp-adapter/dist/abort.js.map +1 -0
  33. package/node_modules/pi-mcp-adapter/dist/bearer-command-resolver.d.ts +6 -0
  34. package/node_modules/pi-mcp-adapter/dist/bearer-command-resolver.js +194 -0
  35. package/node_modules/pi-mcp-adapter/dist/bearer-command-resolver.js.map +1 -0
  36. package/node_modules/pi-mcp-adapter/dist/config.d.ts +5 -0
  37. package/node_modules/pi-mcp-adapter/dist/config.js +75 -6
  38. package/node_modules/pi-mcp-adapter/dist/config.js.map +1 -1
  39. package/node_modules/pi-mcp-adapter/dist/consent-manager.d.ts +18 -0
  40. package/node_modules/pi-mcp-adapter/dist/consent-manager.js +88 -0
  41. package/node_modules/pi-mcp-adapter/dist/consent-manager.js.map +1 -0
  42. package/node_modules/pi-mcp-adapter/dist/elicitation-handler.d.ts +17 -0
  43. package/node_modules/pi-mcp-adapter/dist/elicitation-handler.js +316 -0
  44. package/node_modules/pi-mcp-adapter/dist/elicitation-handler.js.map +1 -0
  45. package/node_modules/pi-mcp-adapter/dist/errors.d.ts +131 -0
  46. package/node_modules/pi-mcp-adapter/dist/errors.js +278 -0
  47. package/node_modules/pi-mcp-adapter/dist/errors.js.map +1 -0
  48. package/node_modules/pi-mcp-adapter/dist/http-ca.d.ts +8 -0
  49. package/node_modules/pi-mcp-adapter/dist/http-ca.js +88 -0
  50. package/node_modules/pi-mcp-adapter/dist/http-ca.js.map +1 -0
  51. package/node_modules/pi-mcp-adapter/dist/jev-client.d.ts +15 -0
  52. package/node_modules/pi-mcp-adapter/dist/jev-client.js +340 -0
  53. package/node_modules/pi-mcp-adapter/dist/jev-client.js.map +1 -0
  54. package/node_modules/pi-mcp-adapter/dist/jev-contracts.d.ts +77 -0
  55. package/node_modules/pi-mcp-adapter/dist/jev-contracts.js +2 -0
  56. package/node_modules/pi-mcp-adapter/dist/jev-contracts.js.map +1 -0
  57. package/node_modules/pi-mcp-adapter/dist/jev-key-store.d.ts +22 -0
  58. package/node_modules/pi-mcp-adapter/dist/jev-key-store.js +85 -0
  59. package/node_modules/pi-mcp-adapter/dist/jev-key-store.js.map +1 -0
  60. package/node_modules/pi-mcp-adapter/dist/json-schema-validator.d.ts +2 -0
  61. package/node_modules/pi-mcp-adapter/dist/json-schema-validator.js +55 -0
  62. package/node_modules/pi-mcp-adapter/dist/json-schema-validator.js.map +1 -0
  63. package/node_modules/pi-mcp-adapter/dist/lifecycle.d.ts +56 -0
  64. package/node_modules/pi-mcp-adapter/dist/lifecycle.js +410 -0
  65. package/node_modules/pi-mcp-adapter/dist/lifecycle.js.map +1 -0
  66. package/node_modules/pi-mcp-adapter/dist/logger.d.ts +51 -0
  67. package/node_modules/pi-mcp-adapter/dist/logger.js +131 -0
  68. package/node_modules/pi-mcp-adapter/dist/logger.js.map +1 -0
  69. package/node_modules/pi-mcp-adapter/dist/mcp-auth-fetch.d.ts +26 -0
  70. package/node_modules/pi-mcp-adapter/dist/mcp-auth-fetch.js +103 -0
  71. package/node_modules/pi-mcp-adapter/dist/mcp-auth-fetch.js.map +1 -0
  72. package/node_modules/pi-mcp-adapter/dist/mcp-auth-flow.d.ts +118 -0
  73. package/node_modules/pi-mcp-adapter/dist/mcp-auth-flow.js +1098 -0
  74. package/node_modules/pi-mcp-adapter/dist/mcp-auth-flow.js.map +1 -0
  75. package/node_modules/pi-mcp-adapter/dist/mcp-auth.d.ts +168 -0
  76. package/node_modules/pi-mcp-adapter/dist/mcp-auth.js +1117 -0
  77. package/node_modules/pi-mcp-adapter/dist/mcp-auth.js.map +1 -0
  78. package/node_modules/pi-mcp-adapter/dist/mcp-bearer-store.js +7 -140
  79. package/node_modules/pi-mcp-adapter/dist/mcp-bearer-store.js.map +1 -1
  80. package/node_modules/pi-mcp-adapter/dist/mcp-callback-server.d.ts +53 -0
  81. package/node_modules/pi-mcp-adapter/dist/mcp-callback-server.js +441 -0
  82. package/node_modules/pi-mcp-adapter/dist/mcp-callback-server.js.map +1 -0
  83. package/node_modules/pi-mcp-adapter/dist/mcp-oauth-provider.d.ts +148 -0
  84. package/node_modules/pi-mcp-adapter/dist/mcp-oauth-provider.js +689 -0
  85. package/node_modules/pi-mcp-adapter/dist/mcp-oauth-provider.js.map +1 -0
  86. package/node_modules/pi-mcp-adapter/dist/mcp-probe.d.ts +6 -0
  87. package/node_modules/pi-mcp-adapter/dist/mcp-probe.js +166 -0
  88. package/node_modules/pi-mcp-adapter/dist/mcp-probe.js.map +1 -0
  89. package/node_modules/pi-mcp-adapter/dist/mcp-tasks.d.ts +102 -0
  90. package/node_modules/pi-mcp-adapter/dist/mcp-tasks.js +369 -0
  91. package/node_modules/pi-mcp-adapter/dist/mcp-tasks.js.map +1 -0
  92. package/node_modules/pi-mcp-adapter/dist/mcp-trace.d.ts +95 -0
  93. package/node_modules/pi-mcp-adapter/dist/mcp-trace.js +242 -0
  94. package/node_modules/pi-mcp-adapter/dist/mcp-trace.js.map +1 -0
  95. package/node_modules/pi-mcp-adapter/dist/metadata-cache.js +2 -11
  96. package/node_modules/pi-mcp-adapter/dist/metadata-cache.js.map +1 -1
  97. package/node_modules/pi-mcp-adapter/dist/npx-resolver.d.ts +6 -0
  98. package/node_modules/pi-mcp-adapter/dist/npx-resolver.js +505 -0
  99. package/node_modules/pi-mcp-adapter/dist/npx-resolver.js.map +1 -0
  100. package/node_modules/pi-mcp-adapter/dist/request-headers-command.d.ts +10 -0
  101. package/node_modules/pi-mcp-adapter/dist/request-headers-command.js +312 -0
  102. package/node_modules/pi-mcp-adapter/dist/request-headers-command.js.map +1 -0
  103. package/node_modules/pi-mcp-adapter/dist/runtime-owner.d.ts +13 -0
  104. package/node_modules/pi-mcp-adapter/dist/runtime-owner.js +97 -0
  105. package/node_modules/pi-mcp-adapter/dist/runtime-owner.js.map +1 -0
  106. package/node_modules/pi-mcp-adapter/dist/sampling-handler.d.ts +17 -0
  107. package/node_modules/pi-mcp-adapter/dist/sampling-handler.js +203 -0
  108. package/node_modules/pi-mcp-adapter/dist/sampling-handler.js.map +1 -0
  109. package/node_modules/pi-mcp-adapter/dist/secure-keyring.d.ts +11 -0
  110. package/node_modules/pi-mcp-adapter/dist/secure-keyring.js +172 -0
  111. package/node_modules/pi-mcp-adapter/dist/secure-keyring.js.map +1 -0
  112. package/node_modules/pi-mcp-adapter/dist/server-manager.d.ts +157 -0
  113. package/node_modules/pi-mcp-adapter/dist/server-manager.js +1743 -0
  114. package/node_modules/pi-mcp-adapter/dist/server-manager.js.map +1 -0
  115. package/node_modules/pi-mcp-adapter/dist/session-approvals.d.ts +34 -0
  116. package/node_modules/pi-mcp-adapter/dist/session-approvals.js +140 -0
  117. package/node_modules/pi-mcp-adapter/dist/session-approvals.js.map +1 -0
  118. package/node_modules/pi-mcp-adapter/dist/session-recovery.d.ts +26 -0
  119. package/node_modules/pi-mcp-adapter/dist/session-recovery.js +142 -0
  120. package/node_modules/pi-mcp-adapter/dist/session-recovery.js.map +1 -0
  121. package/node_modules/pi-mcp-adapter/dist/state.d.ts +73 -0
  122. package/node_modules/pi-mcp-adapter/dist/state.js +2 -0
  123. package/node_modules/pi-mcp-adapter/dist/state.js.map +1 -0
  124. package/node_modules/pi-mcp-adapter/dist/types.d.ts +36 -0
  125. package/node_modules/pi-mcp-adapter/dist/types.js +18 -0
  126. package/node_modules/pi-mcp-adapter/dist/types.js.map +1 -1
  127. package/node_modules/pi-mcp-adapter/dist/ui-resource-handler.d.ts +17 -0
  128. package/node_modules/pi-mcp-adapter/dist/ui-resource-handler.js +219 -0
  129. package/node_modules/pi-mcp-adapter/dist/ui-resource-handler.js.map +1 -0
  130. package/node_modules/pi-mcp-adapter/dist/unix-socket-transport.d.ts +15 -0
  131. package/node_modules/pi-mcp-adapter/dist/unix-socket-transport.js +85 -0
  132. package/node_modules/pi-mcp-adapter/dist/unix-socket-transport.js.map +1 -0
  133. package/node_modules/pi-mcp-adapter/dist/utils.d.ts +1 -0
  134. package/node_modules/pi-mcp-adapter/dist/utils.js +13 -0
  135. package/node_modules/pi-mcp-adapter/dist/utils.js.map +1 -1
  136. package/node_modules/pi-mcp-adapter/elicitation-handler.ts +29 -19
  137. package/node_modules/pi-mcp-adapter/examples/jev-accessibility-loop.mjs +85 -0
  138. package/node_modules/pi-mcp-adapter/examples/jev-semantic-filter.mjs +34 -0
  139. package/node_modules/pi-mcp-adapter/index.ts +129 -6
  140. package/node_modules/pi-mcp-adapter/init.ts +6 -19
  141. package/node_modules/pi-mcp-adapter/jev-client.ts +273 -0
  142. package/node_modules/pi-mcp-adapter/jev-contracts.ts +53 -0
  143. package/node_modules/pi-mcp-adapter/jev-key-store.ts +79 -0
  144. package/node_modules/pi-mcp-adapter/lifecycle.ts +14 -3
  145. package/node_modules/pi-mcp-adapter/mcp-bearer-store.ts +7 -142
  146. package/node_modules/pi-mcp-adapter/mcp-code.ts +119 -11
  147. package/node_modules/pi-mcp-adapter/mcp-output-guard.ts +4 -0
  148. package/node_modules/pi-mcp-adapter/mcp-references.ts +5 -3
  149. package/node_modules/pi-mcp-adapter/mcp-script-worker.mjs +5 -0
  150. package/node_modules/pi-mcp-adapter/mcp-tasks.ts +467 -0
  151. package/node_modules/pi-mcp-adapter/metadata-cache.ts +2 -13
  152. package/node_modules/pi-mcp-adapter/namespace-tools.ts +1 -1
  153. package/node_modules/pi-mcp-adapter/package.json +17 -9
  154. package/node_modules/pi-mcp-adapter/proxy-modes.ts +319 -166
  155. package/node_modules/pi-mcp-adapter/request-headers-command.ts +6 -3
  156. package/node_modules/pi-mcp-adapter/search-ranking.ts +38 -7
  157. package/node_modules/pi-mcp-adapter/secure-keyring.ts +170 -0
  158. package/node_modules/pi-mcp-adapter/semantic-search.ts +186 -0
  159. package/node_modules/pi-mcp-adapter/server-manager.ts +293 -50
  160. package/node_modules/pi-mcp-adapter/session-approvals.ts +14 -0
  161. package/node_modules/pi-mcp-adapter/skills/mcp-scripting/SKILL.md +2 -0
  162. package/node_modules/pi-mcp-adapter/state.ts +3 -1
  163. package/node_modules/pi-mcp-adapter/tool-approval.ts +26 -4
  164. package/node_modules/pi-mcp-adapter/tool-metadata.ts +6 -14
  165. package/node_modules/pi-mcp-adapter/tool-registrar.ts +8 -6
  166. package/node_modules/pi-mcp-adapter/tool-result-renderer.ts +4 -1
  167. package/node_modules/pi-mcp-adapter/types.ts +53 -0
  168. package/node_modules/pi-mcp-adapter/ui-server.ts +17 -9
  169. package/node_modules/pi-mcp-adapter/utils.ts +16 -0
  170. package/node_modules/pi-multi-account/CHANGELOG.md +23 -0
  171. package/node_modules/pi-multi-account/README.md +38 -11
  172. package/node_modules/pi-multi-account/index.ts +371 -96
  173. package/node_modules/pi-multi-account/package.json +5 -4
  174. package/node_modules/pi-multi-account/provider-payload-stream.ts +36 -28
  175. package/node_modules/pi-multi-account/usage.ts +31 -2
  176. package/node_modules/pi-typesafe/README.md +3 -1
  177. package/node_modules/pi-typesafe/dist/auth.d.ts +10 -3
  178. package/node_modules/pi-typesafe/dist/auth.js +13 -7
  179. package/node_modules/pi-typesafe/dist/backends.d.ts +31 -0
  180. package/node_modules/pi-typesafe/dist/backends.js +33 -0
  181. package/node_modules/pi-typesafe/dist/client.d.ts +3 -9
  182. package/node_modules/pi-typesafe/dist/client.js +63 -39
  183. package/node_modules/pi-typesafe/dist/credentials.d.ts +11 -5
  184. package/node_modules/pi-typesafe/dist/credentials.js +15 -8
  185. package/node_modules/pi-typesafe/dist/extension.js +4 -1
  186. package/node_modules/pi-typesafe/dist/index.d.ts +1 -1
  187. package/node_modules/pi-typesafe/dist/index.js +1 -1
  188. package/node_modules/pi-typesafe/dist/login.d.ts +13 -7
  189. package/node_modules/pi-typesafe/dist/login.js +17 -8
  190. package/node_modules/pi-typesafe/dist/schema.js +19 -13
  191. package/node_modules/pi-typesafe/package.json +1 -1
  192. package/package.json +11 -10
@@ -0,0 +1,369 @@
1
+ import { ProtocolError } from "@modelcontextprotocol/client";
2
+ import { createApplicationInputHandler, createTaskSessionEndpointId, createTaskSessionFromClient, DispatchError, JsonRpcResponseError, resultFromTaskOutcome, TaskCancelledError, TaskFailedError, } from "@modelcontextprotocol/ext-tasks/client";
3
+ import { logger } from "./logger.js";
4
+ export const TASKS_EXTENSION_ID = "io.modelcontextprotocol/tasks";
5
+ const RAW_REQUEST_ID_PREFIX = "pi-mcp-tasks-";
6
+ const DEFAULT_RAW_REQUEST_TIMEOUT_MS = 60_000;
7
+ const LATE_TASK_HANDLE_GRACE_MS = 60_000;
8
+ function abortError(signal) {
9
+ return signal.reason instanceof Error
10
+ ? signal.reason
11
+ : new DOMException("This operation was aborted", "AbortError");
12
+ }
13
+ function isJsonRecord(value) {
14
+ return value !== null && typeof value === "object" && !Array.isArray(value);
15
+ }
16
+ /**
17
+ * Raw JSON-RPC request channel chained onto a connected transport. Raw
18
+ * requests use string ids with a reserved prefix, which cannot collide with
19
+ * the SDK's numeric ids; their responses are consumed before the SDK client
20
+ * sees them. All other traffic passes through unchanged.
21
+ *
22
+ * Attach only after Client.connect has installed its transport handlers:
23
+ * attach() reads the current onmessage/onclose (through any wrapper
24
+ * accessors) and chains in front of them.
25
+ */
26
+ export class RawRequestChannel {
27
+ transport;
28
+ pending = new Map();
29
+ requestCounter = 0;
30
+ defaultTimeoutMs;
31
+ attached = false;
32
+ constructor(transport, defaultTimeoutMs) {
33
+ this.transport = transport;
34
+ this.defaultTimeoutMs = defaultTimeoutMs ?? DEFAULT_RAW_REQUEST_TIMEOUT_MS;
35
+ }
36
+ attach() {
37
+ if (this.attached)
38
+ return;
39
+ this.attached = true;
40
+ const previousOnMessage = this.transport.onmessage;
41
+ this.transport.onmessage = (message, extra) => {
42
+ if (this.consumeRawResponse(message))
43
+ return;
44
+ previousOnMessage?.(message, extra);
45
+ };
46
+ const previousOnClose = this.transport.onclose;
47
+ this.transport.onclose = () => {
48
+ this.rejectAll("MCP connection closed");
49
+ previousOnClose?.();
50
+ };
51
+ }
52
+ /** Sends a raw JSON-RPC request and resolves with its correlated response. */
53
+ rawDispatch = (request, options) => {
54
+ if (!isJsonRecord(request) || typeof request.method !== "string") {
55
+ return Promise.reject(new Error("Raw MCP request must be a JSON object with a string method"));
56
+ }
57
+ const params = request.params;
58
+ if (params !== undefined && !isJsonRecord(params)) {
59
+ return Promise.reject(new Error("Raw MCP request params must be a JSON object"));
60
+ }
61
+ const signal = options?.signal;
62
+ if (signal?.aborted)
63
+ return Promise.reject(abortError(signal));
64
+ const id = `${RAW_REQUEST_ID_PREFIX}${++this.requestCounter}`;
65
+ const message = {
66
+ jsonrpc: "2.0",
67
+ id,
68
+ method: request.method,
69
+ ...(params === undefined ? {} : { params }),
70
+ };
71
+ const timeoutMs = options?.context?.requestTimeoutMs ?? this.defaultTimeoutMs;
72
+ return new Promise((resolve, reject) => {
73
+ const entry = {
74
+ id,
75
+ resolve,
76
+ reject,
77
+ state: "pending",
78
+ ...(signal === undefined ? {} : { signal }),
79
+ ...(message.method === "tools/call" ? { observationController: new AbortController() } : {}),
80
+ };
81
+ entry.requestTimer = setTimeout(() => {
82
+ this.terminateLocalWait(entry, new Error(`MCP task request "${message.method}" timed out after ${timeoutMs}ms`));
83
+ }, timeoutMs);
84
+ this.pending.set(id, entry);
85
+ if (signal) {
86
+ entry.onAbort = () => {
87
+ this.terminateLocalWait(entry, abortError(signal));
88
+ };
89
+ signal.addEventListener("abort", entry.onAbort, { once: true });
90
+ }
91
+ const fail = (error) => {
92
+ if (this.pending.get(id) !== entry)
93
+ return;
94
+ const wasPending = entry.state === "pending";
95
+ this.finishRequest(entry, true);
96
+ if (wasPending)
97
+ reject(error instanceof Error ? error : new Error(String(error)));
98
+ };
99
+ // ext-tasks supplies the SEP-2663 Mcp-Name routing header (and any
100
+ // host headers) via context.headers. A task-producing tools/call owns
101
+ // its transport observation independently from the caller's wait.
102
+ try {
103
+ Promise.resolve(this.transport.send(message, {
104
+ ...(options?.context?.headers === undefined ? {} : { headers: options.context.headers }),
105
+ ...(entry.observationController !== undefined
106
+ ? { requestSignal: entry.observationController.signal }
107
+ : signal === undefined ? {} : { requestSignal: signal }),
108
+ })).catch(fail);
109
+ }
110
+ catch (error) {
111
+ fail(error);
112
+ }
113
+ });
114
+ };
115
+ consumeRawResponse(message) {
116
+ if (!("id" in message) || typeof message.id !== "string")
117
+ return false;
118
+ if (!message.id.startsWith(RAW_REQUEST_ID_PREFIX))
119
+ return false;
120
+ // A server request may choose any string ID, including our client-side
121
+ // prefix. Method-bearing frames still belong to the SDK request handler.
122
+ if ("method" in message)
123
+ return false;
124
+ const entry = this.pending.get(message.id);
125
+ if (!entry)
126
+ return true; // ours, but already completed or its bounded observation expired
127
+ const wasObserving = entry.state === "observing";
128
+ this.finishRequest(entry, true);
129
+ if (wasObserving) {
130
+ this.cancelLateTask(message);
131
+ return true;
132
+ }
133
+ if ("error" in message) {
134
+ const { code, message: errorMessage, data } = message.error;
135
+ entry.resolve({ kind: "error", error: { code, message: errorMessage, ...(data === undefined ? {} : { data }) } });
136
+ }
137
+ else if ("result" in message) {
138
+ entry.resolve({ kind: "result", result: message.result });
139
+ }
140
+ else {
141
+ entry.reject(new Error("MCP task response frame carried neither result nor error"));
142
+ }
143
+ return true;
144
+ }
145
+ terminateLocalWait(entry, error) {
146
+ if (this.pending.get(entry.id) !== entry || entry.state !== "pending")
147
+ return;
148
+ if (entry.observationController === undefined) {
149
+ this.finishRequest(entry, false);
150
+ }
151
+ else {
152
+ entry.state = "observing";
153
+ this.clearLocalWait(entry);
154
+ entry.observationTimer = setTimeout(() => {
155
+ this.finishRequest(entry, true);
156
+ }, LATE_TASK_HANDLE_GRACE_MS);
157
+ }
158
+ entry.reject(error);
159
+ }
160
+ clearLocalWait(entry) {
161
+ if (entry.requestTimer !== undefined)
162
+ clearTimeout(entry.requestTimer);
163
+ delete entry.requestTimer;
164
+ if (entry.signal !== undefined && entry.onAbort !== undefined) {
165
+ entry.signal.removeEventListener("abort", entry.onAbort);
166
+ }
167
+ delete entry.onAbort;
168
+ }
169
+ finishRequest(entry, abortObservation) {
170
+ if (this.pending.get(entry.id) !== entry)
171
+ return;
172
+ this.clearLocalWait(entry);
173
+ if (entry.observationTimer !== undefined)
174
+ clearTimeout(entry.observationTimer);
175
+ delete entry.observationTimer;
176
+ this.pending.delete(entry.id);
177
+ if (abortObservation)
178
+ entry.observationController?.abort();
179
+ }
180
+ cancelLateTask(message) {
181
+ if (!("result" in message) || !isJsonRecord(message.result))
182
+ return;
183
+ if (message.result.resultType !== "task" || typeof message.result.taskId !== "string")
184
+ return;
185
+ const taskId = message.result.taskId;
186
+ void this.rawDispatch({ method: "tasks/cancel", params: { taskId } }, { context: { headers: { "Mcp-Name": taskId } } }).catch(() => { });
187
+ }
188
+ rejectAll(reason) {
189
+ const entries = [...this.pending.values()];
190
+ for (const entry of entries) {
191
+ const wasPending = entry.state === "pending";
192
+ this.finishRequest(entry, true);
193
+ if (wasPending)
194
+ entry.reject(new Error(reason));
195
+ }
196
+ }
197
+ }
198
+ /**
199
+ * Whether a connected 2026-07-28 server advertises the tasks extension.
200
+ * ext-tasks requires the exact empty-object declaration; anything else
201
+ * downgrades the session, so mirror its check here.
202
+ *
203
+ * Legacy (2025-11-25) task servers are intentionally out of scope: driving
204
+ * their opt-in `task` parameter requires per-tool taskSupport declarations,
205
+ * which this integration does not supply.
206
+ */
207
+ export function serverAdvertisesTasks(client) {
208
+ if (client.getProtocolEra?.() !== "modern")
209
+ return false;
210
+ const capabilities = client.getServerCapabilities();
211
+ const extension = capabilities?.extensions?.[TASKS_EXTENSION_ID];
212
+ return isJsonRecord(extension) && Object.keys(extension).length === 0;
213
+ }
214
+ /** Missing-handler errors carry the SDK's capability shape so call sites can classify them. */
215
+ function capabilityNotSupportedError(serverName, method, inputKey) {
216
+ const error = new Error(`MCP server ${serverName} requested ${method} during a task, but this session has no handler for it`);
217
+ Object.assign(error, {
218
+ code: "CAPABILITY_NOT_SUPPORTED",
219
+ data: { key: inputKey ?? "unknown", method },
220
+ });
221
+ return error;
222
+ }
223
+ /**
224
+ * Attaches an ext-tasks requester session to a connected client when the
225
+ * server advertises task support. Returns undefined otherwise, leaving the
226
+ * plain callTool path untouched.
227
+ */
228
+ export async function attachTaskSession(options) {
229
+ const { client, serverName, definition } = options;
230
+ if (!serverAdvertisesTasks(client))
231
+ return undefined;
232
+ const protocolVersion = client.getNegotiatedProtocolVersion?.();
233
+ if (protocolVersion === undefined)
234
+ return undefined;
235
+ const endpointId = await createTaskSessionEndpointId("pi-mcp-adapter", {
236
+ server: serverName,
237
+ ...(definition.command !== undefined
238
+ ? { transport: { type: "stdio", command: definition.command, args: definition.args ?? [] } }
239
+ : definition.url !== undefined
240
+ ? { transport: { type: "http", url: definition.url } }
241
+ : { transport: { type: "unix-socket", socket: definition.socket ?? null } }),
242
+ });
243
+ const onInputRequest = createApplicationInputHandler({
244
+ elicitation: async (request, context) => {
245
+ if (options.onElicitation === undefined) {
246
+ throw capabilityNotSupportedError(serverName, "elicitation/create", context.inputId);
247
+ }
248
+ const result = await options.onElicitation({ method: "elicitation/create", params: request.params }, context.signal);
249
+ return {
250
+ action: result.action,
251
+ ...(result.content === undefined
252
+ ? {}
253
+ : { content: result.content }),
254
+ };
255
+ },
256
+ sampling: async (request, context) => {
257
+ if (options.onSampling === undefined) {
258
+ throw capabilityNotSupportedError(serverName, "sampling/createMessage", context.inputId);
259
+ }
260
+ const result = await options.onSampling({ method: "sampling/createMessage", params: request.params }, context.signal);
261
+ return {
262
+ model: result.model,
263
+ role: result.role,
264
+ content: result.content,
265
+ };
266
+ },
267
+ roots: async () => ({ roots: [] }),
268
+ });
269
+ const channel = new RawRequestChannel(options.transport, options.requestTimeoutMs);
270
+ channel.attach();
271
+ const session = createTaskSessionFromClient(client, {
272
+ endpointId,
273
+ rawDispatch: channel.rawDispatch,
274
+ v2RequestFraming: {
275
+ protocolVersion,
276
+ clientInfo: options.clientInfo,
277
+ clientCapabilities: options.clientCapabilities,
278
+ },
279
+ // Tool declarations only add per-tool taskSupport hints; the adapter's
280
+ // cached tools/list is not threaded through here, and a miss is harmless.
281
+ tools: { currentTool: () => undefined },
282
+ onInputRequest,
283
+ onError: (error) => logger.warn(`ext-tasks background error for ${serverName}: ${String(error)}`),
284
+ });
285
+ // A downgraded session (server capability mismatch) would silently route
286
+ // plain tool calls through the generic request path, skipping callTool's
287
+ // output-schema validation. Refuse it and keep the plain path instead.
288
+ if (!session.capabilities.execution) {
289
+ await session.close().catch(() => { });
290
+ channel.rejectAll("Task session downgraded");
291
+ return undefined;
292
+ }
293
+ return { session, channel };
294
+ }
295
+ /**
296
+ * Preserves error-shape parity with Client.callTool: JSON-RPC failures —
297
+ * immediate (JsonRpcResponseError) or terminal task errors (TaskFailedError
298
+ * with a code) — become the typed ProtocolError subclass for their code, so
299
+ * call-site handling (session recovery, URL elicitation) keeps working.
300
+ * Local failures rethrow their original cause instead of being relabeled.
301
+ */
302
+ function toCallToolError(error, signal) {
303
+ if (error instanceof TaskCancelledError && signal?.aborted === true) {
304
+ return abortError(signal);
305
+ }
306
+ if (error instanceof JsonRpcResponseError) {
307
+ return ProtocolError.fromError(error.code, error.message, error.data);
308
+ }
309
+ if (error instanceof TaskFailedError && error.code !== undefined) {
310
+ return ProtocolError.fromError(error.code, error.message, error.data);
311
+ }
312
+ if (error instanceof DispatchError) {
313
+ return error.cause ?? error;
314
+ }
315
+ return error;
316
+ }
317
+ /**
318
+ * Calls a tool through a task session and blocks until settlement, preserving
319
+ * the plain callTool contract: an immediate result returns as-is, a
320
+ * task-backed execution is polled to completion, a failed task throws the
321
+ * underlying JSON-RPC error, and aborting cancels the remote task.
322
+ */
323
+ export async function callToolViaTaskSession(session, options) {
324
+ const { signal } = options;
325
+ let execution;
326
+ try {
327
+ execution = await session.callTool(options.name, options.args, {
328
+ ...(options.meta === undefined ? {} : { metadata: options.meta }),
329
+ ...(signal === undefined ? {} : { signal }),
330
+ ...(options.requestTimeoutMs === undefined ? {} : { requestTimeoutMs: options.requestTimeoutMs }),
331
+ });
332
+ }
333
+ catch (error) {
334
+ throw toCallToolError(error, signal);
335
+ }
336
+ let onAbort;
337
+ if (signal !== undefined && execution.kind === "task") {
338
+ // settle({ signal }) only stops waiting locally; propagate the abort as a
339
+ // cooperative remote cancellation so the server can stop the work. The
340
+ // signal may have aborted while callTool was resolving, so fire directly
341
+ // for an abort the listener would miss.
342
+ onAbort = () => {
343
+ void execution.cancel().catch(() => { });
344
+ };
345
+ if (signal.aborted)
346
+ onAbort();
347
+ else
348
+ signal.addEventListener("abort", onAbort, { once: true });
349
+ }
350
+ try {
351
+ const settlement = await execution.settle(signal === undefined ? {} : { signal });
352
+ const result = resultFromTaskOutcome(settlement.outcome);
353
+ if (!isJsonRecord(result)) {
354
+ throw new Error(`Task-backed tool ${options.name} settled with a non-object result`);
355
+ }
356
+ // Match Client.callTool's lifted shape, which strips the wire-only discriminator.
357
+ const { resultType: _resultType, ...lifted } = result;
358
+ return lifted;
359
+ }
360
+ catch (error) {
361
+ throw toCallToolError(error, signal);
362
+ }
363
+ finally {
364
+ if (signal !== undefined && onAbort !== undefined)
365
+ signal.removeEventListener("abort", onAbort);
366
+ await execution.close().catch(() => { });
367
+ }
368
+ }
369
+ //# sourceMappingURL=mcp-tasks.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mcp-tasks.js","sourceRoot":"","sources":["../mcp-tasks.ts"],"names":[],"mappings":"AAiCA,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EACL,6BAA6B,EAC7B,2BAA2B,EAC3B,2BAA2B,EAC3B,aAAa,EACb,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,EAClB,eAAe,GAIhB,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,CAAC,MAAM,kBAAkB,GAAG,+BAA+B,CAAC;AAElE,MAAM,qBAAqB,GAAG,eAAe,CAAC;AAE9C,MAAM,8BAA8B,GAAG,MAAM,CAAC;AAC9C,MAAM,yBAAyB,GAAG,MAAM,CAAC;AAmBzC,SAAS,UAAU,CAAC,MAAmB;IACrC,OAAO,MAAM,CAAC,MAAM,YAAY,KAAK;QACnC,CAAC,CAAC,MAAM,CAAC,MAAM;QACf,CAAC,CAAC,IAAI,YAAY,CAAC,4BAA4B,EAAE,YAAY,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,OAAO,iBAAiB;IACX,SAAS,CAAY;IACrB,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;IACxD,cAAc,GAAG,CAAC,CAAC;IACV,gBAAgB,CAAS;IAClC,QAAQ,GAAG,KAAK,CAAC;IAEzB,YAAY,SAAoB,EAAE,gBAAyB;QACzD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,IAAI,8BAA8B,CAAC;IAC7E,CAAC;IAED,MAAM;QACJ,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;QACnD,IAAI,CAAC,SAAS,CAAC,SAAS,GAAG,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YAC5C,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;gBAAE,OAAO;YAC7C,iBAAiB,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACtC,CAAC,CAAC;QACF,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;QAC/C,IAAI,CAAC,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE;YAC5B,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YACxC,eAAe,EAAE,EAAE,CAAC;QACtB,CAAC,CAAC;IACJ,CAAC;IAED,8EAA8E;IAC9E,WAAW,GAAsB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE;QACpD,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACjE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC,CAAC;QACjG,CAAC;QACD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,MAAM,KAAK,SAAS,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC,CAAC;QACnF,CAAC;QACD,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAC/B,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;QAE/D,MAAM,EAAE,GAAG,GAAG,qBAAqB,GAAG,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;QAC9D,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,KAAK;YACd,EAAE;YACF,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;SACf,CAAC;QAC/B,MAAM,SAAS,GAAG,OAAO,EAAE,OAAO,EAAE,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAE9E,OAAO,IAAI,OAAO,CAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3D,MAAM,KAAK,GAAsB;gBAC/B,EAAE;gBACF,OAAO;gBACP,MAAM;gBACN,KAAK,EAAE,SAAS;gBAChB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;gBAC3C,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE,qBAAqB,EAAE,IAAI,eAAe,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC7F,CAAC;YACF,KAAK,CAAC,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;gBACnC,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAI,KAAK,CAAC,qBAAqB,OAAO,CAAC,MAAM,qBAAqB,SAAS,IAAI,CAAC,CAAC,CAAC;YACnH,CAAC,EAAE,SAAS,CAAC,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YAC5B,IAAI,MAAM,EAAE,CAAC;gBACX,KAAK,CAAC,OAAO,GAAG,GAAG,EAAE;oBACnB,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;gBACrD,CAAC,CAAC;gBACF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,KAAc,EAAE,EAAE;gBAC9B,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,KAAK;oBAAE,OAAO;gBAC3C,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC;gBAC7C,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAChC,IAAI,UAAU;oBAAE,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACpF,CAAC,CAAC;YACF,mEAAmE;YACnE,sEAAsE;YACtE,kEAAkE;YAClE,IAAI,CAAC;gBACH,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE;oBAC3C,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBACxF,GAAG,CAAC,KAAK,CAAC,qBAAqB,KAAK,SAAS;wBAC3C,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,qBAAqB,CAAC,MAAM,EAAE;wBACvD,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;iBAC3D,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,KAAK,CAAC,CAAC;YACd,CAAC;QACH,CAAC,CAAkC,CAAC;IACtC,CAAC,CAAC;IAEM,kBAAkB,CAAC,OAAuB;QAChD,IAAI,CAAC,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,OAAO,OAAO,CAAC,EAAE,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QACvE,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,qBAAqB,CAAC;YAAE,OAAO,KAAK,CAAC;QAChE,uEAAuE;QACvE,yEAAyE;QACzE,IAAI,QAAQ,IAAI,OAAO;YAAE,OAAO,KAAK,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC,CAAC,iEAAiE;QAC1F,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,KAAK,WAAW,CAAC;QACjD,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAChC,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAC7B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,OAAO,IAAI,OAAO,EAAE,CAAC;YACvB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC;YAC5D,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACpH,CAAC;aAAM,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;YAC/B,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5D,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC,CAAC;QACtF,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,kBAAkB,CAAC,KAAwB,EAAE,KAAY;QAC/D,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;YAAE,OAAO;QAC9E,IAAI,KAAK,CAAC,qBAAqB,KAAK,SAAS,EAAE,CAAC;YAC9C,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,KAAK,GAAG,WAAW,CAAC;YAC1B,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC3B,KAAK,CAAC,gBAAgB,GAAG,UAAU,CAAC,GAAG,EAAE;gBACvC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAClC,CAAC,EAAE,yBAAyB,CAAC,CAAC;QAChC,CAAC;QACD,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IAEO,cAAc,CAAC,KAAwB;QAC7C,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;YAAE,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACvE,OAAO,KAAK,CAAC,YAAY,CAAC;QAC1B,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC9D,KAAK,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC3D,CAAC;QACD,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IAEO,aAAa,CAAC,KAAwB,EAAE,gBAAyB;QACvE,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,KAAK;YAAE,OAAO;QACjD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC3B,IAAI,KAAK,CAAC,gBAAgB,KAAK,SAAS;YAAE,YAAY,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC/E,OAAO,KAAK,CAAC,gBAAgB,CAAC;QAC9B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9B,IAAI,gBAAgB;YAAE,KAAK,CAAC,qBAAqB,EAAE,KAAK,EAAE,CAAC;IAC7D,CAAC;IAEO,cAAc,CAAC,OAAuB;QAC5C,IAAI,CAAC,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO;QACpE,IAAI,OAAO,CAAC,MAAM,CAAC,UAAU,KAAK,MAAM,IAAI,OAAO,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,QAAQ;YAAE,OAAO;QAC9F,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;QACrC,KAAK,IAAI,CAAC,WAAW,CACnB,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,EAAE,EAC9C,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE,CACjD,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACpB,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC;YAC7C,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAChC,IAAI,UAAU;gBAAE,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;CACF;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAAc;IAClD,IAAI,MAAM,CAAC,cAAc,EAAE,EAAE,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACzD,MAAM,YAAY,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;IACpD,MAAM,SAAS,GAAI,YAAY,EAAE,UAAkD,EAAE,CAAC,kBAAkB,CAAC,CAAC;IAC1G,OAAO,YAAY,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC;AACxE,CAAC;AAED,+FAA+F;AAC/F,SAAS,2BAA2B,CAAC,UAAkB,EAAE,MAAc,EAAE,QAA4B;IACnG,MAAM,KAAK,GAAG,IAAI,KAAK,CACrB,cAAc,UAAU,cAAc,MAAM,wDAAwD,CACrG,CAAC;IACF,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE;QACnB,IAAI,EAAE,0BAA0B;QAChC,IAAI,EAAE,EAAE,GAAG,EAAE,QAAQ,IAAI,SAAS,EAAE,MAAM,EAAE;KAC7C,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAmBD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAiC;IAEjC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IACnD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC;QAAE,OAAO,SAAS,CAAC;IACrD,MAAM,eAAe,GAAG,MAAM,CAAC,4BAA4B,EAAE,EAAE,CAAC;IAChE,IAAI,eAAe,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAEpD,MAAM,UAAU,GAAG,MAAM,2BAA2B,CAAC,gBAAgB,EAAE;QACrE,MAAM,EAAE,UAAU;QAClB,GAAG,CAAC,UAAU,CAAC,OAAO,KAAK,SAAS;YAClC,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,EAAE,EAAE,EAAE;YAC5F,CAAC,CAAC,UAAU,CAAC,GAAG,KAAK,SAAS;gBAC5B,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,GAAG,EAAE,EAAE;gBACtD,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,IAAI,EAAE,EAAE,CAAC;KACjF,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,6BAA6B,CAAC;QACnD,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;YACtC,IAAI,OAAO,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;gBACxC,MAAM,2BAA2B,CAAC,UAAU,EAAE,oBAAoB,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YACvF,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,CACxC,EAAE,MAAM,EAAE,oBAAoB,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAA8B,EACpF,OAAO,CAAC,MAAM,CACf,CAAC;YACF,OAAO;gBACL,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS;oBAC9B,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAkE,EAAE,CAAC;aAC5F,CAAC;QACJ,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;YACnC,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBACrC,MAAM,2BAA2B,CAAC,UAAU,EAAE,wBAAwB,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YAC3F,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CACrC,EAAE,MAAM,EAAE,wBAAwB,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAqC,EAC/F,OAAO,CAAC,MAAM,CACf,CAAC;YACF,OAAO;gBACL,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,OAAO,EAAE,MAAM,CAAC,OAAgB;aACjC,CAAC;QACJ,CAAC;QACD,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;KACnC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACnF,OAAO,CAAC,MAAM,EAAE,CAAC;IACjB,MAAM,OAAO,GAAG,2BAA2B,CAAC,MAAM,EAAE;QAClD,UAAU;QACV,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,gBAAgB,EAAE;YAChB,eAAe;YACf,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,kBAAkB,EAAE,OAAO,CAAC,kBAA2B;SACxD;QACD,uEAAuE;QACvE,0EAA0E;QAC1E,KAAK,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,SAAS,EAAE;QACvC,cAAc;QACd,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,kCAAkC,UAAU,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;KAClG,CAAC,CAAC;IACH,yEAAyE;IACzE,yEAAyE;IACzE,uEAAuE;IACvE,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;QACpC,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtC,OAAO,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QAC7C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AAC9B,CAAC;AAUD;;;;;;GAMG;AACH,SAAS,eAAe,CAAC,KAAc,EAAE,MAA+B;IACtE,IAAI,KAAK,YAAY,kBAAkB,IAAI,MAAM,EAAE,OAAO,KAAK,IAAI,EAAE,CAAC;QACpE,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IACD,IAAI,KAAK,YAAY,oBAAoB,EAAE,CAAC;QAC1C,OAAO,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACxE,CAAC;IACD,IAAI,KAAK,YAAY,eAAe,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACjE,OAAO,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACxE,CAAC;IACD,IAAI,KAAK,YAAY,aAAa,EAAE,CAAC;QACnC,OAAO,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC;IAC9B,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,OAA2B,EAC3B,OAA4B;IAE5B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAC3B,IAAI,SAAS,CAAC;IACd,IAAI,CAAC;QACH,SAAS,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAa,EAAE;YACtE,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,IAAa,EAAE,CAAC;YAC1E,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YAC3C,GAAG,CAAC,OAAO,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,EAAE,CAAC;SAClG,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IACD,IAAI,OAAiC,CAAC;IACtC,IAAI,MAAM,KAAK,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACtD,0EAA0E;QAC1E,uEAAuE;QACvE,yEAAyE;QACzE,wCAAwC;QACxC,OAAO,GAAG,GAAG,EAAE;YACb,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC;QACF,IAAI,MAAM,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;;YACzB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACjE,CAAC;IACD,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAClF,MAAM,MAAM,GAAG,qBAAqB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,oBAAoB,OAAO,CAAC,IAAI,mCAAmC,CAAC,CAAC;QACvF,CAAC;QACD,kFAAkF;QAClF,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC;QACtD,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;YAAS,CAAC;QACT,IAAI,MAAM,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAChG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC"}
@@ -0,0 +1,95 @@
1
+ import type { Transport } from "@modelcontextprotocol/client";
2
+ import type { JSONRPCMessage } from "@modelcontextprotocol/client";
3
+ export declare const MCP_TRACE_SCHEMA_VERSION = 1;
4
+ export declare const DEFAULT_MCP_TRACE_MAX_BYTES: number;
5
+ export declare const DEFAULT_MCP_TRACE_MAX_EVENTS = 10000;
6
+ export type McpTraceDirection = "outbound" | "inbound";
7
+ export type McpTraceTransport = "stdio" | "unix-socket" | "sse" | "streamable-http" | "unknown";
8
+ export type McpTraceMessageKind = "request" | "response" | "notification";
9
+ export interface McpTraceSettings {
10
+ /** Enable metadata-only protocol tracing for all servers unless overridden. */
11
+ enabled?: boolean;
12
+ /** JSONL destination. Relative paths are resolved from the session cwd. */
13
+ file?: string;
14
+ /** Maximum bytes retained in the per-session JSONL file. */
15
+ maxBytes?: number;
16
+ /** Maximum events retained in the per-session JSONL file. */
17
+ maxEvents?: number;
18
+ }
19
+ export interface McpTraceEvent {
20
+ version: typeof MCP_TRACE_SCHEMA_VERSION;
21
+ timestamp: string;
22
+ direction: McpTraceDirection;
23
+ server: string;
24
+ transport: McpTraceTransport;
25
+ kind: McpTraceMessageKind;
26
+ status: "sent" | "received" | "error";
27
+ method?: string;
28
+ id?: string | number | null;
29
+ relatedRequestId?: string | number;
30
+ errorCode?: number | string;
31
+ bytes?: number;
32
+ durationMs?: number;
33
+ }
34
+ export interface McpTraceWriterOptions {
35
+ filePath: string;
36
+ maxBytes?: number;
37
+ maxEvents?: number;
38
+ appendFile?: (path: string, data: string, options: {
39
+ encoding: "utf8";
40
+ }) => Promise<void>;
41
+ writeFile?: (path: string, data: string, options: {
42
+ encoding: "utf8";
43
+ }) => Promise<void>;
44
+ mkdir?: (path: string, options: {
45
+ recursive: true;
46
+ }) => Promise<string | undefined>;
47
+ }
48
+ /** Redact strings before they can become durable trace metadata. */
49
+ export declare function redactTraceText(value: string, maxLength?: number): string;
50
+ export declare function createMcpTraceEvent(direction: McpTraceDirection, server: string, transport: McpTraceTransport, message: JSONRPCMessage, status: "sent" | "received" | "error", options?: {
51
+ relatedRequestId?: unknown;
52
+ durationMs?: number;
53
+ }): McpTraceEvent;
54
+ export declare class McpTraceWriter {
55
+ private readonly options;
56
+ private readonly maxBytes;
57
+ private readonly maxEvents;
58
+ private readonly append;
59
+ private readonly resetFile;
60
+ private readonly makeDirectory;
61
+ private bytesWritten;
62
+ private eventsWritten;
63
+ private queue;
64
+ private disabled;
65
+ private initializationFailed;
66
+ private readonly fileReady;
67
+ constructor(options: McpTraceWriterOptions);
68
+ get filePath(): string;
69
+ get isDisabled(): boolean;
70
+ get stats(): {
71
+ bytes: number;
72
+ events: number;
73
+ };
74
+ write(event: McpTraceEvent): void;
75
+ flush(): Promise<void>;
76
+ }
77
+ export declare function createMcpTraceWriter(sessionCwd: string | undefined, settings?: McpTraceSettings, randomSuffix?: string): McpTraceWriter;
78
+ export interface McpTraceObserver {
79
+ record(event: McpTraceEvent): void;
80
+ }
81
+ export declare function isMcpTraceEnabled(definition: {
82
+ trace?: boolean;
83
+ }, settings?: McpTraceSettings): boolean;
84
+ /**
85
+ * Compose the SDK's transport callbacks in place instead of replacing the
86
+ * transport object. SDK v2 detects its base stdio transport before connect so
87
+ * it can run `server/discover` on a disposable sibling process; returning a
88
+ * wrapper hides that identity and makes tracing change negotiation behavior.
89
+ */
90
+ export declare function wrapTransportWithMcpTrace(transport: Transport, server: string, transportKind: McpTraceTransport, observer: McpTraceObserver): Transport;
91
+ export declare function traceTransportKind(definition: {
92
+ command?: string;
93
+ url?: string;
94
+ socket?: string;
95
+ }, transport: Transport): McpTraceTransport;