@meetkai/mka1 0.51.1 → 0.51.2

package/src/react-query/servingTenantsGet.ts

@@ -0,0 +1,141 @@
+/*
+ * 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 {
+  buildServingTenantsGetQuery,
+  prefetchServingTenantsGet,
+  queryKeyServingTenantsGet,
+  ServingTenantsGetQueryData,
+} from "./servingTenantsGet.core.js";
+export {
+  buildServingTenantsGetQuery,
+  prefetchServingTenantsGet,
+  queryKeyServingTenantsGet,
+  type ServingTenantsGetQueryData,
+};
+
+export type ServingTenantsGetQueryError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get a serving tenant
+ *
+ * @remarks
+ * Get one organization's serving enablement record. Cluster administrators only.
+ */
+export function useServingTenantsGet(
+  request: operations.GetServingTenantRequest,
+  options?: QueryHookOptions<
+    ServingTenantsGetQueryData,
+    ServingTenantsGetQueryError
+  >,
+): UseQueryResult<ServingTenantsGetQueryData, ServingTenantsGetQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingTenantsGetQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get a serving tenant
+ *
+ * @remarks
+ * Get one organization's serving enablement record. Cluster administrators only.
+ */
+export function useServingTenantsGetSuspense(
+  request: operations.GetServingTenantRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingTenantsGetQueryData,
+    ServingTenantsGetQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingTenantsGetQueryData,
+  ServingTenantsGetQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingTenantsGetQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingTenantsGetData(
+  client: QueryClient,
+  queryKeyBase: [
+    orgId: string,
+    parameters: { xOnBehalfOf?: string | undefined },
+  ],
+  data: ServingTenantsGetQueryData,
+): ServingTenantsGetQueryData | undefined {
+  const key = queryKeyServingTenantsGet(...queryKeyBase);
+
+  return client.setQueryData<ServingTenantsGetQueryData>(key, data);
+}
+
+export function invalidateServingTenantsGet(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [orgId: string, parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "tenants", "get", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingTenantsGet(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "tenants", "get"],
+  });
+}

package/src/react-query/servingTenantsList.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 { servingTenantsList } from "../funcs/servingTenantsList.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 ServingTenantsListQueryData = components.PageServingTenant;
+
+export function prefetchServingTenantsList(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListServingTenantsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingTenantsListQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingTenantsListQuery(
+  client$: SDKCore,
+  request: operations.ListServingTenantsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingTenantsListQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingTenantsList({
+      cursor: request.cursor,
+      limit: request.limit,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingTenantsListQueryFn(
+      ctx,
+    ): Promise<ServingTenantsListQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingTenantsList(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingTenantsList(
+  parameters: {
+    cursor?: string | null | undefined;
+    limit?: number | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
+): QueryKey {
+  return ["@meetkai/mka1", "tenants", "list", parameters];
+}

package/src/react-query/servingTenantsList.ts

@@ -0,0 +1,148 @@
+/*
+ * 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 {
+  buildServingTenantsListQuery,
+  prefetchServingTenantsList,
+  queryKeyServingTenantsList,
+  ServingTenantsListQueryData,
+} from "./servingTenantsList.core.js";
+export {
+  buildServingTenantsListQuery,
+  prefetchServingTenantsList,
+  queryKeyServingTenantsList,
+  type ServingTenantsListQueryData,
+};
+
+export type ServingTenantsListQueryError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List serving tenants
+ *
+ * @remarks
+ * List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned.
+ */
+export function useServingTenantsList(
+  request: operations.ListServingTenantsRequest,
+  options?: QueryHookOptions<
+    ServingTenantsListQueryData,
+    ServingTenantsListQueryError
+  >,
+): UseQueryResult<ServingTenantsListQueryData, ServingTenantsListQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingTenantsListQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List serving tenants
+ *
+ * @remarks
+ * List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned.
+ */
+export function useServingTenantsListSuspense(
+  request: operations.ListServingTenantsRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingTenantsListQueryData,
+    ServingTenantsListQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingTenantsListQueryData,
+  ServingTenantsListQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingTenantsListQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingTenantsListData(
+  client: QueryClient,
+  queryKeyBase: [
+    parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
+  ],
+  data: ServingTenantsListQueryData,
+): ServingTenantsListQueryData | undefined {
+  const key = queryKeyServingTenantsList(...queryKeyBase);
+
+  return client.setQueryData<ServingTenantsListQueryData>(key, data);
+}
+
+export function invalidateServingTenantsList(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
+      xOnBehalfOf?: string | undefined;
+    }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "tenants", "list", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingTenantsList(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "tenants", "list"],
+  });
+}

package/src/react-query/servingTenantsSet.ts

@@ -0,0 +1,112 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingTenantsSet } from "../funcs/servingTenantsSet.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 ServingTenantsSetMutationVariables = {
+  request: operations.SetServingTenantRequest;
+  options?: RequestOptions;
+};
+
+export type ServingTenantsSetMutationData = components.ServingTenant;
+
+export type ServingTenantsSetMutationError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set serving tenant enablement
+ *
+ * @remarks
+ * Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only.
+ */
+export function useServingTenantsSetMutation(
+  options?: MutationHookOptions<
+    ServingTenantsSetMutationData,
+    ServingTenantsSetMutationError,
+    ServingTenantsSetMutationVariables
+  >,
+): UseMutationResult<
+  ServingTenantsSetMutationData,
+  ServingTenantsSetMutationError,
+  ServingTenantsSetMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildServingTenantsSetMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyServingTenantsSet(): MutationKey {
+  return ["@meetkai/mka1", "tenants", "set"];
+}
+
+export function buildServingTenantsSetMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ServingTenantsSetMutationVariables,
+  ) => Promise<ServingTenantsSetMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyServingTenantsSet(),
+    mutationFn: function servingTenantsSetMutationFn({
+      request,
+      options,
+    }): Promise<ServingTenantsSetMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(servingTenantsSet(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/sdk/serving.ts

@@ -9,6 +9,7 @@ import { FinetuneJobs } from "./finetunejobs.js";
 import { SDKImages } from "./sdkimages.js";
 import { SDKModels } from "./sdkmodels.js";
 import { Secrets } from "./secrets.js";
+import { Tenants } from "./tenants.js";
 import { Volumes } from "./volumes.js";
 
 export class Serving extends ClientSDK {
@@ -42,6 +43,11 @@ export class Serving extends ClientSDK {
     return (this._secrets ??= new Secrets(this._options));
   }
 
+  private _tenants?: Tenants;
+  get tenants(): Tenants {
+    return (this._tenants ??= new Tenants(this._options));
+  }
+
   private _volumes?: Volumes;
   get volumes(): Volumes {
     return (this._volumes ??= new Volumes(this._options));

package/src/sdk/tenants.ts

@@ -0,0 +1,64 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { servingTenantsGet } from "../funcs/servingTenantsGet.js";
+import { servingTenantsList } from "../funcs/servingTenantsList.js";
+import { servingTenantsSet } from "../funcs/servingTenantsSet.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 Tenants extends ClientSDK {
+  /**
+   * List serving tenants
+   *
+   * @remarks
+   * List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned.
+   */
+  async list(
+    request: operations.ListServingTenantsRequest,
+    options?: RequestOptions,
+  ): Promise<components.PageServingTenant> {
+    return unwrapAsync(servingTenantsList(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Get a serving tenant
+   *
+   * @remarks
+   * Get one organization's serving enablement record. Cluster administrators only.
+   */
+  async get(
+    request: operations.GetServingTenantRequest,
+    options?: RequestOptions,
+  ): Promise<components.ServingTenant> {
+    return unwrapAsync(servingTenantsGet(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Set serving tenant enablement
+   *
+   * @remarks
+   * Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only.
+   */
+  async set(
+    request: operations.SetServingTenantRequest,
+    options?: RequestOptions,
+  ): Promise<components.ServingTenant> {
+    return unwrapAsync(servingTenantsSet(
+      this,
+      request,
+      options,
+    ));
+  }
+}