@meetkai/mka1 0.55.29 → 0.55.31

package/src/models/operations/setorgbudgetenforcement.ts

@@ -0,0 +1,280 @@
+/*
+ * 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 { ClosedEnum } from "../../types/enums.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export const OnCheckFailure = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type OnCheckFailure = ClosedEnum<typeof OnCheckFailure>;
+
+export type SetOrgBudgetEnforcementRequestBody = {
+  onCheckFailure: OnCheckFailure | null;
+};
+
+export type SetOrgBudgetEnforcementRequest = {
+  orgId: string;
+  requestBody: SetOrgBudgetEnforcementRequestBody;
+};
+
+export const SetOrgBudgetEnforcementOnCheckFailure = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type SetOrgBudgetEnforcementOnCheckFailure = ClosedEnum<
+  typeof SetOrgBudgetEnforcementOnCheckFailure
+>;
+
+export const SetOrgBudgetEnforcementClusterDefault = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+export type SetOrgBudgetEnforcementClusterDefault = ClosedEnum<
+  typeof SetOrgBudgetEnforcementClusterDefault
+>;
+
+/**
+ * What the spend gate does for this organization when the budget check is unavailable
+ */
+export const SetOrgBudgetEnforcementEffective = {
+  Allow: "allow",
+  Block: "block",
+} as const;
+/**
+ * What the spend gate does for this organization when the budget check is unavailable
+ */
+export type SetOrgBudgetEnforcementEffective = ClosedEnum<
+  typeof SetOrgBudgetEnforcementEffective
+>;
+
+/**
+ * OK
+ */
+export type SetOrgBudgetEnforcementResponseBody = {
+  orgId: string;
+  /**
+   * The organization's own setting; null inherits the cluster default
+   */
+  onCheckFailure: SetOrgBudgetEnforcementOnCheckFailure | null;
+  clusterDefault: SetOrgBudgetEnforcementClusterDefault;
+  /**
+   * What the spend gate does for this organization when the budget check is unavailable
+   */
+  effective: SetOrgBudgetEnforcementEffective;
+};
+
+/** @internal */
+export const OnCheckFailure$inboundSchema: z.ZodNativeEnum<
+  typeof OnCheckFailure
+> = z.nativeEnum(OnCheckFailure);
+/** @internal */
+export const OnCheckFailure$outboundSchema: z.ZodNativeEnum<
+  typeof OnCheckFailure
+> = OnCheckFailure$inboundSchema;
+
+/** @internal */
+export const SetOrgBudgetEnforcementRequestBody$inboundSchema: z.ZodType<
+  SetOrgBudgetEnforcementRequestBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  on_check_failure: z.nullable(OnCheckFailure$inboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    "on_check_failure": "onCheckFailure",
+  });
+});
+/** @internal */
+export type SetOrgBudgetEnforcementRequestBody$Outbound = {
+  on_check_failure: string | null;
+};
+
+/** @internal */
+export const SetOrgBudgetEnforcementRequestBody$outboundSchema: z.ZodType<
+  SetOrgBudgetEnforcementRequestBody$Outbound,
+  z.ZodTypeDef,
+  SetOrgBudgetEnforcementRequestBody
+> = z.object({
+  onCheckFailure: z.nullable(OnCheckFailure$outboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    onCheckFailure: "on_check_failure",
+  });
+});
+
+export function setOrgBudgetEnforcementRequestBodyToJSON(
+  setOrgBudgetEnforcementRequestBody: SetOrgBudgetEnforcementRequestBody,
+): string {
+  return JSON.stringify(
+    SetOrgBudgetEnforcementRequestBody$outboundSchema.parse(
+      setOrgBudgetEnforcementRequestBody,
+    ),
+  );
+}
+export function setOrgBudgetEnforcementRequestBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<SetOrgBudgetEnforcementRequestBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      SetOrgBudgetEnforcementRequestBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetOrgBudgetEnforcementRequestBody' from JSON`,
+  );
+}
+
+/** @internal */
+export const SetOrgBudgetEnforcementRequest$inboundSchema: z.ZodType<
+  SetOrgBudgetEnforcementRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  RequestBody: z.lazy(() => SetOrgBudgetEnforcementRequestBody$inboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "RequestBody": "requestBody",
+  });
+});
+/** @internal */
+export type SetOrgBudgetEnforcementRequest$Outbound = {
+  org_id: string;
+  RequestBody: SetOrgBudgetEnforcementRequestBody$Outbound;
+};
+
+/** @internal */
+export const SetOrgBudgetEnforcementRequest$outboundSchema: z.ZodType<
+  SetOrgBudgetEnforcementRequest$Outbound,
+  z.ZodTypeDef,
+  SetOrgBudgetEnforcementRequest
+> = z.object({
+  orgId: z.string(),
+  requestBody: z.lazy(() => SetOrgBudgetEnforcementRequestBody$outboundSchema),
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    requestBody: "RequestBody",
+  });
+});
+
+export function setOrgBudgetEnforcementRequestToJSON(
+  setOrgBudgetEnforcementRequest: SetOrgBudgetEnforcementRequest,
+): string {
+  return JSON.stringify(
+    SetOrgBudgetEnforcementRequest$outboundSchema.parse(
+      setOrgBudgetEnforcementRequest,
+    ),
+  );
+}
+export function setOrgBudgetEnforcementRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<SetOrgBudgetEnforcementRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => SetOrgBudgetEnforcementRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetOrgBudgetEnforcementRequest' from JSON`,
+  );
+}
+
+/** @internal */
+export const SetOrgBudgetEnforcementOnCheckFailure$inboundSchema:
+  z.ZodNativeEnum<typeof SetOrgBudgetEnforcementOnCheckFailure> = z.nativeEnum(
+    SetOrgBudgetEnforcementOnCheckFailure,
+  );
+/** @internal */
+export const SetOrgBudgetEnforcementOnCheckFailure$outboundSchema:
+  z.ZodNativeEnum<typeof SetOrgBudgetEnforcementOnCheckFailure> =
+    SetOrgBudgetEnforcementOnCheckFailure$inboundSchema;
+
+/** @internal */
+export const SetOrgBudgetEnforcementClusterDefault$inboundSchema:
+  z.ZodNativeEnum<typeof SetOrgBudgetEnforcementClusterDefault> = z.nativeEnum(
+    SetOrgBudgetEnforcementClusterDefault,
+  );
+/** @internal */
+export const SetOrgBudgetEnforcementClusterDefault$outboundSchema:
+  z.ZodNativeEnum<typeof SetOrgBudgetEnforcementClusterDefault> =
+    SetOrgBudgetEnforcementClusterDefault$inboundSchema;
+
+/** @internal */
+export const SetOrgBudgetEnforcementEffective$inboundSchema: z.ZodNativeEnum<
+  typeof SetOrgBudgetEnforcementEffective
+> = z.nativeEnum(SetOrgBudgetEnforcementEffective);
+/** @internal */
+export const SetOrgBudgetEnforcementEffective$outboundSchema: z.ZodNativeEnum<
+  typeof SetOrgBudgetEnforcementEffective
+> = SetOrgBudgetEnforcementEffective$inboundSchema;
+
+/** @internal */
+export const SetOrgBudgetEnforcementResponseBody$inboundSchema: z.ZodType<
+  SetOrgBudgetEnforcementResponseBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  on_check_failure: z.nullable(
+    SetOrgBudgetEnforcementOnCheckFailure$inboundSchema,
+  ),
+  cluster_default: SetOrgBudgetEnforcementClusterDefault$inboundSchema,
+  effective: SetOrgBudgetEnforcementEffective$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "on_check_failure": "onCheckFailure",
+    "cluster_default": "clusterDefault",
+  });
+});
+/** @internal */
+export type SetOrgBudgetEnforcementResponseBody$Outbound = {
+  org_id: string;
+  on_check_failure: string | null;
+  cluster_default: string;
+  effective: string;
+};
+
+/** @internal */
+export const SetOrgBudgetEnforcementResponseBody$outboundSchema: z.ZodType<
+  SetOrgBudgetEnforcementResponseBody$Outbound,
+  z.ZodTypeDef,
+  SetOrgBudgetEnforcementResponseBody
+> = z.object({
+  orgId: z.string(),
+  onCheckFailure: z.nullable(
+    SetOrgBudgetEnforcementOnCheckFailure$outboundSchema,
+  ),
+  clusterDefault: SetOrgBudgetEnforcementClusterDefault$outboundSchema,
+  effective: SetOrgBudgetEnforcementEffective$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    onCheckFailure: "on_check_failure",
+    clusterDefault: "cluster_default",
+  });
+});
+
+export function setOrgBudgetEnforcementResponseBodyToJSON(
+  setOrgBudgetEnforcementResponseBody: SetOrgBudgetEnforcementResponseBody,
+): string {
+  return JSON.stringify(
+    SetOrgBudgetEnforcementResponseBody$outboundSchema.parse(
+      setOrgBudgetEnforcementResponseBody,
+    ),
+  );
+}
+export function setOrgBudgetEnforcementResponseBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<SetOrgBudgetEnforcementResponseBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      SetOrgBudgetEnforcementResponseBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetOrgBudgetEnforcementResponseBody' from JSON`,
+  );
+}

package/src/react-query/index.ts

@@ -352,6 +352,7 @@ export * from "./llmModelsDeleteCatalogEntry.js";
 export * from "./llmModelsDeleteOrgAutoModel.js";
 export * from "./llmModelsGet.js";
 export * from "./llmModelsGetClusterCurrency.js";
+export * from "./llmModelsGetOrgBudgetEnforcement.js";
 export * from "./llmModelsGetOrgTokenQuota.js";
 export * from "./llmModelsGetQuota.js";
 export * from "./llmModelsList.js";
@@ -361,12 +362,14 @@ export * from "./llmModelsListCatalogOrgAccess.js";
 export * from "./llmModelsListClusterPrices.js";
 export * from "./llmModelsListEffectivePrices.js";
 export * from "./llmModelsListOrgAutoModels.js";
+export * from "./llmModelsListOrgBudgetEnforcements.js";
 export * from "./llmModelsListOrgPrices.js";
 export * from "./llmModelsListRegistry.js";
 export * from "./llmModelsPutOrgAutoModel.js";
 export * from "./llmModelsReplaceCatalogOrgAccess.js";
 export * from "./llmModelsSetClusterCurrency.js";
 export * from "./llmModelsSetClusterPrice.js";
+export * from "./llmModelsSetOrgBudgetEnforcement.js";
 export * from "./llmModelsSetOrgPrice.js";
 export * from "./llmModelsSetOrgTokenQuota.js";
 export * from "./llmModelsUpdateCatalogEntry.js";
@@ -490,6 +493,8 @@ export * from "./sandboxNetworkPoliciesEffective.js";
 export * from "./sandboxNetworkPoliciesGet.js";
 export * from "./sandboxNetworkPoliciesSet.js";
 export * from "./sandboxProxyBrowserPortRequest.js";
+export * from "./sandboxResourcePoliciesGet.js";
+export * from "./sandboxResourcePoliciesSet.js";
 export * from "./sandboxRunCode.js";
 export * from "./sandboxRunCommand.js";
 export * from "./sandboxSetPricing.js";
@@ -498,8 +503,6 @@ export * from "./sandboxUploadArchive.js";
 export * from "./sandboxUploadFile.js";
 export * from "./sessionExchangeSessionToken.js";
 export * from "./sessionGetPendingInvitation.js";
-export * from "./sessionsGetTenantResourcePolicy.js";
-export * from "./sessionsSetTenantResourcePolicy.js";
 export * from "./trackingExperimentsCreate.js";
 export * from "./trackingExperimentsDelete.js";
 export * from "./trackingExperimentsGet.js";

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

@@ -0,0 +1,73 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { llmModelsGetOrgBudgetEnforcement } from "../funcs/llmModelsGetOrgBudgetEnforcement.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 LlmModelsGetOrgBudgetEnforcementQueryData =
+  operations.GetOrgBudgetEnforcementResponseBody;
+
+export function prefetchLlmModelsGetOrgBudgetEnforcement(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsGetOrgBudgetEnforcementQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsGetOrgBudgetEnforcementQuery(
+  client$: SDKCore,
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsGetOrgBudgetEnforcementQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsGetOrgBudgetEnforcement(request.orgId),
+    queryFn: async function llmModelsGetOrgBudgetEnforcementQueryFn(
+      ctx,
+    ): Promise<LlmModelsGetOrgBudgetEnforcementQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsGetOrgBudgetEnforcement(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsGetOrgBudgetEnforcement(
+  orgId: string,
+): QueryKey {
+  return ["@meetkai/mka1", "models", "getOrgBudgetEnforcement", orgId];
+}

package/src/react-query/llmModelsGetOrgBudgetEnforcement.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 {
+  buildLlmModelsGetOrgBudgetEnforcementQuery,
+  LlmModelsGetOrgBudgetEnforcementQueryData,
+  prefetchLlmModelsGetOrgBudgetEnforcement,
+  queryKeyLlmModelsGetOrgBudgetEnforcement,
+} from "./llmModelsGetOrgBudgetEnforcement.core.js";
+export {
+  buildLlmModelsGetOrgBudgetEnforcementQuery,
+  type LlmModelsGetOrgBudgetEnforcementQueryData,
+  prefetchLlmModelsGetOrgBudgetEnforcement,
+  queryKeyLlmModelsGetOrgBudgetEnforcement,
+};
+
+export type LlmModelsGetOrgBudgetEnforcementQueryError =
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Get an organization's budget-enforcement setting
+ *
+ * @remarks
+ * What the spend gate does for this organization's billable requests when the budget check cannot be made (budgeting unreachable or degraded): `allow` admits them, `block` refuses them with 503 `budget_check_unavailable`. Returns the organization's own setting (null = inherits), the cluster default and the effective mode. Cluster admins may read any organization; organization admins their own (never through an exchanged end-user token). API keys need `read:budgets`.
+ */
+export function useLlmModelsGetOrgBudgetEnforcement(
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: QueryHookOptions<
+    LlmModelsGetOrgBudgetEnforcementQueryData,
+    LlmModelsGetOrgBudgetEnforcementQueryError
+  >,
+): UseQueryResult<
+  LlmModelsGetOrgBudgetEnforcementQueryData,
+  LlmModelsGetOrgBudgetEnforcementQueryError
+> {
+  const client = useSDKContext();
+  return useQuery({
+    ...buildLlmModelsGetOrgBudgetEnforcementQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get an organization's budget-enforcement setting
+ *
+ * @remarks
+ * What the spend gate does for this organization's billable requests when the budget check cannot be made (budgeting unreachable or degraded): `allow` admits them, `block` refuses them with 503 `budget_check_unavailable`. Returns the organization's own setting (null = inherits), the cluster default and the effective mode. Cluster admins may read any organization; organization admins their own (never through an exchanged end-user token). API keys need `read:budgets`.
+ */
+export function useLlmModelsGetOrgBudgetEnforcementSuspense(
+  request: operations.GetOrgBudgetEnforcementRequest,
+  options?: SuspenseQueryHookOptions<
+    LlmModelsGetOrgBudgetEnforcementQueryData,
+    LlmModelsGetOrgBudgetEnforcementQueryError
+  >,
+): UseSuspenseQueryResult<
+  LlmModelsGetOrgBudgetEnforcementQueryData,
+  LlmModelsGetOrgBudgetEnforcementQueryError
+> {
+  const client = useSDKContext();
+  return useSuspenseQuery({
+    ...buildLlmModelsGetOrgBudgetEnforcementQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function setLlmModelsGetOrgBudgetEnforcementData(
+  client: QueryClient,
+  queryKeyBase: [orgId: string],
+  data: LlmModelsGetOrgBudgetEnforcementQueryData,
+): LlmModelsGetOrgBudgetEnforcementQueryData | undefined {
+  const key = queryKeyLlmModelsGetOrgBudgetEnforcement(...queryKeyBase);
+
+  return client.setQueryData<LlmModelsGetOrgBudgetEnforcementQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateLlmModelsGetOrgBudgetEnforcement(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[orgId: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@meetkai/mka1",
+      "models",
+      "getOrgBudgetEnforcement",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllLlmModelsGetOrgBudgetEnforcement(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@meetkai/mka1", "models", "getOrgBudgetEnforcement"],
+  });
+}

package/src/react-query/llmModelsListOrgBudgetEnforcements.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 { llmModelsListOrgBudgetEnforcements } from "../funcs/llmModelsListOrgBudgetEnforcements.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 LlmModelsListOrgBudgetEnforcementsQueryData =
+  operations.ListOrgBudgetEnforcementsResponseBody;
+
+export function prefetchLlmModelsListOrgBudgetEnforcements(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildLlmModelsListOrgBudgetEnforcementsQuery(
+      client$,
+      options,
+    ),
+  });
+}
+
+export function buildLlmModelsListOrgBudgetEnforcementsQuery(
+  client$: SDKCore,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<LlmModelsListOrgBudgetEnforcementsQueryData>;
+} {
+  return {
+    queryKey: queryKeyLlmModelsListOrgBudgetEnforcements(),
+    queryFn: async function llmModelsListOrgBudgetEnforcementsQueryFn(
+      ctx,
+    ): Promise<LlmModelsListOrgBudgetEnforcementsQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(llmModelsListOrgBudgetEnforcements(
+        client$,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyLlmModelsListOrgBudgetEnforcements(): QueryKey {
+  return ["@meetkai/mka1", "models", "listOrgBudgetEnforcements"];
+}