@meetkai/mka1 0.55.4 → 0.55.5

package/src/models/operations/setorgtokenquota.ts

@@ -0,0 +1,287 @@
+/*
+ * 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 Two = {
+  enabled: false;
+};
+
+export const RequestBodyPeriod = {
+  Minute: "minute",
+  Day: "day",
+  Month: "month",
+} as const;
+export type RequestBodyPeriod = ClosedEnum<typeof RequestBodyPeriod>;
+
+export type One = {
+  limit: number;
+  period: RequestBodyPeriod;
+  enabled: true;
+};
+
+export type SetOrgTokenQuotaRequestBody = One | Two;
+
+export type SetOrgTokenQuotaRequest = {
+  orgId: string;
+  requestBody: One | Two;
+};
+
+export const SetOrgTokenQuotaPeriod = {
+  Minute: "minute",
+  Day: "day",
+  Month: "month",
+} as const;
+export type SetOrgTokenQuotaPeriod = ClosedEnum<typeof SetOrgTokenQuotaPeriod>;
+
+/**
+ * OK
+ */
+export type SetOrgTokenQuotaResponseBody = {
+  orgId: string;
+  enabled: boolean;
+  limit: number | null;
+  period: SetOrgTokenQuotaPeriod | null;
+};
+
+/** @internal */
+export const Two$inboundSchema: z.ZodType<Two, z.ZodTypeDef, unknown> = z
+  .object({
+    enabled: z.literal(false),
+  });
+/** @internal */
+export type Two$Outbound = {
+  enabled: false;
+};
+
+/** @internal */
+export const Two$outboundSchema: z.ZodType<Two$Outbound, z.ZodTypeDef, Two> = z
+  .object({
+    enabled: z.literal(false),
+  });
+
+export function twoToJSON(two: Two): string {
+  return JSON.stringify(Two$outboundSchema.parse(two));
+}
+export function twoFromJSON(
+  jsonString: string,
+): SafeParseResult<Two, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => Two$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'Two' from JSON`,
+  );
+}
+
+/** @internal */
+export const RequestBodyPeriod$inboundSchema: z.ZodNativeEnum<
+  typeof RequestBodyPeriod
+> = z.nativeEnum(RequestBodyPeriod);
+/** @internal */
+export const RequestBodyPeriod$outboundSchema: z.ZodNativeEnum<
+  typeof RequestBodyPeriod
+> = RequestBodyPeriod$inboundSchema;
+
+/** @internal */
+export const One$inboundSchema: z.ZodType<One, z.ZodTypeDef, unknown> = z
+  .object({
+    limit: z.number().int(),
+    period: RequestBodyPeriod$inboundSchema,
+    enabled: z.literal(true),
+  });
+/** @internal */
+export type One$Outbound = {
+  limit: number;
+  period: string;
+  enabled: true;
+};
+
+/** @internal */
+export const One$outboundSchema: z.ZodType<One$Outbound, z.ZodTypeDef, One> = z
+  .object({
+    limit: z.number().int(),
+    period: RequestBodyPeriod$outboundSchema,
+    enabled: z.literal(true),
+  });
+
+export function oneToJSON(one: One): string {
+  return JSON.stringify(One$outboundSchema.parse(one));
+}
+export function oneFromJSON(
+  jsonString: string,
+): SafeParseResult<One, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => One$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'One' from JSON`,
+  );
+}
+
+/** @internal */
+export const SetOrgTokenQuotaRequestBody$inboundSchema: z.ZodType<
+  SetOrgTokenQuotaRequestBody,
+  z.ZodTypeDef,
+  unknown
+> = z.union([z.lazy(() => One$inboundSchema), z.lazy(() => Two$inboundSchema)]);
+/** @internal */
+export type SetOrgTokenQuotaRequestBody$Outbound = One$Outbound | Two$Outbound;
+
+/** @internal */
+export const SetOrgTokenQuotaRequestBody$outboundSchema: z.ZodType<
+  SetOrgTokenQuotaRequestBody$Outbound,
+  z.ZodTypeDef,
+  SetOrgTokenQuotaRequestBody
+> = z.union([
+  z.lazy(() => One$outboundSchema),
+  z.lazy(() => Two$outboundSchema),
+]);
+
+export function setOrgTokenQuotaRequestBodyToJSON(
+  setOrgTokenQuotaRequestBody: SetOrgTokenQuotaRequestBody,
+): string {
+  return JSON.stringify(
+    SetOrgTokenQuotaRequestBody$outboundSchema.parse(
+      setOrgTokenQuotaRequestBody,
+    ),
+  );
+}
+export function setOrgTokenQuotaRequestBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<SetOrgTokenQuotaRequestBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => SetOrgTokenQuotaRequestBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetOrgTokenQuotaRequestBody' from JSON`,
+  );
+}
+
+/** @internal */
+export const SetOrgTokenQuotaRequest$inboundSchema: z.ZodType<
+  SetOrgTokenQuotaRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  RequestBody: z.union([
+    z.lazy(() => One$inboundSchema),
+    z.lazy(() => Two$inboundSchema),
+  ]),
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "RequestBody": "requestBody",
+  });
+});
+/** @internal */
+export type SetOrgTokenQuotaRequest$Outbound = {
+  org_id: string;
+  RequestBody: One$Outbound | Two$Outbound;
+};
+
+/** @internal */
+export const SetOrgTokenQuotaRequest$outboundSchema: z.ZodType<
+  SetOrgTokenQuotaRequest$Outbound,
+  z.ZodTypeDef,
+  SetOrgTokenQuotaRequest
+> = z.object({
+  orgId: z.string(),
+  requestBody: z.union([
+    z.lazy(() => One$outboundSchema),
+    z.lazy(() => Two$outboundSchema),
+  ]),
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    requestBody: "RequestBody",
+  });
+});
+
+export function setOrgTokenQuotaRequestToJSON(
+  setOrgTokenQuotaRequest: SetOrgTokenQuotaRequest,
+): string {
+  return JSON.stringify(
+    SetOrgTokenQuotaRequest$outboundSchema.parse(setOrgTokenQuotaRequest),
+  );
+}
+export function setOrgTokenQuotaRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<SetOrgTokenQuotaRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => SetOrgTokenQuotaRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetOrgTokenQuotaRequest' from JSON`,
+  );
+}
+
+/** @internal */
+export const SetOrgTokenQuotaPeriod$inboundSchema: z.ZodNativeEnum<
+  typeof SetOrgTokenQuotaPeriod
+> = z.nativeEnum(SetOrgTokenQuotaPeriod);
+/** @internal */
+export const SetOrgTokenQuotaPeriod$outboundSchema: z.ZodNativeEnum<
+  typeof SetOrgTokenQuotaPeriod
+> = SetOrgTokenQuotaPeriod$inboundSchema;
+
+/** @internal */
+export const SetOrgTokenQuotaResponseBody$inboundSchema: z.ZodType<
+  SetOrgTokenQuotaResponseBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  enabled: z.boolean(),
+  limit: z.nullable(z.number().int()),
+  period: z.nullable(SetOrgTokenQuotaPeriod$inboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+  });
+});
+/** @internal */
+export type SetOrgTokenQuotaResponseBody$Outbound = {
+  org_id: string;
+  enabled: boolean;
+  limit: number | null;
+  period: string | null;
+};
+
+/** @internal */
+export const SetOrgTokenQuotaResponseBody$outboundSchema: z.ZodType<
+  SetOrgTokenQuotaResponseBody$Outbound,
+  z.ZodTypeDef,
+  SetOrgTokenQuotaResponseBody
+> = z.object({
+  orgId: z.string(),
+  enabled: z.boolean(),
+  limit: z.nullable(z.number().int()),
+  period: z.nullable(SetOrgTokenQuotaPeriod$outboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+  });
+});
+
+export function setOrgTokenQuotaResponseBodyToJSON(
+  setOrgTokenQuotaResponseBody: SetOrgTokenQuotaResponseBody,
+): string {
+  return JSON.stringify(
+    SetOrgTokenQuotaResponseBody$outboundSchema.parse(
+      setOrgTokenQuotaResponseBody,
+    ),
+  );
+}
+export function setOrgTokenQuotaResponseBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<SetOrgTokenQuotaResponseBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => SetOrgTokenQuotaResponseBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetOrgTokenQuotaResponseBody' from JSON`,
+  );
+}

package/src/react-query/index.ts

@@ -338,6 +338,7 @@ export * from "./llmModelsDeleteCatalogEntry.js";
 export * from "./llmModelsDeleteOrgAutoModel.js";
 export * from "./llmModelsGet.js";
 export * from "./llmModelsGetClusterCurrency.js";
+export * from "./llmModelsGetOrgTokenQuota.js";
 export * from "./llmModelsGetQuota.js";
 export * from "./llmModelsList.js";
 export * from "./llmModelsListCatalog.js";
@@ -353,6 +354,7 @@ export * from "./llmModelsReplaceCatalogOrgAccess.js";
 export * from "./llmModelsSetClusterCurrency.js";
 export * from "./llmModelsSetClusterPrice.js";
 export * from "./llmModelsSetOrgPrice.js";
+export * from "./llmModelsSetOrgTokenQuota.js";
 export * from "./llmModelsUpdateCatalogEntry.js";
 export * from "./llmPromptsCreate.js";
 export * from "./llmPromptsCreateVersion.js";

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

@@ -0,0 +1,71 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsGetOrgTokenQuota } from "../funcs/llmModelsGetOrgTokenQuota.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 LlmModelsGetOrgTokenQuotaQueryData =
+  operations.GetOrgTokenQuotaResponseBody;
+
+export function prefetchLlmModelsGetOrgTokenQuota(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetOrgTokenQuotaRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsGetOrgTokenQuotaQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsGetOrgTokenQuotaQuery(
+  client$: SDKCore,
+  request: operations.GetOrgTokenQuotaRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsGetOrgTokenQuotaQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsGetOrgTokenQuota(request.orgId),
+    queryFn: async function llmModelsGetOrgTokenQuotaQueryFn(
+      ctx,
+    ): Promise<LlmModelsGetOrgTokenQuotaQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsGetOrgTokenQuota(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsGetOrgTokenQuota(orgId: string): QueryKey {
+  return ["@meetkai/mka1", "models", "getOrgTokenQuota", orgId];
+}

package/src/react-query/llmModelsGetOrgTokenQuota.ts

@@ -0,0 +1,136 @@
+/*
+ * 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 {
+  buildLlmModelsGetOrgTokenQuotaQuery,
+  LlmModelsGetOrgTokenQuotaQueryData,
+  prefetchLlmModelsGetOrgTokenQuota,
+  queryKeyLlmModelsGetOrgTokenQuota,
+} from "./llmModelsGetOrgTokenQuota.core.js";
+export {
+  buildLlmModelsGetOrgTokenQuotaQuery,
+  type LlmModelsGetOrgTokenQuotaQueryData,
+  prefetchLlmModelsGetOrgTokenQuota,
+  queryKeyLlmModelsGetOrgTokenQuota,
+};
+
+export type LlmModelsGetOrgTokenQuotaQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get an organization's token quota
+ *
+ * @remarks
+ * Cluster-admin read of the aggregate token policy across all models. An absent or disabled quota returns enabled: false with null limit and period. Does not require a model or change the active organization.
+ */
+export function useLlmModelsGetOrgTokenQuota(
+  request: operations.GetOrgTokenQuotaRequest,
+  options?: QueryHookOptions<
+    LlmModelsGetOrgTokenQuotaQueryData,
+    LlmModelsGetOrgTokenQuotaQueryError
+  >,
+): UseQueryResult<
+  LlmModelsGetOrgTokenQuotaQueryData,
+  LlmModelsGetOrgTokenQuotaQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsGetOrgTokenQuotaQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get an organization's token quota
+ *
+ * @remarks
+ * Cluster-admin read of the aggregate token policy across all models. An absent or disabled quota returns enabled: false with null limit and period. Does not require a model or change the active organization.
+ */
+export function useLlmModelsGetOrgTokenQuotaSuspense(
+  request: operations.GetOrgTokenQuotaRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsGetOrgTokenQuotaQueryData,
+    LlmModelsGetOrgTokenQuotaQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsGetOrgTokenQuotaQueryData,
+  LlmModelsGetOrgTokenQuotaQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsGetOrgTokenQuotaQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsGetOrgTokenQuotaData(
+  client: QueryClient,
+  queryKeyBase: [orgId: string],
+  data: LlmModelsGetOrgTokenQuotaQueryData,
+): LlmModelsGetOrgTokenQuotaQueryData | undefined {
+  const key = queryKeyLlmModelsGetOrgTokenQuota(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsGetOrgTokenQuotaQueryData>(key, data);
+}
+
+export function invalidateLlmModelsGetOrgTokenQuota(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[orgId: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getOrgTokenQuota", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmModelsGetOrgTokenQuota(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getOrgTokenQuota"],
+  });
+}

package/src/react-query/llmModelsSetOrgTokenQuota.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 { llmModelsSetOrgTokenQuota } from "../funcs/llmModelsSetOrgTokenQuota.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 LlmModelsSetOrgTokenQuotaMutationVariables = {
+  request: operations.SetOrgTokenQuotaRequest;
+  options?: RequestOptions;
+};
+
+export type LlmModelsSetOrgTokenQuotaMutationData =
+  operations.SetOrgTokenQuotaResponseBody;
+
+export type LlmModelsSetOrgTokenQuotaMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set an organization's token quota
+ *
+ * @remarks
+ * Cluster admins may create or replace the minute/day/month token allowance, or disable it with enabled: false. Changes apply to new admissions within 10 seconds; admitted requests finish. Limit changes preserve usage. Period changes use that period's calendar bucket; usage while disabled is not counted. The cluster organization cannot be capped.
+ */
+export function useLlmModelsSetOrgTokenQuotaMutation(
+  options?: MutationHookOptions<
+    LlmModelsSetOrgTokenQuotaMutationData,
+    LlmModelsSetOrgTokenQuotaMutationError,
+    LlmModelsSetOrgTokenQuotaMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsSetOrgTokenQuotaMutationData,
+  LlmModelsSetOrgTokenQuotaMutationError,
+  LlmModelsSetOrgTokenQuotaMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsSetOrgTokenQuotaMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsSetOrgTokenQuota(): MutationKey {
+  return ["@meetkai/mka1", "models", "setOrgTokenQuota"];
+}
+
+export function buildLlmModelsSetOrgTokenQuotaMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsSetOrgTokenQuotaMutationVariables,
+  ) => Promise<LlmModelsSetOrgTokenQuotaMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsSetOrgTokenQuota(),
+    mutationFn: function llmModelsSetOrgTokenQuotaMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsSetOrgTokenQuotaMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsSetOrgTokenQuota(
+        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 { llmModelsGetOrgTokenQuota } from "../funcs/llmModelsGetOrgTokenQuota.js";
 import { llmModelsGetQuota } from "../funcs/llmModelsGetQuota.js";
 import { llmModelsList } from "../funcs/llmModelsList.js";
 import { llmModelsListCatalog } from "../funcs/llmModelsListCatalog.js";
@@ -28,6 +29,7 @@ import { llmModelsReplaceCatalogOrgAccess } from "../funcs/llmModelsReplaceCatal
 import { llmModelsSetClusterCurrency } from "../funcs/llmModelsSetClusterCurrency.js";
 import { llmModelsSetClusterPrice } from "../funcs/llmModelsSetClusterPrice.js";
 import { llmModelsSetOrgPrice } from "../funcs/llmModelsSetOrgPrice.js";
+import { llmModelsSetOrgTokenQuota } from "../funcs/llmModelsSetOrgTokenQuota.js";
 import { llmModelsUpdateCatalogEntry } from "../funcs/llmModelsUpdateCatalogEntry.js";
 import { ClientSDK, RequestOptions } from "../lib/sdks.js";
 import * as components from "../models/components/index.js";
@@ -86,6 +88,40 @@ export class Models extends ClientSDK {
     ));
   }
 
+  /**
+   * Get an organization's token quota
+   *
+   * @remarks
+   * Cluster-admin read of the aggregate token policy across all models. An absent or disabled quota returns enabled: false with null limit and period. Does not require a model or change the active organization.
+   */
+  async getOrgTokenQuota(
+    request: operations.GetOrgTokenQuotaRequest,
+    options?: RequestOptions,
+  ): Promise<operations.GetOrgTokenQuotaResponseBody> {
+    return unwrapAsync(llmModelsGetOrgTokenQuota(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Set an organization's token quota
+   *
+   * @remarks
+   * Cluster admins may create or replace the minute/day/month token allowance, or disable it with enabled: false. Changes apply to new admissions within 10 seconds; admitted requests finish. Limit changes preserve usage. Period changes use that period's calendar bucket; usage while disabled is not counted. The cluster organization cannot be capped.
+   */
+  async setOrgTokenQuota(
+    request: operations.SetOrgTokenQuotaRequest,
+    options?: RequestOptions,
+  ): Promise<operations.SetOrgTokenQuotaResponseBody> {
+    return unwrapAsync(llmModelsSetOrgTokenQuota(
+      this,
+      request,
+      options,
+    ));
+  }
+
   /**
    * List the organization's model catalog
    *