@nulab/datadog-ai-guard-middleware-ai-v2 0.1.0

package/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2026 Nulab Inc.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

package/README.md

@@ -0,0 +1,177 @@
+# @nulab/datadog-ai-guard-middleware-ai-v2
+
+Datadog AI Guard middleware for [Vercel AI SDK](https://sdk.vercel.ai/) v5 (`@ai-sdk/provider@^2`).
+
+> Using AI SDK v6? See [`@nulab/datadog-ai-guard-middleware-ai-v3`](https://www.npmjs.com/package/@nulab/datadog-ai-guard-middleware-ai-v3) instead.
+
+## Installation
+
+```bash
+npm install @nulab/datadog-ai-guard-middleware-ai-v2
+```
+
+### Peer Dependencies
+
+- `@ai-sdk/provider` ^2.0.0
+
+## Usage
+
+### Initializing dd-trace
+
+Initialize dd-trace with AI Guard enabled and pass `ddtracer.aiguard` (the `AIGuardEvaluator`) to the middleware.
+
+```ts
+import ddtracer from "dd-trace";
+
+ddtracer.init({
+  experimental: {
+    aiguard: { enabled: true },
+  },
+});
+```
+
+> Authentication follows **dd-trace environment variables** (`DD_API_KEY`, `DD_APP_KEY`, etc.). The AI Guard endpoint is **resolved by dd-trace from `DD_SITE`** (e.g., `ap1.datadoghq.com` -> `https://app.ap1.datadoghq.com/api/v2/ai-guard`).
+
+### Creating a Guarded Model
+
+Use `wrapLanguageModel` to create a middleware-applied model that can be reused with both `generateText` and `streamText`.
+
+```ts
+import { wrapLanguageModel } from "ai";
+
+const guardedModel = wrapLanguageModel({
+  model: yourModel,
+  middleware: [
+    new AIGuardMiddleware({
+      evaluator: ddtracer.aiguard,
+    }),
+  ],
+});
+```
+
+### Non-Streaming
+
+```ts
+import { generateText } from "ai";
+import {
+  AIGuardMiddleware,
+  AIGuardMiddlewareAbortError,
+} from "@nulab/datadog-ai-guard-middleware-ai-v2";
+
+try {
+  const result = await generateText({
+    model: guardedModel,
+    prompt: "Hello",
+  });
+  console.log(result.text);
+} catch (err) {
+  if (err instanceof AIGuardMiddlewareAbortError) {
+    // Blocked by AI Guard security policy
+  } else {
+    throw err;
+  }
+}
+```
+
+- The prompt is evaluated **before** being sent to the LLM. If Datadog's **blocking policy** is active and the prompt violates it, an `AIGuardMiddlewareAbortError` is thrown.
+- After the LLM responds, the **entire assistant turn** is evaluated once. If Datadog returns `DENY` / `ABORT` without a blocking policy, only a **warning** is logged.
+
+### Streaming
+
+```ts
+import { streamText } from "ai";
+
+const { stream } = await streamText({
+  model: guardedModel,
+  prompt: "Hello",
+});
+
+for await (const part of stream) {
+  if (part.type === "text-delta") process.stdout.write(part.text);
+  if (part.type === "error") {
+    // AIGuardStreamTripWire error: tool call blocked by AI Guard
+    console.error("blocked:", String(part.error?.message));
+  }
+}
+```
+
+- In streaming mode, **text-only assistant output is not sent to AI Guard**. AI Guard evaluates only prompts and `tool-call` chunks.
+- When evaluating a `tool-call`, **preceding sibling tool calls from the same turn** are included in the same evaluation request.
+
+## Options
+
+```ts
+interface AIGuardMiddlewareOptions {
+  evaluator: AIGuardEvaluator;
+  allowOnFailure?: boolean;
+  createAbortError?: (kind: AIGuardKind, message: string) => Error;
+  logger?: AIGuardLogger;
+}
+```
+
+| Option | Type | Default | Description |
+|--------|------|---------|-------------|
+| `evaluator` | `AIGuardEvaluator` | (required) | Object with an `evaluate(messages, opts)` method. Pass `ddtracer.aiguard` when using dd-trace. Any object satisfying the `AIGuardEvaluator` interface works. |
+| `allowOnFailure` | `boolean` | `true` | `true`: fail-open (continue on evaluation errors). `false`: throw `AIGuardMiddlewareClientError` when evaluation is unavailable. Covers both API failures and message normalization/conversion errors. |
+| `createAbortError` | `(kind, message) => Error` | — | Factory to create a custom error when AI Guard blocks a request. `markAsNonRetryable()` is automatically applied. Useful for framework integration (e.g., injecting Mastra's `TripWire`). |
+| `logger` | `AIGuardLogger` | `console` | Logger with `info`, `warn`, `error` methods. Defaults to `console` so security events are always visible. Pass a no-op logger (`{ info() {}, warn() {}, error() {} }`) to suppress output. |
+
+> When Datadog returns `DENY` / `ABORT` without a blocking policy, the middleware logs a **warning** and continues regardless of `allowOnFailure`.
+
+## API
+
+### Classes
+
+| Export | Description |
+|--------|-------------|
+| `AIGuardMiddleware` | Main middleware class. Pass to `wrapLanguageModel` for use. |
+| `AIGuardMiddlewareAbortError` | Thrown when Datadog AI Guard's blocking policy blocks a request (default when `createAbortError` is not specified). Use `instanceof` to identify policy blocks. |
+| `AIGuardMiddlewareClientError` | Thrown when AI Guard evaluation cannot be performed (`allowOnFailure: false`). Distinguishes service failures from policy blocks (`AIGuardMiddlewareAbortError`). |
+| `AIGuardStreamTripWire` | Error used in streaming error chunks (`{ type: "error" }`). Identifies AI Guard-specific errors on the stream consumer side. |
+
+### Types
+
+| Export | Description |
+|--------|-------------|
+| `AIGuardMiddlewareOptions` | Constructor argument type for `AIGuardMiddleware`. |
+| `AIGuardEvaluator` | Interface with an `evaluate(messages, opts)` method. `ddtracer.aiguard` satisfies this interface. |
+| `AIGuardLogger` | Interface with `info`, `warn`, `error` methods. |
+| `AIGuardKind` | `"Prompt" \| "Assistant response" \| "Tool call"` union type. |
+| `AIGuardMessage` | Message type sent to AI Guard. Argument type for `AIGuardEvaluator.evaluate()`. |
+
+### Error Classes
+
+| Class | Code | Condition |
+|-------|------|-----------|
+| `AIGuardMiddlewareAbortError` | `AI_GUARD_MIDDLEWARE_ABORT` | Datadog AI Guard blocked the request via blocking policy |
+| `AIGuardMiddlewareClientError` | `AI_GUARD_MIDDLEWARE_CLIENT_ERROR` | AI Guard evaluation unavailable + `allowOnFailure=false` |
+| `AIGuardStreamTripWire` | — | Streaming error chunk: policy block or fail-close evaluation failure |
+
+> **Security**: These errors intentionally hide details from the original AI Guard SDK error (`reason`, `tags`, etc.).
+
+## Evaluation Scope
+
+- **Prompt**: Evaluated once before sending to the LLM
+- **Assistant response (non-streaming)**: Evaluated once per assistant turn after `doGenerate`
+- **Tool calls (streaming)**: Evaluated on each `tool-call` chunk with accumulated sibling tool calls
+- **Tool execution results**: Automatically evaluated as part of the next turn's prompt in agent loops
+
+> Blocking decisions are made by **Datadog AI Guard's blocking policy**. The middleware sends `block: true` but only blocks when Datadog returns `AIGuardAbortError`.
+
+## Known Limitations
+
+- **dd-trace AI Guard experimental feature** must be enabled (use a compatible version).
+- **Blocking requires a Datadog-side blocking policy**. Without one, `DENY` / `ABORT` only produces warnings.
+- Streaming **text-only output is not protected by AI Guard**. Use non-streaming for strict blocking of all assistant output.
+- **Multimodal evaluation supports images (`image/*`) only**. Non-image files (e.g., PDF) are not sent to AI Guard.
+- Evaluation API calls increase **per assistant turn (non-streaming)** or **per `tool-call` chunk (streaming)**.
+
+## Requirements
+
+- Node.js >= 22
+- `@ai-sdk/provider` ^2.0.0
+- [dd-trace](https://github.com/DataDog/dd-trace-js) >= 5 with AI Guard enabled
+
+## License
+
+[Apache-2.0](https://github.com/nulab/datadog-ai-guard-middleware/blob/main/LICENSE)

package/dist/ai.d.ts

@@ -0,0 +1,24 @@
+import type { LanguageModelV2, LanguageModelV2CallOptions } from "@ai-sdk/provider";
+import type { AIGuardMiddlewareOptions } from "@nulab/datadog-ai-guard-middleware-core";
+export type { AIGuardEvaluator, AIGuardKind, AIGuardLogger, AIGuardMiddlewareOptions, } from "@nulab/datadog-ai-guard-middleware-core";
+export { AIGuardClientError as AIGuardMiddlewareClientError, AIGuardStreamTripWire, } from "@nulab/datadog-ai-guard-middleware-core";
+export { AIGuardMiddlewareAbortError } from "./marker.js";
+type DoGenerateResult = Awaited<ReturnType<LanguageModelV2["doGenerate"]>>;
+type DoStreamResult = Awaited<ReturnType<LanguageModelV2["doStream"]>>;
+/** AI Guard middleware for Vercel AI SDK (LanguageModelV2Middleware compatible). */
+export declare class AIGuardMiddleware {
+    #private;
+    constructor(options: AIGuardMiddlewareOptions);
+    wrapGenerate({ doGenerate, params, }: {
+        doGenerate: () => ReturnType<LanguageModelV2["doGenerate"]>;
+        doStream: () => ReturnType<LanguageModelV2["doStream"]>;
+        params: LanguageModelV2CallOptions;
+        model: LanguageModelV2;
+    }): Promise<DoGenerateResult>;
+    wrapStream({ doStream, params, }: {
+        doGenerate: () => ReturnType<LanguageModelV2["doGenerate"]>;
+        doStream: () => ReturnType<LanguageModelV2["doStream"]>;
+        params: LanguageModelV2CallOptions;
+        model: LanguageModelV2;
+    }): Promise<DoStreamResult>;
+}

package/dist/ai.js

@@ -0,0 +1,94 @@
+import { AIGuardEvaluationEngine } from "@nulab/datadog-ai-guard-middleware-core";
+import { convertToAIGuardFormat } from "./convert.js";
+import { AIGuardMiddlewareAbortError, markAsNonRetryable } from "./marker.js";
+export { AIGuardClientError as AIGuardMiddlewareClientError, AIGuardStreamTripWire, } from "@nulab/datadog-ai-guard-middleware-core";
+export { AIGuardMiddlewareAbortError } from "./marker.js";
+// NOTE: async/await is required by Vercel AI SDK middleware interface
+/** AI Guard middleware for Vercel AI SDK (LanguageModelV2Middleware compatible). */
+export class AIGuardMiddleware {
+    #engine;
+    constructor(options) {
+        const { evaluator, allowOnFailure = true, createAbortError, logger = console } = options || {};
+        if (!evaluator) {
+            throw new TypeError("AIGuardMiddleware: evaluator is required");
+        }
+        this.#engine = new AIGuardEvaluationEngine({
+            evaluator,
+            allowOnFailure,
+            logger,
+            buildAbortError: (kind) => {
+                if (createAbortError) {
+                    const customError = createAbortError(kind, `${kind} blocked by AI Guard security policy`);
+                    markAsNonRetryable(customError);
+                    return customError;
+                }
+                return new AIGuardMiddlewareAbortError(kind);
+            },
+        });
+        this.wrapGenerate = this.wrapGenerate.bind(this);
+        this.wrapStream = this.wrapStream.bind(this);
+    }
+    async #evaluatePrompt(prompt) {
+        const roles = prompt.map((m) => m.role);
+        const meta = `roles=[${roles.join(",")}]`;
+        const messages = this.#engine.convertOrHandleUnavailable("Prompt", meta, () => convertToAIGuardFormat(prompt));
+        if (!messages) {
+            return undefined;
+        }
+        await this.#engine.evaluate(messages, "Prompt", meta);
+        return messages;
+    }
+    async wrapGenerate({ doGenerate, params, }) {
+        const baseMessages = await this.#evaluatePrompt(params.prompt);
+        const result = await doGenerate();
+        if (result.content.length > 0) {
+            await this.#engine.evaluateAssistantResponse(result.content, baseMessages);
+        }
+        return result;
+    }
+    async wrapStream({ doStream, params, }) {
+        const baseMessages = await this.#evaluatePrompt(params.prompt);
+        const result = await doStream();
+        let stopped = false;
+        const seenToolCalls = [];
+        const transform = new TransformStream({
+            transform: async (chunk, controller) => {
+                if (stopped) {
+                    return;
+                }
+                if (chunk.type === "tool-call") {
+                    try {
+                        const converted = await this.#engine.evaluateToolCall(chunk, baseMessages, seenToolCalls);
+                        if (converted) {
+                            seenToolCalls.push(converted);
+                        }
+                    }
+                    catch (error) {
+                        stopped = true;
+                        if (this.#engine.isBlockedByPolicy(error)) {
+                            this.#engine.logger.warn("[AI Guard] wrapStream — tool call blocked, terminating stream");
+                        }
+                        else {
+                            this.#engine.logger.warn("[AI Guard] wrapStream — tool call could not be evaluated, terminating stream");
+                        }
+                        controller.enqueue({
+                            type: "error",
+                            error: this.#engine.createStreamErrorForAbort(error),
+                        });
+                        controller.terminate();
+                        return;
+                    }
+                }
+                if (chunk.type === "finish") {
+                    seenToolCalls.length = 0;
+                }
+                controller.enqueue(chunk);
+            },
+        });
+        const wrappedStream = result.stream.pipeThrough(transform);
+        return {
+            ...result,
+            stream: wrappedStream,
+        };
+    }
+}

package/dist/convert.d.ts

@@ -0,0 +1,4 @@
+import type { LanguageModelV2Message } from "@ai-sdk/provider";
+import type { AIGuardMessage } from "@nulab/datadog-ai-guard-middleware-core";
+export type { AIGuardMessage } from "@nulab/datadog-ai-guard-middleware-core";
+export declare function convertToAIGuardFormat(prompt: LanguageModelV2Message[]): AIGuardMessage[];

package/dist/convert.js

@@ -0,0 +1,54 @@
+import { convertAssistantContentToAIGuardMessages, convertPromptContent, extractToolResultContent, } from "@nulab/datadog-ai-guard-middleware-core";
+function convertMessage(message) {
+    switch (message.role) {
+        case "system":
+            return {
+                role: message.role,
+                content: convertPromptContent(message.content),
+            };
+        case "user":
+            return {
+                role: message.role,
+                content: convertPromptContent(message.content),
+            };
+        case "assistant":
+            return convertAssistantContentToAIGuardMessages(message.content);
+        case "tool": {
+            if (!Array.isArray(message.content)) {
+                return { role: message.role, content: "" };
+            }
+            const toolMessages = [];
+            for (const part of message.content) {
+                if (part.type === "tool-result") {
+                    if (typeof part.toolCallId !== "string" || part.toolCallId.trim() === "") {
+                        throw new TypeError("Tool result must have a non-empty tool_call_id");
+                    }
+                    toolMessages.push({
+                        role: "tool",
+                        tool_call_id: part.toolCallId,
+                        content: extractToolResultContent(part),
+                    });
+                }
+            }
+            return toolMessages;
+        }
+        default: {
+            // Defensive fallback for unexpected roles at runtime
+            const msg = message;
+            return { role: msg.role, content: convertPromptContent(msg.content) };
+        }
+    }
+}
+export function convertToAIGuardFormat(prompt) {
+    const result = [];
+    for (const message of prompt) {
+        const converted = convertMessage(message);
+        if (Array.isArray(converted)) {
+            result.push(...converted);
+        }
+        else {
+            result.push(converted);
+        }
+    }
+    return result;
+}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export type { AIGuardEvaluator, AIGuardKind, AIGuardLogger, AIGuardMiddlewareOptions, } from "./ai.js";
+export { AIGuardMiddleware, AIGuardMiddlewareAbortError, AIGuardMiddlewareClientError, AIGuardStreamTripWire, } from "./ai.js";
+export type { AIGuardMessage } from "./convert.js";

package/dist/index.js

@@ -0,0 +1 @@
+export { AIGuardMiddleware, AIGuardMiddlewareAbortError, AIGuardMiddlewareClientError, AIGuardStreamTripWire, } from "./ai.js";

package/dist/marker.d.ts

@@ -0,0 +1,30 @@
+import type { AIGuardKind } from "@nulab/datadog-ai-guard-middleware-core";
+import { AIGuardAbortError } from "@nulab/datadog-ai-guard-middleware-core";
+/**
+ * Applies Vercel AI SDK non-retryable markers to an error.
+ *
+ * After calling this, the error will be recognized by the AI SDK's
+ * `APICallError.isInstance()` duck-type check and skipped by
+ * `retryWithExponentialBackoff` due to `isRetryable=false`.
+ *
+ * Any existing `isRetryable` property is overwritten to guarantee
+ * the error will not be retried.
+ */
+export declare function markAsNonRetryable(error: Error): void;
+/**
+ * Non-retryable error thrown when AI Guard blocks a request by security policy.
+ *
+ * The Vercel AI SDK APICallError markers with isRetryable=false prevent
+ * retryWithExponentialBackoff from retrying within the same model.
+ *
+ * For framework integration (e.g. Mastra), use the `createAbortError` option
+ * in `AIGuardMiddlewareOptions` to supply a custom error class that your
+ * framework recognizes. The middleware automatically calls `markAsNonRetryable`
+ * on the custom error.
+ */
+export declare class AIGuardMiddlewareAbortError extends AIGuardAbortError {
+    readonly name = "AIGuardMiddlewareAbortError";
+    readonly code = "AI_GUARD_MIDDLEWARE_ABORT";
+    readonly isRetryable = false;
+    constructor(kind: AIGuardKind);
+}

package/dist/marker.js

@@ -0,0 +1,41 @@
+import { AIGuardAbortError } from "@nulab/datadog-ai-guard-middleware-core";
+// Vercel AI SDK marker symbols used by APICallError.isInstance() for duck-type checking.
+// When both markers are present and isRetryable is false, the AI SDK's
+// retryWithExponentialBackoff skips retries for non-transient security policy blocks.
+const AI_SDK_ERROR_MARKER = Symbol.for("vercel.ai.error.AI_SDKError");
+const API_CALL_ERROR_MARKER = Symbol.for("vercel.ai.error.AI_APICallError");
+/**
+ * Applies Vercel AI SDK non-retryable markers to an error.
+ *
+ * After calling this, the error will be recognized by the AI SDK's
+ * `APICallError.isInstance()` duck-type check and skipped by
+ * `retryWithExponentialBackoff` due to `isRetryable=false`.
+ *
+ * Any existing `isRetryable` property is overwritten to guarantee
+ * the error will not be retried.
+ */
+export function markAsNonRetryable(error) {
+    Object.defineProperty(error, AI_SDK_ERROR_MARKER, { value: true });
+    Object.defineProperty(error, API_CALL_ERROR_MARKER, { value: true });
+    Object.defineProperty(error, "isRetryable", { value: false, configurable: true });
+}
+/**
+ * Non-retryable error thrown when AI Guard blocks a request by security policy.
+ *
+ * The Vercel AI SDK APICallError markers with isRetryable=false prevent
+ * retryWithExponentialBackoff from retrying within the same model.
+ *
+ * For framework integration (e.g. Mastra), use the `createAbortError` option
+ * in `AIGuardMiddlewareOptions` to supply a custom error class that your
+ * framework recognizes. The middleware automatically calls `markAsNonRetryable`
+ * on the custom error.
+ */
+export class AIGuardMiddlewareAbortError extends AIGuardAbortError {
+    name = "AIGuardMiddlewareAbortError";
+    code = "AI_GUARD_MIDDLEWARE_ABORT";
+    isRetryable = false;
+    constructor(kind) {
+        super(kind);
+        markAsNonRetryable(this);
+    }
+}

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@nulab/datadog-ai-guard-middleware-ai-v2",
+  "version": "0.1.0",
+  "type": "module",
+  "license": "Apache-2.0",
+  "engines": {
+    "node": ">=22"
+  },
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      }
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "@nulab/datadog-ai-guard-middleware-core": "0.1.0"
+  },
+  "peerDependencies": {
+    "@ai-sdk/provider": "^2.0.0"
+  },
+  "devDependencies": {
+    "@ai-sdk/openai": "^2.0.0",
+    "@ai-sdk/provider": "^2.0.0",
+    "@types/node": "^25.5.0",
+    "ai": "^5.0.0",
+    "@nulab/datadog-ai-guard-middleware-internal": "0.0.0"
+  },
+  "scripts": {
+    "clean": "rm -rf dist",
+    "build": "pnpm run clean && tsc",
+    "test": "vitest run",
+    "test:coverage": "vitest run --coverage",
+    "test:integration": "vitest run --config vitest.integration.config.ts"
+  }
+}