@meetkai/mka1 0.48.26 → 0.48.28

package/src/models/components/listguardrailspoliciesresponse.ts

@@ -0,0 +1,63 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+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 {
+  GuardrailsPolicy,
+  GuardrailsPolicy$inboundSchema,
+  GuardrailsPolicy$Outbound,
+  GuardrailsPolicy$outboundSchema,
+} from "./guardrailspolicy.js";
+
+export type ListGuardrailsPoliciesResponse = {
+  object: "list";
+  data: Array<GuardrailsPolicy>;
+};
+
+/** @internal */
+export const ListGuardrailsPoliciesResponse$inboundSchema: z.ZodType<
+  ListGuardrailsPoliciesResponse,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  object: z.literal("list"),
+  data: z.array(GuardrailsPolicy$inboundSchema),
+});
+/** @internal */
+export type ListGuardrailsPoliciesResponse$Outbound = {
+  object: "list";
+  data: Array<GuardrailsPolicy$Outbound>;
+};
+
+/** @internal */
+export const ListGuardrailsPoliciesResponse$outboundSchema: z.ZodType<
+  ListGuardrailsPoliciesResponse$Outbound,
+  z.ZodTypeDef,
+  ListGuardrailsPoliciesResponse
+> = z.object({
+  object: z.literal("list"),
+  data: z.array(GuardrailsPolicy$outboundSchema),
+});
+
+export function listGuardrailsPoliciesResponseToJSON(
+  listGuardrailsPoliciesResponse: ListGuardrailsPoliciesResponse,
+): string {
+  return JSON.stringify(
+    ListGuardrailsPoliciesResponse$outboundSchema.parse(
+      listGuardrailsPoliciesResponse,
+    ),
+  );
+}
+export function listGuardrailsPoliciesResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ListGuardrailsPoliciesResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ListGuardrailsPoliciesResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ListGuardrailsPoliciesResponse' from JSON`,
+  );
+}

package/src/models/components/promptinjectionguardrail.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 PromptInjectionGuardrail = {
    * Whether this guardrail is enabled
    */
   enabled?: boolean | undefined;
+  /**
+   * Also apply this guardrail to model output, not just the request
+   */
+  checkOutput?: boolean | undefined;
   /**
    * Configuration for prompt injection detection guardrail
    */
@@ -36,12 +41,18 @@ export const PromptInjectionGuardrail$inboundSchema: z.ZodType<
 > = z.object({
   mode: z.literal("prompt_injection"),
   enabled: z.boolean().default(true),
+  check_output: z.boolean().optional(),
   config: PromptInjectionConfig$inboundSchema.optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "check_output": "checkOutput",
+  });
 });
 /** @internal */
 export type PromptInjectionGuardrail$Outbound = {
   mode: "prompt_injection";
   enabled: boolean;
+  check_output?: boolean | undefined;
   config?: PromptInjectionConfig$Outbound | undefined;
 };
 
@@ -53,7 +64,12 @@ export const PromptInjectionGuardrail$outboundSchema: z.ZodType<
 > = z.object({
   mode: z.literal("prompt_injection"),
   enabled: z.boolean().default(true),
+  checkOutput: z.boolean().optional(),
   config: PromptInjectionConfig$outboundSchema.optional(),
+}).transform((v) => {
+  return remap$(v, {
+    checkOutput: "check_output",
+  });
 });
 
 export function promptInjectionGuardrailToJSON(

package/src/models/components/updateguardrailsrequest.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";
@@ -18,6 +19,10 @@ export type UpdateGuardrailsRequest = {
    * List of guardrails to configure
    */
   guardrails: Array<GuardrailConfig>;
+  /**
+   * Target team for this policy. Omit or pass null to set the org-wide policy that applies to every team (org admin only). Non-admin callers may only target their own team.
+   */
+  teamId?: string | null | undefined;
 };
 
 /** @internal */
@@ -27,10 +32,16 @@ export const UpdateGuardrailsRequest$inboundSchema: z.ZodType<
   unknown
 > = z.object({
   guardrails: z.array(GuardrailConfig$inboundSchema),
+  team_id: z.nullable(z.string()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "team_id": "teamId",
+  });
 });
 /** @internal */
 export type UpdateGuardrailsRequest$Outbound = {
   guardrails: Array<GuardrailConfig$Outbound>;
+  team_id?: string | null | undefined;
 };
 
 /** @internal */
@@ -40,6 +51,11 @@ export const UpdateGuardrailsRequest$outboundSchema: z.ZodType<
   UpdateGuardrailsRequest
 > = z.object({
   guardrails: z.array(GuardrailConfig$outboundSchema),
+  teamId: z.nullable(z.string()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    teamId: "team_id",
+  });
 });
 
 export function updateGuardrailsRequestToJSON(

package/src/models/components/updateguardrailsresponse.ts

@@ -18,7 +18,11 @@ export type UpdateGuardrailsResponse = {
   object: "guardrails_settings";
   guardrails: Array<GuardrailConfig>;
   /**
-   * Unix timestamp of when settings were updated
+   * Team this policy targets; null for the org-wide policy
+   */
+  teamId: string | null;
+  /**
+   * Unix timestamp of when the policy was updated
    */
   updatedAt: number;
 };
@@ -31,9 +35,11 @@ export const UpdateGuardrailsResponse$inboundSchema: z.ZodType<
 > = z.object({
   object: z.literal("guardrails_settings"),
   guardrails: z.array(GuardrailConfig$inboundSchema),
+  team_id: z.nullable(z.string()),
   updated_at: z.number(),
 }).transform((v) => {
   return remap$(v, {
+    "team_id": "teamId",
     "updated_at": "updatedAt",
   });
 });
@@ -41,6 +47,7 @@ export const UpdateGuardrailsResponse$inboundSchema: z.ZodType<
 export type UpdateGuardrailsResponse$Outbound = {
   object: "guardrails_settings";
   guardrails: Array<GuardrailConfig$Outbound>;
+  team_id: string | null;
   updated_at: number;
 };
 
@@ -52,9 +59,11 @@ export const UpdateGuardrailsResponse$outboundSchema: z.ZodType<
 > = z.object({
   object: z.literal("guardrails_settings"),
   guardrails: z.array(GuardrailConfig$outboundSchema),
+  teamId: z.nullable(z.string()),
   updatedAt: z.number(),
 }).transform((v) => {
   return remap$(v, {
+    teamId: "team_id",
     updatedAt: "updated_at",
   });
 });

package/src/models/operations/deleteguardrails.ts

@@ -0,0 +1,69 @@
+/*
+ * 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 DeleteGuardrailsRequest = {
+  teamId?: string | undefined;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const DeleteGuardrailsRequest$inboundSchema: z.ZodType<
+  DeleteGuardrailsRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  team_id: z.string().optional(),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "team_id": "teamId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type DeleteGuardrailsRequest$Outbound = {
+  team_id?: string | undefined;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const DeleteGuardrailsRequest$outboundSchema: z.ZodType<
+  DeleteGuardrailsRequest$Outbound,
+  z.ZodTypeDef,
+  DeleteGuardrailsRequest
+> = z.object({
+  teamId: z.string().optional(),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    teamId: "team_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function deleteGuardrailsRequestToJSON(
+  deleteGuardrailsRequest: DeleteGuardrailsRequest,
+): string {
+  return JSON.stringify(
+    DeleteGuardrailsRequest$outboundSchema.parse(deleteGuardrailsRequest),
+  );
+}
+export function deleteGuardrailsRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<DeleteGuardrailsRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => DeleteGuardrailsRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'DeleteGuardrailsRequest' from JSON`,
+  );
+}

package/src/models/operations/index.ts

@@ -66,6 +66,7 @@ export * from "./deleteevalsuite.js";
 export * from "./deleteextractschema.js";
 export * from "./deletefile.js";
 export * from "./deletegraphragstore.js";
+export * from "./deleteguardrails.js";
 export * from "./deletemcpcredential.js";
 export * from "./deletemcpserver.js";
 export * from "./deletememoryentry.js";
@@ -166,6 +167,7 @@ export * from "./listfiles.js";
 export * from "./listfinetuningcheckpoints.js";
 export * from "./listfinetuningevents.js";
 export * from "./listfinetuningjobs.js";
+export * from "./listguardrailspolicies.js";
 export * from "./listindices.js";
 export * from "./listinputitems.js";
 export * from "./listmcpcredentials.js";

package/src/models/operations/listguardrailspolicies.ts

@@ -0,0 +1,65 @@
+/*
+ * 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 ListGuardrailsPoliciesRequest = {
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const ListGuardrailsPoliciesRequest$inboundSchema: z.ZodType<
+  ListGuardrailsPoliciesRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type ListGuardrailsPoliciesRequest$Outbound = {
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const ListGuardrailsPoliciesRequest$outboundSchema: z.ZodType<
+  ListGuardrailsPoliciesRequest$Outbound,
+  z.ZodTypeDef,
+  ListGuardrailsPoliciesRequest
+> = z.object({
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function listGuardrailsPoliciesRequestToJSON(
+  listGuardrailsPoliciesRequest: ListGuardrailsPoliciesRequest,
+): string {
+  return JSON.stringify(
+    ListGuardrailsPoliciesRequest$outboundSchema.parse(
+      listGuardrailsPoliciesRequest,
+    ),
+  );
+}
+export function listGuardrailsPoliciesRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<ListGuardrailsPoliciesRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ListGuardrailsPoliciesRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ListGuardrailsPoliciesRequest' from JSON`,
+  );
+}

package/src/react-query/guardrailsDeleteGuardrails.ts

@@ -0,0 +1,110 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { guardrailsDeleteGuardrails } from "../funcs/guardrailsDeleteGuardrails.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 { unwrapAsync } from "../types/fp.js";
+import { useSDKContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export type GuardrailsDeleteGuardrailsMutationVariables = {
+  request: operations.DeleteGuardrailsRequest;
+  options?: RequestOptions;
+};
+
+export type GuardrailsDeleteGuardrailsMutationData =
+  components.DeleteGuardrailsResponse;
+
+export type GuardrailsDeleteGuardrailsMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Delete guardrails policy
+ *
+ * @remarks
+ * 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.
+ */
+export function useGuardrailsDeleteGuardrailsMutation(
+  options?: MutationHookOptions<
+    GuardrailsDeleteGuardrailsMutationData,
+    GuardrailsDeleteGuardrailsMutationError,
+    GuardrailsDeleteGuardrailsMutationVariables
+  >,
+): UseMutationResult<
+  GuardrailsDeleteGuardrailsMutationData,
+  GuardrailsDeleteGuardrailsMutationError,
+  GuardrailsDeleteGuardrailsMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildGuardrailsDeleteGuardrailsMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyGuardrailsDeleteGuardrails(): MutationKey {
+  return ["@meetkai/mka1", "Guardrails", "deleteGuardrails"];
+}
+
+export function buildGuardrailsDeleteGuardrailsMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: GuardrailsDeleteGuardrailsMutationVariables,
+  ) => Promise<GuardrailsDeleteGuardrailsMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyGuardrailsDeleteGuardrails(),
+    mutationFn: function guardrailsDeleteGuardrailsMutationFn({
+      request,
+      options,
+    }): Promise<GuardrailsDeleteGuardrailsMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(guardrailsDeleteGuardrails(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/guardrailsGetGuardrails.ts

@@ -51,10 +51,10 @@ export type GuardrailsGetGuardrailsQueryError =
   | SDKValidationError;
 
 /**
- * Get guardrails settings
+ * Get effective guardrails
  *
  * @remarks
- * 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.
  */
 export function useGuardrailsGetGuardrails(
   request: operations.GetGuardrailsRequest,
@@ -78,10 +78,10 @@ export function useGuardrailsGetGuardrails(
 }
 
 /**
- * Get guardrails settings
+ * Get effective guardrails
  *
  * @remarks
- * 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.
  */
 export function useGuardrailsGetGuardrailsSuspense(
   request: operations.GetGuardrailsRequest,

package/src/react-query/guardrailsListGuardrailsPolicies.core.ts

@@ -0,0 +1,76 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { guardrailsListGuardrailsPolicies } from "../funcs/guardrailsListGuardrailsPolicies.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as components from "../models/components/index.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+export type GuardrailsListGuardrailsPoliciesQueryData =
+  components.ListGuardrailsPoliciesResponse;
+
+export function prefetchGuardrailsListGuardrailsPolicies(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListGuardrailsPoliciesRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildGuardrailsListGuardrailsPoliciesQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildGuardrailsListGuardrailsPoliciesQuery(
+  client$: SDKCore,
+  request: operations.ListGuardrailsPoliciesRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<GuardrailsListGuardrailsPoliciesQueryData>;
+} {
+  return {
+    queryKey: queryKeyGuardrailsListGuardrailsPolicies({
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function guardrailsListGuardrailsPoliciesQueryFn(
+      ctx,
+    ): Promise<GuardrailsListGuardrailsPoliciesQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(guardrailsListGuardrailsPolicies(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyGuardrailsListGuardrailsPolicies(
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "Guardrails", "listGuardrailsPolicies", parameters];
+}