@contractspec/lib.ai-agent 3.0.0 → 3.2.0

package/dist/tools/tool-adapter.js

@@ -2263,21 +2263,39 @@ var init_json_schema_to_zod = () => {};
 // src/tools/tool-adapter.ts
 import { tool } from "ai";
 function specToolToAISDKTool(specTool, handler, context = {}) {
+  let lastInvocationAt;
   return tool({
     description: specTool.description ?? specTool.name,
     inputSchema: jsonSchemaToZodSafe(specTool.schema),
     needsApproval: specTool.requiresApproval ?? !specTool.automationSafe,
     execute: async (input) => {
-      const result = await handler(input, {
-        agentId: context.agentId ?? "unknown",
-        sessionId: context.sessionId ?? "unknown",
-        tenantId: context.tenantId,
-        actorId: context.actorId,
-        locale: context.locale,
-        metadata: context.metadata,
-        signal: context.signal
-      });
-      return typeof result === "string" ? result : JSON.stringify(result);
+      const now = Date.now();
+      const cooldownMs = normalizeDuration(specTool.cooldownMs);
+      if (cooldownMs && lastInvocationAt !== undefined) {
+        const elapsed = now - lastInvocationAt;
+        if (elapsed < cooldownMs) {
+          const retryAfterMs = cooldownMs - elapsed;
+          throw createToolExecutionError(`Tool "${specTool.name}" is cooling down. Retry in ${retryAfterMs}ms.`, "TOOL_COOLDOWN_ACTIVE", retryAfterMs);
+        }
+      }
+      const timeoutMs = normalizeDuration(specTool.timeoutMs);
+      const { signal, dispose } = createTimeoutSignal(context.signal, timeoutMs);
+      try {
+        const execution = handler(input, {
+          agentId: context.agentId ?? "unknown",
+          sessionId: context.sessionId ?? "unknown",
+          tenantId: context.tenantId,
+          actorId: context.actorId,
+          locale: context.locale,
+          metadata: context.metadata,
+          signal
+        });
+        const result = timeoutMs ? await withTimeout(execution, timeoutMs, specTool.name) : await execution;
+        return typeof result === "string" ? result : JSON.stringify(result);
+      } finally {
+        dispose();
+        lastInvocationAt = Date.now();
+      }
     }
   });
 }
@@ -2302,6 +2320,64 @@ function createToolHandler(handler) {
 function buildToolHandlers(handlersObj) {
   return new Map(Object.entries(handlersObj));
 }
+function normalizeDuration(value) {
+  if (value === undefined) {
+    return;
+  }
+  if (!Number.isFinite(value)) {
+    return;
+  }
+  if (value <= 0) {
+    return;
+  }
+  return Math.round(value);
+}
+function withTimeout(execution, timeoutMs, toolName) {
+  return new Promise((resolve, reject) => {
+    const timeoutHandle = setTimeout(() => {
+      reject(createToolExecutionError(`Tool "${toolName}" timed out after ${timeoutMs}ms.`, "TOOL_EXECUTION_TIMEOUT"));
+    }, timeoutMs);
+    execution.then((result) => {
+      clearTimeout(timeoutHandle);
+      resolve(result);
+    }).catch((error) => {
+      clearTimeout(timeoutHandle);
+      reject(error);
+    });
+  });
+}
+function createTimeoutSignal(signal, timeoutMs) {
+  const controller = new AbortController;
+  const abortFromSource = () => controller.abort();
+  if (signal) {
+    if (signal.aborted) {
+      controller.abort();
+    } else {
+      signal.addEventListener("abort", abortFromSource);
+    }
+  }
+  const timeoutHandle = timeoutMs !== undefined ? setTimeout(() => {
+    controller.abort();
+  }, timeoutMs) : undefined;
+  return {
+    signal: controller.signal,
+    dispose: () => {
+      if (timeoutHandle !== undefined) {
+        clearTimeout(timeoutHandle);
+      }
+      if (signal) {
+        signal.removeEventListener("abort", abortFromSource);
+      }
+    }
+  };
+}
+function createToolExecutionError(message, code, retryAfterMs) {
+  return Object.assign(new Error(message), {
+    code,
+    kind: "retryable",
+    retryAfterMs
+  });
+}
 var init_tool_adapter = __esm(() => {
   init_json_schema_to_zod();
   init_i18n();

package/dist/tools/tool-adapter.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/types.d.ts

@@ -36,6 +36,13 @@ export interface AgentMessage {
 export type AgentStatus = 'idle' | 'running' | 'waiting' | 'completed' | 'failed' | 'escalated';
 export type AgentEventName = 'agent.session.created' | 'agent.session.updated' | 'agent.step.started' | 'agent.step.completed' | 'agent.tool.called' | 'agent.tool.completed' | 'agent.tool.failed' | 'agent.tool.approval_requested' | 'agent.escalated' | 'agent.completed' | 'agent.failed';
 export type AgentFinishReason = 'stop' | 'tool-calls' | 'length' | 'content-filter' | 'error' | 'other' | 'unknown';
+export type AgentExecutionErrorKind = 'fatal' | 'retryable' | 'timeout' | 'guard_rejected' | 'policy_blocked';
+export interface AgentExecutionError {
+    kind: AgentExecutionErrorKind;
+    message: string;
+    code?: string;
+    retryAfterMs?: number;
+}
 export interface AgentEventPayload {
     sessionId: string;
     agentId: string;
@@ -59,6 +66,14 @@ export interface AgentCallOptions {
     metadata?: Record<string, string>;
     /** Locale override for this call */
     locale?: string;
+    /** Model selection context for ranking-driven model routing */
+    selectionContext?: import('@contractspec/lib.ai-providers/selector-types').ModelSelectionContext;
+    /** Transport preference for provider calls. */
+    transport?: 'rest' | 'mcp' | 'sdk';
+    /** Auth method to use for provider calls. */
+    authMethod?: 'api-key' | 'oauth2' | 'bearer';
+    /** Custom auth headers for provider calls. */
+    authHeaders?: Record<string, string>;
 }
 export interface AgentSessionState {
     sessionId: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contractspec/lib.ai-agent",
-  "version": "3.0.0",
+  "version": "3.2.0",
   "description": "AI agent orchestration with MCP and tool support",
   "keywords": [
     "contractspec",
@@ -33,18 +33,18 @@
     "typecheck": "tsc --noEmit"
   },
   "dependencies": {
-    "ai": "6.0.103",
-    "@ai-sdk/mcp": "1.0.21",
-    "@ai-sdk/anthropic": "3.0.48",
-    "@ai-sdk/google": "3.0.33",
-    "@ai-sdk/mistral": "3.0.20",
-    "@ai-sdk/openai": "3.0.36",
+    "ai": "6.0.116",
+    "@ai-sdk/mcp": "1.0.25",
+    "@ai-sdk/anthropic": "3.0.58",
+    "@ai-sdk/google": "3.0.43",
+    "@ai-sdk/mistral": "3.0.24",
+    "@ai-sdk/openai": "3.0.41",
     "@modelcontextprotocol/sdk": "^1.27.1",
-    "@posthog/react": "^1.8.1",
-    "@posthog/ai": "latest",
-    "@contractspec/lib.contracts-spec": "3.0.0",
-    "@contractspec/lib.ai-providers": "3.0.0",
-    "@contractspec/lib.knowledge": "3.0.0",
+    "@posthog/react": "^1.8.2",
+    "@posthog/ai": "7.9.5",
+    "@contractspec/lib.contracts-spec": "3.2.0",
+    "@contractspec/lib.ai-providers": "3.2.0",
+    "@contractspec/lib.knowledge": "3.2.0",
     "compare-versions": "^6.1.1",
     "zod": "^4.3.5"
   },
@@ -68,9 +68,9 @@
     }
   },
   "devDependencies": {
-    "@contractspec/tool.typescript": "3.0.0",
+    "@contractspec/tool.typescript": "3.2.0",
     "typescript": "^5.9.3",
-    "@contractspec/tool.bun": "3.0.0"
+    "@contractspec/tool.bun": "3.2.0"
   },
   "exports": {
     ".": {
@@ -235,6 +235,12 @@
       "node": "./dist/node/interop/index.js",
       "default": "./dist/interop/index.js"
     },
+    "./interop/runtime-adapters": {
+      "types": "./dist/interop/runtime-adapters.d.ts",
+      "bun": "./dist/interop/runtime-adapters.js",
+      "node": "./dist/node/interop/runtime-adapters.js",
+      "default": "./dist/interop/runtime-adapters.js"
+    },
     "./interop/spec-consumer": {
       "types": "./dist/interop/spec-consumer.d.ts",
       "bun": "./dist/interop/spec-consumer.js",
@@ -689,6 +695,12 @@
         "node": "./dist/node/interop/index.js",
         "default": "./dist/interop/index.js"
       },
+      "./interop/runtime-adapters": {
+        "types": "./dist/interop/runtime-adapters.d.ts",
+        "bun": "./dist/interop/runtime-adapters.js",
+        "node": "./dist/node/interop/runtime-adapters.js",
+        "default": "./dist/interop/runtime-adapters.js"
+      },
       "./interop/spec-consumer": {
         "types": "./dist/interop/spec-consumer.d.ts",
         "bun": "./dist/interop/spec-consumer.js",