@meetkai/mka1 0.55.3 → 0.55.5

package/src/react-query/llmVectorStoresGetPermission.ts

@@ -0,0 +1,151 @@
+/*
+ * 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 {
+  buildLlmVectorStoresGetPermissionQuery,
+  LlmVectorStoresGetPermissionQueryData,
+  prefetchLlmVectorStoresGetPermission,
+  queryKeyLlmVectorStoresGetPermission,
+} from "./llmVectorStoresGetPermission.core.js";
+export {
+  buildLlmVectorStoresGetPermissionQuery,
+  type LlmVectorStoresGetPermissionQueryData,
+  prefetchLlmVectorStoresGetPermission,
+  queryKeyLlmVectorStoresGetPermission,
+};
+
+export type LlmVectorStoresGetPermissionQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Retrieve a vector store permission
+ *
+ * @remarks
+ * Returns the role a principal effectively holds on the vector store and where it comes from: `owner`, a permission granted to it (`grant`), the store's `team` permission, or `none`. A principal with `none` can still reach the store as an organization admin of its team; that is decided per request from the caller's role. Anyone who can read the store can call this. Returns 410 once the vector store has expired.
+ */
+export function useLlmVectorStoresGetPermission(
+  request: operations.GetVectorStorePermissionRequest,
+  options?: QueryHookOptions<
+    LlmVectorStoresGetPermissionQueryData,
+    LlmVectorStoresGetPermissionQueryError
+  >,
+): UseQueryResult<
+  LlmVectorStoresGetPermissionQueryData,
+  LlmVectorStoresGetPermissionQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmVectorStoresGetPermissionQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Retrieve a vector store permission
+ *
+ * @remarks
+ * Returns the role a principal effectively holds on the vector store and where it comes from: `owner`, a permission granted to it (`grant`), the store's `team` permission, or `none`. A principal with `none` can still reach the store as an organization admin of its team; that is decided per request from the caller's role. Anyone who can read the store can call this. Returns 410 once the vector store has expired.
+ */
+export function useLlmVectorStoresGetPermissionSuspense(
+  request: operations.GetVectorStorePermissionRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmVectorStoresGetPermissionQueryData,
+    LlmVectorStoresGetPermissionQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmVectorStoresGetPermissionQueryData,
+  LlmVectorStoresGetPermissionQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmVectorStoresGetPermissionQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmVectorStoresGetPermissionData(
+  client: QueryClient,
+  queryKeyBase: [
+    vectorStoreId: string,
+    userId: string,
+    parameters: { xOnBehalfOf?: string | undefined },
+  ],
+  data: LlmVectorStoresGetPermissionQueryData,
+): LlmVectorStoresGetPermissionQueryData | undefined {
+  const key = queryKeyLlmVectorStoresGetPermission(...queryKeyBase);
+
+  return client.setQueryData<LlmVectorStoresGetPermissionQueryData>(key, data);
+}
+
+export function invalidateLlmVectorStoresGetPermission(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [
+      vectorStoreId: string,
+      userId: string,
+      parameters: { xOnBehalfOf?: string | undefined },
+    ]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "vectorStores",
+      "getPermission",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllLlmVectorStoresGetPermission(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "vectorStores", "getPermission"],
+  });
+}

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

@@ -0,0 +1,83 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmVectorStoresListPermissions } from "../funcs/llmVectorStoresListPermissions.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 LlmVectorStoresListPermissionsQueryData =
+  components.VectorStorePermissionList;
+
+export function prefetchLlmVectorStoresListPermissions(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListVectorStorePermissionsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmVectorStoresListPermissionsQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmVectorStoresListPermissionsQuery(
+  client$: SDKCore,
+  request: operations.ListVectorStorePermissionsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmVectorStoresListPermissionsQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmVectorStoresListPermissions(request.vectorStoreId, {
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function llmVectorStoresListPermissionsQueryFn(
+      ctx,
+    ): Promise<LlmVectorStoresListPermissionsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmVectorStoresListPermissions(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmVectorStoresListPermissions(
+  vectorStoreId: string,
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return [
+    "@meetkai/mka1",
+    "vectorStores",
+    "listPermissions",
+    vectorStoreId,
+    parameters,
+  ];
+}

package/src/react-query/llmVectorStoresListPermissions.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 {
+  buildLlmVectorStoresListPermissionsQuery,
+  LlmVectorStoresListPermissionsQueryData,
+  prefetchLlmVectorStoresListPermissions,
+  queryKeyLlmVectorStoresListPermissions,
+} from "./llmVectorStoresListPermissions.core.js";
+export {
+  buildLlmVectorStoresListPermissionsQuery,
+  type LlmVectorStoresListPermissionsQueryData,
+  prefetchLlmVectorStoresListPermissions,
+  queryKeyLlmVectorStoresListPermissions,
+};
+
+export type LlmVectorStoresListPermissionsQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List vector store permissions
+ *
+ * @remarks
+ * Lists who holds a role on the vector store: its owner, then its team when that permission is set, then individual principals. Anyone who can read the store can list its permissions. Organization admins of the store's team can always read it and are not listed. Returns 410 once the vector store has expired.
+ */
+export function useLlmVectorStoresListPermissions(
+  request: operations.ListVectorStorePermissionsRequest,
+  options?: QueryHookOptions<
+    LlmVectorStoresListPermissionsQueryData,
+    LlmVectorStoresListPermissionsQueryError
+  >,
+): UseQueryResult<
+  LlmVectorStoresListPermissionsQueryData,
+  LlmVectorStoresListPermissionsQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmVectorStoresListPermissionsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List vector store permissions
+ *
+ * @remarks
+ * Lists who holds a role on the vector store: its owner, then its team when that permission is set, then individual principals. Anyone who can read the store can list its permissions. Organization admins of the store's team can always read it and are not listed. Returns 410 once the vector store has expired.
+ */
+export function useLlmVectorStoresListPermissionsSuspense(
+  request: operations.ListVectorStorePermissionsRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmVectorStoresListPermissionsQueryData,
+    LlmVectorStoresListPermissionsQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmVectorStoresListPermissionsQueryData,
+  LlmVectorStoresListPermissionsQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmVectorStoresListPermissionsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmVectorStoresListPermissionsData(
+  client: QueryClient,
+  queryKeyBase: [
+    vectorStoreId: string,
+    parameters: { xOnBehalfOf?: string | undefined },
+  ],
+  data: LlmVectorStoresListPermissionsQueryData,
+): LlmVectorStoresListPermissionsQueryData | undefined {
+  const key = queryKeyLlmVectorStoresListPermissions(...queryKeyBase);
+
+  return client.setQueryData<LlmVectorStoresListPermissionsQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateLlmVectorStoresListPermissions(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [vectorStoreId: string, parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "vectorStores",
+      "listPermissions",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllLlmVectorStoresListPermissions(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "vectorStores", "listPermissions"],
+  });
+}

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";
@@ -73,7 +75,7 @@ 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.
+   * Read applicable token limits without consuming tokens or starting a window. Organization limits may be shared across models. 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 model windows have reset_at: null; organization calendar windows always report their next UTC reset. Unavailable policies or counters return 503.
    */
   async getQuota(
     request: operations.GetModelQuotaRequest,
@@ -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
    *

package/src/sdk/vectorstores.ts

@@ -6,15 +6,19 @@ import { llmVectorStoresCancelFileBatch } from "../funcs/llmVectorStoresCancelFi
 import { llmVectorStoresCreate } from "../funcs/llmVectorStoresCreate.js";
 import { llmVectorStoresCreateFile } from "../funcs/llmVectorStoresCreateFile.js";
 import { llmVectorStoresCreateFileBatch } from "../funcs/llmVectorStoresCreateFileBatch.js";
+import { llmVectorStoresCreatePermission } from "../funcs/llmVectorStoresCreatePermission.js";
 import { llmVectorStoresDelete } from "../funcs/llmVectorStoresDelete.js";
 import { llmVectorStoresDeleteFile } from "../funcs/llmVectorStoresDeleteFile.js";
+import { llmVectorStoresDeletePermission } from "../funcs/llmVectorStoresDeletePermission.js";
 import { llmVectorStoresGet } from "../funcs/llmVectorStoresGet.js";
 import { llmVectorStoresGetFile } from "../funcs/llmVectorStoresGetFile.js";
 import { llmVectorStoresGetFileBatch } from "../funcs/llmVectorStoresGetFileBatch.js";
 import { llmVectorStoresGetFileContent } from "../funcs/llmVectorStoresGetFileContent.js";
+import { llmVectorStoresGetPermission } from "../funcs/llmVectorStoresGetPermission.js";
 import { llmVectorStoresList } from "../funcs/llmVectorStoresList.js";
 import { llmVectorStoresListFiles } from "../funcs/llmVectorStoresListFiles.js";
 import { llmVectorStoresListFilesInBatch } from "../funcs/llmVectorStoresListFilesInBatch.js";
+import { llmVectorStoresListPermissions } from "../funcs/llmVectorStoresListPermissions.js";
 import { llmVectorStoresSearch } from "../funcs/llmVectorStoresSearch.js";
 import { llmVectorStoresUpdate } from "../funcs/llmVectorStoresUpdate.js";
 import { llmVectorStoresUpdateFile } from "../funcs/llmVectorStoresUpdateFile.js";
@@ -295,4 +299,72 @@ export class VectorStores extends ClientSDK {
       options,
     ));
   }
+
+  /**
+   * List vector store permissions
+   *
+   * @remarks
+   * Lists who holds a role on the vector store: its owner, then its team when that permission is set, then individual principals. Anyone who can read the store can list its permissions. Organization admins of the store's team can always read it and are not listed. Returns 410 once the vector store has expired.
+   */
+  async listPermissions(
+    request: operations.ListVectorStorePermissionsRequest,
+    options?: RequestOptions,
+  ): Promise<components.VectorStorePermissionList> {
+    return unwrapAsync(llmVectorStoresListPermissions(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Grant a vector store permission
+   *
+   * @remarks
+   * Grants `reader` or `writer` on the vector store to one principal, or to every member of the store's team (`grantee_type: team`), replacing any role that principal already held. Only the store's owner or an organization admin of its team can grant (403 otherwise). The owner's role is fixed at creation and cannot be granted or changed. Returns 410 once the vector store has expired.
+   */
+  async createPermission(
+    request: operations.CreateVectorStorePermissionRequest,
+    options?: RequestOptions,
+  ): Promise<components.VectorStorePermission> {
+    return unwrapAsync(llmVectorStoresCreatePermission(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Retrieve a vector store permission
+   *
+   * @remarks
+   * Returns the role a principal effectively holds on the vector store and where it comes from: `owner`, a permission granted to it (`grant`), the store's `team` permission, or `none`. A principal with `none` can still reach the store as an organization admin of its team; that is decided per request from the caller's role. Anyone who can read the store can call this. Returns 410 once the vector store has expired.
+   */
+  async getPermission(
+    request: operations.GetVectorStorePermissionRequest,
+    options?: RequestOptions,
+  ): Promise<components.VectorStoreEffectivePermission> {
+    return unwrapAsync(llmVectorStoresGetPermission(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Remove a vector store permission
+   *
+   * @remarks
+   * Removes every role the principal holds on the vector store, or the `team` permission. Only the store's owner or an organization admin of its team can remove permissions (403 otherwise); the owner's own permission cannot be removed. 404 when the principal holds no permission. Returns 410 once the vector store has expired.
+   */
+  async deletePermission(
+    request: operations.DeleteVectorStorePermissionRequest,
+    options?: RequestOptions,
+  ): Promise<components.VectorStorePermissionDeleted> {
+    return unwrapAsync(llmVectorStoresDeletePermission(
+      this,
+      request,
+      options,
+    ));
+  }
 }