@meetkai/mka1 0.48.26 → 0.48.28

package/src/funcs/guardrailsListGuardrailsPolicies.ts

@@ -0,0 +1,178 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { SDKCore } from "../core.js";
+import { encodeSimple } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
+import { SDKError } from "../models/errors/sdkerror.js";
+import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
+import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * List guardrails policies
+ *
+ * @remarks
+ * List the stored guardrails policies visible to the caller: org admins see every policy in their org (the org-wide policy has team_id null, plus each team's additional policy); other callers see the org-wide policy and their own team's policy. Unlike GET /guardrails, rows are returned per scope so a team can see which rules it owns versus which the org mandates.
+ */
+export function guardrailsListGuardrailsPolicies(
+  client: SDKCore,
+  request: operations.ListGuardrailsPoliciesRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    components.ListGuardrailsPoliciesResponse,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.ListGuardrailsPoliciesRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.ListGuardrailsPoliciesResponse,
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) =>
+      operations.ListGuardrailsPoliciesRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = null;
+
+  const path = pathToFunc("/api/v1/llm/guardrails/policies")();
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+    "X-On-Behalf-Of": encodeSimple(
+      "X-On-Behalf-Of",
+      payload["X-On-Behalf-Of"],
+      { explode: false, charEncoding: "none" },
+    ),
+  }));
+
+  const secConfig = await extractSecurity(client._options.bearerAuth);
+  const securityInput = secConfig == null ? {} : { bearerAuth: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "listGuardrailsPolicies",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.bearerAuth,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || {
+        strategy: "backoff",
+        backoff: {
+          initialInterval: 500,
+          maxInterval: 5000,
+          exponent: 1.5,
+          maxElapsedTime: 15000,
+        },
+        retryConnectionErrors: true,
+      }
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "GET",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    body: body,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const [result] = await M.match<
+    components.ListGuardrailsPoliciesResponse,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, components.ListGuardrailsPoliciesResponse$inboundSchema),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req);
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/guardrailsTestGuardrails.ts

@@ -30,7 +30,7 @@ import { Result } from "../types/fp.js";
  * Test content against guardrails
  *
  * @remarks
- * Test a piece of content against the configured guardrails without making an actual API call. Useful for debugging and validating guardrail configurations. Returns a report indicating whether the content passed and which guardrail was triggered if any.
+ * Test a piece of content against the caller's effective guardrails (org-wide policy plus team policy) without making an actual API call. Useful for debugging and validating guardrail configurations. Returns a report indicating whether the content passed and which guardrail was triggered if any.
  */
 export function guardrailsTestGuardrails(
   client: SDKCore,

package/src/funcs/guardrailsUpdateGuardrails.ts

@@ -27,10 +27,10 @@ import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
 
 /**
- * Update guardrails settings
+ * Update guardrails policy
  *
  * @remarks
- * Update the guardrails configuration for the authenticated user. Configure multiple guardrail modes including ban words (custom word list), prompt injection detection, and system prompt leakage detection. Each guardrail can be individually enabled/disabled with custom thresholds and rejection messages.
+ * Replace a guardrails policy. Omit team_id to set the org-wide policy (org admin only); provide team_id to set that team's additional policy (org admins may target any team in their org, other callers only their own team). The org-wide policy applies to every team; a team policy is enforced in addition to it — configuring a team policy never weakens the org-wide one.
  */
 export function guardrailsUpdateGuardrails(
   client: SDKCore,

package/src/lib/config.ts

@@ -65,7 +65,7 @@ export function serverURLFromOptions(options: SDKOptions): URL | null {
 export const SDK_METADATA = {
   language: "typescript",
   openapiDocVersion: "1.1.0",
-  sdkVersion: "0.48.26",
-  genVersion: "2.915.0",
-  userAgent: "speakeasy-sdk/typescript 0.48.26 2.915.0 1.1.0 @meetkai/mka1",
+  sdkVersion: "0.48.28",
+  genVersion: "2.916.2",
+  userAgent: "speakeasy-sdk/typescript 0.48.28 2.916.2 1.1.0 @meetkai/mka1",
 } as const;

package/src/mcp-server/mcp-server.ts

@@ -19,7 +19,7 @@ const routes = buildRouteMap({
 export const app = buildApplication(routes, {
   name: "mcp",
   versionInfo: {
-    currentVersion: "0.48.26",
+    currentVersion: "0.48.28",
   },
 });
 

package/src/mcp-server/server.ts

@@ -34,7 +34,9 @@ import { tool$agentVersionsGetAgentVersion } from "./tools/agentVersionsGetAgent
 import { tool$agentVersionsListAgentVersions } from "./tools/agentVersionsListAgentVersions.js";
 import { tool$agentVersionsRollbackAgentVersion } from "./tools/agentVersionsRollbackAgentVersion.js";
 import { tool$authApiKeyGetJwtFromKey } from "./tools/authApiKeyGetJwtFromKey.js";
+import { tool$guardrailsDeleteGuardrails } from "./tools/guardrailsDeleteGuardrails.js";
 import { tool$guardrailsGetGuardrails } from "./tools/guardrailsGetGuardrails.js";
+import { tool$guardrailsListGuardrailsPolicies } from "./tools/guardrailsListGuardrailsPolicies.js";
 import { tool$guardrailsTestGuardrails } from "./tools/guardrailsTestGuardrails.js";
 import { tool$guardrailsUpdateGuardrails } from "./tools/guardrailsUpdateGuardrails.js";
 import { tool$llmBatchesCancel } from "./tools/llmBatchesCancel.js";
@@ -274,7 +276,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.48.26",
+    version: "0.48.28",
   });
 
   const client = new SDKCore({
@@ -306,6 +308,8 @@ export function createMCPServer(deps: {
 
   tool(tool$guardrailsGetGuardrails);
   tool(tool$guardrailsUpdateGuardrails);
+  tool(tool$guardrailsDeleteGuardrails);
+  tool(tool$guardrailsListGuardrailsPolicies);
   tool(tool$guardrailsTestGuardrails);
   tool(tool$sandboxCreate);
   tool(tool$sandboxList);

package/src/mcp-server/tools/guardrailsDeleteGuardrails.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { guardrailsDeleteGuardrails } from "../../funcs/guardrailsDeleteGuardrails.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.DeleteGuardrailsRequest$inboundSchema,
+};
+
+export const tool$guardrailsDeleteGuardrails: ToolDefinition<typeof args> = {
+  name: "guardrails-delete-guardrails",
+  description: `Delete guardrails policy
+
+Delete a guardrails policy. Omit team_id to delete the org-wide policy (org admin only); provide team_id to delete that team's policy (org admins may target any team in their org, other callers only their own team). Idempotent: deleting a policy that doesn't exist returns deleted: false.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await guardrailsDeleteGuardrails(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/guardrailsGetGuardrails.ts

@@ -12,9 +12,9 @@ const args = {
 
 export const tool$guardrailsGetGuardrails: ToolDefinition<typeof args> = {
   name: "guardrails-get-guardrails",
-  description: `Get guardrails settings
+  description: `Get effective guardrails
 
-Retrieve the current guardrails configuration for the authenticated user. Returns all configured guardrails including ban words, prompt injection detection, and leakage detection settings.`,
+Retrieve the guardrails that apply to the caller's requests: the org-wide policy plus the caller's team policy, combined. Both are enforced when both exist. Policies are managed by org admins via PUT /guardrails.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await guardrailsGetGuardrails(

package/src/mcp-server/tools/guardrailsListGuardrailsPolicies.ts

@@ -0,0 +1,39 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { guardrailsListGuardrailsPolicies } from "../../funcs/guardrailsListGuardrailsPolicies.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.ListGuardrailsPoliciesRequest$inboundSchema,
+};
+
+export const tool$guardrailsListGuardrailsPolicies: ToolDefinition<
+  typeof args
+> = {
+  name: "guardrails-list-guardrails-policies",
+  description: `List guardrails policies
+
+List the stored guardrails policies visible to the caller: org admins see every policy in their org (the org-wide policy has team_id null, plus each team's additional policy); other callers see the org-wide policy and their own team's policy. Unlike GET /guardrails, rows are returned per scope so a team can see which rules it owns versus which the org mandates.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await guardrailsListGuardrailsPolicies(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/guardrailsTestGuardrails.ts

@@ -14,7 +14,7 @@ export const tool$guardrailsTestGuardrails: ToolDefinition<typeof args> = {
   name: "guardrails-test-guardrails",
   description: `Test content against guardrails
 
-Test a piece of content against the configured guardrails without making an actual API call. Useful for debugging and validating guardrail configurations. Returns a report indicating whether the content passed and which guardrail was triggered if any.`,
+Test a piece of content against the caller's effective guardrails (org-wide policy plus team policy) without making an actual API call. Useful for debugging and validating guardrail configurations. Returns a report indicating whether the content passed and which guardrail was triggered if any.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await guardrailsTestGuardrails(

package/src/mcp-server/tools/guardrailsUpdateGuardrails.ts

@@ -12,9 +12,9 @@ const args = {
 
 export const tool$guardrailsUpdateGuardrails: ToolDefinition<typeof args> = {
   name: "guardrails-update-guardrails",
-  description: `Update guardrails settings
+  description: `Update guardrails policy
 
-Update the guardrails configuration for the authenticated user. Configure multiple guardrail modes including ban words (custom word list), prompt injection detection, and system prompt leakage detection. Each guardrail can be individually enabled/disabled with custom thresholds and rejection messages.`,
+Replace a guardrails policy. Omit team_id to set the org-wide policy (org admin only); provide team_id to set that team's additional policy (org admins may target any team in their org, other callers only their own team). The org-wide policy applies to every team; a team policy is enforced in addition to it — configuring a team policy never weakens the org-wide one.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await guardrailsUpdateGuardrails(

package/src/models/components/banwordsguardrail.ts

@@ -3,6 +3,7 @@
  */
 
 import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -22,6 +23,10 @@ export type BanWordsGuardrail = {
    * Whether this guardrail is enabled
    */
   enabled?: boolean | undefined;
+  /**
+   * Also apply this guardrail to model output, not just the request
+   */
+  checkOutput?: boolean | undefined;
   /**
    * Configuration for banned words guardrail
    */
@@ -36,12 +41,18 @@ export const BanWordsGuardrail$inboundSchema: z.ZodType<
 > = z.object({
   mode: z.literal("ban_words"),
   enabled: z.boolean().default(true),
+  check_output: z.boolean().optional(),
   config: BanWordsConfig$inboundSchema.optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "check_output": "checkOutput",
+  });
 });
 /** @internal */
 export type BanWordsGuardrail$Outbound = {
   mode: "ban_words";
   enabled: boolean;
+  check_output?: boolean | undefined;
   config?: BanWordsConfig$Outbound | undefined;
 };
 
@@ -53,7 +64,12 @@ export const BanWordsGuardrail$outboundSchema: z.ZodType<
 > = z.object({
   mode: z.literal("ban_words"),
   enabled: z.boolean().default(true),
+  checkOutput: z.boolean().optional(),
   config: BanWordsConfig$outboundSchema.optional(),
+}).transform((v) => {
+  return remap$(v, {
+    checkOutput: "check_output",
+  });
 });
 
 export function banWordsGuardrailToJSON(

package/src/models/components/deleteguardrailsresponse.ts

@@ -0,0 +1,74 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type DeleteGuardrailsResponse = {
+  object: "guardrails_settings";
+  /**
+   * Whether a policy row was deleted
+   */
+  deleted: boolean;
+  /**
+   * Team the deleted policy targeted; null for the org-wide policy
+   */
+  teamId: string | null;
+};
+
+/** @internal */
+export const DeleteGuardrailsResponse$inboundSchema: z.ZodType<
+  DeleteGuardrailsResponse,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  object: z.literal("guardrails_settings"),
+  deleted: z.boolean(),
+  team_id: z.nullable(z.string()),
+}).transform((v) => {
+  return remap$(v, {
+    "team_id": "teamId",
+  });
+});
+/** @internal */
+export type DeleteGuardrailsResponse$Outbound = {
+  object: "guardrails_settings";
+  deleted: boolean;
+  team_id: string | null;
+};
+
+/** @internal */
+export const DeleteGuardrailsResponse$outboundSchema: z.ZodType<
+  DeleteGuardrailsResponse$Outbound,
+  z.ZodTypeDef,
+  DeleteGuardrailsResponse
+> = z.object({
+  object: z.literal("guardrails_settings"),
+  deleted: z.boolean(),
+  teamId: z.nullable(z.string()),
+}).transform((v) => {
+  return remap$(v, {
+    teamId: "team_id",
+  });
+});
+
+export function deleteGuardrailsResponseToJSON(
+  deleteGuardrailsResponse: DeleteGuardrailsResponse,
+): string {
+  return JSON.stringify(
+    DeleteGuardrailsResponse$outboundSchema.parse(deleteGuardrailsResponse),
+  );
+}
+export function deleteGuardrailsResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<DeleteGuardrailsResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => DeleteGuardrailsResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'DeleteGuardrailsResponse' from JSON`,
+  );
+}

package/src/models/components/guardrailspolicy.ts

@@ -0,0 +1,82 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  GuardrailConfig,
+  GuardrailConfig$inboundSchema,
+  GuardrailConfig$Outbound,
+  GuardrailConfig$outboundSchema,
+} from "./guardrailconfig.js";
+
+export type GuardrailsPolicy = {
+  /**
+   * Team this policy targets; null for the org-wide policy
+   */
+  teamId: string | null;
+  guardrails: Array<GuardrailConfig>;
+  /**
+   * Unix timestamp of when the policy was last updated
+   */
+  updatedAt: number;
+};
+
+/** @internal */
+export const GuardrailsPolicy$inboundSchema: z.ZodType<
+  GuardrailsPolicy,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  team_id: z.nullable(z.string()),
+  guardrails: z.array(GuardrailConfig$inboundSchema),
+  updated_at: z.number(),
+}).transform((v) => {
+  return remap$(v, {
+    "team_id": "teamId",
+    "updated_at": "updatedAt",
+  });
+});
+/** @internal */
+export type GuardrailsPolicy$Outbound = {
+  team_id: string | null;
+  guardrails: Array<GuardrailConfig$Outbound>;
+  updated_at: number;
+};
+
+/** @internal */
+export const GuardrailsPolicy$outboundSchema: z.ZodType<
+  GuardrailsPolicy$Outbound,
+  z.ZodTypeDef,
+  GuardrailsPolicy
+> = z.object({
+  teamId: z.nullable(z.string()),
+  guardrails: z.array(GuardrailConfig$outboundSchema),
+  updatedAt: z.number(),
+}).transform((v) => {
+  return remap$(v, {
+    teamId: "team_id",
+    updatedAt: "updated_at",
+  });
+});
+
+export function guardrailsPolicyToJSON(
+  guardrailsPolicy: GuardrailsPolicy,
+): string {
+  return JSON.stringify(
+    GuardrailsPolicy$outboundSchema.parse(guardrailsPolicy),
+  );
+}
+export function guardrailsPolicyFromJSON(
+  jsonString: string,
+): SafeParseResult<GuardrailsPolicy, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => GuardrailsPolicy$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GuardrailsPolicy' from JSON`,
+  );
+}

package/src/models/components/index.ts

@@ -132,6 +132,7 @@ export * from "./deletedataresponse.js";
 export * from "./deletedskill.js";
 export * from "./deletedskillversion.js";
 export * from "./deletegraphragstoreresponse.js";
+export * from "./deleteguardrailsresponse.js";
 export * from "./deletememoryentryrequest.js";
 export * from "./deletemodelpriceresponse.js";
 export * from "./deleteorgautomodelresponse.js";
@@ -237,6 +238,7 @@ export * from "./graphragqueryresult.js";
 export * from "./graphragscorebreakdown.js";
 export * from "./guardrailconfig.js";
 export * from "./guardrailmode.js";
+export * from "./guardrailspolicy.js";
 export * from "./healthcheckresponse.js";
 export * from "./historytooldefinition.js";
 export * from "./hnswpqindexconfig.js";
@@ -294,6 +296,7 @@ export * from "./listelementtype.js";
 export * from "./listeventsresponse.js";
 export * from "./listfield.js";
 export * from "./listfinetuningjobsresponse.js";
+export * from "./listguardrailspoliciesresponse.js";
 export * from "./listindicesresponse.js";
 export * from "./listmodelpricesresponse.js";
 export * from "./listorgautomodelsresponse.js";

package/src/models/components/leakageguardrail.ts

@@ -3,6 +3,7 @@
  */
 
 import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -22,6 +23,10 @@ export type LeakageGuardrail = {
    * Whether this guardrail is enabled
    */
   enabled?: boolean | undefined;
+  /**
+   * Also apply this guardrail to model output, not just the request
+   */
+  checkOutput?: boolean | undefined;
   /**
    * Configuration for system prompt leakage detection guardrail
    */
@@ -36,12 +41,18 @@ export const LeakageGuardrail$inboundSchema: z.ZodType<
 > = z.object({
   mode: z.literal("leakage"),
   enabled: z.boolean().default(true),
+  check_output: z.boolean().optional(),
   config: LeakageConfig$inboundSchema.optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "check_output": "checkOutput",
+  });
 });
 /** @internal */
 export type LeakageGuardrail$Outbound = {
   mode: "leakage";
   enabled: boolean;
+  check_output?: boolean | undefined;
   config?: LeakageConfig$Outbound | undefined;
 };
 
@@ -53,7 +64,12 @@ export const LeakageGuardrail$outboundSchema: z.ZodType<
 > = z.object({
   mode: z.literal("leakage"),
   enabled: z.boolean().default(true),
+  checkOutput: z.boolean().optional(),
   config: LeakageConfig$outboundSchema.optional(),
+}).transform((v) => {
+  return remap$(v, {
+    checkOutput: "check_output",
+  });
 });
 
 export function leakageGuardrailToJSON(