@meetkai/mka1 0.51.17 → 0.51.19

package/src/react-query/servingDatasetsGet.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 {
+  buildServingDatasetsGetQuery,
+  prefetchServingDatasetsGet,
+  queryKeyServingDatasetsGet,
+  ServingDatasetsGetQueryData,
+} from "./servingDatasetsGet.core.js";
+export {
+  buildServingDatasetsGetQuery,
+  prefetchServingDatasetsGet,
+  queryKeyServingDatasetsGet,
+  type ServingDatasetsGetQueryData,
+};
+
+export type ServingDatasetsGetQueryError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get a dataset
+ *
+ * @remarks
+ * Retrieve a dataset, including its immutable manifest once committed.
+ */
+export function useServingDatasetsGet(
+  request: operations.GetDatasetRequest,
+  options?: QueryHookOptions<
+    ServingDatasetsGetQueryData,
+    ServingDatasetsGetQueryError
+  >,
+): UseQueryResult<ServingDatasetsGetQueryData, ServingDatasetsGetQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingDatasetsGetQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get a dataset
+ *
+ * @remarks
+ * Retrieve a dataset, including its immutable manifest once committed.
+ */
+export function useServingDatasetsGetSuspense(
+  request: operations.GetDatasetRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingDatasetsGetQueryData,
+    ServingDatasetsGetQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingDatasetsGetQueryData,
+  ServingDatasetsGetQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingDatasetsGetQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingDatasetsGetData(
+  client: QueryClient,
+  queryKeyBase: [
+    datasetId: string,
+    parameters: { xOnBehalfOf?: string | undefined },
+  ],
+  data: ServingDatasetsGetQueryData,
+): ServingDatasetsGetQueryData | undefined {
+  const key = queryKeyServingDatasetsGet(...queryKeyBase);
+
+  return client.setQueryData<ServingDatasetsGetQueryData>(key, data);
+}
+
+export function invalidateServingDatasetsGet(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [datasetId: string, parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "datasets", "get", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingDatasetsGet(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "datasets", "get"],
+  });
+}

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

@@ -0,0 +1,85 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingDatasetsList } from "../funcs/servingDatasetsList.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 ServingDatasetsListQueryData = components.PageDataset;
+
+export function prefetchServingDatasetsList(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListDatasetsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingDatasetsListQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingDatasetsListQuery(
+  client$: SDKCore,
+  request: operations.ListDatasetsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingDatasetsListQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingDatasetsList({
+      cursor: request.cursor,
+      limit: request.limit,
+      name: request.name,
+      status: request.status,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingDatasetsListQueryFn(
+      ctx,
+    ): Promise<ServingDatasetsListQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingDatasetsList(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingDatasetsList(
+  parameters: {
+    cursor?: string | null | undefined;
+    limit?: number | undefined;
+    name?: string | null | undefined;
+    status?: operations.ListDatasetsQueryParamStatus | null | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
+): QueryKey {
+  return ["@meetkai/mka1", "datasets", "list", parameters];
+}

package/src/react-query/servingDatasetsList.ts

@@ -0,0 +1,152 @@
+/*
+ * 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 {
+  buildServingDatasetsListQuery,
+  prefetchServingDatasetsList,
+  queryKeyServingDatasetsList,
+  ServingDatasetsListQueryData,
+} from "./servingDatasetsList.core.js";
+export {
+  buildServingDatasetsListQuery,
+  prefetchServingDatasetsList,
+  queryKeyServingDatasetsList,
+  type ServingDatasetsListQueryData,
+};
+
+export type ServingDatasetsListQueryError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List datasets
+ *
+ * @remarks
+ * List your datasets.
+ */
+export function useServingDatasetsList(
+  request: operations.ListDatasetsRequest,
+  options?: QueryHookOptions<
+    ServingDatasetsListQueryData,
+    ServingDatasetsListQueryError
+  >,
+): UseQueryResult<ServingDatasetsListQueryData, ServingDatasetsListQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingDatasetsListQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List datasets
+ *
+ * @remarks
+ * List your datasets.
+ */
+export function useServingDatasetsListSuspense(
+  request: operations.ListDatasetsRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingDatasetsListQueryData,
+    ServingDatasetsListQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingDatasetsListQueryData,
+  ServingDatasetsListQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingDatasetsListQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingDatasetsListData(
+  client: QueryClient,
+  queryKeyBase: [
+    parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
+      name?: string | null | undefined;
+      status?: operations.ListDatasetsQueryParamStatus | null | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
+  ],
+  data: ServingDatasetsListQueryData,
+): ServingDatasetsListQueryData | undefined {
+  const key = queryKeyServingDatasetsList(...queryKeyBase);
+
+  return client.setQueryData<ServingDatasetsListQueryData>(key, data);
+}
+
+export function invalidateServingDatasetsList(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
+      name?: string | null | undefined;
+      status?: operations.ListDatasetsQueryParamStatus | null | undefined;
+      xOnBehalfOf?: string | undefined;
+    }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "datasets", "list", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingDatasetsList(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "datasets", "list"],
+  });
+}

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

@@ -14,7 +14,7 @@ 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 ServingDeploymentsGetLogsQueryData = Array<components.LogLine>;
+export type ServingDeploymentsGetLogsQueryData = components.PageLogLine;
 
 export function prefetchServingDeploymentsGetLogs(
   queryClient: QueryClient,

package/src/react-query/servingFinetuneJobsCancel.ts

@@ -11,6 +11,7 @@ import { SDKCore } from "../core.js";
 import { servingFinetuneJobsCancel } from "../funcs/servingFinetuneJobsCancel.js";
 import { combineSignals } from "../lib/primitives.js";
 import { RequestOptions } from "../lib/sdks.js";
+import * as components from "../models/components/index.js";
 import {
   ConnectionError,
   InvalidRequestError,
@@ -32,7 +33,7 @@ export type ServingFinetuneJobsCancelMutationVariables = {
   options?: RequestOptions;
 };
 
-export type ServingFinetuneJobsCancelMutationData = void;
+export type ServingFinetuneJobsCancelMutationData = components.FinetuneJob;
 
 export type ServingFinetuneJobsCancelMutationError =
   | errors.ServingError
@@ -50,7 +51,7 @@ export type ServingFinetuneJobsCancelMutationError =
  * Cancel a fine-tune job
  *
  * @remarks
- * Cancel a running fine-tune job.
+ * Cancel a running fine-tune job and return the updated job. Fine-tune records are never deleted: the cancelled job, its events, and its spend history stay readable.
  */
 export function useServingFinetuneJobsCancelMutation(
   options?: MutationHookOptions<

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

@@ -14,7 +14,7 @@ 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 ServingFinetuneJobsGetEventsQueryData = Array<components.JobEvent>;
+export type ServingFinetuneJobsGetEventsQueryData = components.PageJobEvent;
 
 export function prefetchServingFinetuneJobsGetEvents(
   queryClient: QueryClient,
@@ -43,6 +43,8 @@ export function buildServingFinetuneJobsGetEventsQuery(
 } {
   return {
     queryKey: queryKeyServingFinetuneJobsGetEvents(request.jobId, {
+      cursor: request.cursor,
+      limit: request.limit,
       includeLineage: request.includeLineage,
       xOnBehalfOf: request.xOnBehalfOf,
     }),
@@ -72,6 +74,8 @@ export function buildServingFinetuneJobsGetEventsQuery(
 export function queryKeyServingFinetuneJobsGetEvents(
   jobId: string,
   parameters: {
+    cursor?: string | null | undefined;
+    limit?: number | undefined;
     includeLineage?: boolean | undefined;
     xOnBehalfOf?: string | undefined;
   },

package/src/react-query/servingFinetuneJobsGetEvents.ts

@@ -112,6 +112,8 @@ export function setServingFinetuneJobsGetEventsData(
   queryKeyBase: [
     jobId: string,
     parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
       includeLineage?: boolean | undefined;
       xOnBehalfOf?: string | undefined;
     },
@@ -129,6 +131,8 @@ export function invalidateServingFinetuneJobsGetEvents(
     [
       jobId: string,
       parameters: {
+        cursor?: string | null | undefined;
+        limit?: number | undefined;
         includeLineage?: boolean | undefined;
         xOnBehalfOf?: string | undefined;
       },

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

@@ -14,7 +14,7 @@ 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 ServingFinetuneJobsGetLogsQueryData = Array<components.LogLine>;
+export type ServingFinetuneJobsGetLogsQueryData = components.PageLogLine;
 
 export function prefetchServingFinetuneJobsGetLogs(
   queryClient: QueryClient,

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

@@ -14,9 +14,8 @@ 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 ServingFinetuneJobsGetMetricsQueryData = Array<
-  components.TrainingMetricPoint
->;
+export type ServingFinetuneJobsGetMetricsQueryData =
+  components.PageTrainingMetricPoint;
 
 export function prefetchServingFinetuneJobsGetMetrics(
   queryClient: QueryClient,
@@ -45,6 +44,8 @@ export function buildServingFinetuneJobsGetMetricsQuery(
 } {
   return {
     queryKey: queryKeyServingFinetuneJobsGetMetrics(request.jobId, {
+      cursor: request.cursor,
+      limit: request.limit,
       xOnBehalfOf: request.xOnBehalfOf,
     }),
     queryFn: async function servingFinetuneJobsGetMetricsQueryFn(
@@ -72,7 +73,11 @@ export function buildServingFinetuneJobsGetMetricsQuery(
 
 export function queryKeyServingFinetuneJobsGetMetrics(
   jobId: string,
-  parameters: { xOnBehalfOf?: string | undefined },
+  parameters: {
+    cursor?: string | null | undefined;
+    limit?: number | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
 ): QueryKey {
   return ["@meetkai/mka1", "finetuneJobs", "getMetrics", jobId, parameters];
 }

package/src/react-query/servingFinetuneJobsGetMetrics.ts

@@ -111,7 +111,11 @@ export function setServingFinetuneJobsGetMetricsData(
   client: QueryClient,
   queryKeyBase: [
     jobId: string,
-    parameters: { xOnBehalfOf?: string | undefined },
+    parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
   ],
   data: ServingFinetuneJobsGetMetricsQueryData,
 ): ServingFinetuneJobsGetMetricsQueryData | undefined {
@@ -123,7 +127,14 @@ export function setServingFinetuneJobsGetMetricsData(
 export function invalidateServingFinetuneJobsGetMetrics(
   client: QueryClient,
   queryKeyBase: TupleToPrefixes<
-    [jobId: string, parameters: { xOnBehalfOf?: string | undefined }]
+    [
+      jobId: string,
+      parameters: {
+        cursor?: string | null | undefined;
+        limit?: number | undefined;
+        xOnBehalfOf?: string | undefined;
+      },
+    ]
   >,
   filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
 ): Promise<void> {

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

@@ -14,9 +14,8 @@ 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 ServingFinetuneJobsListCheckpointsQueryData = Array<
-  components.Checkpoint
->;
+export type ServingFinetuneJobsListCheckpointsQueryData =
+  components.PageCheckpoint;
 
 export function prefetchServingFinetuneJobsListCheckpoints(
   queryClient: QueryClient,
@@ -45,6 +44,8 @@ export function buildServingFinetuneJobsListCheckpointsQuery(
 } {
   return {
     queryKey: queryKeyServingFinetuneJobsListCheckpoints(request.jobId, {
+      cursor: request.cursor,
+      limit: request.limit,
       xOnBehalfOf: request.xOnBehalfOf,
     }),
     queryFn: async function servingFinetuneJobsListCheckpointsQueryFn(
@@ -72,7 +73,11 @@ export function buildServingFinetuneJobsListCheckpointsQuery(
 
 export function queryKeyServingFinetuneJobsListCheckpoints(
   jobId: string,
-  parameters: { xOnBehalfOf?: string | undefined },
+  parameters: {
+    cursor?: string | null | undefined;
+    limit?: number | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
 ): QueryKey {
   return [
     "@meetkai/mka1",

package/src/react-query/servingFinetuneJobsListCheckpoints.ts

@@ -111,7 +111,11 @@ export function setServingFinetuneJobsListCheckpointsData(
   client: QueryClient,
   queryKeyBase: [
     jobId: string,
-    parameters: { xOnBehalfOf?: string | undefined },
+    parameters: {
+      cursor?: string | null | undefined;
+      limit?: number | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
   ],
   data: ServingFinetuneJobsListCheckpointsQueryData,
 ): ServingFinetuneJobsListCheckpointsQueryData | undefined {
@@ -126,7 +130,14 @@ export function setServingFinetuneJobsListCheckpointsData(
 export function invalidateServingFinetuneJobsListCheckpoints(
   client: QueryClient,
   queryKeyBase: TupleToPrefixes<
-    [jobId: string, parameters: { xOnBehalfOf?: string | undefined }]
+    [
+      jobId: string,
+      parameters: {
+        cursor?: string | null | undefined;
+        limit?: number | undefined;
+        xOnBehalfOf?: string | undefined;
+      },
+    ]
   >,
   filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
 ): Promise<void> {

package/src/react-query/servingModelsDelete.ts

@@ -28,7 +28,7 @@ import { useSDKContext } from "./_context.js";
 import { MutationHookOptions } from "./_types.js";
 
 export type ServingModelsDeleteMutationVariables = {
-  request: operations.DeleteModelRequest;
+  request: operations.DeleteServingModelRequest;
   options?: RequestOptions;
 };
 

package/src/react-query/servingModelsRegister.ts

@@ -29,7 +29,7 @@ import { useSDKContext } from "./_context.js";
 import { MutationHookOptions } from "./_types.js";
 
 export type ServingModelsRegisterMutationVariables = {
-  request: operations.RegisterModelRequest;
+  request: operations.RegisterServingModelRequest;
   options?: RequestOptions;
 };