@meetkai/mka1 0.55.1 → 0.55.3

package/src/models/operations/gettenantresourcepolicy.ts

@@ -0,0 +1,100 @@
+/*
+ * 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 GetTenantResourcePolicyPathParamScope = {
+  Organization: "organization",
+  Team: "team",
+} as const;
+export type GetTenantResourcePolicyPathParamScope = ClosedEnum<
+  typeof GetTenantResourcePolicyPathParamScope
+>;
+
+export type GetTenantResourcePolicyRequest = {
+  scope: GetTenantResourcePolicyPathParamScope;
+  expectedOrganizationId: string;
+  expectedTeamId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const GetTenantResourcePolicyPathParamScope$inboundSchema:
+  z.ZodNativeEnum<typeof GetTenantResourcePolicyPathParamScope> = z.nativeEnum(
+    GetTenantResourcePolicyPathParamScope,
+  );
+/** @internal */
+export const GetTenantResourcePolicyPathParamScope$outboundSchema:
+  z.ZodNativeEnum<typeof GetTenantResourcePolicyPathParamScope> =
+    GetTenantResourcePolicyPathParamScope$inboundSchema;
+
+/** @internal */
+export const GetTenantResourcePolicyRequest$inboundSchema: z.ZodType<
+  GetTenantResourcePolicyRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  scope: GetTenantResourcePolicyPathParamScope$inboundSchema,
+  expected_organization_id: z.string(),
+  expected_team_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "expected_organization_id": "expectedOrganizationId",
+    "expected_team_id": "expectedTeamId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type GetTenantResourcePolicyRequest$Outbound = {
+  scope: string;
+  expected_organization_id: string;
+  expected_team_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const GetTenantResourcePolicyRequest$outboundSchema: z.ZodType<
+  GetTenantResourcePolicyRequest$Outbound,
+  z.ZodTypeDef,
+  GetTenantResourcePolicyRequest
+> = z.object({
+  scope: GetTenantResourcePolicyPathParamScope$outboundSchema,
+  expectedOrganizationId: z.string(),
+  expectedTeamId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    expectedOrganizationId: "expected_organization_id",
+    expectedTeamId: "expected_team_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function getTenantResourcePolicyRequestToJSON(
+  getTenantResourcePolicyRequest: GetTenantResourcePolicyRequest,
+): string {
+  return JSON.stringify(
+    GetTenantResourcePolicyRequest$outboundSchema.parse(
+      getTenantResourcePolicyRequest,
+    ),
+  );
+}
+export function getTenantResourcePolicyRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<GetTenantResourcePolicyRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => GetTenantResourcePolicyRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GetTenantResourcePolicyRequest' from JSON`,
+  );
+}

package/src/models/operations/index.ts

@@ -215,6 +215,7 @@ export * from "./getmcpserver.js";
 export * from "./getmemoryentry.js";
 export * from "./getmemorystore.js";
 export * from "./getmodel.js";
+export * from "./getmodelquota.js";
 export * from "./getnetworkpolicyaccess.js";
 export * from "./getorganizationbilling.js";
 export * from "./getorgbudgets.js";
@@ -247,6 +248,7 @@ export * from "./getteam.js";
 export * from "./getteambudgets.js";
 export * from "./getteamdefaultbudgets.js";
 export * from "./gettenantnetworkpolicy.js";
+export * from "./gettenantresourcepolicy.js";
 export * from "./gettexttospeechhistory.js";
 export * from "./gettexttospeechhistorycontent.js";
 export * from "./getuserbudgets.js";
@@ -404,6 +406,7 @@ export * from "./setsandboxconnectortools.js";
 export * from "./setteambudget.js";
 export * from "./setteamdefaultbudget.js";
 export * from "./settenantnetworkpolicy.js";
+export * from "./settenantresourcepolicy.js";
 export * from "./setupwhatsappappwebhook.js";
 export * from "./setuserbudget.js";
 export * from "./setuserdefaultbudget.js";

package/src/models/operations/listscopedapikeys.ts

@@ -37,6 +37,9 @@ export type ResponseBody = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -118,6 +121,9 @@ export const ResponseBody$inboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 /** @internal */
@@ -132,6 +138,9 @@ export type ResponseBody$Outbound = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -151,6 +160,9 @@ export const ResponseBody$outboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 

package/src/models/operations/listserviceaccountapikeys.ts

@@ -37,6 +37,9 @@ export type ListServiceAccountApiKeysResponseBody = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -120,6 +123,9 @@ export const ListServiceAccountApiKeysResponseBody$inboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 /** @internal */
@@ -134,6 +140,9 @@ export type ListServiceAccountApiKeysResponseBody$Outbound = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -153,6 +162,9 @@ export const ListServiceAccountApiKeysResponseBody$outboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 

package/src/models/operations/reissueownerkey.ts

@@ -42,6 +42,9 @@ export type ApiKey = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -175,6 +178,9 @@ export const ApiKey$inboundSchema: z.ZodType<ApiKey, z.ZodTypeDef, unknown> = z
     orgId: z.nullable(z.string()),
     teamId: z.nullable(z.string()),
     scopes: z.array(z.string()),
+    rateLimitEnabled: z.nullable(z.boolean()).optional(),
+    rateLimitTimeWindow: z.nullable(z.number()).optional(),
+    rateLimitMax: z.nullable(z.number()).optional(),
     expiresAt: z.nullable(z.string()).optional(),
   });
 /** @internal */
@@ -189,6 +195,9 @@ export type ApiKey$Outbound = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -208,6 +217,9 @@ export const ApiKey$outboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 

package/src/models/operations/settenantresourcepolicy.ts

@@ -0,0 +1,107 @@
+/*
+ * 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 * as components from "../components/index.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export const SetTenantResourcePolicyPathParamScope = {
+  Organization: "organization",
+  Team: "team",
+} as const;
+export type SetTenantResourcePolicyPathParamScope = ClosedEnum<
+  typeof SetTenantResourcePolicyPathParamScope
+>;
+
+export type SetTenantResourcePolicyRequest = {
+  scope: SetTenantResourcePolicyPathParamScope;
+  expectedOrganizationId: string;
+  expectedTeamId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+  saveTenantResourcePolicy: components.SaveTenantResourcePolicy;
+};
+
+/** @internal */
+export const SetTenantResourcePolicyPathParamScope$inboundSchema:
+  z.ZodNativeEnum<typeof SetTenantResourcePolicyPathParamScope> = z.nativeEnum(
+    SetTenantResourcePolicyPathParamScope,
+  );
+/** @internal */
+export const SetTenantResourcePolicyPathParamScope$outboundSchema:
+  z.ZodNativeEnum<typeof SetTenantResourcePolicyPathParamScope> =
+    SetTenantResourcePolicyPathParamScope$inboundSchema;
+
+/** @internal */
+export const SetTenantResourcePolicyRequest$inboundSchema: z.ZodType<
+  SetTenantResourcePolicyRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  scope: SetTenantResourcePolicyPathParamScope$inboundSchema,
+  expected_organization_id: z.string(),
+  expected_team_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+  SaveTenantResourcePolicy: components.SaveTenantResourcePolicy$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "expected_organization_id": "expectedOrganizationId",
+    "expected_team_id": "expectedTeamId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+    "SaveTenantResourcePolicy": "saveTenantResourcePolicy",
+  });
+});
+/** @internal */
+export type SetTenantResourcePolicyRequest$Outbound = {
+  scope: string;
+  expected_organization_id: string;
+  expected_team_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+  SaveTenantResourcePolicy: components.SaveTenantResourcePolicy$Outbound;
+};
+
+/** @internal */
+export const SetTenantResourcePolicyRequest$outboundSchema: z.ZodType<
+  SetTenantResourcePolicyRequest$Outbound,
+  z.ZodTypeDef,
+  SetTenantResourcePolicyRequest
+> = z.object({
+  scope: SetTenantResourcePolicyPathParamScope$outboundSchema,
+  expectedOrganizationId: z.string(),
+  expectedTeamId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+  saveTenantResourcePolicy: components.SaveTenantResourcePolicy$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    expectedOrganizationId: "expected_organization_id",
+    expectedTeamId: "expected_team_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+    saveTenantResourcePolicy: "SaveTenantResourcePolicy",
+  });
+});
+
+export function setTenantResourcePolicyRequestToJSON(
+  setTenantResourcePolicyRequest: SetTenantResourcePolicyRequest,
+): string {
+  return JSON.stringify(
+    SetTenantResourcePolicyRequest$outboundSchema.parse(
+      setTenantResourcePolicyRequest,
+    ),
+  );
+}
+export function setTenantResourcePolicyRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<SetTenantResourcePolicyRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => SetTenantResourcePolicyRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetTenantResourcePolicyRequest' from JSON`,
+  );
+}

package/src/models/operations/updatescopedapikey.ts

@@ -48,6 +48,9 @@ export type UpdateScopedApiKeyResponseBody = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -186,6 +189,9 @@ export const UpdateScopedApiKeyResponseBody$inboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 /** @internal */
@@ -200,6 +206,9 @@ export type UpdateScopedApiKeyResponseBody$Outbound = {
   orgId: string | null;
   teamId: string | null;
   scopes: Array<string>;
+  rateLimitEnabled?: boolean | null | undefined;
+  rateLimitTimeWindow?: number | null | undefined;
+  rateLimitMax?: number | null | undefined;
   expiresAt?: string | null | undefined;
 };
 
@@ -219,6 +228,9 @@ export const UpdateScopedApiKeyResponseBody$outboundSchema: z.ZodType<
   orgId: z.nullable(z.string()),
   teamId: z.nullable(z.string()),
   scopes: z.array(z.string()),
+  rateLimitEnabled: z.nullable(z.boolean()).optional(),
+  rateLimitTimeWindow: z.nullable(z.number()).optional(),
+  rateLimitMax: z.nullable(z.number()).optional(),
   expiresAt: z.nullable(z.string()).optional(),
 });
 

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 "./llmModelsGetQuota.js";
 export * from "./llmModelsList.js";
 export * from "./llmModelsListCatalog.js";
 export * from "./llmModelsListCatalogModelAccess.js";
@@ -477,6 +478,8 @@ export * from "./sandboxUploadArchive.js";
 export * from "./sandboxUploadFile.js";
 export * from "./sessionExchangeSessionToken.js";
 export * from "./sessionGetPendingInvitation.js";
+export * from "./sessionsGetTenantResourcePolicy.js";
+export * from "./sessionsSetTenantResourcePolicy.js";
 export * from "./trackingExperimentsCreate.js";
 export * from "./trackingExperimentsDelete.js";
 export * from "./trackingExperimentsGet.js";

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

@@ -0,0 +1,81 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsGetQuota } from "../funcs/llmModelsGetQuota.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 LlmModelsGetQuotaQueryData = operations.GetModelQuotaResponseBody;
+
+export function prefetchLlmModelsGetQuota(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetModelQuotaRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsGetQuotaQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsGetQuotaQuery(
+  client$: SDKCore,
+  request: operations.GetModelQuotaRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsGetQuotaQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsGetQuota(request.modelId, {
+      apiKeyId: request.apiKeyId,
+      serviceAccountId: request.serviceAccountId,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function llmModelsGetQuotaQueryFn(
+      ctx,
+    ): Promise<LlmModelsGetQuotaQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsGetQuota(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsGetQuota(
+  modelId: string,
+  parameters: {
+    apiKeyId?: string | undefined;
+    serviceAccountId?: string | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
+): QueryKey {
+  return ["@meetkai/mka1", "models", "getQuota", modelId, parameters];
+}

package/src/react-query/llmModelsGetQuota.ts

@@ -0,0 +1,149 @@
+/*
+ * 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 {
+  buildLlmModelsGetQuotaQuery,
+  LlmModelsGetQuotaQueryData,
+  prefetchLlmModelsGetQuota,
+  queryKeyLlmModelsGetQuota,
+} from "./llmModelsGetQuota.core.js";
+export {
+  buildLlmModelsGetQuotaQuery,
+  type LlmModelsGetQuotaQueryData,
+  prefetchLlmModelsGetQuota,
+  queryKeyLlmModelsGetQuota,
+};
+
+export type LlmModelsGetQuotaQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Read a model's token quota
+ *
+ * @remarks
+ * Read current tokens-per-minute limits without consuming tokens or starting a window. Use a concrete active model, not auto. Omit selectors for the authenticated caller's context, including its delegated external user. Human sessions may select an authorized api_key_id, plus service_account_id for a service-account key; selected keys use their ordinary, non-delegated context. Scopes with missing subject identities are omitted to avoid exposing shared fallback usage; their enforcement is unchanged. An empty limits array means no reportable token rules for this context. Untouched windows have reset_at: null; unavailable counters return 503.
+ */
+export function useLlmModelsGetQuota(
+  request: operations.GetModelQuotaRequest,
+  options?: QueryHookOptions<
+    LlmModelsGetQuotaQueryData,
+    LlmModelsGetQuotaQueryError
+  >,
+): UseQueryResult<LlmModelsGetQuotaQueryData, LlmModelsGetQuotaQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsGetQuotaQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Read a model's token quota
+ *
+ * @remarks
+ * Read current tokens-per-minute limits without consuming tokens or starting a window. Use a concrete active model, not auto. Omit selectors for the authenticated caller's context, including its delegated external user. Human sessions may select an authorized api_key_id, plus service_account_id for a service-account key; selected keys use their ordinary, non-delegated context. Scopes with missing subject identities are omitted to avoid exposing shared fallback usage; their enforcement is unchanged. An empty limits array means no reportable token rules for this context. Untouched windows have reset_at: null; unavailable counters return 503.
+ */
+export function useLlmModelsGetQuotaSuspense(
+  request: operations.GetModelQuotaRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsGetQuotaQueryData,
+    LlmModelsGetQuotaQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsGetQuotaQueryData,
+  LlmModelsGetQuotaQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsGetQuotaQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsGetQuotaData(
+  client: QueryClient,
+  queryKeyBase: [
+    modelId: string,
+    parameters: {
+      apiKeyId?: string | undefined;
+      serviceAccountId?: string | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
+  ],
+  data: LlmModelsGetQuotaQueryData,
+): LlmModelsGetQuotaQueryData | undefined {
+  const key = queryKeyLlmModelsGetQuota(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsGetQuotaQueryData>(key, data);
+}
+
+export function invalidateLlmModelsGetQuota(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [
+      modelId: string,
+      parameters: {
+        apiKeyId?: string | undefined;
+        serviceAccountId?: string | undefined;
+        xOnBehalfOf?: string | undefined;
+      },
+    ]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getQuota", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmModelsGetQuota(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getQuota"],
+  });
+}