@alint-js/agent-apeira 0.0.25 → 0.0.28

package/dist/index.d.mts

@@ -1,7 +1,6 @@
+import { AgentAdapter, AgentTool } from "@alint-js/core/agent";
 import { Runner, Tool } from "apeira";
 import { ResolvedModel } from "@alint-js/core";
-import { AgentAdapter, AgentTool } from "@alint-js/core/agent";
-
 //#region src/index.d.ts
 interface ApeiraAdapterOptions {
   createRunner: (model: ResolvedModel, maxSteps: number) => Runner;
@@ -9,6 +8,6 @@ interface ApeiraAdapterOptions {
 }
 declare function createApeiraAdapter(options?: Partial<ApeiraAdapterOptions>): AgentAdapter;
 declare function createApeiraRunner(model: ResolvedModel, maxSteps?: number): Runner;
-declare function toRunnerTools(tools: AgentTool[]): Tool[];
+declare function toRunnerTools(tools: AgentTool[], onToolStart?: () => void): Tool[];
 //#endregion
 export { ApeiraAdapterOptions, createApeiraAdapter, createApeiraRunner, toRunnerTools };

package/dist/index.mjs

@@ -1,4 +1,30 @@
+import { RetryableAgentError } from "@alint-js/core/agent";
 import { chat, rawTool, stepCountAtLeast, user } from "apeira";
+//#region src/retry.ts
+function isRetryableApeiraFailure(error) {
+	const seen = /* @__PURE__ */ new Set();
+	let candidate = error;
+	let retryable = false;
+	while (isObject(candidate) && !seen.has(candidate)) {
+		seen.add(candidate);
+		if (readProperty(candidate, "name") === "AbortError") return false;
+		const statusCode = readProperty(candidate, "statusCode");
+		if (typeof statusCode === "number" && (statusCode === 408 || statusCode === 429 || Number.isInteger(statusCode) && statusCode >= 500 && statusCode <= 599)) retryable = true;
+		candidate = readProperty(candidate, "cause");
+	}
+	return retryable;
+}
+function isObject(value) {
+	return typeof value === "object" && value !== null || typeof value === "function";
+}
+function readProperty(value, property) {
+	try {
+		return Reflect.get(value, property);
+	} catch {
+		return;
+	}
+}
+//#endregion
 //#region src/index.ts
 const defaultMaxSteps = 8;
 function createApeiraAdapter(options = {}) {
@@ -6,11 +32,20 @@ function createApeiraAdapter(options = {}) {
 	const maxSteps = options.maxSteps ?? defaultMaxSteps;
 	if (!Number.isInteger(maxSteps) || maxSteps < 1) throw new TypeError("Apeira adapter maxSteps must be a positive integer");
 	return async (request) => {
-		const result = await createRunner(request.model, maxSteps)(buildRunnerContext(request));
-		return {
-			answer: extractAnswer(result.output),
-			usage: mapUsage(result.usage)
-		};
+		let toolStarted = false;
+		const runner = createRunner(request.model, maxSteps);
+		try {
+			const result = await runner(buildRunnerContext(request, () => {
+				toolStarted = true;
+			}));
+			return {
+				answer: extractAnswer(result.output),
+				usage: mapUsage(result.usage)
+			};
+		} catch (error) {
+			if (!toolStarted && !request.signal?.aborted && isRetryableApeiraFailure(error)) throw new RetryableAgentError("Apeira agent invocation can be safely replayed", { cause: error });
+			throw error;
+		}
 	};
 }
 function createApeiraRunner(model, maxSteps = defaultMaxSteps) {
@@ -22,22 +57,24 @@ function createApeiraRunner(model, maxSteps = defaultMaxSteps) {
 		streamOptions: { includeUsage: true }
 	});
 }
-function toRunnerTools(tools) {
+function toRunnerTools(tools, onToolStart = () => {}) {
 	return tools.map((agentTool) => rawTool({
 		description: agentTool.description,
 		execute: async (input) => {
+			onToolStart();
 			return await agentTool.execute(input) ?? "";
 		},
 		name: agentTool.name,
 		parameters: agentTool.parameters
 	}));
 }
-function buildRunnerContext(request) {
+function buildRunnerContext(request, onToolStart) {
 	return {
+		abortSignal: request.signal,
 		channel: noopChannel(),
 		input: [user(request.prompt)],
 		instructions: request.instructions,
-		tools: toRunnerTools(request.tools),
+		tools: toRunnerTools(request.tools, onToolStart),
 		turnId: "alint"
 	};
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@alint-js/agent-apeira",
   "type": "module",
-  "version": "0.0.25",
+  "version": "0.0.28",
   "exports": {
     ".": {
       "types": "./dist/index.d.mts",
@@ -13,13 +13,13 @@
     "dist"
   ],
   "peerDependencies": {
-    "@alint-js/core": "0.0.25"
+    "@alint-js/core": "0.0.28"
   },
   "dependencies": {
     "apeira": "^0.0.7"
   },
   "devDependencies": {
-    "@alint-js/core": "0.0.25"
+    "@alint-js/core": "0.0.28"
   },
   "scripts": {
     "build": "tsdown",