@meetkai/mka1 0.48.33 → 0.48.34

package/src/models/operations/updatescaling.ts

@@ -0,0 +1,76 @@
+/*
+ * 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 { Result as SafeParseResult } from "../../types/fp.js";
+import * as components from "../components/index.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type UpdateScalingRequest = {
+  deploymentId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+  scaling: components.Scaling;
+};
+
+/** @internal */
+export const UpdateScalingRequest$inboundSchema: z.ZodType<
+  UpdateScalingRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  deployment_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+  Scaling: components.Scaling$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "deployment_id": "deploymentId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+    "Scaling": "scaling",
+  });
+});
+/** @internal */
+export type UpdateScalingRequest$Outbound = {
+  deployment_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+  Scaling: components.Scaling$Outbound;
+};
+
+/** @internal */
+export const UpdateScalingRequest$outboundSchema: z.ZodType<
+  UpdateScalingRequest$Outbound,
+  z.ZodTypeDef,
+  UpdateScalingRequest
+> = z.object({
+  deploymentId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+  scaling: components.Scaling$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    deploymentId: "deployment_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+    scaling: "Scaling",
+  });
+});
+
+export function updateScalingRequestToJSON(
+  updateScalingRequest: UpdateScalingRequest,
+): string {
+  return JSON.stringify(
+    UpdateScalingRequest$outboundSchema.parse(updateScalingRequest),
+  );
+}
+export function updateScalingRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateScalingRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateScalingRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateScalingRequest' from JSON`,
+  );
+}

package/src/react-query/index.ts

@@ -295,4 +295,38 @@ export * from "./searchTextStoreCreateTextStore.js";
 export * from "./searchTextStoreDeleteTexts.js";
 export * from "./searchTextStoreDeleteTextStore.js";
 export * from "./searchTextStoreSearchTexts.js";
+export * from "./servingAcceleratorsList.js";
+export * from "./servingDeploymentsCreate.js";
+export * from "./servingDeploymentsDelete.js";
+export * from "./servingDeploymentsGet.js";
+export * from "./servingDeploymentsGetLogs.js";
+export * from "./servingDeploymentsGetMetrics.js";
+export * from "./servingDeploymentsList.js";
+export * from "./servingDeploymentsListReplicas.js";
+export * from "./servingDeploymentsListRevisions.js";
+export * from "./servingDeploymentsRollback.js";
+export * from "./servingDeploymentsUpdate.js";
+export * from "./servingDeploymentsUpdateScaling.js";
+export * from "./servingFinetuneJobsCancel.js";
+export * from "./servingFinetuneJobsCreate.js";
+export * from "./servingFinetuneJobsGet.js";
+export * from "./servingFinetuneJobsGetEvents.js";
+export * from "./servingFinetuneJobsGetLogs.js";
+export * from "./servingFinetuneJobsList.js";
+export * from "./servingFinetuneJobsListCheckpoints.js";
+export * from "./servingImagesBuild.js";
+export * from "./servingImagesDelete.js";
+export * from "./servingImagesGet.js";
+export * from "./servingImagesList.js";
+export * from "./servingModelsDelete.js";
+export * from "./servingModelsGet.js";
+export * from "./servingModelsList.js";
+export * from "./servingModelsRegister.js";
+export * from "./servingSecretsCreate.js";
+export * from "./servingSecretsDelete.js";
+export * from "./servingSecretsList.js";
+export * from "./servingVolumesCreate.js";
+export * from "./servingVolumesDelete.js";
+export * from "./servingVolumesGet.js";
+export * from "./servingVolumesList.js";
 export * from "./usageCosts.js";

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

@@ -0,0 +1,75 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingAcceleratorsList } from "../funcs/servingAcceleratorsList.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 ServingAcceleratorsListQueryData = components.PageAcceleratorType;
+
+export function prefetchServingAcceleratorsList(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListAcceleratorsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingAcceleratorsListQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingAcceleratorsListQuery(
+  client$: SDKCore,
+  request: operations.ListAcceleratorsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingAcceleratorsListQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingAcceleratorsList({
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingAcceleratorsListQueryFn(
+      ctx,
+    ): Promise<ServingAcceleratorsListQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingAcceleratorsList(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingAcceleratorsList(
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "accelerators", "list", parameters];
+}

package/src/react-query/servingAcceleratorsList.ts

@@ -0,0 +1,140 @@
+/*
+ * 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 {
+  buildServingAcceleratorsListQuery,
+  prefetchServingAcceleratorsList,
+  queryKeyServingAcceleratorsList,
+  ServingAcceleratorsListQueryData,
+} from "./servingAcceleratorsList.core.js";
+export {
+  buildServingAcceleratorsListQuery,
+  prefetchServingAcceleratorsList,
+  queryKeyServingAcceleratorsList,
+  type ServingAcceleratorsListQueryData,
+};
+
+export type ServingAcceleratorsListQueryError =
+  | errors.ServingError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List available accelerator types
+ *
+ * @remarks
+ * List the accelerator types available to your organization (GPU, NPU, TPU).
+ */
+export function useServingAcceleratorsList(
+  request: operations.ListAcceleratorsRequest,
+  options?: QueryHookOptions<
+    ServingAcceleratorsListQueryData,
+    ServingAcceleratorsListQueryError
+  >,
+): UseQueryResult<
+  ServingAcceleratorsListQueryData,
+  ServingAcceleratorsListQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingAcceleratorsListQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List available accelerator types
+ *
+ * @remarks
+ * List the accelerator types available to your organization (GPU, NPU, TPU).
+ */
+export function useServingAcceleratorsListSuspense(
+  request: operations.ListAcceleratorsRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingAcceleratorsListQueryData,
+    ServingAcceleratorsListQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingAcceleratorsListQueryData,
+  ServingAcceleratorsListQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingAcceleratorsListQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingAcceleratorsListData(
+  client: QueryClient,
+  queryKeyBase: [parameters: { xOnBehalfOf?: string | undefined }],
+  data: ServingAcceleratorsListQueryData,
+): ServingAcceleratorsListQueryData | undefined {
+  const key = queryKeyServingAcceleratorsList(...queryKeyBase);
+
+  return client.setQueryData<ServingAcceleratorsListQueryData>(key, data);
+}
+
+export function invalidateServingAcceleratorsList(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "accelerators", "list", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingAcceleratorsList(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "accelerators", "list"],
+  });
+}

package/src/react-query/servingDeploymentsCreate.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 { servingDeploymentsCreate } from "../funcs/servingDeploymentsCreate.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 ServingDeploymentsCreateMutationVariables = {
+  request: operations.CreateDeploymentRequest;
+  options?: RequestOptions;
+};
+
+export type ServingDeploymentsCreateMutationData = components.Deployment;
+
+export type ServingDeploymentsCreateMutationError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Create an inference deployment
+ *
+ * @remarks
+ * Create and start an inference deployment for a model, and return its endpoint.
+ */
+export function useServingDeploymentsCreateMutation(
+  options?: MutationHookOptions<
+    ServingDeploymentsCreateMutationData,
+    ServingDeploymentsCreateMutationError,
+    ServingDeploymentsCreateMutationVariables
+  >,
+): UseMutationResult<
+  ServingDeploymentsCreateMutationData,
+  ServingDeploymentsCreateMutationError,
+  ServingDeploymentsCreateMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildServingDeploymentsCreateMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyServingDeploymentsCreate(): MutationKey {
+  return ["@meetkai/mka1", "deployments", "create"];
+}
+
+export function buildServingDeploymentsCreateMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ServingDeploymentsCreateMutationVariables,
+  ) => Promise<ServingDeploymentsCreateMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyServingDeploymentsCreate(),
+    mutationFn: function servingDeploymentsCreateMutationFn({
+      request,
+      options,
+    }): Promise<ServingDeploymentsCreateMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(servingDeploymentsCreate(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/servingDeploymentsDelete.ts

@@ -0,0 +1,111 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingDeploymentsDelete } from "../funcs/servingDeploymentsDelete.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 ServingDeploymentsDeleteMutationVariables = {
+  request: operations.DeleteDeploymentRequest;
+  options?: RequestOptions;
+};
+
+export type ServingDeploymentsDeleteMutationData = void;
+
+export type ServingDeploymentsDeleteMutationError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Tear down a deployment
+ *
+ * @remarks
+ * Tear down a deployment and release its resources.
+ */
+export function useServingDeploymentsDeleteMutation(
+  options?: MutationHookOptions<
+    ServingDeploymentsDeleteMutationData,
+    ServingDeploymentsDeleteMutationError,
+    ServingDeploymentsDeleteMutationVariables
+  >,
+): UseMutationResult<
+  ServingDeploymentsDeleteMutationData,
+  ServingDeploymentsDeleteMutationError,
+  ServingDeploymentsDeleteMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildServingDeploymentsDeleteMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyServingDeploymentsDelete(): MutationKey {
+  return ["@meetkai/mka1", "deployments", "delete"];
+}
+
+export function buildServingDeploymentsDeleteMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ServingDeploymentsDeleteMutationVariables,
+  ) => Promise<ServingDeploymentsDeleteMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyServingDeploymentsDelete(),
+    mutationFn: function servingDeploymentsDeleteMutationFn({
+      request,
+      options,
+    }): Promise<ServingDeploymentsDeleteMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(servingDeploymentsDelete(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

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

@@ -0,0 +1,76 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingDeploymentsGet } from "../funcs/servingDeploymentsGet.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 ServingDeploymentsGetQueryData = components.Deployment;
+
+export function prefetchServingDeploymentsGet(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetDeploymentRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingDeploymentsGetQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingDeploymentsGetQuery(
+  client$: SDKCore,
+  request: operations.GetDeploymentRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingDeploymentsGetQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingDeploymentsGet(request.deploymentId, {
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingDeploymentsGetQueryFn(
+      ctx,
+    ): Promise<ServingDeploymentsGetQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingDeploymentsGet(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingDeploymentsGet(
+  deploymentId: string,
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "deployments", "get", deploymentId, parameters];
+}