@meetkai/mka1 0.55.0 → 0.55.2

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"],
+  });
+}

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 { llmModelsGetQuota } from "../funcs/llmModelsGetQuota.js";
 import { llmModelsList } from "../funcs/llmModelsList.js";
 import { llmModelsListCatalog } from "../funcs/llmModelsListCatalog.js";
 import { llmModelsListCatalogModelAccess } from "../funcs/llmModelsListCatalogModelAccess.js";
@@ -68,6 +69,23 @@ export class Models extends ClientSDK {
     ));
   }
 
+  /**
+   * 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.
+   */
+  async getQuota(
+    request: operations.GetModelQuotaRequest,
+    options?: RequestOptions,
+  ): Promise<operations.GetModelQuotaResponseBody> {
+    return unwrapAsync(llmModelsGetQuota(
+      this,
+      request,
+      options,
+    ));
+  }
+
   /**
    * List the organization's model catalog
    *