@meetkai/mka1 0.48.24 → 0.48.26

package/src/react-query/llmModelsListOrgPrices.ts

@@ -0,0 +1,138 @@
+/*
+ * 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 {
+  buildLlmModelsListOrgPricesQuery,
+  LlmModelsListOrgPricesQueryData,
+  prefetchLlmModelsListOrgPrices,
+  queryKeyLlmModelsListOrgPrices,
+} from "./llmModelsListOrgPrices.core.js";
+export {
+  buildLlmModelsListOrgPricesQuery,
+  type LlmModelsListOrgPricesQueryData,
+  prefetchLlmModelsListOrgPrices,
+  queryKeyLlmModelsListOrgPrices,
+};
+
+export type LlmModelsListOrgPricesQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List an org's price overrides
+ *
+ * @remarks
+ * Without `model`, lists the org's current overrides. With `model`, returns that override's dated history.
+ */
+export function useLlmModelsListOrgPrices(
+  request: operations.ListOrgPricesRequest,
+  options?: QueryHookOptions<
+    LlmModelsListOrgPricesQueryData,
+    LlmModelsListOrgPricesQueryError
+  >,
+): UseQueryResult<
+  LlmModelsListOrgPricesQueryData,
+  LlmModelsListOrgPricesQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsListOrgPricesQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List an org's price overrides
+ *
+ * @remarks
+ * Without `model`, lists the org's current overrides. With `model`, returns that override's dated history.
+ */
+export function useLlmModelsListOrgPricesSuspense(
+  request: operations.ListOrgPricesRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsListOrgPricesQueryData,
+    LlmModelsListOrgPricesQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsListOrgPricesQueryData,
+  LlmModelsListOrgPricesQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsListOrgPricesQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsListOrgPricesData(
+  client: QueryClient,
+  queryKeyBase: [orgId: string, parameters: { model?: string | undefined }],
+  data: LlmModelsListOrgPricesQueryData,
+): LlmModelsListOrgPricesQueryData | undefined {
+  const key = queryKeyLlmModelsListOrgPrices(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsListOrgPricesQueryData>(key, data);
+}
+
+export function invalidateLlmModelsListOrgPrices(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [orgId: string, parameters: { model?: string | undefined }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "listOrgPrices", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmModelsListOrgPrices(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "listOrgPrices"],
+  });
+}

package/src/react-query/llmModelsSetClusterCurrency.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 { llmModelsSetClusterCurrency } from "../funcs/llmModelsSetClusterCurrency.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 LlmModelsSetClusterCurrencyMutationVariables = {
+  request: components.PutClusterCurrencyBody;
+  options?: RequestOptions;
+};
+
+export type LlmModelsSetClusterCurrencyMutationData =
+  components.ClusterCurrencyResponse;
+
+export type LlmModelsSetClusterCurrencyMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set the cluster currency
+ *
+ * @remarks
+ * Sets the cluster-wide currency. Changing it does not auto-convert existing prices.
+ */
+export function useLlmModelsSetClusterCurrencyMutation(
+  options?: MutationHookOptions<
+    LlmModelsSetClusterCurrencyMutationData,
+    LlmModelsSetClusterCurrencyMutationError,
+    LlmModelsSetClusterCurrencyMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsSetClusterCurrencyMutationData,
+  LlmModelsSetClusterCurrencyMutationError,
+  LlmModelsSetClusterCurrencyMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsSetClusterCurrencyMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsSetClusterCurrency(): MutationKey {
+  return ["@meetkai/mka1", "models", "setClusterCurrency"];
+}
+
+export function buildLlmModelsSetClusterCurrencyMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsSetClusterCurrencyMutationVariables,
+  ) => Promise<LlmModelsSetClusterCurrencyMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsSetClusterCurrency(),
+    mutationFn: function llmModelsSetClusterCurrencyMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsSetClusterCurrencyMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsSetClusterCurrency(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/llmModelsSetClusterPrice.ts

@@ -0,0 +1,108 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsSetClusterPrice } from "../funcs/llmModelsSetClusterPrice.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 LlmModelsSetClusterPriceMutationVariables = {
+  request: components.PutModelPriceBody;
+  options?: RequestOptions;
+};
+
+export type LlmModelsSetClusterPriceMutationData = components.ModelPriceRow;
+
+export type LlmModelsSetClusterPriceMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set a cluster-default model price
+ *
+ * @remarks
+ * Upserts the cluster-default price for a model as a new dated version. `rates` is a flexible metric→rate map; omitted metrics are not charged.
+ */
+export function useLlmModelsSetClusterPriceMutation(
+  options?: MutationHookOptions<
+    LlmModelsSetClusterPriceMutationData,
+    LlmModelsSetClusterPriceMutationError,
+    LlmModelsSetClusterPriceMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsSetClusterPriceMutationData,
+  LlmModelsSetClusterPriceMutationError,
+  LlmModelsSetClusterPriceMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsSetClusterPriceMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsSetClusterPrice(): MutationKey {
+  return ["@meetkai/mka1", "models", "setClusterPrice"];
+}
+
+export function buildLlmModelsSetClusterPriceMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsSetClusterPriceMutationVariables,
+  ) => Promise<LlmModelsSetClusterPriceMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsSetClusterPrice(),
+    mutationFn: function llmModelsSetClusterPriceMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsSetClusterPriceMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsSetClusterPrice(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/llmModelsSetOrgPrice.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 { llmModelsSetOrgPrice } from "../funcs/llmModelsSetOrgPrice.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 LlmModelsSetOrgPriceMutationVariables = {
+  request: operations.SetOrgPriceRequest;
+  options?: RequestOptions;
+};
+
+export type LlmModelsSetOrgPriceMutationData = components.ModelPriceRow;
+
+export type LlmModelsSetOrgPriceMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set an org price override
+ *
+ * @remarks
+ * Upserts a per-org price override for a model (new dated version). Resolution at spend time is org override → cluster default → $0.
+ */
+export function useLlmModelsSetOrgPriceMutation(
+  options?: MutationHookOptions<
+    LlmModelsSetOrgPriceMutationData,
+    LlmModelsSetOrgPriceMutationError,
+    LlmModelsSetOrgPriceMutationVariables
+  >,
+): UseMutationResult<
+  LlmModelsSetOrgPriceMutationData,
+  LlmModelsSetOrgPriceMutationError,
+  LlmModelsSetOrgPriceMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildLlmModelsSetOrgPriceMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyLlmModelsSetOrgPrice(): MutationKey {
+  return ["@meetkai/mka1", "models", "setOrgPrice"];
+}
+
+export function buildLlmModelsSetOrgPriceMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: LlmModelsSetOrgPriceMutationVariables,
+  ) => Promise<LlmModelsSetOrgPriceMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyLlmModelsSetOrgPrice(),
+    mutationFn: function llmModelsSetOrgPriceMutationFn({
+      request,
+      options,
+    }): Promise<LlmModelsSetOrgPriceMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(llmModelsSetOrgPrice(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

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

@@ -0,0 +1,83 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmUsageCosts } from "../funcs/llmUsageCosts.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 LlmUsageCostsQueryData = components.UsageCostsResponse;
+
+export function prefetchLlmUsageCosts(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetCostsUsageRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmUsageCostsQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmUsageCostsQuery(
+  client$: SDKCore,
+  request: operations.GetCostsUsageRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (context: QueryFunctionContext) => Promise<LlmUsageCostsQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmUsageCosts({
+      startTime: request.startTime,
+      endTime: request.endTime,
+      orgId: request.orgId,
+      groupBy: request.groupBy,
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function llmUsageCostsQueryFn(
+      ctx,
+    ): Promise<LlmUsageCostsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmUsageCosts(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmUsageCosts(
+  parameters: {
+    startTime: number;
+    endTime: number;
+    orgId?: string | undefined;
+    groupBy?: operations.GetCostsUsageQueryParamGroupBy | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
+): QueryKey {
+  return ["@meetkai/mka1", "usage", "costs", parameters];
+}

package/src/react-query/llmUsageCosts.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 {
+  buildLlmUsageCostsQuery,
+  LlmUsageCostsQueryData,
+  prefetchLlmUsageCosts,
+  queryKeyLlmUsageCosts,
+} from "./llmUsageCosts.core.js";
+export {
+  buildLlmUsageCostsQuery,
+  type LlmUsageCostsQueryData,
+  prefetchLlmUsageCosts,
+  queryKeyLlmUsageCosts,
+};
+
+export type LlmUsageCostsQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get cost by model
+ *
+ * @remarks
+ * Aggregates spend by model over a time range, priced in the cluster currency from the model price book. Org-scoped: org admins see their org; cluster admins may pass `org_id`. Includes billable work with no usage row of its own (store=false responses, agent tool calls), so totals reconcile with the live budget gauge.
+ */
+export function useLlmUsageCosts(
+  request: operations.GetCostsUsageRequest,
+  options?: QueryHookOptions<LlmUsageCostsQueryData, LlmUsageCostsQueryError>,
+): UseQueryResult<LlmUsageCostsQueryData, LlmUsageCostsQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmUsageCostsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get cost by model
+ *
+ * @remarks
+ * Aggregates spend by model over a time range, priced in the cluster currency from the model price book. Org-scoped: org admins see their org; cluster admins may pass `org_id`. Includes billable work with no usage row of its own (store=false responses, agent tool calls), so totals reconcile with the live budget gauge.
+ */
+export function useLlmUsageCostsSuspense(
+  request: operations.GetCostsUsageRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmUsageCostsQueryData,
+    LlmUsageCostsQueryError
+  >,
+): UseSuspenseQueryResult<LlmUsageCostsQueryData, LlmUsageCostsQueryError> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmUsageCostsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmUsageCostsData(
+  client: QueryClient,
+  queryKeyBase: [
+    parameters: {
+      startTime: number;
+      endTime: number;
+      orgId?: string | undefined;
+      groupBy?: operations.GetCostsUsageQueryParamGroupBy | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
+  ],
+  data: LlmUsageCostsQueryData,
+): LlmUsageCostsQueryData | undefined {
+  const key = queryKeyLlmUsageCosts(...queryKeyBase);
+
+  return client.setQueryData<LlmUsageCostsQueryData>(key, data);
+}
+
+export function invalidateLlmUsageCosts(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [parameters: {
+      startTime: number;
+      endTime: number;
+      orgId?: string | undefined;
+      groupBy?: operations.GetCostsUsageQueryParamGroupBy | undefined;
+      xOnBehalfOf?: string | undefined;
+    }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "usage", "costs", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmUsageCosts(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "usage", "costs"],
+  });
+}