@meetkai/mka1 0.55.29 → 0.55.30

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

@@ -0,0 +1,39 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsSetOrgBudgetEnforcement } from "../../funcs/llmModelsSetOrgBudgetEnforcement.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.SetOrgBudgetEnforcementRequest$inboundSchema,
+};
+
+export const tool$llmModelsSetOrgBudgetEnforcement: ToolDefinition<
+  typeof args
+> = {
+  name: "llm-models-set-org-budget-enforcement",
+  description: `Set an organization's budget-enforcement setting
+
+Choose \`block\` to refuse this organization's billable requests whenever the budget check is unavailable (fail closed), \`allow\` to admit them (fail open), or \`null\` to inherit the cluster default again. Applies to new admissions within 10 seconds on every gateway replica. Cluster admins may set any organization to anything; organization admins may only tighten their own — \`block\`, or \`null\` to inherit while that does not loosen an effective \`block\` — and never \`allow\`, so neither a cluster-wide nor a cluster-admin-pinned \`block\` can be switched off or pre-empted by a tenant (never through an exchanged end-user token). API keys need \`write:budgets\`. The cluster organization has no override: it is the operator's own identity and always fails open.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsSetOrgBudgetEnforcement(
+      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/operations/getorgbudgetenforcement.ts

@@ -0,0 +1,198 @@
+/*
+ * 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";
+
+export type GetOrgBudgetEnforcementRequest = {
+  orgId: string;
+};
+
+export const GetOrgBudgetEnforcementOnCheckFailure = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type GetOrgBudgetEnforcementOnCheckFailure = ClosedEnum<
+  typeof GetOrgBudgetEnforcementOnCheckFailure
+>;
+
+export const ClusterDefault = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type ClusterDefault = ClosedEnum<typeof ClusterDefault>;
+
+/**
+ * What the spend gate does for this organization when the budget check is unavailable
+ */
+export const Effective = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+/**
+ * What the spend gate does for this organization when the budget check is unavailable
+ */
+export type Effective = ClosedEnum<typeof Effective>;
+
+/**
+ * OK
+ */
+export type GetOrgBudgetEnforcementResponseBody = {
+  orgId: string;
+  /**
+   * The organization's own setting; null inherits the cluster default
+   */
+  onCheckFailure: GetOrgBudgetEnforcementOnCheckFailure | null;
+  clusterDefault: ClusterDefault;
+  /**
+   * What the spend gate does for this organization when the budget check is unavailable
+   */
+  effective: Effective;
+};
+
+/** @internal */
+export const GetOrgBudgetEnforcementRequest$inboundSchema: z.ZodType<
+  GetOrgBudgetEnforcementRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+  });
+});
+/** @internal */
+export type GetOrgBudgetEnforcementRequest$Outbound = {
+  org_id: string;
+};
+
+/** @internal */
+export const GetOrgBudgetEnforcementRequest$outboundSchema: z.ZodType<
+  GetOrgBudgetEnforcementRequest$Outbound,
+  z.ZodTypeDef,
+  GetOrgBudgetEnforcementRequest
+> = z.object({
+  orgId: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+  });
+});
+
+export function getOrgBudgetEnforcementRequestToJSON(
+  getOrgBudgetEnforcementRequest: GetOrgBudgetEnforcementRequest,
+): string {
+  return JSON.stringify(
+    GetOrgBudgetEnforcementRequest$outboundSchema.parse(
+      getOrgBudgetEnforcementRequest,
+    ),
+  );
+}
+export function getOrgBudgetEnforcementRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<GetOrgBudgetEnforcementRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => GetOrgBudgetEnforcementRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GetOrgBudgetEnforcementRequest' from JSON`,
+  );
+}
+
+/** @internal */
+export const GetOrgBudgetEnforcementOnCheckFailure$inboundSchema:
+  z.ZodNativeEnum<typeof GetOrgBudgetEnforcementOnCheckFailure> = z.nativeEnum(
+    GetOrgBudgetEnforcementOnCheckFailure,
+  );
+/** @internal */
+export const GetOrgBudgetEnforcementOnCheckFailure$outboundSchema:
+  z.ZodNativeEnum<typeof GetOrgBudgetEnforcementOnCheckFailure> =
+    GetOrgBudgetEnforcementOnCheckFailure$inboundSchema;
+
+/** @internal */
+export const ClusterDefault$inboundSchema: z.ZodNativeEnum<
+  typeof ClusterDefault
+> = z.nativeEnum(ClusterDefault);
+/** @internal */
+export const ClusterDefault$outboundSchema: z.ZodNativeEnum<
+  typeof ClusterDefault
+> = ClusterDefault$inboundSchema;
+
+/** @internal */
+export const Effective$inboundSchema: z.ZodNativeEnum<typeof Effective> = z
+  .nativeEnum(Effective);
+/** @internal */
+export const Effective$outboundSchema: z.ZodNativeEnum<typeof Effective> =
+  Effective$inboundSchema;
+
+/** @internal */
+export const GetOrgBudgetEnforcementResponseBody$inboundSchema: z.ZodType<
+  GetOrgBudgetEnforcementResponseBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  on_check_failure: z.nullable(
+    GetOrgBudgetEnforcementOnCheckFailure$inboundSchema,
+  ),
+  cluster_default: ClusterDefault$inboundSchema,
+  effective: Effective$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "on_check_failure": "onCheckFailure",
+    "cluster_default": "clusterDefault",
+  });
+});
+/** @internal */
+export type GetOrgBudgetEnforcementResponseBody$Outbound = {
+  org_id: string;
+  on_check_failure: string | null;
+  cluster_default: string;
+  effective: string;
+};
+
+/** @internal */
+export const GetOrgBudgetEnforcementResponseBody$outboundSchema: z.ZodType<
+  GetOrgBudgetEnforcementResponseBody$Outbound,
+  z.ZodTypeDef,
+  GetOrgBudgetEnforcementResponseBody
+> = z.object({
+  orgId: z.string(),
+  onCheckFailure: z.nullable(
+    GetOrgBudgetEnforcementOnCheckFailure$outboundSchema,
+  ),
+  clusterDefault: ClusterDefault$outboundSchema,
+  effective: Effective$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    onCheckFailure: "on_check_failure",
+    clusterDefault: "cluster_default",
+  });
+});
+
+export function getOrgBudgetEnforcementResponseBodyToJSON(
+  getOrgBudgetEnforcementResponseBody: GetOrgBudgetEnforcementResponseBody,
+): string {
+  return JSON.stringify(
+    GetOrgBudgetEnforcementResponseBody$outboundSchema.parse(
+      getOrgBudgetEnforcementResponseBody,
+    ),
+  );
+}
+export function getOrgBudgetEnforcementResponseBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<GetOrgBudgetEnforcementResponseBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      GetOrgBudgetEnforcementResponseBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GetOrgBudgetEnforcementResponseBody' from JSON`,
+  );
+}

package/src/models/operations/index.ts

@@ -224,6 +224,7 @@ export * from "./getmodelquota.js";
 export * from "./getnetworkpolicyaccess.js";
 export * from "./getopenapitoolset.js";
 export * from "./getorganizationbilling.js";
+export * from "./getorgbudgetenforcement.js";
 export * from "./getorgbudgets.js";
 export * from "./getorgbyid.js";
 export * from "./getorginboundrules.js";
@@ -332,6 +333,7 @@ export * from "./listmodelcatalogorgaccess.js";
 export * from "./listmodels.js";
 export * from "./listopenapitoolsets.js";
 export * from "./listorganizationbillinginvoices.js";
+export * from "./listorgbudgetenforcements.js";
 export * from "./listorgbudgetevents.js";
 export * from "./listorgmembers.js";
 export * from "./listorgprices.js";
@@ -418,6 +420,7 @@ export * from "./setexternaluserdefaultbudget.js";
 export * from "./setfeatureflagenvironmentstate.js";
 export * from "./setorganizationbillingenabled.js";
 export * from "./setorgbudget.js";
+export * from "./setorgbudgetenforcement.js";
 export * from "./setorgprice.js";
 export * from "./setorgtokenquota.js";
 export * from "./setsandboxconnectorapikey.js";

package/src/models/operations/listorgbudgetenforcements.ts

@@ -0,0 +1,214 @@
+/*
+ * 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";
+
+export const ListOrgBudgetEnforcementsOnCheckFailure = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type ListOrgBudgetEnforcementsOnCheckFailure = ClosedEnum<
+  typeof ListOrgBudgetEnforcementsOnCheckFailure
+>;
+
+export const ListOrgBudgetEnforcementsClusterDefault = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type ListOrgBudgetEnforcementsClusterDefault = ClosedEnum<
+  typeof ListOrgBudgetEnforcementsClusterDefault
+>;
+
+/**
+ * What the spend gate does for this organization when the budget check is unavailable
+ */
+export const ListOrgBudgetEnforcementsEffective = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+/**
+ * What the spend gate does for this organization when the budget check is unavailable
+ */
+export type ListOrgBudgetEnforcementsEffective = ClosedEnum<
+  typeof ListOrgBudgetEnforcementsEffective
+>;
+
+export type ListOrgBudgetEnforcementsData = {
+  orgId: string;
+  /**
+   * The organization's own setting; null inherits the cluster default
+   */
+  onCheckFailure: ListOrgBudgetEnforcementsOnCheckFailure | null;
+  clusterDefault: ListOrgBudgetEnforcementsClusterDefault;
+  /**
+   * What the spend gate does for this organization when the budget check is unavailable
+   */
+  effective: ListOrgBudgetEnforcementsEffective;
+};
+
+/**
+ * OK
+ */
+export type ListOrgBudgetEnforcementsResponseBody = {
+  object: "list";
+  data: Array<ListOrgBudgetEnforcementsData>;
+  /**
+   * True when more organizations hold an override than the list returns
+   */
+  hasMore: boolean;
+};
+
+/** @internal */
+export const ListOrgBudgetEnforcementsOnCheckFailure$inboundSchema:
+  z.ZodNativeEnum<typeof ListOrgBudgetEnforcementsOnCheckFailure> = z
+    .nativeEnum(ListOrgBudgetEnforcementsOnCheckFailure);
+/** @internal */
+export const ListOrgBudgetEnforcementsOnCheckFailure$outboundSchema:
+  z.ZodNativeEnum<typeof ListOrgBudgetEnforcementsOnCheckFailure> =
+    ListOrgBudgetEnforcementsOnCheckFailure$inboundSchema;
+
+/** @internal */
+export const ListOrgBudgetEnforcementsClusterDefault$inboundSchema:
+  z.ZodNativeEnum<typeof ListOrgBudgetEnforcementsClusterDefault> = z
+    .nativeEnum(ListOrgBudgetEnforcementsClusterDefault);
+/** @internal */
+export const ListOrgBudgetEnforcementsClusterDefault$outboundSchema:
+  z.ZodNativeEnum<typeof ListOrgBudgetEnforcementsClusterDefault> =
+    ListOrgBudgetEnforcementsClusterDefault$inboundSchema;
+
+/** @internal */
+export const ListOrgBudgetEnforcementsEffective$inboundSchema: z.ZodNativeEnum<
+  typeof ListOrgBudgetEnforcementsEffective
+> = z.nativeEnum(ListOrgBudgetEnforcementsEffective);
+/** @internal */
+export const ListOrgBudgetEnforcementsEffective$outboundSchema: z.ZodNativeEnum<
+  typeof ListOrgBudgetEnforcementsEffective
+> = ListOrgBudgetEnforcementsEffective$inboundSchema;
+
+/** @internal */
+export const ListOrgBudgetEnforcementsData$inboundSchema: z.ZodType<
+  ListOrgBudgetEnforcementsData,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  on_check_failure: z.nullable(
+    ListOrgBudgetEnforcementsOnCheckFailure$inboundSchema,
+  ),
+  cluster_default: ListOrgBudgetEnforcementsClusterDefault$inboundSchema,
+  effective: ListOrgBudgetEnforcementsEffective$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "on_check_failure": "onCheckFailure",
+    "cluster_default": "clusterDefault",
+  });
+});
+/** @internal */
+export type ListOrgBudgetEnforcementsData$Outbound = {
+  org_id: string;
+  on_check_failure: string | null;
+  cluster_default: string;
+  effective: string;
+};
+
+/** @internal */
+export const ListOrgBudgetEnforcementsData$outboundSchema: z.ZodType<
+  ListOrgBudgetEnforcementsData$Outbound,
+  z.ZodTypeDef,
+  ListOrgBudgetEnforcementsData
+> = z.object({
+  orgId: z.string(),
+  onCheckFailure: z.nullable(
+    ListOrgBudgetEnforcementsOnCheckFailure$outboundSchema,
+  ),
+  clusterDefault: ListOrgBudgetEnforcementsClusterDefault$outboundSchema,
+  effective: ListOrgBudgetEnforcementsEffective$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    onCheckFailure: "on_check_failure",
+    clusterDefault: "cluster_default",
+  });
+});
+
+export function listOrgBudgetEnforcementsDataToJSON(
+  listOrgBudgetEnforcementsData: ListOrgBudgetEnforcementsData,
+): string {
+  return JSON.stringify(
+    ListOrgBudgetEnforcementsData$outboundSchema.parse(
+      listOrgBudgetEnforcementsData,
+    ),
+  );
+}
+export function listOrgBudgetEnforcementsDataFromJSON(
+  jsonString: string,
+): SafeParseResult<ListOrgBudgetEnforcementsData, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ListOrgBudgetEnforcementsData$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ListOrgBudgetEnforcementsData' from JSON`,
+  );
+}
+
+/** @internal */
+export const ListOrgBudgetEnforcementsResponseBody$inboundSchema: z.ZodType<
+  ListOrgBudgetEnforcementsResponseBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  object: z.literal("list"),
+  data: z.array(z.lazy(() => ListOrgBudgetEnforcementsData$inboundSchema)),
+  has_more: z.boolean(),
+}).transform((v) => {
+  return remap$(v, {
+    "has_more": "hasMore",
+  });
+});
+/** @internal */
+export type ListOrgBudgetEnforcementsResponseBody$Outbound = {
+  object: "list";
+  data: Array<ListOrgBudgetEnforcementsData$Outbound>;
+  has_more: boolean;
+};
+
+/** @internal */
+export const ListOrgBudgetEnforcementsResponseBody$outboundSchema: z.ZodType<
+  ListOrgBudgetEnforcementsResponseBody$Outbound,
+  z.ZodTypeDef,
+  ListOrgBudgetEnforcementsResponseBody
+> = z.object({
+  object: z.literal("list"),
+  data: z.array(z.lazy(() => ListOrgBudgetEnforcementsData$outboundSchema)),
+  hasMore: z.boolean(),
+}).transform((v) => {
+  return remap$(v, {
+    hasMore: "has_more",
+  });
+});
+
+export function listOrgBudgetEnforcementsResponseBodyToJSON(
+  listOrgBudgetEnforcementsResponseBody: ListOrgBudgetEnforcementsResponseBody,
+): string {
+  return JSON.stringify(
+    ListOrgBudgetEnforcementsResponseBody$outboundSchema.parse(
+      listOrgBudgetEnforcementsResponseBody,
+    ),
+  );
+}
+export function listOrgBudgetEnforcementsResponseBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<ListOrgBudgetEnforcementsResponseBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      ListOrgBudgetEnforcementsResponseBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ListOrgBudgetEnforcementsResponseBody' from JSON`,
+  );
+}