@meetkai/mka1 0.48.34 → 0.48.36

package/src/react-query/llmModelsAddClusterRegistryModel.ts

@@ -0,0 +1,109 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsAddClusterRegistryModel } from "../funcs/llmModelsAddClusterRegistryModel.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 { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
+import { SDKError } from "../models/errors/sdkerror.js";
+import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
+import { unwrapAsync } from "../types/fp.js";
+import { useSDKContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export type LlmModelsAddClusterRegistryModelMutationVariables = {
+  request: components.AddRegistryModelRequest;
+  options?: RequestOptions;
+};
+
+export type LlmModelsAddClusterRegistryModelMutationData =
+  components.RegistryModelResponse;
+
+export type LlmModelsAddClusterRegistryModelMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Add a cluster-owned model
+ *
+ * @remarks
+ * Adds an operator-owned, database-sourced model that a cluster admin can grant to orgs (see /models/cluster_registry). Cluster admin only.
+ */
+export function useLlmModelsAddClusterRegistryModelMutation(
+  options?: MutationHookOptions<
+    LlmModelsAddClusterRegistryModelMutationData,
+    LlmModelsAddClusterRegistryModelMutationError,
+    LlmModelsAddClusterRegistryModelMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsAddClusterRegistryModelMutationData,
+  LlmModelsAddClusterRegistryModelMutationError,
+  LlmModelsAddClusterRegistryModelMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsAddClusterRegistryModelMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsAddClusterRegistryModel(): MutationKey {
+  return ["@meetkai/mka1", "models", "addClusterRegistryModel"];
+}
+
+export function buildLlmModelsAddClusterRegistryModelMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsAddClusterRegistryModelMutationVariables,
+  ) => Promise<LlmModelsAddClusterRegistryModelMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsAddClusterRegistryModel(),
+    mutationFn: function llmModelsAddClusterRegistryModelMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsAddClusterRegistryModelMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsAddClusterRegistryModel(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

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

@@ -0,0 +1,72 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsGetClusterRegistryModel } from "../funcs/llmModelsGetClusterRegistryModel.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 LlmModelsGetClusterRegistryModelQueryData =
+  components.RegistryModelResponse;
+
+export function prefetchLlmModelsGetClusterRegistryModel(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetClusterRegistryModelRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsGetClusterRegistryModelQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsGetClusterRegistryModelQuery(
+  client$: SDKCore,
+  request: operations.GetClusterRegistryModelRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsGetClusterRegistryModelQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsGetClusterRegistryModel(request.id),
+    queryFn: async function llmModelsGetClusterRegistryModelQueryFn(
+      ctx,
+    ): Promise<LlmModelsGetClusterRegistryModelQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsGetClusterRegistryModel(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsGetClusterRegistryModel(id: string): QueryKey {
+  return ["@meetkai/mka1", "models", "getClusterRegistryModel", id];
+}

package/src/react-query/llmModelsGetClusterRegistryModel.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 { 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 {
+  buildLlmModelsGetClusterRegistryModelQuery,
+  LlmModelsGetClusterRegistryModelQueryData,
+  prefetchLlmModelsGetClusterRegistryModel,
+  queryKeyLlmModelsGetClusterRegistryModel,
+} from "./llmModelsGetClusterRegistryModel.core.js";
+export {
+  buildLlmModelsGetClusterRegistryModelQuery,
+  type LlmModelsGetClusterRegistryModelQueryData,
+  prefetchLlmModelsGetClusterRegistryModel,
+  queryKeyLlmModelsGetClusterRegistryModel,
+};
+
+export type LlmModelsGetClusterRegistryModelQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get a cluster-owned model
+ *
+ * @remarks
+ * Returns a single operator-owned, API-added model. Cluster admin only.
+ */
+export function useLlmModelsGetClusterRegistryModel(
+  request: operations.GetClusterRegistryModelRequest,
+  options?: QueryHookOptions<
+    LlmModelsGetClusterRegistryModelQueryData,
+    LlmModelsGetClusterRegistryModelQueryError
+  >,
+): UseQueryResult<
+  LlmModelsGetClusterRegistryModelQueryData,
+  LlmModelsGetClusterRegistryModelQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsGetClusterRegistryModelQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get a cluster-owned model
+ *
+ * @remarks
+ * Returns a single operator-owned, API-added model. Cluster admin only.
+ */
+export function useLlmModelsGetClusterRegistryModelSuspense(
+  request: operations.GetClusterRegistryModelRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsGetClusterRegistryModelQueryData,
+    LlmModelsGetClusterRegistryModelQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsGetClusterRegistryModelQueryData,
+  LlmModelsGetClusterRegistryModelQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsGetClusterRegistryModelQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsGetClusterRegistryModelData(
+  client: QueryClient,
+  queryKeyBase: [id: string],
+  data: LlmModelsGetClusterRegistryModelQueryData,
+): LlmModelsGetClusterRegistryModelQueryData | undefined {
+  const key = queryKeyLlmModelsGetClusterRegistryModel(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsGetClusterRegistryModelQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateLlmModelsGetClusterRegistryModel(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[id: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "models",
+      "getClusterRegistryModel",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllLlmModelsGetClusterRegistryModel(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getClusterRegistryModel"],
+  });
+}

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

@@ -0,0 +1,67 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsListClusterRegistryModels } from "../funcs/llmModelsListClusterRegistryModels.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as components from "../models/components/index.js";
+import { unwrapAsync } from "../types/fp.js";
+export type LlmModelsListClusterRegistryModelsQueryData =
+  components.RegistryModelListResponse;
+
+export function prefetchLlmModelsListClusterRegistryModels(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsListClusterRegistryModelsQuery(
+      client$,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsListClusterRegistryModelsQuery(
+  client$: SDKCore,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsListClusterRegistryModelsQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsListClusterRegistryModels(),
+    queryFn: async function llmModelsListClusterRegistryModelsQueryFn(
+      ctx,
+    ): Promise<LlmModelsListClusterRegistryModelsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsListClusterRegistryModels(
+        client$,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsListClusterRegistryModels(): QueryKey {
+  return ["@meetkai/mka1", "models", "listClusterRegistryModels"];
+}

package/src/react-query/llmModelsListClusterRegistryModels.ts

@@ -0,0 +1,118 @@
+/*
+ * 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 { useSDKContext } from "./_context.js";
+import { QueryHookOptions, SuspenseQueryHookOptions } from "./_types.js";
+import {
+  buildLlmModelsListClusterRegistryModelsQuery,
+  LlmModelsListClusterRegistryModelsQueryData,
+  prefetchLlmModelsListClusterRegistryModels,
+  queryKeyLlmModelsListClusterRegistryModels,
+} from "./llmModelsListClusterRegistryModels.core.js";
+export {
+  buildLlmModelsListClusterRegistryModelsQuery,
+  type LlmModelsListClusterRegistryModelsQueryData,
+  prefetchLlmModelsListClusterRegistryModels,
+  queryKeyLlmModelsListClusterRegistryModels,
+};
+
+export type LlmModelsListClusterRegistryModelsQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List cluster-owned models
+ *
+ * @remarks
+ * Lists operator-owned, API-added models (the grantable, database-sourced ones). Cluster admin only.
+ */
+export function useLlmModelsListClusterRegistryModels(
+  options?: QueryHookOptions<
+    LlmModelsListClusterRegistryModelsQueryData,
+    LlmModelsListClusterRegistryModelsQueryError
+  >,
+): UseQueryResult<
+  LlmModelsListClusterRegistryModelsQueryData,
+  LlmModelsListClusterRegistryModelsQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsListClusterRegistryModelsQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List cluster-owned models
+ *
+ * @remarks
+ * Lists operator-owned, API-added models (the grantable, database-sourced ones). Cluster admin only.
+ */
+export function useLlmModelsListClusterRegistryModelsSuspense(
+  options?: SuspenseQueryHookOptions<
+    LlmModelsListClusterRegistryModelsQueryData,
+    LlmModelsListClusterRegistryModelsQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsListClusterRegistryModelsQueryData,
+  LlmModelsListClusterRegistryModelsQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsListClusterRegistryModelsQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsListClusterRegistryModelsData(
+  client: QueryClient,
+  data: LlmModelsListClusterRegistryModelsQueryData,
+): LlmModelsListClusterRegistryModelsQueryData | undefined {
+  const key = queryKeyLlmModelsListClusterRegistryModels();
+
+  return client.setQueryData<LlmModelsListClusterRegistryModelsQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateAllLlmModelsListClusterRegistryModels(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "listClusterRegistryModels"],
+  });
+}

package/src/react-query/llmModelsListEffectivePrices.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 { llmModelsListEffectivePrices } from "../funcs/llmModelsListEffectivePrices.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 LlmModelsListEffectivePricesQueryData =
+  components.EffectivePricesResponse;
+
+export function prefetchLlmModelsListEffectivePrices(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListEffectivePricesRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsListEffectivePricesQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsListEffectivePricesQuery(
+  client$: SDKCore,
+  request: operations.ListEffectivePricesRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsListEffectivePricesQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsListEffectivePrices({
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function llmModelsListEffectivePricesQueryFn(
+      ctx,
+    ): Promise<LlmModelsListEffectivePricesQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsListEffectivePrices(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsListEffectivePrices(
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "models", "listEffectivePrices", parameters];
+}