@meetkai/mka1 0.55.5 → 0.55.8

package/src/funcs/authOrgsGetInboundRules.ts

@@ -0,0 +1,248 @@
+/*
+ * 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 { resolveSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.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";
+
+/**
+ * Get an org's inbound IP rules
+ *
+ * @remarks
+ * Source-IP allowlist for the org's API keys. `available` mirrors the cluster's INBOUND_RULES_ENABLED flag. Rules are ordered by createdAt, then id.
+ */
+export function authOrgsGetInboundRules(
+  client: SDKCore,
+  security: operations.GetOrgInboundRulesSecurity,
+  request: operations.GetOrgInboundRulesRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    operations.GetOrgInboundRulesResponseBody,
+    | errors.GetOrgInboundRulesResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponseResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse404ResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse409ResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse410ResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse429ResponseBody
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    security,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  security: operations.GetOrgInboundRulesSecurity,
+  request: operations.GetOrgInboundRulesRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      operations.GetOrgInboundRulesResponseBody,
+      | errors.GetOrgInboundRulesResponseBody
+      | errors.GetOrgInboundRulesAuthOrgsResponseBody
+      | errors.GetOrgInboundRulesAuthOrgsResponseResponseBody
+      | errors.GetOrgInboundRulesAuthOrgsResponse404ResponseBody
+      | errors.GetOrgInboundRulesAuthOrgsResponse409ResponseBody
+      | errors.GetOrgInboundRulesAuthOrgsResponse410ResponseBody
+      | errors.GetOrgInboundRulesAuthOrgsResponse429ResponseBody
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) => operations.GetOrgInboundRulesRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = null;
+
+  const pathParams = {
+    orgId: encodeSimple("orgId", payload.orgId, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc("/api/v1/authentication/orgs/{orgId}/inbound-rules")(
+    pathParams,
+  );
+
+  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 requestSecurity = resolveSecurity(
+    [
+      {
+        fieldName: "Authorization",
+        type: "http:bearer",
+        value: security?.bearerAuth,
+      },
+    ],
+    [
+      {
+        fieldName: "__Secure-mka1-auth-v2.session_token",
+        type: "apiKey:cookie",
+        value: security?.sessionCookie,
+      },
+    ],
+  );
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "getOrgInboundRules",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: security,
+    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 responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    operations.GetOrgInboundRulesResponseBody,
+    | errors.GetOrgInboundRulesResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponseResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse404ResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse409ResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse410ResponseBody
+    | errors.GetOrgInboundRulesAuthOrgsResponse429ResponseBody
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, operations.GetOrgInboundRulesResponseBody$inboundSchema),
+    M.jsonErr(400, errors.GetOrgInboundRulesResponseBody$inboundSchema),
+    M.jsonErr(401, errors.GetOrgInboundRulesAuthOrgsResponseBody$inboundSchema),
+    M.jsonErr(
+      403,
+      errors.GetOrgInboundRulesAuthOrgsResponseResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      404,
+      errors.GetOrgInboundRulesAuthOrgsResponse404ResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      409,
+      errors.GetOrgInboundRulesAuthOrgsResponse409ResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      410,
+      errors.GetOrgInboundRulesAuthOrgsResponse410ResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      429,
+      errors.GetOrgInboundRulesAuthOrgsResponse429ResponseBody$inboundSchema,
+    ),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/authOrgsUpdateInboundRules.ts

@@ -0,0 +1,246 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { SDKCore } from "../core.js";
+import { encodeJSON, 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 { resolveSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.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";
+
+/**
+ * Replace an org's inbound IP rules (org owner/admin, browser session)
+ *
+ * @remarks
+ * Replaces the whole policy atomically. Each cidr is normalized (bare 0.0.0.0 / :: mean /0, other bare addresses are /32 or /128, host bits are cleared); duplicates after normalization are a 400. A request carrying an Authorization header is refused with 403.
+ */
+export function authOrgsUpdateInboundRules(
+  client: SDKCore,
+  security: operations.UpdateOrgInboundRulesSecurity,
+  request: operations.UpdateOrgInboundRulesRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    operations.UpdateOrgInboundRulesResponseBody,
+    | errors.UpdateOrgInboundRulesResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponseResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse404ResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse409ResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse410ResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse429ResponseBody
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    security,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  security: operations.UpdateOrgInboundRulesSecurity,
+  request: operations.UpdateOrgInboundRulesRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      operations.UpdateOrgInboundRulesResponseBody,
+      | errors.UpdateOrgInboundRulesResponseBody
+      | errors.UpdateOrgInboundRulesAuthOrgsResponseBody
+      | errors.UpdateOrgInboundRulesAuthOrgsResponseResponseBody
+      | errors.UpdateOrgInboundRulesAuthOrgsResponse404ResponseBody
+      | errors.UpdateOrgInboundRulesAuthOrgsResponse409ResponseBody
+      | errors.UpdateOrgInboundRulesAuthOrgsResponse410ResponseBody
+      | errors.UpdateOrgInboundRulesAuthOrgsResponse429ResponseBody
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) =>
+      operations.UpdateOrgInboundRulesRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = encodeJSON("body", payload.RequestBody, { explode: true });
+
+  const pathParams = {
+    orgId: encodeSimple("orgId", payload.orgId, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc("/api/v1/authentication/orgs/{orgId}/inbound-rules")(
+    pathParams,
+  );
+
+  const headers = new Headers(compactMap({
+    "Content-Type": "application/json",
+    Accept: "application/json",
+    "X-On-Behalf-Of": encodeSimple(
+      "X-On-Behalf-Of",
+      payload["X-On-Behalf-Of"],
+      { explode: false, charEncoding: "none" },
+    ),
+  }));
+
+  const requestSecurity = resolveSecurity(
+    [
+      {
+        fieldName: "__Secure-mka1-auth-v2.session_token",
+        type: "apiKey:cookie",
+        value: security?.sessionCookie,
+      },
+    ],
+  );
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "updateOrgInboundRules",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: security,
+    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: "PUT",
+    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 responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    operations.UpdateOrgInboundRulesResponseBody,
+    | errors.UpdateOrgInboundRulesResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponseResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse404ResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse409ResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse410ResponseBody
+    | errors.UpdateOrgInboundRulesAuthOrgsResponse429ResponseBody
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, operations.UpdateOrgInboundRulesResponseBody$inboundSchema),
+    M.jsonErr(400, errors.UpdateOrgInboundRulesResponseBody$inboundSchema),
+    M.jsonErr(
+      401,
+      errors.UpdateOrgInboundRulesAuthOrgsResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      403,
+      errors.UpdateOrgInboundRulesAuthOrgsResponseResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      404,
+      errors.UpdateOrgInboundRulesAuthOrgsResponse404ResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      409,
+      errors.UpdateOrgInboundRulesAuthOrgsResponse409ResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      410,
+      errors.UpdateOrgInboundRulesAuthOrgsResponse410ResponseBody$inboundSchema,
+    ),
+    M.jsonErr(
+      429,
+      errors.UpdateOrgInboundRulesAuthOrgsResponse429ResponseBody$inboundSchema,
+    ),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/llmFilesList.ts

@@ -96,6 +96,7 @@ async function $do(
     "limit": payload.limit,
     "order": payload.order,
     "purpose": payload.purpose,
+    "sha256": payload.sha256,
   });
 
   const headers = new Headers(compactMap({

package/src/funcs/llmVectorStoresCreate.ts

@@ -30,7 +30,7 @@ import { Result } from "../types/fp.js";
  * Create a vector store
  *
  * @remarks
- * Creates a new vector store for storing and searching through document embeddings.
+ * Creates a new vector store for storing and searching through document embeddings. With file_ids, two files of the same content in the request are refused with 409 before the store is created.
  *
  * If set, this operation will use {@link Security.bearerAuth} from the global security.
  */

package/src/funcs/llmVectorStoresCreateFile.ts

@@ -30,7 +30,7 @@ import { Result } from "../types/fp.js";
  * Add a file to a vector store
  *
  * @remarks
- * Adds a file to a vector store for semantic search indexing. Returns 410 once the vector store has expired.
+ * Adds a file to a vector store for semantic search indexing. Returns 409 when the store already holds a file with the same content (the error names that file; the store indexes each content once), 410 once the vector store has expired.
  *
  * If set, this operation will use {@link Security.bearerAuth} from the global security.
  */

package/src/funcs/llmVectorStoresCreateFileBatch.ts

@@ -30,7 +30,7 @@ import { Result } from "../types/fp.js";
  * Batch add multiple files to vector store
  *
  * @remarks
- * Adds multiple files to a vector store in a single batch operation. Returns 410 once the vector store has expired.
+ * Adds multiple files to a vector store in a single batch operation. A file with the same content as one the store already holds, or as an earlier file in the batch, fails the batch with 409 unless skip_duplicates is set, in which case it is recorded as cancelled with duplicate_file and the rest are attached. Returns 410 once the vector store has expired.
  *
  * If set, this operation will use {@link Security.bearerAuth} from the global security.
  */

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.55.5",
+  sdkVersion: "0.55.8",
   genVersion: "2.950.2",
-  userAgent: "speakeasy-sdk/typescript 0.55.5 2.950.2 1.1.0 @meetkai/mka1",
+  userAgent: "speakeasy-sdk/typescript 0.55.8 2.950.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.55.5",
+    currentVersion: "0.55.8",
   },
 });
 

package/src/mcp-server/server.ts

@@ -509,7 +509,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.55.5",
+    version: "0.55.8",
   });
 
   const client = new SDKCore({

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

@@ -14,7 +14,7 @@ export const tool$llmVectorStoresCreate: ToolDefinition<typeof args> = {
   name: "llm-vector-stores-create",
   description: `Create a vector store
 
-Creates a new vector store for storing and searching through document embeddings.`,
+Creates a new vector store for storing and searching through document embeddings. With file_ids, two files of the same content in the request are refused with 409 before the store is created.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await llmVectorStoresCreate(

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

@@ -14,7 +14,7 @@ export const tool$llmVectorStoresCreateFile: ToolDefinition<typeof args> = {
   name: "llm-vector-stores-create-file",
   description: `Add a file to a vector store
 
-Adds a file to a vector store for semantic search indexing. Returns 410 once the vector store has expired.`,
+Adds a file to a vector store for semantic search indexing. Returns 409 when the store already holds a file with the same content (the error names that file; the store indexes each content once), 410 once the vector store has expired.`,
   args,
   tool: async (client, args, ctx) => {
     const [result, apiCall] = await llmVectorStoresCreateFile(

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

@@ -15,7 +15,7 @@ export const tool$llmVectorStoresCreateFileBatch: ToolDefinition<typeof args> =
     name: "llm-vector-stores-create-file-batch",
     description: `Batch add multiple files to vector store
 
-Adds multiple files to a vector store in a single batch operation. Returns 410 once the vector store has expired.`,
+Adds multiple files to a vector store in a single batch operation. A file with the same content as one the store already holds, or as an earlier file in the batch, fails the batch with 409 unless skip_duplicates is set, in which case it is recorded as cancelled with duplicate_file and the rest are attached. Returns 410 once the vector store has expired.`,
     args,
     tool: async (client, args, ctx) => {
       const [result, apiCall] = await llmVectorStoresCreateFileBatch(

package/src/models/components/banwordsguardrail.ts

@@ -13,6 +13,11 @@ import {
   BanWordsConfig$Outbound,
   BanWordsConfig$outboundSchema,
 } from "./banwordsconfig.js";
+import {
+  GuardrailInputSource,
+  GuardrailInputSource$inboundSchema,
+  GuardrailInputSource$outboundSchema,
+} from "./guardrailinputsource.js";
 
 /**
  * Banned words guardrail
@@ -27,6 +32,10 @@ export type BanWordsGuardrail = {
    * Also apply this guardrail to model output, not just the request
    */
   checkOutput?: boolean | undefined;
+  /**
+   * Request sources this guardrail checks before they reach the model. Omit to use the mode's default: every source for classifier; every source except system for ban_words, since instructions often name the terms they forbid; only what the user sends for prompt_injection and leakage, whose keyword patterns match ordinary instructions, replies and documents. An empty list checks no input, so the rule only checks output when check_output is set. Rules saved before this field existed were set to user messages only.
+   */
+  inputSources?: Array<GuardrailInputSource> | undefined;
   /**
    * Configuration for banned words guardrail
    */
@@ -42,10 +51,12 @@ export const BanWordsGuardrail$inboundSchema: z.ZodType<
   mode: z.literal("ban_words"),
   enabled: z.boolean().default(true),
   check_output: z.boolean().optional(),
+  input_sources: z.array(GuardrailInputSource$inboundSchema).optional(),
   config: BanWordsConfig$inboundSchema.optional(),
 }).transform((v) => {
   return remap$(v, {
     "check_output": "checkOutput",
+    "input_sources": "inputSources",
   });
 });
 /** @internal */
@@ -53,6 +64,7 @@ export type BanWordsGuardrail$Outbound = {
   mode: "ban_words";
   enabled: boolean;
   check_output?: boolean | undefined;
+  input_sources?: Array<string> | undefined;
   config?: BanWordsConfig$Outbound | undefined;
 };
 
@@ -65,10 +77,12 @@ export const BanWordsGuardrail$outboundSchema: z.ZodType<
   mode: z.literal("ban_words"),
   enabled: z.boolean().default(true),
   checkOutput: z.boolean().optional(),
+  inputSources: z.array(GuardrailInputSource$outboundSchema).optional(),
   config: BanWordsConfig$outboundSchema.optional(),
 }).transform((v) => {
   return remap$(v, {
     checkOutput: "check_output",
+    inputSources: "input_sources",
   });
 });