@meetkai/mka1 0.48.34 → 0.48.36

package/src/react-query/llmModelsListEffectivePrices.ts

@@ -0,0 +1,143 @@
+/*
+ * 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 {
+  buildLlmModelsListEffectivePricesQuery,
+  LlmModelsListEffectivePricesQueryData,
+  prefetchLlmModelsListEffectivePrices,
+  queryKeyLlmModelsListEffectivePrices,
+} from "./llmModelsListEffectivePrices.core.js";
+export {
+  buildLlmModelsListEffectivePricesQuery,
+  type LlmModelsListEffectivePricesQueryData,
+  prefetchLlmModelsListEffectivePrices,
+  queryKeyLlmModelsListEffectivePrices,
+};
+
+export type LlmModelsListEffectivePricesQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List effective prices for the caller's models
+ *
+ * @remarks
+ * For every model the caller's organization can use, returns the price that spend is actually rated at: the org's override if one exists, else the cluster default, else unpriced ($0). Cluster admins see the full catalog priced against their own org scope.
+ */
+export function useLlmModelsListEffectivePrices(
+  request: operations.ListEffectivePricesRequest,
+  options?: QueryHookOptions<
+    LlmModelsListEffectivePricesQueryData,
+    LlmModelsListEffectivePricesQueryError
+  >,
+): UseQueryResult<
+  LlmModelsListEffectivePricesQueryData,
+  LlmModelsListEffectivePricesQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsListEffectivePricesQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List effective prices for the caller's models
+ *
+ * @remarks
+ * For every model the caller's organization can use, returns the price that spend is actually rated at: the org's override if one exists, else the cluster default, else unpriced ($0). Cluster admins see the full catalog priced against their own org scope.
+ */
+export function useLlmModelsListEffectivePricesSuspense(
+  request: operations.ListEffectivePricesRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsListEffectivePricesQueryData,
+    LlmModelsListEffectivePricesQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsListEffectivePricesQueryData,
+  LlmModelsListEffectivePricesQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsListEffectivePricesQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsListEffectivePricesData(
+  client: QueryClient,
+  queryKeyBase: [parameters: { xOnBehalfOf?: string | undefined }],
+  data: LlmModelsListEffectivePricesQueryData,
+): LlmModelsListEffectivePricesQueryData | undefined {
+  const key = queryKeyLlmModelsListEffectivePrices(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsListEffectivePricesQueryData>(key, data);
+}
+
+export function invalidateLlmModelsListEffectivePrices(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [parameters: { xOnBehalfOf?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "models",
+      "listEffectivePrices",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllLlmModelsListEffectivePrices(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "listEffectivePrices"],
+  });
+}

package/src/react-query/llmModelsRemoveClusterRegistryModel.ts

@@ -0,0 +1,110 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsRemoveClusterRegistryModel } from "../funcs/llmModelsRemoveClusterRegistryModel.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 * 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 LlmModelsRemoveClusterRegistryModelMutationVariables = {
+  request: operations.RemoveClusterRegistryModelRequest;
+  options?: RequestOptions;
+};
+
+export type LlmModelsRemoveClusterRegistryModelMutationData =
+  components.DeleteRegistryModelResponse;
+
+export type LlmModelsRemoveClusterRegistryModelMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Remove a cluster-owned model
+ *
+ * @remarks
+ * Removes an operator-owned, API-added model and cleans up its cluster-default price and all per-org grants for the id. Cluster admin only.
+ */
+export function useLlmModelsRemoveClusterRegistryModelMutation(
+  options?: MutationHookOptions<
+    LlmModelsRemoveClusterRegistryModelMutationData,
+    LlmModelsRemoveClusterRegistryModelMutationError,
+    LlmModelsRemoveClusterRegistryModelMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsRemoveClusterRegistryModelMutationData,
+  LlmModelsRemoveClusterRegistryModelMutationError,
+  LlmModelsRemoveClusterRegistryModelMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsRemoveClusterRegistryModelMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsRemoveClusterRegistryModel(): MutationKey {
+  return ["@meetkai/mka1", "models", "removeClusterRegistryModel"];
+}
+
+export function buildLlmModelsRemoveClusterRegistryModelMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsRemoveClusterRegistryModelMutationVariables,
+  ) => Promise<LlmModelsRemoveClusterRegistryModelMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsRemoveClusterRegistryModel(),
+    mutationFn: function llmModelsRemoveClusterRegistryModelMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsRemoveClusterRegistryModelMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsRemoveClusterRegistryModel(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/llmModelsUpdateClusterRegistryModel.ts

@@ -0,0 +1,110 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsUpdateClusterRegistryModel } from "../funcs/llmModelsUpdateClusterRegistryModel.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 * 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 LlmModelsUpdateClusterRegistryModelMutationVariables = {
+  request: operations.UpdateClusterRegistryModelRequest;
+  options?: RequestOptions;
+};
+
+export type LlmModelsUpdateClusterRegistryModelMutationData =
+  components.RegistryModelResponse;
+
+export type LlmModelsUpdateClusterRegistryModelMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Update a cluster-owned model
+ *
+ * @remarks
+ * Partially updates an operator-owned, API-added model. Only provided fields change. Cluster admin only.
+ */
+export function useLlmModelsUpdateClusterRegistryModelMutation(
+  options?: MutationHookOptions<
+    LlmModelsUpdateClusterRegistryModelMutationData,
+    LlmModelsUpdateClusterRegistryModelMutationError,
+    LlmModelsUpdateClusterRegistryModelMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsUpdateClusterRegistryModelMutationData,
+  LlmModelsUpdateClusterRegistryModelMutationError,
+  LlmModelsUpdateClusterRegistryModelMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsUpdateClusterRegistryModelMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsUpdateClusterRegistryModel(): MutationKey {
+  return ["@meetkai/mka1", "models", "updateClusterRegistryModel"];
+}
+
+export function buildLlmModelsUpdateClusterRegistryModelMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsUpdateClusterRegistryModelMutationVariables,
+  ) => Promise<LlmModelsUpdateClusterRegistryModelMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsUpdateClusterRegistryModel(),
+    mutationFn: function llmModelsUpdateClusterRegistryModelMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsUpdateClusterRegistryModelMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsUpdateClusterRegistryModel(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/sdk/models.ts

@@ -2,6 +2,7 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
+import { llmModelsAddClusterRegistryModel } from "../funcs/llmModelsAddClusterRegistryModel.js";
 import { llmModelsAddRegistryModel } from "../funcs/llmModelsAddRegistryModel.js";
 import { llmModelsCheckRegistryModelHealth } from "../funcs/llmModelsCheckRegistryModelHealth.js";
 import { llmModelsClearClusterPrice } from "../funcs/llmModelsClearClusterPrice.js";
@@ -9,20 +10,25 @@ import { llmModelsClearOrgPrice } from "../funcs/llmModelsClearOrgPrice.js";
 import { llmModelsDeleteOrgAutoModel } from "../funcs/llmModelsDeleteOrgAutoModel.js";
 import { llmModelsGet } from "../funcs/llmModelsGet.js";
 import { llmModelsGetClusterCurrency } from "../funcs/llmModelsGetClusterCurrency.js";
+import { llmModelsGetClusterRegistryModel } from "../funcs/llmModelsGetClusterRegistryModel.js";
 import { llmModelsGetRegistryModel } from "../funcs/llmModelsGetRegistryModel.js";
 import { llmModelsList } from "../funcs/llmModelsList.js";
 import { llmModelsListClusterPrices } from "../funcs/llmModelsListClusterPrices.js";
 import { llmModelsListClusterRegistry } from "../funcs/llmModelsListClusterRegistry.js";
+import { llmModelsListClusterRegistryModels } from "../funcs/llmModelsListClusterRegistryModels.js";
 import { llmModelsListClusterRegistryOrgGrants } from "../funcs/llmModelsListClusterRegistryOrgGrants.js";
+import { llmModelsListEffectivePrices } from "../funcs/llmModelsListEffectivePrices.js";
 import { llmModelsListOrgAutoModels } from "../funcs/llmModelsListOrgAutoModels.js";
 import { llmModelsListOrgPrices } from "../funcs/llmModelsListOrgPrices.js";
 import { llmModelsListRegistryModels } from "../funcs/llmModelsListRegistryModels.js";
 import { llmModelsPutClusterRegistryOrgGrants } from "../funcs/llmModelsPutClusterRegistryOrgGrants.js";
 import { llmModelsPutOrgAutoModel } from "../funcs/llmModelsPutOrgAutoModel.js";
+import { llmModelsRemoveClusterRegistryModel } from "../funcs/llmModelsRemoveClusterRegistryModel.js";
 import { llmModelsRemoveRegistryModel } from "../funcs/llmModelsRemoveRegistryModel.js";
 import { llmModelsSetClusterCurrency } from "../funcs/llmModelsSetClusterCurrency.js";
 import { llmModelsSetClusterPrice } from "../funcs/llmModelsSetClusterPrice.js";
 import { llmModelsSetOrgPrice } from "../funcs/llmModelsSetOrgPrice.js";
+import { llmModelsUpdateClusterRegistryModel } from "../funcs/llmModelsUpdateClusterRegistryModel.js";
 import { llmModelsUpdateRegistryModel } from "../funcs/llmModelsUpdateRegistryModel.js";
 import { ClientSDK, RequestOptions } from "../lib/sdks.js";
 import * as components from "../models/components/index.js";
@@ -164,6 +170,89 @@ export class Models extends ClientSDK {
     ));
   }
 
+  /**
+   * 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.
+   */
+  async addClusterRegistryModel(
+    request: components.AddRegistryModelRequest,
+    options?: RequestOptions,
+  ): Promise<components.RegistryModelResponse> {
+    return unwrapAsync(llmModelsAddClusterRegistryModel(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List cluster-owned models
+   *
+   * @remarks
+   * Lists operator-owned, API-added models (the grantable, database-sourced ones). Cluster admin only.
+   */
+  async listClusterRegistryModels(
+    options?: RequestOptions,
+  ): Promise<components.RegistryModelListResponse> {
+    return unwrapAsync(llmModelsListClusterRegistryModels(
+      this,
+      options,
+    ));
+  }
+
+  /**
+   * Get a cluster-owned model
+   *
+   * @remarks
+   * Returns a single operator-owned, API-added model. Cluster admin only.
+   */
+  async getClusterRegistryModel(
+    request: operations.GetClusterRegistryModelRequest,
+    options?: RequestOptions,
+  ): Promise<components.RegistryModelResponse> {
+    return unwrapAsync(llmModelsGetClusterRegistryModel(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Update a cluster-owned model
+   *
+   * @remarks
+   * Partially updates an operator-owned, API-added model. Only provided fields change. Cluster admin only.
+   */
+  async updateClusterRegistryModel(
+    request: operations.UpdateClusterRegistryModelRequest,
+    options?: RequestOptions,
+  ): Promise<components.RegistryModelResponse> {
+    return unwrapAsync(llmModelsUpdateClusterRegistryModel(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Remove a cluster-owned model
+   *
+   * @remarks
+   * Removes an operator-owned, API-added model and cleans up its cluster-default price and all per-org grants for the id. Cluster admin only.
+   */
+  async removeClusterRegistryModel(
+    request: operations.RemoveClusterRegistryModelRequest,
+    options?: RequestOptions,
+  ): Promise<components.DeleteRegistryModelResponse> {
+    return unwrapAsync(llmModelsRemoveClusterRegistryModel(
+      this,
+      request,
+      options,
+    ));
+  }
+
   /**
    * List the cluster registry catalog
    *
@@ -395,4 +484,21 @@ export class Models extends ClientSDK {
       options,
     ));
   }
+
+  /**
+   * List effective prices for the caller's models
+   *
+   * @remarks
+   * For every model the caller's organization can use, returns the price that spend is actually rated at: the org's override if one exists, else the cluster default, else unpriced ($0). Cluster admins see the full catalog priced against their own org scope.
+   */
+  async listEffectivePrices(
+    request: operations.ListEffectivePricesRequest,
+    options?: RequestOptions,
+  ): Promise<components.EffectivePricesResponse> {
+    return unwrapAsync(llmModelsListEffectivePrices(
+      this,
+      request,
+      options,
+    ));
+  }
 }