@meetkai/mka1 0.48.25 → 0.48.26

package/src/react-query/llmBudgetsGetApiKey.ts

@@ -0,0 +1,133 @@
+/*
+ * 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 {
+  buildLlmBudgetsGetApiKeyQuery,
+  LlmBudgetsGetApiKeyQueryData,
+  prefetchLlmBudgetsGetApiKey,
+  queryKeyLlmBudgetsGetApiKey,
+} from "./llmBudgetsGetApiKey.core.js";
+export {
+  buildLlmBudgetsGetApiKeyQuery,
+  type LlmBudgetsGetApiKeyQueryData,
+  prefetchLlmBudgetsGetApiKey,
+  queryKeyLlmBudgetsGetApiKey,
+};
+
+export type LlmBudgetsGetApiKeyQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List apikey budgets
+ *
+ * @remarks
+ * Returns the apikey's budgets (self-budget and/or ceiling) with each one's live spend gauge.
+ */
+export function useLlmBudgetsGetApiKey(
+  request: operations.GetApiKeyBudgetsRequest,
+  options?: QueryHookOptions<
+    LlmBudgetsGetApiKeyQueryData,
+    LlmBudgetsGetApiKeyQueryError
+  >,
+): UseQueryResult<LlmBudgetsGetApiKeyQueryData, LlmBudgetsGetApiKeyQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmBudgetsGetApiKeyQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List apikey budgets
+ *
+ * @remarks
+ * Returns the apikey's budgets (self-budget and/or ceiling) with each one's live spend gauge.
+ */
+export function useLlmBudgetsGetApiKeySuspense(
+  request: operations.GetApiKeyBudgetsRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmBudgetsGetApiKeyQueryData,
+    LlmBudgetsGetApiKeyQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmBudgetsGetApiKeyQueryData,
+  LlmBudgetsGetApiKeyQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmBudgetsGetApiKeyQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmBudgetsGetApiKeyData(
+  client: QueryClient,
+  queryKeyBase: [apiKeyId: string],
+  data: LlmBudgetsGetApiKeyQueryData,
+): LlmBudgetsGetApiKeyQueryData | undefined {
+  const key = queryKeyLlmBudgetsGetApiKey(...queryKeyBase);
+
+  return client.setQueryData<LlmBudgetsGetApiKeyQueryData>(key, data);
+}
+
+export function invalidateLlmBudgetsGetApiKey(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[apiKeyId: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "getApiKey", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmBudgetsGetApiKey(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "getApiKey"],
+  });
+}

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

@@ -0,0 +1,71 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmBudgetsGetOrg } from "../funcs/llmBudgetsGetOrg.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 LlmBudgetsGetOrgQueryData = components.ListBudgetsResponse;
+
+export function prefetchLlmBudgetsGetOrg(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetOrgBudgetsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmBudgetsGetOrgQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmBudgetsGetOrgQuery(
+  client$: SDKCore,
+  request: operations.GetOrgBudgetsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmBudgetsGetOrgQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmBudgetsGetOrg(request.orgId),
+    queryFn: async function llmBudgetsGetOrgQueryFn(
+      ctx,
+    ): Promise<LlmBudgetsGetOrgQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmBudgetsGetOrg(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmBudgetsGetOrg(orgId: string): QueryKey {
+  return ["@meetkai/mka1", "budgets", "getOrg", orgId];
+}

package/src/react-query/llmBudgetsGetOrg.ts

@@ -0,0 +1,133 @@
+/*
+ * 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 {
+  buildLlmBudgetsGetOrgQuery,
+  LlmBudgetsGetOrgQueryData,
+  prefetchLlmBudgetsGetOrg,
+  queryKeyLlmBudgetsGetOrg,
+} from "./llmBudgetsGetOrg.core.js";
+export {
+  buildLlmBudgetsGetOrgQuery,
+  type LlmBudgetsGetOrgQueryData,
+  prefetchLlmBudgetsGetOrg,
+  queryKeyLlmBudgetsGetOrg,
+};
+
+export type LlmBudgetsGetOrgQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List org budgets
+ *
+ * @remarks
+ * Returns the org's budgets (self-budget and/or ceiling) with each one's live spend gauge.
+ */
+export function useLlmBudgetsGetOrg(
+  request: operations.GetOrgBudgetsRequest,
+  options?: QueryHookOptions<
+    LlmBudgetsGetOrgQueryData,
+    LlmBudgetsGetOrgQueryError
+  >,
+): UseQueryResult<LlmBudgetsGetOrgQueryData, LlmBudgetsGetOrgQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmBudgetsGetOrgQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List org budgets
+ *
+ * @remarks
+ * Returns the org's budgets (self-budget and/or ceiling) with each one's live spend gauge.
+ */
+export function useLlmBudgetsGetOrgSuspense(
+  request: operations.GetOrgBudgetsRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmBudgetsGetOrgQueryData,
+    LlmBudgetsGetOrgQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmBudgetsGetOrgQueryData,
+  LlmBudgetsGetOrgQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmBudgetsGetOrgQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmBudgetsGetOrgData(
+  client: QueryClient,
+  queryKeyBase: [orgId: string],
+  data: LlmBudgetsGetOrgQueryData,
+): LlmBudgetsGetOrgQueryData | undefined {
+  const key = queryKeyLlmBudgetsGetOrg(...queryKeyBase);
+
+  return client.setQueryData<LlmBudgetsGetOrgQueryData>(key, data);
+}
+
+export function invalidateLlmBudgetsGetOrg(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[orgId: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "getOrg", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmBudgetsGetOrg(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "getOrg"],
+  });
+}

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

@@ -0,0 +1,64 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmBudgetsList } from "../funcs/llmBudgetsList.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 LlmBudgetsListQueryData = components.ListBudgetsResponse;
+
+export function prefetchLlmBudgetsList(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmBudgetsListQuery(
+      client$,
+      options,
+    ),
+  });
+}
+
+export function buildLlmBudgetsListQuery(
+  client$: SDKCore,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (context: QueryFunctionContext) => Promise<LlmBudgetsListQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmBudgetsList(),
+    queryFn: async function llmBudgetsListQueryFn(
+      ctx,
+    ): Promise<LlmBudgetsListQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmBudgetsList(
+        client$,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmBudgetsList(): QueryKey {
+  return ["@meetkai/mka1", "budgets", "list"];
+}

package/src/react-query/llmBudgetsList.ts

@@ -0,0 +1,106 @@
+/*
+ * 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 { useSDKContext } from "./_context.js";
+import { QueryHookOptions, SuspenseQueryHookOptions } from "./_types.js";
+import {
+  buildLlmBudgetsListQuery,
+  LlmBudgetsListQueryData,
+  prefetchLlmBudgetsList,
+  queryKeyLlmBudgetsList,
+} from "./llmBudgetsList.core.js";
+export {
+  buildLlmBudgetsListQuery,
+  type LlmBudgetsListQueryData,
+  prefetchLlmBudgetsList,
+  queryKeyLlmBudgetsList,
+};
+
+export type LlmBudgetsListQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List budgets
+ *
+ * @remarks
+ * Cluster admins see all budgets; org admins see their org's budgets.
+ */
+export function useLlmBudgetsList(
+  options?: QueryHookOptions<LlmBudgetsListQueryData, LlmBudgetsListQueryError>,
+): UseQueryResult<LlmBudgetsListQueryData, LlmBudgetsListQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmBudgetsListQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List budgets
+ *
+ * @remarks
+ * Cluster admins see all budgets; org admins see their org's budgets.
+ */
+export function useLlmBudgetsListSuspense(
+  options?: SuspenseQueryHookOptions<
+    LlmBudgetsListQueryData,
+    LlmBudgetsListQueryError
+  >,
+): UseSuspenseQueryResult<LlmBudgetsListQueryData, LlmBudgetsListQueryError> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmBudgetsListQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmBudgetsListData(
+  client: QueryClient,
+  data: LlmBudgetsListQueryData,
+): LlmBudgetsListQueryData | undefined {
+  const key = queryKeyLlmBudgetsList();
+
+  return client.setQueryData<LlmBudgetsListQueryData>(key, data);
+}
+
+export function invalidateAllLlmBudgetsList(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "list"],
+  });
+}

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

@@ -0,0 +1,71 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmBudgetsOrgEvents } from "../funcs/llmBudgetsOrgEvents.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 LlmBudgetsOrgEventsQueryData = components.ListBudgetEventsResponse;
+
+export function prefetchLlmBudgetsOrgEvents(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.ListOrgBudgetEventsRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmBudgetsOrgEventsQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmBudgetsOrgEventsQuery(
+  client$: SDKCore,
+  request: operations.ListOrgBudgetEventsRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmBudgetsOrgEventsQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmBudgetsOrgEvents(request.orgId),
+    queryFn: async function llmBudgetsOrgEventsQueryFn(
+      ctx,
+    ): Promise<LlmBudgetsOrgEventsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmBudgetsOrgEvents(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmBudgetsOrgEvents(orgId: string): QueryKey {
+  return ["@meetkai/mka1", "budgets", "orgEvents", orgId];
+}

package/src/react-query/llmBudgetsOrgEvents.ts

@@ -0,0 +1,133 @@
+/*
+ * 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 {
+  buildLlmBudgetsOrgEventsQuery,
+  LlmBudgetsOrgEventsQueryData,
+  prefetchLlmBudgetsOrgEvents,
+  queryKeyLlmBudgetsOrgEvents,
+} from "./llmBudgetsOrgEvents.core.js";
+export {
+  buildLlmBudgetsOrgEventsQuery,
+  type LlmBudgetsOrgEventsQueryData,
+  prefetchLlmBudgetsOrgEvents,
+  queryKeyLlmBudgetsOrgEvents,
+};
+
+export type LlmBudgetsOrgEventsQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * List org budget events
+ *
+ * @remarks
+ * Returns recent threshold-alert and block events for the scope's budgets.
+ */
+export function useLlmBudgetsOrgEvents(
+  request: operations.ListOrgBudgetEventsRequest,
+  options?: QueryHookOptions<
+    LlmBudgetsOrgEventsQueryData,
+    LlmBudgetsOrgEventsQueryError
+  >,
+): UseQueryResult<LlmBudgetsOrgEventsQueryData, LlmBudgetsOrgEventsQueryError> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmBudgetsOrgEventsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List org budget events
+ *
+ * @remarks
+ * Returns recent threshold-alert and block events for the scope's budgets.
+ */
+export function useLlmBudgetsOrgEventsSuspense(
+  request: operations.ListOrgBudgetEventsRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmBudgetsOrgEventsQueryData,
+    LlmBudgetsOrgEventsQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmBudgetsOrgEventsQueryData,
+  LlmBudgetsOrgEventsQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmBudgetsOrgEventsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmBudgetsOrgEventsData(
+  client: QueryClient,
+  queryKeyBase: [orgId: string],
+  data: LlmBudgetsOrgEventsQueryData,
+): LlmBudgetsOrgEventsQueryData | undefined {
+  const key = queryKeyLlmBudgetsOrgEvents(...queryKeyBase);
+
+  return client.setQueryData<LlmBudgetsOrgEventsQueryData>(key, data);
+}
+
+export function invalidateLlmBudgetsOrgEvents(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[orgId: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "orgEvents", ...queryKeyBase],
+  });
+}
+
+export function invalidateAllLlmBudgetsOrgEvents(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "budgets", "orgEvents"],
+  });
+}