@meetkai/mka1 0.48.34 → 0.48.35

package/src/models/operations/listeffectiveprices.ts

@@ -0,0 +1,63 @@
+/*
+ * 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 { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type ListEffectivePricesRequest = {
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const ListEffectivePricesRequest$inboundSchema: z.ZodType<
+  ListEffectivePricesRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type ListEffectivePricesRequest$Outbound = {
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const ListEffectivePricesRequest$outboundSchema: z.ZodType<
+  ListEffectivePricesRequest$Outbound,
+  z.ZodTypeDef,
+  ListEffectivePricesRequest
+> = z.object({
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function listEffectivePricesRequestToJSON(
+  listEffectivePricesRequest: ListEffectivePricesRequest,
+): string {
+  return JSON.stringify(
+    ListEffectivePricesRequest$outboundSchema.parse(listEffectivePricesRequest),
+  );
+}
+export function listEffectivePricesRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<ListEffectivePricesRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ListEffectivePricesRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ListEffectivePricesRequest' from JSON`,
+  );
+}

package/src/models/operations/removeclusterregistrymodel.ts

@@ -0,0 +1,53 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type RemoveClusterRegistryModelRequest = {
+  id: string;
+};
+
+/** @internal */
+export const RemoveClusterRegistryModelRequest$inboundSchema: z.ZodType<
+  RemoveClusterRegistryModelRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  id: z.string(),
+});
+/** @internal */
+export type RemoveClusterRegistryModelRequest$Outbound = {
+  id: string;
+};
+
+/** @internal */
+export const RemoveClusterRegistryModelRequest$outboundSchema: z.ZodType<
+  RemoveClusterRegistryModelRequest$Outbound,
+  z.ZodTypeDef,
+  RemoveClusterRegistryModelRequest
+> = z.object({
+  id: z.string(),
+});
+
+export function removeClusterRegistryModelRequestToJSON(
+  removeClusterRegistryModelRequest: RemoveClusterRegistryModelRequest,
+): string {
+  return JSON.stringify(
+    RemoveClusterRegistryModelRequest$outboundSchema.parse(
+      removeClusterRegistryModelRequest,
+    ),
+  );
+}
+export function removeClusterRegistryModelRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<RemoveClusterRegistryModelRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => RemoveClusterRegistryModelRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'RemoveClusterRegistryModelRequest' from JSON`,
+  );
+}

package/src/models/operations/updateclusterregistrymodel.ts

@@ -0,0 +1,69 @@
+/*
+ * 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 UpdateClusterRegistryModelRequest = {
+  id: string;
+  updateRegistryModelRequest: components.UpdateRegistryModelRequest;
+};
+
+/** @internal */
+export const UpdateClusterRegistryModelRequest$inboundSchema: z.ZodType<
+  UpdateClusterRegistryModelRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  id: z.string(),
+  UpdateRegistryModelRequest:
+    components.UpdateRegistryModelRequest$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "UpdateRegistryModelRequest": "updateRegistryModelRequest",
+  });
+});
+/** @internal */
+export type UpdateClusterRegistryModelRequest$Outbound = {
+  id: string;
+  UpdateRegistryModelRequest: components.UpdateRegistryModelRequest$Outbound;
+};
+
+/** @internal */
+export const UpdateClusterRegistryModelRequest$outboundSchema: z.ZodType<
+  UpdateClusterRegistryModelRequest$Outbound,
+  z.ZodTypeDef,
+  UpdateClusterRegistryModelRequest
+> = z.object({
+  id: z.string(),
+  updateRegistryModelRequest:
+    components.UpdateRegistryModelRequest$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    updateRegistryModelRequest: "UpdateRegistryModelRequest",
+  });
+});
+
+export function updateClusterRegistryModelRequestToJSON(
+  updateClusterRegistryModelRequest: UpdateClusterRegistryModelRequest,
+): string {
+  return JSON.stringify(
+    UpdateClusterRegistryModelRequest$outboundSchema.parse(
+      updateClusterRegistryModelRequest,
+    ),
+  );
+}
+export function updateClusterRegistryModelRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateClusterRegistryModelRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateClusterRegistryModelRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateClusterRegistryModelRequest' from JSON`,
+  );
+}

package/src/react-query/index.ts

@@ -170,6 +170,7 @@ export * from "./llmMemoryStoresList.js";
 export * from "./llmMemoryStoresListEntries.js";
 export * from "./llmMemoryStoresUpdate.js";
 export * from "./llmMemoryStoresUpdateEntry.js";
+export * from "./llmModelsAddClusterRegistryModel.js";
 export * from "./llmModelsAddRegistryModel.js";
 export * from "./llmModelsCheckRegistryModelHealth.js";
 export * from "./llmModelsClearClusterPrice.js";
@@ -177,20 +178,25 @@ export * from "./llmModelsClearOrgPrice.js";
 export * from "./llmModelsDeleteOrgAutoModel.js";
 export * from "./llmModelsGet.js";
 export * from "./llmModelsGetClusterCurrency.js";
+export * from "./llmModelsGetClusterRegistryModel.js";
 export * from "./llmModelsGetRegistryModel.js";
 export * from "./llmModelsList.js";
 export * from "./llmModelsListClusterPrices.js";
 export * from "./llmModelsListClusterRegistry.js";
+export * from "./llmModelsListClusterRegistryModels.js";
 export * from "./llmModelsListClusterRegistryOrgGrants.js";
+export * from "./llmModelsListEffectivePrices.js";
 export * from "./llmModelsListOrgAutoModels.js";
 export * from "./llmModelsListOrgPrices.js";
 export * from "./llmModelsListRegistryModels.js";
 export * from "./llmModelsPutClusterRegistryOrgGrants.js";
 export * from "./llmModelsPutOrgAutoModel.js";
+export * from "./llmModelsRemoveClusterRegistryModel.js";
 export * from "./llmModelsRemoveRegistryModel.js";
 export * from "./llmModelsSetClusterCurrency.js";
 export * from "./llmModelsSetClusterPrice.js";
 export * from "./llmModelsSetOrgPrice.js";
+export * from "./llmModelsUpdateClusterRegistryModel.js";
 export * from "./llmModelsUpdateRegistryModel.js";
 export * from "./llmPromptsCreate.js";
 export * from "./llmPromptsCreateVersion.js";

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"];
+}