68 lines
1.7 KiB
TypeScript
68 lines
1.7 KiB
TypeScript
import { bindAbortRelay } from "../utils/fetch-timeout.js";
|
|
import type { AnyAgentTool } from "./pi-tools.types.js";
|
|
|
|
function throwAbortError(): never {
|
|
const err = new Error("Aborted");
|
|
err.name = "AbortError";
|
|
throw err;
|
|
}
|
|
|
|
/**
|
|
* Checks if an object is a valid AbortSignal using structural typing.
|
|
* This is more reliable than `instanceof` across different realms (VM, iframe, etc.)
|
|
* where the AbortSignal constructor may differ.
|
|
*/
|
|
function isAbortSignal(obj: unknown): obj is AbortSignal {
|
|
return obj instanceof AbortSignal;
|
|
}
|
|
|
|
function combineAbortSignals(a?: AbortSignal, b?: AbortSignal): AbortSignal | undefined {
|
|
if (!a && !b) {
|
|
return undefined;
|
|
}
|
|
if (a && !b) {
|
|
return a;
|
|
}
|
|
if (b && !a) {
|
|
return b;
|
|
}
|
|
if (a?.aborted) {
|
|
return a;
|
|
}
|
|
if (b?.aborted) {
|
|
return b;
|
|
}
|
|
if (typeof AbortSignal.any === "function" && isAbortSignal(a) && isAbortSignal(b)) {
|
|
return AbortSignal.any([a, b]);
|
|
}
|
|
|
|
const controller = new AbortController();
|
|
const onAbort = bindAbortRelay(controller);
|
|
a?.addEventListener("abort", onAbort, { once: true });
|
|
b?.addEventListener("abort", onAbort, { once: true });
|
|
return controller.signal;
|
|
}
|
|
|
|
export function wrapToolWithAbortSignal(
|
|
tool: AnyAgentTool,
|
|
abortSignal?: AbortSignal,
|
|
): AnyAgentTool {
|
|
if (!abortSignal) {
|
|
return tool;
|
|
}
|
|
const execute = tool.execute;
|
|
if (!execute) {
|
|
return tool;
|
|
}
|
|
return {
|
|
...tool,
|
|
execute: async (toolCallId, params, signal, onUpdate) => {
|
|
const combined = combineAbortSignals(signal, abortSignal);
|
|
if (combined?.aborted) {
|
|
throwAbortError();
|
|
}
|
|
return await execute(toolCallId, params, combined, onUpdate);
|
|
},
|
|
};
|
|
}
|