@meetkai/mka1 0.55.5 → 0.55.6

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(