@meetkai/mka1 0.48.25 → 0.48.26

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

package/src/sdk/budgets.ts

@@ -0,0 +1,170 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmBudgetsApiKeyEvents } from "../funcs/llmBudgetsApiKeyEvents.js";
+import { llmBudgetsDeleteApiKey } from "../funcs/llmBudgetsDeleteApiKey.js";
+import { llmBudgetsDeleteOrg } from "../funcs/llmBudgetsDeleteOrg.js";
+import { llmBudgetsGetApiKey } from "../funcs/llmBudgetsGetApiKey.js";
+import { llmBudgetsGetOrg } from "../funcs/llmBudgetsGetOrg.js";
+import { llmBudgetsList } from "../funcs/llmBudgetsList.js";
+import { llmBudgetsOrgEvents } from "../funcs/llmBudgetsOrgEvents.js";
+import { llmBudgetsSetApiKey } from "../funcs/llmBudgetsSetApiKey.js";
+import { llmBudgetsSetOrg } from "../funcs/llmBudgetsSetOrg.js";
+import { ClientSDK, 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 class Budgets extends ClientSDK {
+  /**
+   * List budgets
+   *
+   * @remarks
+   * Cluster admins see all budgets; org admins see their org's budgets.
+   */
+  async list(
+    options?: RequestOptions,
+  ): Promise<components.ListBudgetsResponse> {
+    return unwrapAsync(llmBudgetsList(
+      this,
+      options,
+    ));
+  }
+
+  /**
+   * List org budgets
+   *
+   * @remarks
+   * Returns the org's budgets (self-budget and/or ceiling) with each one's live spend gauge.
+   */
+  async getOrg(
+    request: operations.GetOrgBudgetsRequest,
+    options?: RequestOptions,
+  ): Promise<components.ListBudgetsResponse> {
+    return unwrapAsync(llmBudgetsGetOrg(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Set a org budget
+   *
+   * @remarks
+   * Creates or updates a budget for the org. One budget per (scope, owner); editing re-evaluates against live spend.
+   */
+  async setOrg(
+    request: operations.SetOrgBudgetRequest,
+    options?: RequestOptions,
+  ): Promise<components.BudgetResponse> {
+    return unwrapAsync(llmBudgetsSetOrg(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Delete a org budget
+   *
+   * @remarks
+   * Removes the org's budget for the given owner; enforcement and alerts stop.
+   */
+  async deleteOrg(
+    request: operations.DeleteOrgBudgetRequest,
+    options?: RequestOptions,
+  ): Promise<components.DeleteBudgetResponse> {
+    return unwrapAsync(llmBudgetsDeleteOrg(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List org budget events
+   *
+   * @remarks
+   * Returns recent threshold-alert and block events for the scope's budgets.
+   */
+  async orgEvents(
+    request: operations.ListOrgBudgetEventsRequest,
+    options?: RequestOptions,
+  ): Promise<components.ListBudgetEventsResponse> {
+    return unwrapAsync(llmBudgetsOrgEvents(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List apikey budgets
+   *
+   * @remarks
+   * Returns the apikey's budgets (self-budget and/or ceiling) with each one's live spend gauge.
+   */
+  async getApiKey(
+    request: operations.GetApiKeyBudgetsRequest,
+    options?: RequestOptions,
+  ): Promise<components.ListBudgetsResponse> {
+    return unwrapAsync(llmBudgetsGetApiKey(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Set a apikey budget
+   *
+   * @remarks
+   * Creates or updates a budget for the apikey. One budget per (scope, owner); editing re-evaluates against live spend.
+   */
+  async setApiKey(
+    request: operations.SetApiKeyBudgetRequest,
+    options?: RequestOptions,
+  ): Promise<components.BudgetResponse> {
+    return unwrapAsync(llmBudgetsSetApiKey(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Delete a apikey budget
+   *
+   * @remarks
+   * Removes the apikey's budget for the given owner; enforcement and alerts stop.
+   */
+  async deleteApiKey(
+    request: operations.DeleteApiKeyBudgetRequest,
+    options?: RequestOptions,
+  ): Promise<components.DeleteBudgetResponse> {
+    return unwrapAsync(llmBudgetsDeleteApiKey(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List apikey budget events
+   *
+   * @remarks
+   * Returns recent threshold-alert and block events for the scope's budgets.
+   */
+  async apiKeyEvents(
+    request: operations.ListApiKeyBudgetEventsRequest,
+    options?: RequestOptions,
+  ): Promise<components.ListBudgetEventsResponse> {
+    return unwrapAsync(llmBudgetsApiKeyEvents(
+      this,
+      request,
+      options,
+    ));
+  }
+}

package/src/sdk/llm.ts

@@ -4,6 +4,7 @@
 
 import { ClientSDK } from "../lib/sdks.js";
 import { Batches } from "./batches.js";
+import { Budgets } from "./budgets.js";
 import { Chat } from "./chat.js";
 import { Classify } from "./classify.js";
 import { Conversations } from "./conversations.js";
@@ -105,6 +106,11 @@ export class Llm extends ClientSDK {
     return (this._usage ??= new Usage(this._options));
   }
 
+  private _budgets?: Budgets;
+  get budgets(): Budgets {
+    return (this._budgets ??= new Budgets(this._options));
+  }
+
   private _classify?: Classify;
   get classify(): Classify {
     return (this._classify ??= new Classify(this._options));

package/src/sdk/models.ts

@@ -4,17 +4,25 @@
 
 import { llmModelsAddRegistryModel } from "../funcs/llmModelsAddRegistryModel.js";
 import { llmModelsCheckRegistryModelHealth } from "../funcs/llmModelsCheckRegistryModelHealth.js";
+import { llmModelsClearClusterPrice } from "../funcs/llmModelsClearClusterPrice.js";
+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 { llmModelsGetRegistryModel } from "../funcs/llmModelsGetRegistryModel.js";
 import { llmModelsList } from "../funcs/llmModelsList.js";
+import { llmModelsListClusterPrices } from "../funcs/llmModelsListClusterPrices.js";
 import { llmModelsListClusterRegistry } from "../funcs/llmModelsListClusterRegistry.js";
 import { llmModelsListClusterRegistryOrgGrants } from "../funcs/llmModelsListClusterRegistryOrgGrants.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 { llmModelsRemoveRegistryModel } from "../funcs/llmModelsRemoveRegistryModel.js";
+import { llmModelsSetClusterCurrency } from "../funcs/llmModelsSetClusterCurrency.js";
+import { llmModelsSetClusterPrice } from "../funcs/llmModelsSetClusterPrice.js";
+import { llmModelsSetOrgPrice } from "../funcs/llmModelsSetOrgPrice.js";
 import { llmModelsUpdateRegistryModel } from "../funcs/llmModelsUpdateRegistryModel.js";
 import { ClientSDK, RequestOptions } from "../lib/sdks.js";
 import * as components from "../models/components/index.js";
@@ -253,4 +261,138 @@ export class Models extends ClientSDK {
       options,
     ));
   }
+
+  /**
+   * Get the cluster currency
+   *
+   * @remarks
+   * Returns the single cluster-wide ISO-4217 currency code (null if not set).
+   */
+  async getClusterCurrency(
+    options?: RequestOptions,
+  ): Promise<components.ClusterCurrencyResponse> {
+    return unwrapAsync(llmModelsGetClusterCurrency(
+      this,
+      options,
+    ));
+  }
+
+  /**
+   * Set the cluster currency
+   *
+   * @remarks
+   * Sets the cluster-wide currency. Changing it does not auto-convert existing prices.
+   */
+  async setClusterCurrency(
+    request: components.PutClusterCurrencyBody,
+    options?: RequestOptions,
+  ): Promise<components.ClusterCurrencyResponse> {
+    return unwrapAsync(llmModelsSetClusterCurrency(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List cluster-default model prices
+   *
+   * @remarks
+   * Without `model`, lists every model's current cluster-default price. With `model`, returns that model's full dated history.
+   */
+  async listClusterPrices(
+    request: operations.ListClusterPricesRequest,
+    options?: RequestOptions,
+  ): Promise<components.ListModelPricesResponse> {
+    return unwrapAsync(llmModelsListClusterPrices(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * 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.
+   */
+  async setClusterPrice(
+    request: components.PutModelPriceBody,
+    options?: RequestOptions,
+  ): Promise<components.ModelPriceRow> {
+    return unwrapAsync(llmModelsSetClusterPrice(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Clear a cluster-default model price
+   *
+   * @remarks
+   * Removes the cluster-default price for a model; it becomes unpriced ($0) unless an org override exists.
+   */
+  async clearClusterPrice(
+    request: operations.ClearClusterPriceRequest,
+    options?: RequestOptions,
+  ): Promise<components.DeleteModelPriceResponse> {
+    return unwrapAsync(llmModelsClearClusterPrice(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List an org's price overrides
+   *
+   * @remarks
+   * Without `model`, lists the org's current overrides. With `model`, returns that override's dated history.
+   */
+  async listOrgPrices(
+    request: operations.ListOrgPricesRequest,
+    options?: RequestOptions,
+  ): Promise<components.ListModelPricesResponse> {
+    return unwrapAsync(llmModelsListOrgPrices(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * 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.
+   */
+  async setOrgPrice(
+    request: operations.SetOrgPriceRequest,
+    options?: RequestOptions,
+  ): Promise<components.ModelPriceRow> {
+    return unwrapAsync(llmModelsSetOrgPrice(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Clear an org price override
+   *
+   * @remarks
+   * Removes a per-org override for a model; resolution falls back to the cluster default.
+   */
+  async clearOrgPrice(
+    request: operations.ClearOrgPriceRequest,
+    options?: RequestOptions,
+  ): Promise<components.DeleteModelPriceResponse> {
+    return unwrapAsync(llmModelsClearOrgPrice(
+      this,
+      request,
+      options,
+    ));
+  }
 }

package/src/sdk/usage.ts

@@ -5,12 +5,14 @@
 import { llmUsageClassify } from "../funcs/llmUsageClassify.js";
 import { llmUsageCompletions } from "../funcs/llmUsageCompletions.js";
 import { llmUsageConversations } from "../funcs/llmUsageConversations.js";
+import { llmUsageCosts } from "../funcs/llmUsageCosts.js";
 import { llmUsageEmbeddings } from "../funcs/llmUsageEmbeddings.js";
 import { llmUsageExtract } from "../funcs/llmUsageExtract.js";
 import { llmUsageFiles } from "../funcs/llmUsageFiles.js";
 import { llmUsageResponses } from "../funcs/llmUsageResponses.js";
 import { llmUsageVectorStores } from "../funcs/llmUsageVectorStores.js";
 import { ClientSDK, 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";
 
@@ -150,4 +152,21 @@ export class Usage extends ClientSDK {
       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.
+   */
+  async costs(
+    request: operations.GetCostsUsageRequest,
+    options?: RequestOptions,
+  ): Promise<components.UsageCostsResponse> {
+    return unwrapAsync(llmUsageCosts(
+      this,
+      request,
+      options,
+    ));
+  }
 }