@meetkai/mka1 0.55.4 → 0.55.6

package/src/funcs/llmModelsSetOrgTokenQuota.ts

@@ -0,0 +1,182 @@
+/*
+ * 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 { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.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";
+
+/**
+ * Set an organization's token quota
+ *
+ * @remarks
+ * Cluster admins may create or replace the minute/day/month token allowance, or disable it with enabled: false. Changes apply to new admissions within 10 seconds; admitted requests finish. Limit changes preserve usage. Period changes use that period's calendar bucket; usage while disabled is not counted. The cluster organization cannot be capped.
+ *
+ * If set, this operation will use {@link Security.bearerAuth} from the global security.
+ */
+export function llmModelsSetOrgTokenQuota(
+  client: SDKCore,
+  request: operations.SetOrgTokenQuotaRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    operations.SetOrgTokenQuotaResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.SetOrgTokenQuotaRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      operations.SetOrgTokenQuotaResponseBody,
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) => operations.SetOrgTokenQuotaRequest$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 = {
+    org_id: encodeSimple("org_id", payload.org_id, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc("/api/v1/llm/models/orgs/{org_id}/token-quota")(
+    pathParams,
+  );
+
+  const headers = new Headers(compactMap({
+    "Content-Type": "application/json",
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.bearerAuth);
+  const securityInput = secConfig == null ? {} : { bearerAuth: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput, [0]);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "setOrgTokenQuota",
+    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: "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 [result] = await M.match<
+    operations.SetOrgTokenQuotaResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, operations.SetOrgTokenQuotaResponseBody$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/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.4",
+  sdkVersion: "0.55.6",
   genVersion: "2.950.2",
-  userAgent: "speakeasy-sdk/typescript 0.55.4 2.950.2 1.1.0 @meetkai/mka1",
+  userAgent: "speakeasy-sdk/typescript 0.55.6 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.4",
+    currentVersion: "0.55.6",
   },
 });
 

package/src/mcp-server/server.ts

@@ -333,6 +333,7 @@ import { tool$llmModelsDeleteCatalogEntry } from "./tools/llmModelsDeleteCatalog
 import { tool$llmModelsDeleteOrgAutoModel } from "./tools/llmModelsDeleteOrgAutoModel.js";
 import { tool$llmModelsGet } from "./tools/llmModelsGet.js";
 import { tool$llmModelsGetClusterCurrency } from "./tools/llmModelsGetClusterCurrency.js";
+import { tool$llmModelsGetOrgTokenQuota } from "./tools/llmModelsGetOrgTokenQuota.js";
 import { tool$llmModelsGetQuota } from "./tools/llmModelsGetQuota.js";
 import { tool$llmModelsList } from "./tools/llmModelsList.js";
 import { tool$llmModelsListCatalog } from "./tools/llmModelsListCatalog.js";
@@ -348,6 +349,7 @@ import { tool$llmModelsReplaceCatalogOrgAccess } from "./tools/llmModelsReplaceC
 import { tool$llmModelsSetClusterCurrency } from "./tools/llmModelsSetClusterCurrency.js";
 import { tool$llmModelsSetClusterPrice } from "./tools/llmModelsSetClusterPrice.js";
 import { tool$llmModelsSetOrgPrice } from "./tools/llmModelsSetOrgPrice.js";
+import { tool$llmModelsSetOrgTokenQuota } from "./tools/llmModelsSetOrgTokenQuota.js";
 import { tool$llmModelsUpdateCatalogEntry } from "./tools/llmModelsUpdateCatalogEntry.js";
 import { tool$llmPromptsCreate } from "./tools/llmPromptsCreate.js";
 import { tool$llmPromptsCreateVersion } from "./tools/llmPromptsCreateVersion.js";
@@ -507,7 +509,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.55.4",
+    version: "0.55.6",
   });
 
   const client = new SDKCore({
@@ -808,6 +810,8 @@ export function createMCPServer(deps: {
   tool(tool$llmModelsList);
   tool(tool$llmModelsGet);
   tool(tool$llmModelsGetQuota);
+  tool(tool$llmModelsGetOrgTokenQuota);
+  tool(tool$llmModelsSetOrgTokenQuota);
   tool(tool$llmModelsListCatalog);
   tool(tool$llmModelsAddCatalogEntry);
   tool(tool$llmModelsUpdateCatalogEntry);

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

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsGetOrgTokenQuota } from "../../funcs/llmModelsGetOrgTokenQuota.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.GetOrgTokenQuotaRequest$inboundSchema,
+};
+
+export const tool$llmModelsGetOrgTokenQuota: ToolDefinition<typeof args> = {
+  name: "llm-models-get-org-token-quota",
+  description: `Get an organization's token quota
+
+Cluster-admin read of the aggregate token policy across all models. An absent or disabled quota returns enabled: false with null limit and period. Does not require a model or change the active organization.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsGetOrgTokenQuota(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.mcpReq.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/llmModelsSetOrgTokenQuota.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsSetOrgTokenQuota } from "../../funcs/llmModelsSetOrgTokenQuota.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.SetOrgTokenQuotaRequest$inboundSchema,
+};
+
+export const tool$llmModelsSetOrgTokenQuota: ToolDefinition<typeof args> = {
+  name: "llm-models-set-org-token-quota",
+  description: `Set an organization's token quota
+
+Cluster admins may create or replace the minute/day/month token allowance, or disable it with enabled: false. Changes apply to new admissions within 10 seconds; admitted requests finish. Limit changes preserve usage. Period changes use that period's calendar bucket; usage while disabled is not counted. The cluster organization cannot be capped.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsSetOrgTokenQuota(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.mcpReq.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/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",
   });
 });
 

package/src/models/components/classifierguardrail.ts

@@ -13,6 +13,11 @@ import {
   ClassifierConfig$Outbound,
   ClassifierConfig$outboundSchema,
 } from "./classifierconfig.js";
+import {
+  GuardrailInputSource,
+  GuardrailInputSource$inboundSchema,
+  GuardrailInputSource$outboundSchema,
+} from "./guardrailinputsource.js";
 
 export type ClassifierGuardrail = {
   mode: "classifier";
@@ -24,6 +29,10 @@ export type ClassifierGuardrail = {
    * 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;
   config: ClassifierConfig;
 };
 
@@ -36,10 +45,12 @@ export const ClassifierGuardrail$inboundSchema: z.ZodType<
   mode: z.literal("classifier"),
   enabled: z.boolean().default(true),
   check_output: z.boolean().optional(),
+  input_sources: z.array(GuardrailInputSource$inboundSchema).optional(),
   config: ClassifierConfig$inboundSchema,
 }).transform((v) => {
   return remap$(v, {
     "check_output": "checkOutput",
+    "input_sources": "inputSources",
   });
 });
 /** @internal */
@@ -47,6 +58,7 @@ export type ClassifierGuardrail$Outbound = {
   mode: "classifier";
   enabled: boolean;
   check_output?: boolean | undefined;
+  input_sources?: Array<string> | undefined;
   config: ClassifierConfig$Outbound;
 };
 
@@ -59,10 +71,12 @@ export const ClassifierGuardrail$outboundSchema: z.ZodType<
   mode: z.literal("classifier"),
   enabled: z.boolean().default(true),
   checkOutput: z.boolean().optional(),
+  inputSources: z.array(GuardrailInputSource$outboundSchema).optional(),
   config: ClassifierConfig$outboundSchema,
 }).transform((v) => {
   return remap$(v, {
     checkOutput: "check_output",
+    inputSources: "input_sources",
   });
 });
 

package/src/models/components/clusterdefaultrulesummary.ts

@@ -7,6 +7,11 @@ 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 {
+  GuardrailInputSource,
+  GuardrailInputSource$inboundSchema,
+  GuardrailInputSource$outboundSchema,
+} from "./guardrailinputsource.js";
 import {
   GuardrailMode,
   GuardrailMode$inboundSchema,
@@ -23,6 +28,10 @@ export type ClusterDefaultRuleSummary = {
    * Whether this guardrail also checks model output
    */
   checkOutput?: boolean | undefined;
+  /**
+   * Request sources this guardrail checks
+   */
+  inputSources: Array<GuardrailInputSource>;
   /**
    * The message returned when this guardrail blocks a request
    */
@@ -38,10 +47,12 @@ export const ClusterDefaultRuleSummary$inboundSchema: z.ZodType<
   mode: GuardrailMode$inboundSchema,
   enabled: z.boolean(),
   check_output: z.boolean().optional(),
+  input_sources: z.array(GuardrailInputSource$inboundSchema),
   rejection_message: z.string().optional(),
 }).transform((v) => {
   return remap$(v, {
     "check_output": "checkOutput",
+    "input_sources": "inputSources",
     "rejection_message": "rejectionMessage",
   });
 });
@@ -50,6 +61,7 @@ export type ClusterDefaultRuleSummary$Outbound = {
   mode: string;
   enabled: boolean;
   check_output?: boolean | undefined;
+  input_sources: Array<string>;
   rejection_message?: string | undefined;
 };
 
@@ -62,10 +74,12 @@ export const ClusterDefaultRuleSummary$outboundSchema: z.ZodType<
   mode: GuardrailMode$outboundSchema,
   enabled: z.boolean(),
   checkOutput: z.boolean().optional(),
+  inputSources: z.array(GuardrailInputSource$outboundSchema),
   rejectionMessage: z.string().optional(),
 }).transform((v) => {
   return remap$(v, {
     checkOutput: "check_output",
+    inputSources: "input_sources",
     rejectionMessage: "rejection_message",
   });
 });

package/src/models/components/guardrailinputsource.ts

@@ -0,0 +1,32 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { ClosedEnum } from "../../types/enums.js";
+
+/**
+ * A source of model input: user (user messages and text sent to a sleeping response), system (system and developer messages, instructions and compaction summaries), assistant (earlier assistant messages), tool (function, custom tool, MCP, shell and code interpreter results), retrieval (file search, web search, web page, uploaded file and history search results), image (text read from images in messages) or audio (transcripts of audio in messages)
+ */
+export const GuardrailInputSource = {
+  User: "user",
+  System: "system",
+  Assistant: "assistant",
+  Tool: "tool",
+  Retrieval: "retrieval",
+  Image: "image",
+  Audio: "audio",
+} as const;
+/**
+ * A source of model input: user (user messages and text sent to a sleeping response), system (system and developer messages, instructions and compaction summaries), assistant (earlier assistant messages), tool (function, custom tool, MCP, shell and code interpreter results), retrieval (file search, web search, web page, uploaded file and history search results), image (text read from images in messages) or audio (transcripts of audio in messages)
+ */
+export type GuardrailInputSource = ClosedEnum<typeof GuardrailInputSource>;
+
+/** @internal */
+export const GuardrailInputSource$inboundSchema: z.ZodNativeEnum<
+  typeof GuardrailInputSource
+> = z.nativeEnum(GuardrailInputSource);
+/** @internal */
+export const GuardrailInputSource$outboundSchema: z.ZodNativeEnum<
+  typeof GuardrailInputSource
+> = GuardrailInputSource$inboundSchema;

package/src/models/components/index.ts

@@ -404,6 +404,7 @@ export * from "./grammarformat.js";
 export * from "./grantablevectorstorepermissionrole.js";
 export * from "./grantpermissionrequest.js";
 export * from "./guardrailconfig.js";
+export * from "./guardrailinputsource.js";
 export * from "./guardrailmode.js";
 export * from "./guardrailspolicy.js";
 export * from "./historytooldefinition.js";

package/src/models/components/leakageguardrail.ts

@@ -7,6 +7,11 @@ 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 {
+  GuardrailInputSource,
+  GuardrailInputSource$inboundSchema,
+  GuardrailInputSource$outboundSchema,
+} from "./guardrailinputsource.js";
 import {
   LeakageConfig,
   LeakageConfig$inboundSchema,
@@ -27,6 +32,10 @@ export type LeakageGuardrail = {
    * 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 system prompt leakage detection guardrail
    */
@@ -42,10 +51,12 @@ export const LeakageGuardrail$inboundSchema: z.ZodType<
   mode: z.literal("leakage"),
   enabled: z.boolean().default(true),
   check_output: z.boolean().optional(),
+  input_sources: z.array(GuardrailInputSource$inboundSchema).optional(),
   config: LeakageConfig$inboundSchema.optional(),
 }).transform((v) => {
   return remap$(v, {
     "check_output": "checkOutput",
+    "input_sources": "inputSources",
   });
 });
 /** @internal */
@@ -53,6 +64,7 @@ export type LeakageGuardrail$Outbound = {
   mode: "leakage";
   enabled: boolean;
   check_output?: boolean | undefined;
+  input_sources?: Array<string> | undefined;
   config?: LeakageConfig$Outbound | undefined;
 };
 
@@ -65,10 +77,12 @@ export const LeakageGuardrail$outboundSchema: z.ZodType<
   mode: z.literal("leakage"),
   enabled: z.boolean().default(true),
   checkOutput: z.boolean().optional(),
+  inputSources: z.array(GuardrailInputSource$outboundSchema).optional(),
   config: LeakageConfig$outboundSchema.optional(),
 }).transform((v) => {
   return remap$(v, {
     checkOutput: "check_output",
+    inputSources: "input_sources",
   });
 });
 

package/src/models/components/promptinjectionguardrail.ts

@@ -7,6 +7,11 @@ 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 {
+  GuardrailInputSource,
+  GuardrailInputSource$inboundSchema,
+  GuardrailInputSource$outboundSchema,
+} from "./guardrailinputsource.js";
 import {
   PromptInjectionConfig,
   PromptInjectionConfig$inboundSchema,
@@ -27,6 +32,10 @@ export type PromptInjectionGuardrail = {
    * 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 prompt injection detection guardrail
    */
@@ -42,10 +51,12 @@ export const PromptInjectionGuardrail$inboundSchema: z.ZodType<
   mode: z.literal("prompt_injection"),
   enabled: z.boolean().default(true),
   check_output: z.boolean().optional(),
+  input_sources: z.array(GuardrailInputSource$inboundSchema).optional(),
   config: PromptInjectionConfig$inboundSchema.optional(),
 }).transform((v) => {
   return remap$(v, {
     "check_output": "checkOutput",
+    "input_sources": "inputSources",
   });
 });
 /** @internal */
@@ -53,6 +64,7 @@ export type PromptInjectionGuardrail$Outbound = {
   mode: "prompt_injection";
   enabled: boolean;
   check_output?: boolean | undefined;
+  input_sources?: Array<string> | undefined;
   config?: PromptInjectionConfig$Outbound | undefined;
 };
 
@@ -65,10 +77,12 @@ export const PromptInjectionGuardrail$outboundSchema: z.ZodType<
   mode: z.literal("prompt_injection"),
   enabled: z.boolean().default(true),
   checkOutput: z.boolean().optional(),
+  inputSources: z.array(GuardrailInputSource$outboundSchema).optional(),
   config: PromptInjectionConfig$outboundSchema.optional(),
 }).transform((v) => {
   return remap$(v, {
     checkOutput: "check_output",
+    inputSources: "input_sources",
   });
 });
 

package/src/models/components/testguardrailsrequest.ts

@@ -6,12 +6,21 @@ import * as z from "zod/v3";
 import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  GuardrailInputSource,
+  GuardrailInputSource$inboundSchema,
+  GuardrailInputSource$outboundSchema,
+} from "./guardrailinputsource.js";
 
 export type TestGuardrailsRequest = {
   /**
    * Content to test against guardrails
    */
   content: string;
+  /**
+   * A source of model input: user (user messages and text sent to a sleeping response), system (system and developer messages, instructions and compaction summaries), assistant (earlier assistant messages), tool (function, custom tool, MCP, shell and code interpreter results), retrieval (file search, web search, web page, uploaded file and history search results), image (text read from images in messages) or audio (transcripts of audio in messages)
+   */
+  source?: GuardrailInputSource | undefined;
 };
 
 /** @internal */
@@ -21,10 +30,12 @@ export const TestGuardrailsRequest$inboundSchema: z.ZodType<
   unknown
 > = z.object({
   content: z.string(),
+  source: GuardrailInputSource$inboundSchema.optional(),
 });
 /** @internal */
 export type TestGuardrailsRequest$Outbound = {
   content: string;
+  source?: string | undefined;
 };
 
 /** @internal */
@@ -34,6 +45,7 @@ export const TestGuardrailsRequest$outboundSchema: z.ZodType<
   TestGuardrailsRequest
 > = z.object({
   content: z.string(),
+  source: GuardrailInputSource$outboundSchema.optional(),
 });
 
 export function testGuardrailsRequestToJSON(