@meetkai/mka1 0.55.1 → 0.55.3

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

@@ -0,0 +1,89 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { sessionsGetTenantResourcePolicy } from "../funcs/sessionsGetTenantResourcePolicy.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 =
+  components.TenantResourcePolicy;
+
+export function prefetchSessionsGetTenantResourcePolicy(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetTenantResourcePolicyRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildSessionsGetTenantResourcePolicyQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildSessionsGetTenantResourcePolicyQuery(
+  client$: SDKCore,
+  request: operations.GetTenantResourcePolicyRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<SessionsGetTenantResourcePolicyQueryData>;
+} {
+  return {
+    queryKey: queryKeySessionsGetTenantResourcePolicy(request.scope, {
+      expectedOrganizationId: request.expectedOrganizationId,
+      expectedTeamId: request.expectedTeamId,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function sessionsGetTenantResourcePolicyQueryFn(
+      ctx,
+    ): Promise<SessionsGetTenantResourcePolicyQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(sessionsGetTenantResourcePolicy(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeySessionsGetTenantResourcePolicy(
+  scope: operations.GetTenantResourcePolicyPathParamScope,
+  parameters: {
+    expectedOrganizationId: string;
+    expectedTeamId: string;
+    xOnBehalfOf?: string | undefined;
+  },
+): QueryKey {
+  return [
+    "@meetkai/mka1",
+    "Sessions",
+    "getTenantResourcePolicy",
+    scope,
+    parameters,
+  ];
+}

package/src/react-query/sessionsGetTenantResourcePolicy.ts

@@ -0,0 +1,156 @@
+/*
+ * 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 * as errors from "../models/errors/index.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 {
+  buildSessionsGetTenantResourcePolicyQuery,
+  prefetchSessionsGetTenantResourcePolicy,
+  queryKeySessionsGetTenantResourcePolicy,
+  SessionsGetTenantResourcePolicyQueryData,
+} from "./sessionsGetTenantResourcePolicy.core.js";
+export {
+  buildSessionsGetTenantResourcePolicyQuery,
+  prefetchSessionsGetTenantResourcePolicy,
+  queryKeySessionsGetTenantResourcePolicy,
+  type SessionsGetTenantResourcePolicyQueryData,
+};
+
+export type SessionsGetTenantResourcePolicyQueryError =
+  | errors.HTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get
+ */
+export function useSessionsGetTenantResourcePolicy(
+  request: operations.GetTenantResourcePolicyRequest,
+  options?: QueryHookOptions<
+    SessionsGetTenantResourcePolicyQueryData,
+    SessionsGetTenantResourcePolicyQueryError
+  >,
+): UseQueryResult<
+  SessionsGetTenantResourcePolicyQueryData,
+  SessionsGetTenantResourcePolicyQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildSessionsGetTenantResourcePolicyQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get
+ */
+export function useSessionsGetTenantResourcePolicySuspense(
+  request: operations.GetTenantResourcePolicyRequest,
+  options?: SuspenseQueryHookOptions<
+    SessionsGetTenantResourcePolicyQueryData,
+    SessionsGetTenantResourcePolicyQueryError
+  >,
+): UseSuspenseQueryResult<
+  SessionsGetTenantResourcePolicyQueryData,
+  SessionsGetTenantResourcePolicyQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildSessionsGetTenantResourcePolicyQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setSessionsGetTenantResourcePolicyData(
+  client: QueryClient,
+  queryKeyBase: [
+    scope: operations.GetTenantResourcePolicyPathParamScope,
+    parameters: {
+      expectedOrganizationId: string;
+      expectedTeamId: string;
+      xOnBehalfOf?: string | undefined;
+    },
+  ],
+  data: SessionsGetTenantResourcePolicyQueryData,
+): SessionsGetTenantResourcePolicyQueryData | undefined {
+  const key = queryKeySessionsGetTenantResourcePolicy(...queryKeyBase);
+
+  return client.setQueryData<SessionsGetTenantResourcePolicyQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateSessionsGetTenantResourcePolicy(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [
+      scope: operations.GetTenantResourcePolicyPathParamScope,
+      parameters: {
+        expectedOrganizationId: string;
+        expectedTeamId: string;
+        xOnBehalfOf?: string | undefined;
+      },
+    ]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "Sessions",
+      "getTenantResourcePolicy",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllSessionsGetTenantResourcePolicy(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "Sessions", "getTenantResourcePolicy"],
+  });
+}

package/src/react-query/sessionsSetTenantResourcePolicy.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 { sessionsSetTenantResourcePolicy } from "../funcs/sessionsSetTenantResourcePolicy.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.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 SessionsSetTenantResourcePolicyMutationVariables = {
+  request: operations.SetTenantResourcePolicyRequest;
+  options?: RequestOptions;
+};
+
+export type SessionsSetTenantResourcePolicyMutationData =
+  components.TenantResourcePolicy;
+
+export type SessionsSetTenantResourcePolicyMutationError =
+  | errors.HTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Save
+ */
+export function useSessionsSetTenantResourcePolicyMutation(
+  options?: MutationHookOptions<
+    SessionsSetTenantResourcePolicyMutationData,
+    SessionsSetTenantResourcePolicyMutationError,
+    SessionsSetTenantResourcePolicyMutationVariables
+  >,
+): UseMutationResult<
+  SessionsSetTenantResourcePolicyMutationData,
+  SessionsSetTenantResourcePolicyMutationError,
+  SessionsSetTenantResourcePolicyMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildSessionsSetTenantResourcePolicyMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeySessionsSetTenantResourcePolicy(): MutationKey {
+  return ["@meetkai/mka1", "Sessions", "setTenantResourcePolicy"];
+}
+
+export function buildSessionsSetTenantResourcePolicyMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: SessionsSetTenantResourcePolicyMutationVariables,
+  ) => Promise<SessionsSetTenantResourcePolicyMutationData>;
+} {
+  return {
+    mutationKey: mutationKeySessionsSetTenantResourcePolicy(),
+    mutationFn: function sessionsSetTenantResourcePolicyMutationFn({
+      request,
+      options,
+    }): Promise<SessionsSetTenantResourcePolicyMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(sessionsSetTenantResourcePolicy(
+        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 { 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
    *

package/src/sdk/sdk.ts

@@ -28,6 +28,7 @@ import { Permissions } from "./permissions.js";
 import { Repos } from "./repos.js";
 import { Sandbox } from "./sandbox.js";
 import { Session } from "./session.js";
+import { Sessions } from "./sessions.js";
 import { Tracking } from "./tracking.js";
 import { Usage } from "./usage.js";
 
@@ -62,6 +63,11 @@ export class SDK extends ClientSDK {
     return (this._sandbox ??= new Sandbox(this._options));
   }
 
+  private _sessions?: Sessions;
+  get sessions(): Sessions {
+    return (this._sessions ??= new Sessions(this._options));
+  }
+
   private _usage?: Usage;
   get usage(): Usage {
     return (this._usage ??= new Usage(this._options));

package/src/sdk/sessions.ts

@@ -0,0 +1,40 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { sessionsGetTenantResourcePolicy } from "../funcs/sessionsGetTenantResourcePolicy.js";
+import { sessionsSetTenantResourcePolicy } from "../funcs/sessionsSetTenantResourcePolicy.js";
+import { ClientSDK, 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 class Sessions extends ClientSDK {
+  /**
+   * Get
+   */
+  async getTenantResourcePolicy(
+    request: operations.GetTenantResourcePolicyRequest,
+    options?: RequestOptions,
+  ): Promise<components.TenantResourcePolicy> {
+    return unwrapAsync(sessionsGetTenantResourcePolicy(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Save
+   */
+  async setTenantResourcePolicy(
+    request: operations.SetTenantResourcePolicyRequest,
+    options?: RequestOptions,
+  ): Promise<components.TenantResourcePolicy> {
+    return unwrapAsync(sessionsSetTenantResourcePolicy(
+      this,
+      request,
+      options,
+    ));
+  }
+}