@meetkai/mka1 0.55.29 → 0.55.31

package/src/react-query/llmModelsListOrgBudgetEnforcements.ts

@@ -0,0 +1,118 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  InvalidateQueryFilters,
+  QueryClient,
+  useQuery,
+  UseQueryResult,
+  useSuspenseQuery,
+  UseSuspenseQueryResult,
+} from "@tanstack/react-query";
+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 { useSDKContext } from "./_context.js";
+import { QueryHookOptions, SuspenseQueryHookOptions } from "./_types.js";
+import {
+  buildLlmModelsListOrgBudgetEnforcementsQuery,
+  LlmModelsListOrgBudgetEnforcementsQueryData,
+  prefetchLlmModelsListOrgBudgetEnforcements,
+  queryKeyLlmModelsListOrgBudgetEnforcements,
+} from "./llmModelsListOrgBudgetEnforcements.core.js";
+export {
+  buildLlmModelsListOrgBudgetEnforcementsQuery,
+  type LlmModelsListOrgBudgetEnforcementsQueryData,
+  prefetchLlmModelsListOrgBudgetEnforcements,
+  queryKeyLlmModelsListOrgBudgetEnforcements,
+};
+
+export type LlmModelsListOrgBudgetEnforcementsQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List every organization's budget-enforcement override
+ *
+ * @remarks
+ * Cluster-admin inventory of the organizations that hold a budget-enforcement override, with the cluster default and each one's effective mode — what to review before flipping the cluster default, and where to look when an organization is being refused with `budget_check_unavailable`. Organizations that inherit the default are not listed. API keys need `read:budgets`.
+ */
+export function useLlmModelsListOrgBudgetEnforcements(
+  options?: QueryHookOptions<
+    LlmModelsListOrgBudgetEnforcementsQueryData,
+    LlmModelsListOrgBudgetEnforcementsQueryError
+  >,
+): UseQueryResult<
+  LlmModelsListOrgBudgetEnforcementsQueryData,
+  LlmModelsListOrgBudgetEnforcementsQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsListOrgBudgetEnforcementsQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List every organization's budget-enforcement override
+ *
+ * @remarks
+ * Cluster-admin inventory of the organizations that hold a budget-enforcement override, with the cluster default and each one's effective mode — what to review before flipping the cluster default, and where to look when an organization is being refused with `budget_check_unavailable`. Organizations that inherit the default are not listed. API keys need `read:budgets`.
+ */
+export function useLlmModelsListOrgBudgetEnforcementsSuspense(
+  options?: SuspenseQueryHookOptions<
+    LlmModelsListOrgBudgetEnforcementsQueryData,
+    LlmModelsListOrgBudgetEnforcementsQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsListOrgBudgetEnforcementsQueryData,
+  LlmModelsListOrgBudgetEnforcementsQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsListOrgBudgetEnforcementsQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsListOrgBudgetEnforcementsData(
+  client: QueryClient,
+  data: LlmModelsListOrgBudgetEnforcementsQueryData,
+): LlmModelsListOrgBudgetEnforcementsQueryData | undefined {
+  const key = queryKeyLlmModelsListOrgBudgetEnforcements();
+
+  return client.setQueryData<LlmModelsListOrgBudgetEnforcementsQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateAllLlmModelsListOrgBudgetEnforcements(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "listOrgBudgetEnforcements"],
+  });
+}

package/src/react-query/llmModelsSetOrgBudgetEnforcement.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 { llmModelsSetOrgBudgetEnforcement } from "../funcs/llmModelsSetOrgBudgetEnforcement.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 LlmModelsSetOrgBudgetEnforcementMutationVariables = {
+  request: operations.SetOrgBudgetEnforcementRequest;
+  options?: RequestOptions;
+};
+
+export type LlmModelsSetOrgBudgetEnforcementMutationData =
+  operations.SetOrgBudgetEnforcementResponseBody;
+
+export type LlmModelsSetOrgBudgetEnforcementMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set an organization's budget-enforcement setting
+ *
+ * @remarks
+ * 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.
+ */
+export function useLlmModelsSetOrgBudgetEnforcementMutation(
+  options?: MutationHookOptions<
+    LlmModelsSetOrgBudgetEnforcementMutationData,
+    LlmModelsSetOrgBudgetEnforcementMutationError,
+    LlmModelsSetOrgBudgetEnforcementMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsSetOrgBudgetEnforcementMutationData,
+  LlmModelsSetOrgBudgetEnforcementMutationError,
+  LlmModelsSetOrgBudgetEnforcementMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsSetOrgBudgetEnforcementMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsSetOrgBudgetEnforcement(): MutationKey {
+  return ["@meetkai/mka1", "models", "setOrgBudgetEnforcement"];
+}
+
+export function buildLlmModelsSetOrgBudgetEnforcementMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsSetOrgBudgetEnforcementMutationVariables,
+  ) => Promise<LlmModelsSetOrgBudgetEnforcementMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsSetOrgBudgetEnforcement(),
+    mutationFn: function llmModelsSetOrgBudgetEnforcementMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsSetOrgBudgetEnforcementMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsSetOrgBudgetEnforcement(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/{sessionsGetTenantResourcePolicy.core.ts → sandboxResourcePoliciesGet.core.ts}

@@ -8,23 +8,23 @@ import {
   QueryKey,
 } from "@tanstack/react-query";
 import { SDKCore } from "../core.js";
-import { sessionsGetTenantResourcePolicy } from "../funcs/sessionsGetTenantResourcePolicy.js";
+import { sandboxResourcePoliciesGet } from "../funcs/sandboxResourcePoliciesGet.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 SessionsGetTenantResourcePolicyQueryData =
+export type SandboxResourcePoliciesGetQueryData =
   components.TenantResourcePolicy;
 
-export function prefetchSessionsGetTenantResourcePolicy(
+export function prefetchSandboxResourcePoliciesGet(
   queryClient: QueryClient,
   client$: SDKCore,
   request: operations.GetTenantResourcePolicyRequest,
   options?: RequestOptions,
 ): Promise<void> {
   return queryClient.prefetchQuery({
-    ...buildSessionsGetTenantResourcePolicyQuery(
+    ...buildSandboxResourcePoliciesGetQuery(
       client$,
       request,
       options,
@@ -32,7 +32,7 @@ export function prefetchSessionsGetTenantResourcePolicy(
   });
 }
 
-export function buildSessionsGetTenantResourcePolicyQuery(
+export function buildSandboxResourcePoliciesGetQuery(
   client$: SDKCore,
   request: operations.GetTenantResourcePolicyRequest,
   options?: RequestOptions,
@@ -40,17 +40,17 @@ export function buildSessionsGetTenantResourcePolicyQuery(
   queryKey: QueryKey;
   queryFn: (
     context: QueryFunctionContext,
-  ) => Promise<SessionsGetTenantResourcePolicyQueryData>;
+  ) => Promise<SandboxResourcePoliciesGetQueryData>;
 } {
   return {
-    queryKey: queryKeySessionsGetTenantResourcePolicy(request.scope, {
+    queryKey: queryKeySandboxResourcePoliciesGet(request.scope, {
       expectedOrganizationId: request.expectedOrganizationId,
       expectedTeamId: request.expectedTeamId,
       xOnBehalfOf: request.xOnBehalfOf,
     }),
-    queryFn: async function sessionsGetTenantResourcePolicyQueryFn(
+    queryFn: async function sandboxResourcePoliciesGetQueryFn(
       ctx,
-    ): Promise<SessionsGetTenantResourcePolicyQueryData> {
+    ): Promise<SandboxResourcePoliciesGetQueryData> {
       const sig = combineSignals(
         ctx.signal,
         options?.signal,
@@ -62,7 +62,7 @@ export function buildSessionsGetTenantResourcePolicyQuery(
         signal: sig,
       };
 
-      return unwrapAsync(sessionsGetTenantResourcePolicy(
+      return unwrapAsync(sandboxResourcePoliciesGet(
         client$,
         request,
         mergedOptions,
@@ -71,7 +71,7 @@ export function buildSessionsGetTenantResourcePolicyQuery(
   };
 }
 
-export function queryKeySessionsGetTenantResourcePolicy(
+export function queryKeySandboxResourcePoliciesGet(
   scope: operations.GetTenantResourcePolicyPathParamScope,
   parameters: {
     expectedOrganizationId: string;
@@ -79,11 +79,5 @@ export function queryKeySessionsGetTenantResourcePolicy(
     xOnBehalfOf?: string | undefined;
   },
 ): QueryKey {
-  return [
-    "@meetkai/mka1",
-    "Sessions",
-    "getTenantResourcePolicy",
-    scope,
-    parameters,
-  ];
+  return ["@meetkai/mka1", "resourcePolicies", "get", scope, parameters];
 }

package/src/react-query/{sessionsGetTenantResourcePolicy.ts → sandboxResourcePoliciesGet.ts}

@@ -29,19 +29,19 @@ import {
   TupleToPrefixes,
 } from "./_types.js";
 import {
-  buildSessionsGetTenantResourcePolicyQuery,
-  prefetchSessionsGetTenantResourcePolicy,
-  queryKeySessionsGetTenantResourcePolicy,
-  SessionsGetTenantResourcePolicyQueryData,
-} from "./sessionsGetTenantResourcePolicy.core.js";
+  buildSandboxResourcePoliciesGetQuery,
+  prefetchSandboxResourcePoliciesGet,
+  queryKeySandboxResourcePoliciesGet,
+  SandboxResourcePoliciesGetQueryData,
+} from "./sandboxResourcePoliciesGet.core.js";
 export {
-  buildSessionsGetTenantResourcePolicyQuery,
-  prefetchSessionsGetTenantResourcePolicy,
-  queryKeySessionsGetTenantResourcePolicy,
-  type SessionsGetTenantResourcePolicyQueryData,
+  buildSandboxResourcePoliciesGetQuery,
+  prefetchSandboxResourcePoliciesGet,
+  queryKeySandboxResourcePoliciesGet,
+  type SandboxResourcePoliciesGetQueryData,
 };
 
-export type SessionsGetTenantResourcePolicyQueryError =
+export type SandboxResourcePoliciesGetQueryError =
   | errors.HTTPValidationError
   | SDKError
   | ResponseValidationError
@@ -53,21 +53,24 @@ export type SessionsGetTenantResourcePolicyQueryError =
   | SDKValidationError;
 
 /**
- * Get
+ * Get Tenant Resource Policy
+ *
+ * @remarks
+ * Read organization or team resource limits for standard sandbox sessions.
  */
-export function useSessionsGetTenantResourcePolicy(
+export function useSandboxResourcePoliciesGet(
   request: operations.GetTenantResourcePolicyRequest,
   options?: QueryHookOptions<
-    SessionsGetTenantResourcePolicyQueryData,
-    SessionsGetTenantResourcePolicyQueryError
+    SandboxResourcePoliciesGetQueryData,
+    SandboxResourcePoliciesGetQueryError
   >,
 ): UseQueryResult<
-  SessionsGetTenantResourcePolicyQueryData,
-  SessionsGetTenantResourcePolicyQueryError
+  SandboxResourcePoliciesGetQueryData,
+  SandboxResourcePoliciesGetQueryError
 > {
   const client = useSDKContext();
   return useQuery({
-    ...buildSessionsGetTenantResourcePolicyQuery(
+    ...buildSandboxResourcePoliciesGetQuery(
       client,
       request,
       options,
@@ -77,21 +80,24 @@ export function useSessionsGetTenantResourcePolicy(
 }
 
 /**
- * Get
+ * Get Tenant Resource Policy
+ *
+ * @remarks
+ * Read organization or team resource limits for standard sandbox sessions.
  */
-export function useSessionsGetTenantResourcePolicySuspense(
+export function useSandboxResourcePoliciesGetSuspense(
   request: operations.GetTenantResourcePolicyRequest,
   options?: SuspenseQueryHookOptions<
-    SessionsGetTenantResourcePolicyQueryData,
-    SessionsGetTenantResourcePolicyQueryError
+    SandboxResourcePoliciesGetQueryData,
+    SandboxResourcePoliciesGetQueryError
   >,
 ): UseSuspenseQueryResult<
-  SessionsGetTenantResourcePolicyQueryData,
-  SessionsGetTenantResourcePolicyQueryError
+  SandboxResourcePoliciesGetQueryData,
+  SandboxResourcePoliciesGetQueryError
 > {
   const client = useSDKContext();
   return useSuspenseQuery({
-    ...buildSessionsGetTenantResourcePolicyQuery(
+    ...buildSandboxResourcePoliciesGetQuery(
       client,
       request,
       options,
@@ -100,7 +106,7 @@ export function useSessionsGetTenantResourcePolicySuspense(
   });
 }
 
-export function setSessionsGetTenantResourcePolicyData(
+export function setSandboxResourcePoliciesGetData(
   client: QueryClient,
   queryKeyBase: [
     scope: operations.GetTenantResourcePolicyPathParamScope,
@@ -110,17 +116,14 @@ export function setSessionsGetTenantResourcePolicyData(
       xOnBehalfOf?: string | undefined;
     },
   ],
-  data: SessionsGetTenantResourcePolicyQueryData,
-): SessionsGetTenantResourcePolicyQueryData | undefined {
-  const key = queryKeySessionsGetTenantResourcePolicy(...queryKeyBase);
+  data: SandboxResourcePoliciesGetQueryData,
+): SandboxResourcePoliciesGetQueryData | undefined {
+  const key = queryKeySandboxResourcePoliciesGet(...queryKeyBase);
 
-  return client.setQueryData<SessionsGetTenantResourcePolicyQueryData>(
-    key,
-    data,
-  );
+  return client.setQueryData<SandboxResourcePoliciesGetQueryData>(key, data);
 }
 
-export function invalidateSessionsGetTenantResourcePolicy(
+export function invalidateSandboxResourcePoliciesGet(
   client: QueryClient,
   queryKeyBase: TupleToPrefixes<
     [
@@ -136,21 +139,16 @@ export function invalidateSessionsGetTenantResourcePolicy(
 ): Promise<void> {
   return client.invalidateQueries({
     ...filters,
-    queryKey: [
-      "@meetkai/mka1",
-      "Sessions",
-      "getTenantResourcePolicy",
-      ...queryKeyBase,
-    ],
+    queryKey: ["@meetkai/mka1", "resourcePolicies", "get", ...queryKeyBase],
   });
 }
 
-export function invalidateAllSessionsGetTenantResourcePolicy(
+export function invalidateAllSandboxResourcePoliciesGet(
   client: QueryClient,
   filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
 ): Promise<void> {
   return client.invalidateQueries({
     ...filters,
-    queryKey: ["@meetkai/mka1", "Sessions", "getTenantResourcePolicy"],
+    queryKey: ["@meetkai/mka1", "resourcePolicies", "get"],
   });
 }

package/src/react-query/{sessionsSetTenantResourcePolicy.ts → sandboxResourcePoliciesSet.ts}

@@ -8,7 +8,7 @@ import {
   UseMutationResult,
 } from "@tanstack/react-query";
 import { SDKCore } from "../core.js";
-import { sessionsSetTenantResourcePolicy } from "../funcs/sessionsSetTenantResourcePolicy.js";
+import { sandboxResourcePoliciesSet } from "../funcs/sandboxResourcePoliciesSet.js";
 import { combineSignals } from "../lib/primitives.js";
 import { RequestOptions } from "../lib/sdks.js";
 import * as components from "../models/components/index.js";
@@ -28,15 +28,15 @@ import { unwrapAsync } from "../types/fp.js";
 import { useSDKContext } from "./_context.js";
 import { MutationHookOptions } from "./_types.js";
 
-export type SessionsSetTenantResourcePolicyMutationVariables = {
+export type SandboxResourcePoliciesSetMutationVariables = {
   request: operations.SetTenantResourcePolicyRequest;
   options?: RequestOptions;
 };
 
-export type SessionsSetTenantResourcePolicyMutationData =
+export type SandboxResourcePoliciesSetMutationData =
   components.TenantResourcePolicy;
 
-export type SessionsSetTenantResourcePolicyMutationError =
+export type SandboxResourcePoliciesSetMutationError =
   | errors.HTTPValidationError
   | SDKError
   | ResponseValidationError
@@ -48,45 +48,48 @@ export type SessionsSetTenantResourcePolicyMutationError =
   | SDKValidationError;
 
 /**
- * Save
+ * Set Tenant Resource Policy
+ *
+ * @remarks
+ * Update organization or team resource limits for standard sandbox sessions. The expected revision rejects stale writes.
  */
-export function useSessionsSetTenantResourcePolicyMutation(
+export function useSandboxResourcePoliciesSetMutation(
   options?: MutationHookOptions<
-    SessionsSetTenantResourcePolicyMutationData,
-    SessionsSetTenantResourcePolicyMutationError,
-    SessionsSetTenantResourcePolicyMutationVariables
+    SandboxResourcePoliciesSetMutationData,
+    SandboxResourcePoliciesSetMutationError,
+    SandboxResourcePoliciesSetMutationVariables
   >,
 ): UseMutationResult<
-  SessionsSetTenantResourcePolicyMutationData,
-  SessionsSetTenantResourcePolicyMutationError,
-  SessionsSetTenantResourcePolicyMutationVariables
+  SandboxResourcePoliciesSetMutationData,
+  SandboxResourcePoliciesSetMutationError,
+  SandboxResourcePoliciesSetMutationVariables
 > {
   const client = useSDKContext();
   return useMutation({
-    ...buildSessionsSetTenantResourcePolicyMutation(client, options),
+    ...buildSandboxResourcePoliciesSetMutation(client, options),
     ...options,
   });
 }
 
-export function mutationKeySessionsSetTenantResourcePolicy(): MutationKey {
-  return ["@meetkai/mka1", "Sessions", "setTenantResourcePolicy"];
+export function mutationKeySandboxResourcePoliciesSet(): MutationKey {
+  return ["@meetkai/mka1", "resourcePolicies", "set"];
 }
 
-export function buildSessionsSetTenantResourcePolicyMutation(
+export function buildSandboxResourcePoliciesSetMutation(
   client$: SDKCore,
   hookOptions?: RequestOptions,
 ): {
   mutationKey: MutationKey;
   mutationFn: (
-    variables: SessionsSetTenantResourcePolicyMutationVariables,
-  ) => Promise<SessionsSetTenantResourcePolicyMutationData>;
+    variables: SandboxResourcePoliciesSetMutationVariables,
+  ) => Promise<SandboxResourcePoliciesSetMutationData>;
 } {
   return {
-    mutationKey: mutationKeySessionsSetTenantResourcePolicy(),
-    mutationFn: function sessionsSetTenantResourcePolicyMutationFn({
+    mutationKey: mutationKeySandboxResourcePoliciesSet(),
+    mutationFn: function sandboxResourcePoliciesSetMutationFn({
       request,
       options,
-    }): Promise<SessionsSetTenantResourcePolicyMutationData> {
+    }): Promise<SandboxResourcePoliciesSetMutationData> {
       const mergedOptions = {
         ...hookOptions,
         ...options,
@@ -99,7 +102,7 @@ export function buildSessionsSetTenantResourcePolicyMutation(
           ),
         },
       };
-      return unwrapAsync(sessionsSetTenantResourcePolicy(
+      return unwrapAsync(sandboxResourcePoliciesSet(
         client$,
         request,
         mergedOptions,

package/src/sdk/models.ts

@@ -13,6 +13,7 @@ import { llmModelsDeleteCatalogEntry } from "../funcs/llmModelsDeleteCatalogEntr
 import { llmModelsDeleteOrgAutoModel } from "../funcs/llmModelsDeleteOrgAutoModel.js";
 import { llmModelsGet } from "../funcs/llmModelsGet.js";
 import { llmModelsGetClusterCurrency } from "../funcs/llmModelsGetClusterCurrency.js";
+import { llmModelsGetOrgBudgetEnforcement } from "../funcs/llmModelsGetOrgBudgetEnforcement.js";
 import { llmModelsGetOrgTokenQuota } from "../funcs/llmModelsGetOrgTokenQuota.js";
 import { llmModelsGetQuota } from "../funcs/llmModelsGetQuota.js";
 import { llmModelsList } from "../funcs/llmModelsList.js";
@@ -22,12 +23,14 @@ import { llmModelsListCatalogOrgAccess } from "../funcs/llmModelsListCatalogOrgA
 import { llmModelsListClusterPrices } from "../funcs/llmModelsListClusterPrices.js";
 import { llmModelsListEffectivePrices } from "../funcs/llmModelsListEffectivePrices.js";
 import { llmModelsListOrgAutoModels } from "../funcs/llmModelsListOrgAutoModels.js";
+import { llmModelsListOrgBudgetEnforcements } from "../funcs/llmModelsListOrgBudgetEnforcements.js";
 import { llmModelsListOrgPrices } from "../funcs/llmModelsListOrgPrices.js";
 import { llmModelsListRegistry } from "../funcs/llmModelsListRegistry.js";
 import { llmModelsPutOrgAutoModel } from "../funcs/llmModelsPutOrgAutoModel.js";
 import { llmModelsReplaceCatalogOrgAccess } from "../funcs/llmModelsReplaceCatalogOrgAccess.js";
 import { llmModelsSetClusterCurrency } from "../funcs/llmModelsSetClusterCurrency.js";
 import { llmModelsSetClusterPrice } from "../funcs/llmModelsSetClusterPrice.js";
+import { llmModelsSetOrgBudgetEnforcement } from "../funcs/llmModelsSetOrgBudgetEnforcement.js";
 import { llmModelsSetOrgPrice } from "../funcs/llmModelsSetOrgPrice.js";
 import { llmModelsSetOrgTokenQuota } from "../funcs/llmModelsSetOrgTokenQuota.js";
 import { llmModelsUpdateCatalogEntry } from "../funcs/llmModelsUpdateCatalogEntry.js";
@@ -122,6 +125,55 @@ export class Models extends ClientSDK {
     ));
   }
 
+  /**
+   * List every organization's budget-enforcement override
+   *
+   * @remarks
+   * Cluster-admin inventory of the organizations that hold a budget-enforcement override, with the cluster default and each one's effective mode — what to review before flipping the cluster default, and where to look when an organization is being refused with `budget_check_unavailable`. Organizations that inherit the default are not listed. API keys need `read:budgets`.
+   */
+  async listOrgBudgetEnforcements(
+    options?: RequestOptions,
+  ): Promise<operations.ListOrgBudgetEnforcementsResponseBody> {
+    return unwrapAsync(llmModelsListOrgBudgetEnforcements(
+      this,
+      options,
+    ));
+  }
+
+  /**
+   * Get an organization's budget-enforcement setting
+   *
+   * @remarks
+   * What the spend gate does for this organization's billable requests when the budget check cannot be made (budgeting unreachable or degraded): `allow` admits them, `block` refuses them with 503 `budget_check_unavailable`. Returns the organization's own setting (null = inherits), the cluster default and the effective mode. Cluster admins may read any organization; organization admins their own (never through an exchanged end-user token). API keys need `read:budgets`.
+   */
+  async getOrgBudgetEnforcement(
+    request: operations.GetOrgBudgetEnforcementRequest,
+    options?: RequestOptions,
+  ): Promise<operations.GetOrgBudgetEnforcementResponseBody> {
+    return unwrapAsync(llmModelsGetOrgBudgetEnforcement(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Set an organization's budget-enforcement setting
+   *
+   * @remarks
+   * 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.
+   */
+  async setOrgBudgetEnforcement(
+    request: operations.SetOrgBudgetEnforcementRequest,
+    options?: RequestOptions,
+  ): Promise<operations.SetOrgBudgetEnforcementResponseBody> {
+    return unwrapAsync(llmModelsSetOrgBudgetEnforcement(
+      this,
+      request,
+      options,
+    ));
+  }
+
   /**
    * List the organization's model catalog
    *