@meetkai/mka1 0.55.35 → 0.55.36

package/src/models/components/guardrailsresourcetarget.ts

@@ -0,0 +1,97 @@
+/*
+ * 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 { ClosedEnum } from "../../types/enums.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+/**
+ * Saved agent or catalog model endpoint.
+ */
+export const GuardrailsResourceTargetResourceType = {
+  Agent: "agent",
+  Model: "model",
+} as const;
+/**
+ * Saved agent or catalog model endpoint.
+ */
+export type GuardrailsResourceTargetResourceType = ClosedEnum<
+  typeof GuardrailsResourceTargetResourceType
+>;
+
+export type GuardrailsResourceTarget = {
+  /**
+   * Saved agent or catalog model endpoint.
+   */
+  resourceType: GuardrailsResourceTargetResourceType;
+  /**
+   * Canonical saved-agent ID or catalog model ID. Model aliases are resolved before storage.
+   */
+  resourceId: string;
+};
+
+/** @internal */
+export const GuardrailsResourceTargetResourceType$inboundSchema:
+  z.ZodNativeEnum<typeof GuardrailsResourceTargetResourceType> = z.nativeEnum(
+    GuardrailsResourceTargetResourceType,
+  );
+/** @internal */
+export const GuardrailsResourceTargetResourceType$outboundSchema:
+  z.ZodNativeEnum<typeof GuardrailsResourceTargetResourceType> =
+    GuardrailsResourceTargetResourceType$inboundSchema;
+
+/** @internal */
+export const GuardrailsResourceTarget$inboundSchema: z.ZodType<
+  GuardrailsResourceTarget,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  resource_type: GuardrailsResourceTargetResourceType$inboundSchema,
+  resource_id: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "resource_type": "resourceType",
+    "resource_id": "resourceId",
+  });
+});
+/** @internal */
+export type GuardrailsResourceTarget$Outbound = {
+  resource_type: string;
+  resource_id: string;
+};
+
+/** @internal */
+export const GuardrailsResourceTarget$outboundSchema: z.ZodType<
+  GuardrailsResourceTarget$Outbound,
+  z.ZodTypeDef,
+  GuardrailsResourceTarget
+> = z.object({
+  resourceType: GuardrailsResourceTargetResourceType$outboundSchema,
+  resourceId: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    resourceType: "resource_type",
+    resourceId: "resource_id",
+  });
+});
+
+export function guardrailsResourceTargetToJSON(
+  guardrailsResourceTarget: GuardrailsResourceTarget,
+): string {
+  return JSON.stringify(
+    GuardrailsResourceTarget$outboundSchema.parse(guardrailsResourceTarget),
+  );
+}
+export function guardrailsResourceTargetFromJSON(
+  jsonString: string,
+): SafeParseResult<GuardrailsResourceTarget, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => GuardrailsResourceTarget$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GuardrailsResourceTarget' from JSON`,
+  );
+}

package/src/models/components/index.ts

@@ -261,6 +261,7 @@ export * from "./declaration.js";
 export * from "./deletecatalogmodelresponse.js";
 export * from "./deletedskill.js";
 export * from "./deletedskillversion.js";
+export * from "./deleteguardrailsresourcepolicyresponse.js";
 export * from "./deleteguardrailsresponse.js";
 export * from "./deletememoryentryrequest.js";
 export * from "./deletemodelpriceresponse.js";
@@ -402,6 +403,7 @@ export * from "./generatorlist.js";
 export * from "./generatorref.js";
 export * from "./getexportstatusresponse.js";
 export * from "./getfeedbackresponse.js";
+export * from "./getguardrailsresourcepolicyresponse.js";
 export * from "./getguardrailsresponse.js";
 export * from "./getpromptresponse.js";
 export * from "./githubpullrequestrequest.js";
@@ -417,6 +419,8 @@ export * from "./guardrailconfig.js";
 export * from "./guardrailinputsource.js";
 export * from "./guardrailmode.js";
 export * from "./guardrailspolicy.js";
+export * from "./guardrailsresourcepolicy.js";
+export * from "./guardrailsresourcetarget.js";
 export * from "./historytooldefinition.js";
 export * from "./hostedtoolchoice.js";
 export * from "./hostedtooldefinition.js";
@@ -778,6 +782,7 @@ export * from "./updateclusterdefaultguardrailsrequest.js";
 export * from "./updateevalschedulerequest.js";
 export * from "./updatefeedbackresponse.js";
 export * from "./updateguardrailsrequest.js";
+export * from "./updateguardrailsresourcepolicyrequest.js";
 export * from "./updateguardrailsresponse.js";
 export * from "./updatemcpserverrequest.js";
 export * from "./updatememoryentryrequest.js";

package/src/models/components/responsescreaterequest.ts

@@ -279,6 +279,10 @@ export type ResponsesCreateRequest = {
    * The ID of the model to use for generating the response.
    */
   model: string;
+  /**
+   * Apply additional guardrails attached to this saved agent in the caller's team. Saved-agent execution sets this to its canonical agent ID, including pinned versions and background runs. This field only adds policy; it does not authorize access to or execute an agent.
+   */
+  guardrailAgentId?: string | undefined;
   /**
    * Ordered list of model IDs to fall back to when a model call fails (provider error or unhealthy provider), tried in order after `model`. Each entry must be a model available to your organization, including your own registered models, and may not be `auto`. When set, this chain replaces any fallback rules configured on the models themselves; an empty list disables fallback for the request. With `model: "auto"`, the chain is tried after the routed model fails. Once the run has switched to a fallback it stays on it for the remaining steps; a step that has already streamed output does not fall back. The served model is reported in `resolved_model`. Maximum length is configured by the gateway (default 5).
    */
@@ -825,6 +829,7 @@ export const ResponsesCreateRequest$inboundSchema: z.ZodType<
   unknown
 > = z.object({
   model: z.string(),
+  guardrail_agent_id: z.string().optional(),
   fallback_models: z.array(z.string()).optional(),
   input: z.union([z.string(), z.array(Item$inboundSchema)]).optional(),
   instructions: z.string().optional(),
@@ -894,6 +899,7 @@ export const ResponsesCreateRequest$inboundSchema: z.ZodType<
   user: z.string().optional(),
 }).transform((v) => {
   return remap$(v, {
+    "guardrail_agent_id": "guardrailAgentId",
     "fallback_models": "fallbackModels",
     "previous_response_id": "previousResponseId",
     "stream_options": "streamOptions",
@@ -925,6 +931,7 @@ export const ResponsesCreateRequest$inboundSchema: z.ZodType<
 /** @internal */
 export type ResponsesCreateRequest$Outbound = {
   model: string;
+  guardrail_agent_id?: string | undefined;
   fallback_models?: Array<string> | undefined;
   input?: string | Array<Item$Outbound> | undefined;
   instructions?: string | undefined;
@@ -1000,6 +1007,7 @@ export const ResponsesCreateRequest$outboundSchema: z.ZodType<
   ResponsesCreateRequest
 > = z.object({
   model: z.string(),
+  guardrailAgentId: z.string().optional(),
   fallbackModels: z.array(z.string()).optional(),
   input: z.union([z.string(), z.array(Item$outboundSchema)]).optional(),
   instructions: z.string().optional(),
@@ -1069,6 +1077,7 @@ export const ResponsesCreateRequest$outboundSchema: z.ZodType<
   user: z.string().optional(),
 }).transform((v) => {
   return remap$(v, {
+    guardrailAgentId: "guardrail_agent_id",
     fallbackModels: "fallback_models",
     previousResponseId: "previous_response_id",
     streamOptions: "stream_options",

package/src/models/components/updateguardrailsresourcepolicyrequest.ts

@@ -0,0 +1,111 @@
+/*
+ * 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 { ClosedEnum } from "../../types/enums.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";
+
+/**
+ * Saved agent or catalog model endpoint.
+ */
+export const UpdateGuardrailsResourcePolicyRequestResourceType = {
+  Agent: "agent",
+  Model: "model",
+} as const;
+/**
+ * Saved agent or catalog model endpoint.
+ */
+export type UpdateGuardrailsResourcePolicyRequestResourceType = ClosedEnum<
+  typeof UpdateGuardrailsResourcePolicyRequestResourceType
+>;
+
+export type UpdateGuardrailsResourcePolicyRequest = {
+  /**
+   * Saved agent or catalog model endpoint.
+   */
+  resourceType: UpdateGuardrailsResourcePolicyRequestResourceType;
+  /**
+   * Canonical saved-agent ID or catalog model ID. Model aliases are resolved before storage.
+   */
+  resourceId: string;
+  guardrails: Array<GuardrailConfig>;
+};
+
+/** @internal */
+export const UpdateGuardrailsResourcePolicyRequestResourceType$inboundSchema:
+  z.ZodNativeEnum<typeof UpdateGuardrailsResourcePolicyRequestResourceType> = z
+    .nativeEnum(UpdateGuardrailsResourcePolicyRequestResourceType);
+/** @internal */
+export const UpdateGuardrailsResourcePolicyRequestResourceType$outboundSchema:
+  z.ZodNativeEnum<typeof UpdateGuardrailsResourcePolicyRequestResourceType> =
+    UpdateGuardrailsResourcePolicyRequestResourceType$inboundSchema;
+
+/** @internal */
+export const UpdateGuardrailsResourcePolicyRequest$inboundSchema: z.ZodType<
+  UpdateGuardrailsResourcePolicyRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  resource_type:
+    UpdateGuardrailsResourcePolicyRequestResourceType$inboundSchema,
+  resource_id: z.string(),
+  guardrails: z.array(GuardrailConfig$inboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    "resource_type": "resourceType",
+    "resource_id": "resourceId",
+  });
+});
+/** @internal */
+export type UpdateGuardrailsResourcePolicyRequest$Outbound = {
+  resource_type: string;
+  resource_id: string;
+  guardrails: Array<GuardrailConfig$Outbound>;
+};
+
+/** @internal */
+export const UpdateGuardrailsResourcePolicyRequest$outboundSchema: z.ZodType<
+  UpdateGuardrailsResourcePolicyRequest$Outbound,
+  z.ZodTypeDef,
+  UpdateGuardrailsResourcePolicyRequest
+> = z.object({
+  resourceType:
+    UpdateGuardrailsResourcePolicyRequestResourceType$outboundSchema,
+  resourceId: z.string(),
+  guardrails: z.array(GuardrailConfig$outboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    resourceType: "resource_type",
+    resourceId: "resource_id",
+  });
+});
+
+export function updateGuardrailsResourcePolicyRequestToJSON(
+  updateGuardrailsResourcePolicyRequest: UpdateGuardrailsResourcePolicyRequest,
+): string {
+  return JSON.stringify(
+    UpdateGuardrailsResourcePolicyRequest$outboundSchema.parse(
+      updateGuardrailsResourcePolicyRequest,
+    ),
+  );
+}
+export function updateGuardrailsResourcePolicyRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateGuardrailsResourcePolicyRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      UpdateGuardrailsResourcePolicyRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateGuardrailsResourcePolicyRequest' from JSON`,
+  );
+}

package/src/models/operations/getguardrailsresourcepolicy.ts

@@ -0,0 +1,101 @@
+/*
+ * 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 { ClosedEnum } from "../../types/enums.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+/**
+ * Saved agent or catalog model endpoint.
+ */
+export const ResourceType = {
+  Agent: "agent",
+  Model: "model",
+} as const;
+/**
+ * Saved agent or catalog model endpoint.
+ */
+export type ResourceType = ClosedEnum<typeof ResourceType>;
+
+export type GetGuardrailsResourcePolicyRequest = {
+  /**
+   * Saved agent or catalog model endpoint.
+   */
+  resourceType: ResourceType;
+  resourceId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const ResourceType$inboundSchema: z.ZodNativeEnum<typeof ResourceType> =
+  z.nativeEnum(ResourceType);
+/** @internal */
+export const ResourceType$outboundSchema: z.ZodNativeEnum<typeof ResourceType> =
+  ResourceType$inboundSchema;
+
+/** @internal */
+export const GetGuardrailsResourcePolicyRequest$inboundSchema: z.ZodType<
+  GetGuardrailsResourcePolicyRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  resource_type: ResourceType$inboundSchema,
+  resource_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "resource_type": "resourceType",
+    "resource_id": "resourceId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type GetGuardrailsResourcePolicyRequest$Outbound = {
+  resource_type: string;
+  resource_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const GetGuardrailsResourcePolicyRequest$outboundSchema: z.ZodType<
+  GetGuardrailsResourcePolicyRequest$Outbound,
+  z.ZodTypeDef,
+  GetGuardrailsResourcePolicyRequest
+> = z.object({
+  resourceType: ResourceType$outboundSchema,
+  resourceId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    resourceType: "resource_type",
+    resourceId: "resource_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function getGuardrailsResourcePolicyRequestToJSON(
+  getGuardrailsResourcePolicyRequest: GetGuardrailsResourcePolicyRequest,
+): string {
+  return JSON.stringify(
+    GetGuardrailsResourcePolicyRequest$outboundSchema.parse(
+      getGuardrailsResourcePolicyRequest,
+    ),
+  );
+}
+export function getGuardrailsResourcePolicyRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<GetGuardrailsResourcePolicyRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      GetGuardrailsResourcePolicyRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GetGuardrailsResourcePolicyRequest' from JSON`,
+  );
+}

package/src/models/operations/index.ts

@@ -213,6 +213,7 @@ export * from "./getgenerationapiv1datasetsgenerationsgenerationidget.js";
 export * from "./getgeneratorapiv1datasetsgeneratorsgeneratoridget.js";
 export * from "./getgithubconnection.js";
 export * from "./getguardrails.js";
+export * from "./getguardrailsresourcepolicy.js";
 export * from "./getjob.js";
 export * from "./getjwtfromkey.js";
 export * from "./getlogsapiv1datasetsgenerationsgenerationidlogsget.js";

package/src/react-query/guardrailsDeleteGuardrailsResourcePolicy.ts

@@ -0,0 +1,109 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { guardrailsDeleteGuardrailsResourcePolicy } from "../funcs/guardrailsDeleteGuardrailsResourcePolicy.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 { unwrapAsync } from "../types/fp.js";
+import { useSDKContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export type GuardrailsDeleteGuardrailsResourcePolicyMutationVariables = {
+  request: components.GuardrailsResourceTarget;
+  options?: RequestOptions;
+};
+
+export type GuardrailsDeleteGuardrailsResourcePolicyMutationData =
+  components.DeleteGuardrailsResourcePolicyResponse;
+
+export type GuardrailsDeleteGuardrailsResourcePolicyMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Remove a resource guardrails policy
+ *
+ * @remarks
+ * Additional guardrails attached to a saved agent in the active team or a catalog model endpoint throughout the organization. Cluster, organization and team policies are always enforced as well. Model aliases resolve to canonical model IDs; auto is not an attachable endpoint. Only organization admins/owners may change these policies. Policies apply to saved-agent versions and future background executions independently of version snapshots. Configured fallback and image-reroute endpoint policies are checked before dispatch as well, even if a fallback is not eventually needed.
+ */
+export function useGuardrailsDeleteGuardrailsResourcePolicyMutation(
+  options?: MutationHookOptions<
+    GuardrailsDeleteGuardrailsResourcePolicyMutationData,
+    GuardrailsDeleteGuardrailsResourcePolicyMutationError,
+    GuardrailsDeleteGuardrailsResourcePolicyMutationVariables
+  >,
+): UseMutationResult<
+  GuardrailsDeleteGuardrailsResourcePolicyMutationData,
+  GuardrailsDeleteGuardrailsResourcePolicyMutationError,
+  GuardrailsDeleteGuardrailsResourcePolicyMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildGuardrailsDeleteGuardrailsResourcePolicyMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyGuardrailsDeleteGuardrailsResourcePolicy(): MutationKey {
+  return ["@meetkai/mka1", "Guardrails", "deleteGuardrailsResourcePolicy"];
+}
+
+export function buildGuardrailsDeleteGuardrailsResourcePolicyMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: GuardrailsDeleteGuardrailsResourcePolicyMutationVariables,
+  ) => Promise<GuardrailsDeleteGuardrailsResourcePolicyMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyGuardrailsDeleteGuardrailsResourcePolicy(),
+    mutationFn: function guardrailsDeleteGuardrailsResourcePolicyMutationFn({
+      request,
+      options,
+    }): Promise<GuardrailsDeleteGuardrailsResourcePolicyMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(guardrailsDeleteGuardrailsResourcePolicy(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

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

@@ -0,0 +1,87 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { guardrailsGetGuardrailsResourcePolicy } from "../funcs/guardrailsGetGuardrailsResourcePolicy.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 GuardrailsGetGuardrailsResourcePolicyQueryData =
+  components.GetGuardrailsResourcePolicyResponse;
+
+export function prefetchGuardrailsGetGuardrailsResourcePolicy(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetGuardrailsResourcePolicyRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildGuardrailsGetGuardrailsResourcePolicyQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildGuardrailsGetGuardrailsResourcePolicyQuery(
+  client$: SDKCore,
+  request: operations.GetGuardrailsResourcePolicyRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<GuardrailsGetGuardrailsResourcePolicyQueryData>;
+} {
+  return {
+    queryKey: queryKeyGuardrailsGetGuardrailsResourcePolicy({
+      resourceType: request.resourceType,
+      resourceId: request.resourceId,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function guardrailsGetGuardrailsResourcePolicyQueryFn(
+      ctx,
+    ): Promise<GuardrailsGetGuardrailsResourcePolicyQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(guardrailsGetGuardrailsResourcePolicy(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyGuardrailsGetGuardrailsResourcePolicy(
+  parameters: {
+    resourceType: operations.ResourceType;
+    resourceId: string;
+    xOnBehalfOf?: string | undefined;
+  },
+): QueryKey {
+  return [
+    "@meetkai/mka1",
+    "Guardrails",
+    "getGuardrailsResourcePolicy",
+    parameters,
+  ];
+}