@meetkai/mka1 0.48.32 → 0.48.34

package/src/react-query/servingFinetuneJobsCreate.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 { servingFinetuneJobsCreate } from "../funcs/servingFinetuneJobsCreate.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 ServingFinetuneJobsCreateMutationVariables = {
+  request: operations.CreateFinetuneJobRequest;
+  options?: RequestOptions;
+};
+
+export type ServingFinetuneJobsCreateMutationData = components.FinetuneJob;
+
+export type ServingFinetuneJobsCreateMutationError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Submit a fine-tune job
+ *
+ * @remarks
+ * Submit a fine-tune job for a base model and dataset, and return its id.
+ */
+export function useServingFinetuneJobsCreateMutation(
+  options?: MutationHookOptions<
+    ServingFinetuneJobsCreateMutationData,
+    ServingFinetuneJobsCreateMutationError,
+    ServingFinetuneJobsCreateMutationVariables
+  >,
+): UseMutationResult<
+  ServingFinetuneJobsCreateMutationData,
+  ServingFinetuneJobsCreateMutationError,
+  ServingFinetuneJobsCreateMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildServingFinetuneJobsCreateMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyServingFinetuneJobsCreate(): MutationKey {
+  return ["@meetkai/mka1", "finetuneJobs", "create"];
+}
+
+export function buildServingFinetuneJobsCreateMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ServingFinetuneJobsCreateMutationVariables,
+  ) => Promise<ServingFinetuneJobsCreateMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyServingFinetuneJobsCreate(),
+    mutationFn: function servingFinetuneJobsCreateMutationFn({
+      request,
+      options,
+    }): Promise<ServingFinetuneJobsCreateMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(servingFinetuneJobsCreate(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/servingFinetuneJobsGet.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 { servingFinetuneJobsGet } from "../funcs/servingFinetuneJobsGet.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 ServingFinetuneJobsGetQueryData = components.FinetuneJob;
+
+export function prefetchServingFinetuneJobsGet(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetFinetuneJobRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingFinetuneJobsGetQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingFinetuneJobsGetQuery(
+  client$: SDKCore,
+  request: operations.GetFinetuneJobRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingFinetuneJobsGetQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingFinetuneJobsGet(request.jobId, {
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingFinetuneJobsGetQueryFn(
+      ctx,
+    ): Promise<ServingFinetuneJobsGetQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingFinetuneJobsGet(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingFinetuneJobsGet(
+  jobId: string,
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "finetuneJobs", "get", jobId, parameters];
+}

package/src/react-query/servingFinetuneJobsGet.ts

@@ -0,0 +1,144 @@
+/*
+ * 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 {
+  buildServingFinetuneJobsGetQuery,
+  prefetchServingFinetuneJobsGet,
+  queryKeyServingFinetuneJobsGet,
+  ServingFinetuneJobsGetQueryData,
+} from "./servingFinetuneJobsGet.core.js";
+export {
+  buildServingFinetuneJobsGetQuery,
+  prefetchServingFinetuneJobsGet,
+  queryKeyServingFinetuneJobsGet,
+  type ServingFinetuneJobsGetQueryData,
+};
+
+export type ServingFinetuneJobsGetQueryError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get a fine-tune job
+ *
+ * @remarks
+ * Retrieve a fine-tune job, including its status and progress.
+ */
+export function useServingFinetuneJobsGet(
+  request: operations.GetFinetuneJobRequest,
+  options?: QueryHookOptions<
+    ServingFinetuneJobsGetQueryData,
+    ServingFinetuneJobsGetQueryError
+  >,
+): UseQueryResult<
+  ServingFinetuneJobsGetQueryData,
+  ServingFinetuneJobsGetQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingFinetuneJobsGetQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get a fine-tune job
+ *
+ * @remarks
+ * Retrieve a fine-tune job, including its status and progress.
+ */
+export function useServingFinetuneJobsGetSuspense(
+  request: operations.GetFinetuneJobRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingFinetuneJobsGetQueryData,
+    ServingFinetuneJobsGetQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingFinetuneJobsGetQueryData,
+  ServingFinetuneJobsGetQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingFinetuneJobsGetQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingFinetuneJobsGetData(
+  client: QueryClient,
+  queryKeyBase: [
+    jobId: string,
+    parameters: { xOnBehalfOf?: string | undefined },
+  ],
+  data: ServingFinetuneJobsGetQueryData,
+): ServingFinetuneJobsGetQueryData | undefined {
+  const key = queryKeyServingFinetuneJobsGet(...queryKeyBase);
+
+  return client.setQueryData<ServingFinetuneJobsGetQueryData>(key, data);
+}
+
+export function invalidateServingFinetuneJobsGet(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [jobId: string, parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "finetuneJobs", "get", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingFinetuneJobsGet(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "finetuneJobs", "get"],
+  });
+}

package/src/react-query/servingFinetuneJobsGetEvents.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 { servingFinetuneJobsGetEvents } from "../funcs/servingFinetuneJobsGetEvents.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 ServingFinetuneJobsGetEventsQueryData = Array<components.JobEvent>;
+
+export function prefetchServingFinetuneJobsGetEvents(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetFinetuneEventsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingFinetuneJobsGetEventsQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingFinetuneJobsGetEventsQuery(
+  client$: SDKCore,
+  request: operations.GetFinetuneEventsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingFinetuneJobsGetEventsQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingFinetuneJobsGetEvents(request.jobId, {
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingFinetuneJobsGetEventsQueryFn(
+      ctx,
+    ): Promise<ServingFinetuneJobsGetEventsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingFinetuneJobsGetEvents(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingFinetuneJobsGetEvents(
+  jobId: string,
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "finetuneJobs", "getEvents", jobId, parameters];
+}

package/src/react-query/servingFinetuneJobsGetEvents.ts

@@ -0,0 +1,144 @@
+/*
+ * 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 {
+  buildServingFinetuneJobsGetEventsQuery,
+  prefetchServingFinetuneJobsGetEvents,
+  queryKeyServingFinetuneJobsGetEvents,
+  ServingFinetuneJobsGetEventsQueryData,
+} from "./servingFinetuneJobsGetEvents.core.js";
+export {
+  buildServingFinetuneJobsGetEventsQuery,
+  prefetchServingFinetuneJobsGetEvents,
+  queryKeyServingFinetuneJobsGetEvents,
+  type ServingFinetuneJobsGetEventsQueryData,
+};
+
+export type ServingFinetuneJobsGetEventsQueryError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Status timeline
+ *
+ * @remarks
+ * Return the job's status timeline (queued, running, checkpoints, completion).
+ */
+export function useServingFinetuneJobsGetEvents(
+  request: operations.GetFinetuneEventsRequest,
+  options?: QueryHookOptions<
+    ServingFinetuneJobsGetEventsQueryData,
+    ServingFinetuneJobsGetEventsQueryError
+  >,
+): UseQueryResult<
+  ServingFinetuneJobsGetEventsQueryData,
+  ServingFinetuneJobsGetEventsQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildServingFinetuneJobsGetEventsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Status timeline
+ *
+ * @remarks
+ * Return the job's status timeline (queued, running, checkpoints, completion).
+ */
+export function useServingFinetuneJobsGetEventsSuspense(
+  request: operations.GetFinetuneEventsRequest,
+  options?: SuspenseQueryHookOptions<
+    ServingFinetuneJobsGetEventsQueryData,
+    ServingFinetuneJobsGetEventsQueryError
+  >,
+): UseSuspenseQueryResult<
+  ServingFinetuneJobsGetEventsQueryData,
+  ServingFinetuneJobsGetEventsQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildServingFinetuneJobsGetEventsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setServingFinetuneJobsGetEventsData(
+  client: QueryClient,
+  queryKeyBase: [
+    jobId: string,
+    parameters: { xOnBehalfOf?: string | undefined },
+  ],
+  data: ServingFinetuneJobsGetEventsQueryData,
+): ServingFinetuneJobsGetEventsQueryData | undefined {
+  const key = queryKeyServingFinetuneJobsGetEvents(...queryKeyBase);
+
+  return client.setQueryData<ServingFinetuneJobsGetEventsQueryData>(key, data);
+}
+
+export function invalidateServingFinetuneJobsGetEvents(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [jobId: string, parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "finetuneJobs", "getEvents", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllServingFinetuneJobsGetEvents(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "finetuneJobs", "getEvents"],
+  });
+}

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

@@ -0,0 +1,77 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingFinetuneJobsGetLogs } from "../funcs/servingFinetuneJobsGetLogs.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 ServingFinetuneJobsGetLogsQueryData = Array<components.LogLine>;
+
+export function prefetchServingFinetuneJobsGetLogs(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetFinetuneLogsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingFinetuneJobsGetLogsQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingFinetuneJobsGetLogsQuery(
+  client$: SDKCore,
+  request: operations.GetFinetuneLogsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingFinetuneJobsGetLogsQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingFinetuneJobsGetLogs(request.jobId, {
+      tail: request.tail,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingFinetuneJobsGetLogsQueryFn(
+      ctx,
+    ): Promise<ServingFinetuneJobsGetLogsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingFinetuneJobsGetLogs(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingFinetuneJobsGetLogs(
+  jobId: string,
+  parameters: { tail?: number | undefined; xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "finetuneJobs", "getLogs", jobId, parameters];
+}