@meetkai/mka1 0.55.27 → 0.55.30

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

@@ -0,0 +1,73 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsGetOrgBudgetEnforcement } from "../funcs/llmModelsGetOrgBudgetEnforcement.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+export type LlmModelsGetOrgBudgetEnforcementQueryData =
+  operations.GetOrgBudgetEnforcementResponseBody;
+
+export function prefetchLlmModelsGetOrgBudgetEnforcement(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsGetOrgBudgetEnforcementQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsGetOrgBudgetEnforcementQuery(
+  client$: SDKCore,
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsGetOrgBudgetEnforcementQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsGetOrgBudgetEnforcement(request.orgId),
+    queryFn: async function llmModelsGetOrgBudgetEnforcementQueryFn(
+      ctx,
+    ): Promise<LlmModelsGetOrgBudgetEnforcementQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsGetOrgBudgetEnforcement(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsGetOrgBudgetEnforcement(
+  orgId: string,
+): QueryKey {
+  return ["@meetkai/mka1", "models", "getOrgBudgetEnforcement", orgId];
+}

package/src/react-query/llmModelsGetOrgBudgetEnforcement.ts

@@ -0,0 +1,144 @@
+/*
+ * 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 * as operations from "../models/operations/index.js";
+import { useSDKContext } from "./_context.js";
+import {
+  QueryHookOptions,
+  SuspenseQueryHookOptions,
+  TupleToPrefixes,
+} from "./_types.js";
+import {
+  buildLlmModelsGetOrgBudgetEnforcementQuery,
+  LlmModelsGetOrgBudgetEnforcementQueryData,
+  prefetchLlmModelsGetOrgBudgetEnforcement,
+  queryKeyLlmModelsGetOrgBudgetEnforcement,
+} from "./llmModelsGetOrgBudgetEnforcement.core.js";
+export {
+  buildLlmModelsGetOrgBudgetEnforcementQuery,
+  type LlmModelsGetOrgBudgetEnforcementQueryData,
+  prefetchLlmModelsGetOrgBudgetEnforcement,
+  queryKeyLlmModelsGetOrgBudgetEnforcement,
+};
+
+export type LlmModelsGetOrgBudgetEnforcementQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * 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`.
+ */
+export function useLlmModelsGetOrgBudgetEnforcement(
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: QueryHookOptions<
+    LlmModelsGetOrgBudgetEnforcementQueryData,
+    LlmModelsGetOrgBudgetEnforcementQueryError
+  >,
+): UseQueryResult<
+  LlmModelsGetOrgBudgetEnforcementQueryData,
+  LlmModelsGetOrgBudgetEnforcementQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsGetOrgBudgetEnforcementQuery(
+      client,
+      request,
+      options,
+    ),
+    ...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`.
+ */
+export function useLlmModelsGetOrgBudgetEnforcementSuspense(
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsGetOrgBudgetEnforcementQueryData,
+    LlmModelsGetOrgBudgetEnforcementQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsGetOrgBudgetEnforcementQueryData,
+  LlmModelsGetOrgBudgetEnforcementQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsGetOrgBudgetEnforcementQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsGetOrgBudgetEnforcementData(
+  client: QueryClient,
+  queryKeyBase: [orgId: string],
+  data: LlmModelsGetOrgBudgetEnforcementQueryData,
+): LlmModelsGetOrgBudgetEnforcementQueryData | undefined {
+  const key = queryKeyLlmModelsGetOrgBudgetEnforcement(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsGetOrgBudgetEnforcementQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateLlmModelsGetOrgBudgetEnforcement(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[orgId: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "models",
+      "getOrgBudgetEnforcement",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllLlmModelsGetOrgBudgetEnforcement(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getOrgBudgetEnforcement"],
+  });
+}

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

@@ -0,0 +1,67 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsListOrgBudgetEnforcements } from "../funcs/llmModelsListOrgBudgetEnforcements.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+export type LlmModelsListOrgBudgetEnforcementsQueryData =
+  operations.ListOrgBudgetEnforcementsResponseBody;
+
+export function prefetchLlmModelsListOrgBudgetEnforcements(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsListOrgBudgetEnforcementsQuery(
+      client$,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsListOrgBudgetEnforcementsQuery(
+  client$: SDKCore,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsListOrgBudgetEnforcementsQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsListOrgBudgetEnforcements(),
+    queryFn: async function llmModelsListOrgBudgetEnforcementsQueryFn(
+      ctx,
+    ): Promise<LlmModelsListOrgBudgetEnforcementsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsListOrgBudgetEnforcements(
+        client$,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsListOrgBudgetEnforcements(): QueryKey {
+  return ["@meetkai/mka1", "models", "listOrgBudgetEnforcements"];
+}

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/sdk/agents.ts

@@ -6,10 +6,15 @@ import { agentsCreateAgent } from "../funcs/agentsCreateAgent.js";
 import { agentsCreateFunctionTool } from "../funcs/agentsCreateFunctionTool.js";
 import { agentsDeleteAgent } from "../funcs/agentsDeleteAgent.js";
 import { agentsDeleteFunctionTool } from "../funcs/agentsDeleteFunctionTool.js";
+import { agentsDeleteOpenApiToolSet } from "../funcs/agentsDeleteOpenApiToolSet.js";
 import { agentsGetAgent } from "../funcs/agentsGetAgent.js";
 import { agentsGetFunctionTool } from "../funcs/agentsGetFunctionTool.js";
+import { agentsGetOpenApiToolSet } from "../funcs/agentsGetOpenApiToolSet.js";
+import { agentsImportOpenApiToolSet } from "../funcs/agentsImportOpenApiToolSet.js";
 import { agentsListAgents } from "../funcs/agentsListAgents.js";
 import { agentsListFunctionTools } from "../funcs/agentsListFunctionTools.js";
+import { agentsListOpenApiToolSets } from "../funcs/agentsListOpenApiToolSets.js";
+import { agentsReimportOpenApiToolSet } from "../funcs/agentsReimportOpenApiToolSet.js";
 import { agentsRestoreAgent } from "../funcs/agentsRestoreAgent.js";
 import { agentsUpdateAgent } from "../funcs/agentsUpdateAgent.js";
 import { agentsUpdateFunctionTool } from "../funcs/agentsUpdateFunctionTool.js";
@@ -95,6 +100,85 @@ export class Agents extends ClientSDK {
     ));
   }
 
+  /**
+   * List imported OpenAPI tool sets
+   */
+  async listOpenApiToolSets(
+    request: operations.ListOpenApiToolSetsRequest,
+    options?: RequestOptions,
+  ): Promise<components.OpenApiToolSetList> {
+    return unwrapAsync(agentsListOpenApiToolSets(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Import an OpenAPI document as a tool set
+   *
+   * @remarks
+   * Parses an OpenAPI 3.0 or 3.1 document (JSON or YAML, sent in the body; it is never fetched from a URL) and stores its supported operations as tools an agent can call. Operations that cannot be called faithfully are listed in `skipped_operations`; selecting one with `operations` rejects the import instead. Credentials are never part of the document: reference a credential vault server registered for the API's base URL.
+   */
+  async importOpenApiToolSet(
+    request: operations.ImportOpenApiToolSetRequest,
+    options?: RequestOptions,
+  ): Promise<components.OpenApiToolSet> {
+    return unwrapAsync(agentsImportOpenApiToolSet(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Get an imported OpenAPI tool set
+   */
+  async getOpenApiToolSet(
+    request: operations.GetOpenApiToolSetRequest,
+    options?: RequestOptions,
+  ): Promise<components.OpenApiToolSet> {
+    return unwrapAsync(agentsGetOpenApiToolSet(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Replace an OpenAPI tool set with a new import
+   *
+   * @remarks
+   * Applies to every attached agent on its next run. Existing run snapshots are immutable.
+   */
+  async reimportOpenApiToolSet(
+    request: operations.ReimportOpenApiToolSetRequest,
+    options?: RequestOptions,
+  ): Promise<components.OpenApiToolSet> {
+    return unwrapAsync(agentsReimportOpenApiToolSet(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Delete an OpenAPI tool set
+   *
+   * @remarks
+   * Attached agents fail until the unavailable reference is removed. Existing run snapshots remain usable.
+   */
+  async deleteOpenApiToolSet(
+    request: operations.DeleteOpenApiToolSetRequest,
+    options?: RequestOptions,
+  ): Promise<operations.DeleteOpenApiToolSetResponseBody> {
+    return unwrapAsync(agentsDeleteOpenApiToolSet(
+      this,
+      request,
+      options,
+    ));
+  }
+
   /**
    * List agents
    */

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
    *