@meetkai/mka1 0.48.32 → 0.48.33

package/src/react-query/budgetsSetCurrency.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 { budgetsSetCurrency } from "../funcs/budgetsSetCurrency.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 BudgetsSetCurrencyMutationVariables = {
+  request: operations.SetCurrencyRequestBody;
+  options?: RequestOptions;
+};
+
+export type BudgetsSetCurrencyMutationData = operations.SetCurrencyResponseBody;
+
+export type BudgetsSetCurrencyMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set display currency
+ *
+ * @remarks
+ * ISO 4217 code used for display on budgets and cost reports. No conversion is performed anywhere.
+ */
+export function useBudgetsSetCurrencyMutation(
+  options?: MutationHookOptions<
+    BudgetsSetCurrencyMutationData,
+    BudgetsSetCurrencyMutationError,
+    BudgetsSetCurrencyMutationVariables
+  >,
+): UseMutationResult<
+  BudgetsSetCurrencyMutationData,
+  BudgetsSetCurrencyMutationError,
+  BudgetsSetCurrencyMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildBudgetsSetCurrencyMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyBudgetsSetCurrency(): MutationKey {
+  return ["@meetkai/mka1", "budgets", "setCurrency"];
+}
+
+export function buildBudgetsSetCurrencyMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: BudgetsSetCurrencyMutationVariables,
+  ) => Promise<BudgetsSetCurrencyMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyBudgetsSetCurrency(),
+    mutationFn: function budgetsSetCurrencyMutationFn({
+      request,
+      options,
+    }): Promise<BudgetsSetCurrencyMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(budgetsSetCurrency(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/budgetsSetOrg.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 { budgetsSetOrg } from "../funcs/budgetsSetOrg.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 BudgetsSetOrgMutationVariables = {
+  request: operations.SetOrgBudgetRequest;
+  options?: RequestOptions;
+};
+
+export type BudgetsSetOrgMutationData = operations.SetOrgBudgetResponseBody;
+
+export type BudgetsSetOrgMutationError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Set a org budget
+ *
+ * @remarks
+ * Creates or updates a budget for the org. One budget per (scope, owner); editing re-evaluates against live spend.
+ */
+export function useBudgetsSetOrgMutation(
+  options?: MutationHookOptions<
+    BudgetsSetOrgMutationData,
+    BudgetsSetOrgMutationError,
+    BudgetsSetOrgMutationVariables
+  >,
+): UseMutationResult<
+  BudgetsSetOrgMutationData,
+  BudgetsSetOrgMutationError,
+  BudgetsSetOrgMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildBudgetsSetOrgMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyBudgetsSetOrg(): MutationKey {
+  return ["@meetkai/mka1", "budgets", "setOrg"];
+}
+
+export function buildBudgetsSetOrgMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: BudgetsSetOrgMutationVariables,
+  ) => Promise<BudgetsSetOrgMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyBudgetsSetOrg(),
+    mutationFn: function budgetsSetOrgMutationFn({
+      request,
+      options,
+    }): Promise<BudgetsSetOrgMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(budgetsSetOrg(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/index.ts

@@ -62,6 +62,17 @@ export * from "./authTeamsList.js";
 export * from "./authTeamsListMembers.js";
 export * from "./authTeamsRemoveMember.js";
 export * from "./authTeamsUpdate.js";
+export * from "./budgetsApiKeyEvents.js";
+export * from "./budgetsDeleteApiKey.js";
+export * from "./budgetsDeleteOrg.js";
+export * from "./budgetsGetApiKey.js";
+export * from "./budgetsGetCurrency.js";
+export * from "./budgetsGetOrg.js";
+export * from "./budgetsList.js";
+export * from "./budgetsOrgEvents.js";
+export * from "./budgetsSetApiKey.js";
+export * from "./budgetsSetCurrency.js";
+export * from "./budgetsSetOrg.js";
 export * from "./guardrailsDeleteGuardrails.js";
 export * from "./guardrailsGetGuardrails.js";
 export * from "./guardrailsListGuardrailsPolicies.js";
@@ -284,3 +295,4 @@ export * from "./searchTextStoreCreateTextStore.js";
 export * from "./searchTextStoreDeleteTexts.js";
 export * from "./searchTextStoreDeleteTextStore.js";
 export * from "./searchTextStoreSearchTexts.js";
+export * from "./usageCosts.js";

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

@@ -0,0 +1,80 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { usageCosts } from "../funcs/usageCosts.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+export type UsageCostsQueryData = operations.GetCostsResponseBody;
+
+export function prefetchUsageCosts(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetCostsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildUsageCostsQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildUsageCostsQuery(
+  client$: SDKCore,
+  request: operations.GetCostsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (context: QueryFunctionContext) => Promise<UsageCostsQueryData>;
+} {
+  return {
+    queryKey: queryKeyUsageCosts({
+      startTime: request.startTime,
+      endTime: request.endTime,
+      groupBy: request.groupBy,
+      orgId: request.orgId,
+    }),
+    queryFn: async function usageCostsQueryFn(
+      ctx,
+    ): Promise<UsageCostsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(usageCosts(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyUsageCosts(
+  parameters: {
+    startTime: number;
+    endTime: number;
+    groupBy?: string | undefined;
+    orgId?: string | undefined;
+  },
+): QueryKey {
+  return ["@meetkai/mka1", "usage", "costs", parameters];
+}

package/src/react-query/usageCosts.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 {
+  buildUsageCostsQuery,
+  prefetchUsageCosts,
+  queryKeyUsageCosts,
+  UsageCostsQueryData,
+} from "./usageCosts.core.js";
+export {
+  buildUsageCostsQuery,
+  prefetchUsageCosts,
+  queryKeyUsageCosts,
+  type UsageCostsQueryData,
+};
+
+export type UsageCostsQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Aggregate cost by dimension
+ *
+ * @remarks
+ * Sums stored per-entry cost over [start_time, end_time), grouped by any of service, task_type, model, api_key_id, team_id, user_id, external_user_id, org_id. Point-in-time correct for chargeback/showback.
+ */
+export function useUsageCosts(
+  request: operations.GetCostsRequest,
+  options?: QueryHookOptions<UsageCostsQueryData, UsageCostsQueryError>,
+): UseQueryResult<UsageCostsQueryData, UsageCostsQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildUsageCostsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Aggregate cost by dimension
+ *
+ * @remarks
+ * Sums stored per-entry cost over [start_time, end_time), grouped by any of service, task_type, model, api_key_id, team_id, user_id, external_user_id, org_id. Point-in-time correct for chargeback/showback.
+ */
+export function useUsageCostsSuspense(
+  request: operations.GetCostsRequest,
+  options?: SuspenseQueryHookOptions<UsageCostsQueryData, UsageCostsQueryError>,
+): UseSuspenseQueryResult<UsageCostsQueryData, UsageCostsQueryError> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildUsageCostsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setUsageCostsData(
+  client: QueryClient,
+  queryKeyBase: [
+    parameters: {
+      startTime: number;
+      endTime: number;
+      groupBy?: string | undefined;
+      orgId?: string | undefined;
+    },
+  ],
+  data: UsageCostsQueryData,
+): UsageCostsQueryData | undefined {
+  const key = queryKeyUsageCosts(...queryKeyBase);
+
+  return client.setQueryData<UsageCostsQueryData>(key, data);
+}
+
+export function invalidateUsageCosts(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<
+    [parameters: {
+      startTime: number;
+      endTime: number;
+      groupBy?: string | undefined;
+      orgId?: string | undefined;
+    }]
+  >,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "usage", "costs", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllUsageCosts(
+  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,200 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsApiKeyEvents } from "../funcs/budgetsApiKeyEvents.js";
+import { budgetsDeleteApiKey } from "../funcs/budgetsDeleteApiKey.js";
+import { budgetsDeleteOrg } from "../funcs/budgetsDeleteOrg.js";
+import { budgetsGetApiKey } from "../funcs/budgetsGetApiKey.js";
+import { budgetsGetCurrency } from "../funcs/budgetsGetCurrency.js";
+import { budgetsGetOrg } from "../funcs/budgetsGetOrg.js";
+import { budgetsList } from "../funcs/budgetsList.js";
+import { budgetsOrgEvents } from "../funcs/budgetsOrgEvents.js";
+import { budgetsSetApiKey } from "../funcs/budgetsSetApiKey.js";
+import { budgetsSetCurrency } from "../funcs/budgetsSetCurrency.js";
+import { budgetsSetOrg } from "../funcs/budgetsSetOrg.js";
+import { ClientSDK, RequestOptions } from "../lib/sdks.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<operations.ListBudgetsResponseBody> {
+    return unwrapAsync(budgetsList(
+      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<operations.GetOrgBudgetsResponseBody> {
+    return unwrapAsync(budgetsGetOrg(
+      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<operations.SetOrgBudgetResponseBody> {
+    return unwrapAsync(budgetsSetOrg(
+      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<operations.DeleteOrgBudgetResponseBody> {
+    return unwrapAsync(budgetsDeleteOrg(
+      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<operations.ListOrgBudgetEventsResponseBody> {
+    return unwrapAsync(budgetsOrgEvents(
+      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<operations.GetApiKeyBudgetsResponseBody> {
+    return unwrapAsync(budgetsGetApiKey(
+      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<operations.SetApiKeyBudgetResponseBody> {
+    return unwrapAsync(budgetsSetApiKey(
+      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<operations.DeleteApiKeyBudgetResponseBody> {
+    return unwrapAsync(budgetsDeleteApiKey(
+      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<operations.ListApiKeyBudgetEventsResponseBody> {
+    return unwrapAsync(budgetsApiKeyEvents(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Get display currency
+   */
+  async getCurrency(
+    options?: RequestOptions,
+  ): Promise<operations.GetCurrencyResponseBody> {
+    return unwrapAsync(budgetsGetCurrency(
+      this,
+      options,
+    ));
+  }
+
+  /**
+   * Set display currency
+   *
+   * @remarks
+   * ISO 4217 code used for display on budgets and cost reports. No conversion is performed anywhere.
+   */
+  async setCurrency(
+    request: operations.SetCurrencyRequestBody,
+    options?: RequestOptions,
+  ): Promise<operations.SetCurrencyResponseBody> {
+    return unwrapAsync(budgetsSetCurrency(
+      this,
+      request,
+      options,
+    ));
+  }
+}

package/src/sdk/llm.ts

@@ -19,9 +19,9 @@ import { MemoryStores } from "./memorystores.js";
 import { Models } from "./models.js";
 import { Prompts } from "./prompts.js";
 import { Responses } from "./responses.js";
+import { SDKUsage } from "./sdkusage.js";
 import { Skills } from "./skills.js";
 import { Speech } from "./speech.js";
-import { Usage } from "./usage.js";
 import { VectorStores } from "./vectorstores.js";
 
 export class Llm extends ClientSDK {
@@ -100,9 +100,9 @@ export class Llm extends ClientSDK {
     return (this._speech ??= new Speech(this._options));
   }
 
-  private _usage?: Usage;
-  get usage(): Usage {
-    return (this._usage ??= new Usage(this._options));
+  private _usage?: SDKUsage;
+  get usage(): SDKUsage {
+    return (this._usage ??= new SDKUsage(this._options));
   }
 
   private _classify?: Classify;