@meetkai/mka1 0.55.29 → 0.55.31

package/src/funcs/llmModelsSetOrgBudgetEnforcement.ts

@@ -0,0 +1,183 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { SDKCore } from "../core.js";
+import { encodeJSON, encodeSimple } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.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 { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Set an organization's budget-enforcement setting
+ *
+ * @remarks
+ * Choose `block` to refuse this organization's billable requests whenever the budget check is unavailable (fail closed), `allow` to admit them (fail open), or `null` to inherit the cluster default again. Applies to new admissions within 10 seconds on every gateway replica. Cluster admins may set any organization to anything; organization admins may only tighten their own — `block`, or `null` to inherit while that does not loosen an effective `block` — and never `allow`, so neither a cluster-wide nor a cluster-admin-pinned `block` can be switched off or pre-empted by a tenant (never through an exchanged end-user token). API keys need `write:budgets`. The cluster organization has no override: it is the operator's own identity and always fails open.
+ *
+ * If set, this operation will use {@link Security.bearerAuth} from the global security.
+ */
+export function llmModelsSetOrgBudgetEnforcement(
+  client: SDKCore,
+  request: operations.SetOrgBudgetEnforcementRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    operations.SetOrgBudgetEnforcementResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.SetOrgBudgetEnforcementRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      operations.SetOrgBudgetEnforcementResponseBody,
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) =>
+      operations.SetOrgBudgetEnforcementRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = encodeJSON("body", payload.RequestBody, { explode: true });
+
+  const pathParams = {
+    org_id: encodeSimple("org_id", payload.org_id, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc(
+    "/api/v1/llm/models/orgs/{org_id}/budget-enforcement",
+  )(pathParams);
+
+  const headers = new Headers(compactMap({
+    "Content-Type": "application/json",
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.bearerAuth);
+  const securityInput = secConfig == null ? {} : { bearerAuth: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput, [0]);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "setOrgBudgetEnforcement",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.bearerAuth,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || {
+        strategy: "backoff",
+        backoff: {
+          initialInterval: 500,
+          maxInterval: 5000,
+          exponent: 1.5,
+          maxElapsedTime: 15000,
+        },
+        retryConnectionErrors: true,
+      }
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "PUT",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    body: body,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const [result] = await M.match<
+    operations.SetOrgBudgetEnforcementResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, operations.SetOrgBudgetEnforcementResponseBody$inboundSchema),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req);
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/{sessionsGetTenantResourcePolicy.ts → sandboxResourcePoliciesGet.ts}

@@ -28,11 +28,14 @@ import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
 
 /**
- * Get
+ * Get Tenant Resource Policy
+ *
+ * @remarks
+ * Read organization or team resource limits for standard sandbox sessions.
  *
  * If set, this operation will use {@link Security.bearerAuth} from the global security.
  */
-export function sessionsGetTenantResourcePolicy(
+export function sandboxResourcePoliciesGet(
   client: SDKCore,
   request: operations.GetTenantResourcePolicyRequest,
   options?: RequestOptions,

package/src/funcs/{sessionsSetTenantResourcePolicy.ts → sandboxResourcePoliciesSet.ts}

@@ -28,11 +28,14 @@ import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
 
 /**
- * Save
+ * Set Tenant Resource Policy
+ *
+ * @remarks
+ * Update organization or team resource limits for standard sandbox sessions. The expected revision rejects stale writes.
  *
  * If set, this operation will use {@link Security.bearerAuth} from the global security.
  */
-export function sessionsSetTenantResourcePolicy(
+export function sandboxResourcePoliciesSet(
   client: SDKCore,
   request: operations.SetTenantResourcePolicyRequest,
   options?: RequestOptions,

package/src/lib/config.ts

@@ -65,7 +65,7 @@ export function serverURLFromOptions(options: SDKOptions): URL | null {
 export const SDK_METADATA = {
   language: "typescript",
   openapiDocVersion: "1.1.0",
-  sdkVersion: "0.55.29",
+  sdkVersion: "0.55.31",
   genVersion: "2.950.2",
-  userAgent: "speakeasy-sdk/typescript 0.55.29 2.950.2 1.1.0 @meetkai/mka1",
+  userAgent: "speakeasy-sdk/typescript 0.55.31 2.950.2 1.1.0 @meetkai/mka1",
 } as const;

package/src/mcp-server/mcp-server.ts

@@ -19,7 +19,7 @@ const routes = buildRouteMap({
 export const app = buildApplication(routes, {
   name: "mcp",
   versionInfo: {
-    currentVersion: "0.55.29",
+    currentVersion: "0.55.31",
   },
 });
 

package/src/mcp-server/server.ts

@@ -345,6 +345,7 @@ import { tool$llmModelsDeleteCatalogEntry } from "./tools/llmModelsDeleteCatalog
 import { tool$llmModelsDeleteOrgAutoModel } from "./tools/llmModelsDeleteOrgAutoModel.js";
 import { tool$llmModelsGet } from "./tools/llmModelsGet.js";
 import { tool$llmModelsGetClusterCurrency } from "./tools/llmModelsGetClusterCurrency.js";
+import { tool$llmModelsGetOrgBudgetEnforcement } from "./tools/llmModelsGetOrgBudgetEnforcement.js";
 import { tool$llmModelsGetOrgTokenQuota } from "./tools/llmModelsGetOrgTokenQuota.js";
 import { tool$llmModelsGetQuota } from "./tools/llmModelsGetQuota.js";
 import { tool$llmModelsList } from "./tools/llmModelsList.js";
@@ -354,12 +355,14 @@ import { tool$llmModelsListCatalogOrgAccess } from "./tools/llmModelsListCatalog
 import { tool$llmModelsListClusterPrices } from "./tools/llmModelsListClusterPrices.js";
 import { tool$llmModelsListEffectivePrices } from "./tools/llmModelsListEffectivePrices.js";
 import { tool$llmModelsListOrgAutoModels } from "./tools/llmModelsListOrgAutoModels.js";
+import { tool$llmModelsListOrgBudgetEnforcements } from "./tools/llmModelsListOrgBudgetEnforcements.js";
 import { tool$llmModelsListOrgPrices } from "./tools/llmModelsListOrgPrices.js";
 import { tool$llmModelsListRegistry } from "./tools/llmModelsListRegistry.js";
 import { tool$llmModelsPutOrgAutoModel } from "./tools/llmModelsPutOrgAutoModel.js";
 import { tool$llmModelsReplaceCatalogOrgAccess } from "./tools/llmModelsReplaceCatalogOrgAccess.js";
 import { tool$llmModelsSetClusterCurrency } from "./tools/llmModelsSetClusterCurrency.js";
 import { tool$llmModelsSetClusterPrice } from "./tools/llmModelsSetClusterPrice.js";
+import { tool$llmModelsSetOrgBudgetEnforcement } from "./tools/llmModelsSetOrgBudgetEnforcement.js";
 import { tool$llmModelsSetOrgPrice } from "./tools/llmModelsSetOrgPrice.js";
 import { tool$llmModelsSetOrgTokenQuota } from "./tools/llmModelsSetOrgTokenQuota.js";
 import { tool$llmModelsUpdateCatalogEntry } from "./tools/llmModelsUpdateCatalogEntry.js";
@@ -482,14 +485,14 @@ import { tool$sandboxNetworkPoliciesEffective } from "./tools/sandboxNetworkPoli
 import { tool$sandboxNetworkPoliciesGet } from "./tools/sandboxNetworkPoliciesGet.js";
 import { tool$sandboxNetworkPoliciesSet } from "./tools/sandboxNetworkPoliciesSet.js";
 import { tool$sandboxProxyBrowserPortRequest } from "./tools/sandboxProxyBrowserPortRequest.js";
+import { tool$sandboxResourcePoliciesGet } from "./tools/sandboxResourcePoliciesGet.js";
+import { tool$sandboxResourcePoliciesSet } from "./tools/sandboxResourcePoliciesSet.js";
 import { tool$sandboxRunCode } from "./tools/sandboxRunCode.js";
 import { tool$sandboxRunCommand } from "./tools/sandboxRunCommand.js";
 import { tool$sandboxSetPricing } from "./tools/sandboxSetPricing.js";
 import { tool$sandboxTerminate } from "./tools/sandboxTerminate.js";
 import { tool$sandboxUploadArchive } from "./tools/sandboxUploadArchive.js";
 import { tool$sandboxUploadFile } from "./tools/sandboxUploadFile.js";
-import { tool$sessionsGetTenantResourcePolicy } from "./tools/sessionsGetTenantResourcePolicy.js";
-import { tool$sessionsSetTenantResourcePolicy } from "./tools/sessionsSetTenantResourcePolicy.js";
 import { tool$trackingExperimentsCreate } from "./tools/trackingExperimentsCreate.js";
 import { tool$trackingExperimentsDelete } from "./tools/trackingExperimentsDelete.js";
 import { tool$trackingExperimentsGet } from "./tools/trackingExperimentsGet.js";
@@ -521,7 +524,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.55.29",
+    version: "0.55.31",
   });
 
   const client = new SDKCore({
@@ -576,8 +579,6 @@ export function createMCPServer(deps: {
   tool(tool$sandboxUploadFile);
   tool(tool$sandboxDownloadArchive);
   tool(tool$sandboxUploadArchive);
-  tool(tool$sessionsGetTenantResourcePolicy);
-  tool(tool$sessionsSetTenantResourcePolicy);
   tool(tool$usageSandbox);
   tool(tool$usageCosts);
   tool(tool$agentsListFunctionTools);
@@ -831,6 +832,9 @@ export function createMCPServer(deps: {
   tool(tool$llmModelsGetQuota);
   tool(tool$llmModelsGetOrgTokenQuota);
   tool(tool$llmModelsSetOrgTokenQuota);
+  tool(tool$llmModelsListOrgBudgetEnforcements);
+  tool(tool$llmModelsGetOrgBudgetEnforcement);
+  tool(tool$llmModelsSetOrgBudgetEnforcement);
   tool(tool$llmModelsListCatalog);
   tool(tool$llmModelsAddCatalogEntry);
   tool(tool$llmModelsUpdateCatalogEntry);
@@ -993,6 +997,8 @@ export function createMCPServer(deps: {
   tool(tool$sandboxNetworkPoliciesEffective);
   tool(tool$sandboxNetworkPoliciesGet);
   tool(tool$sandboxNetworkPoliciesSet);
+  tool(tool$sandboxResourcePoliciesGet);
+  tool(tool$sandboxResourcePoliciesSet);
   tool(tool$sandboxEnvironmentsList);
   tool(tool$sandboxEnvironmentsCreate);
   tool(tool$sandboxEnvironmentsGet);

package/src/mcp-server/tools/llmModelsGetOrgBudgetEnforcement.ts

@@ -0,0 +1,39 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsGetOrgBudgetEnforcement } from "../../funcs/llmModelsGetOrgBudgetEnforcement.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.GetOrgBudgetEnforcementRequest$inboundSchema,
+};
+
+export const tool$llmModelsGetOrgBudgetEnforcement: ToolDefinition<
+  typeof args
+> = {
+  name: "llm-models-get-org-budget-enforcement",
+  description: `Get an organization's budget-enforcement setting
+
+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\`.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsGetOrgBudgetEnforcement(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/llmModelsListOrgBudgetEnforcements.ts

@@ -0,0 +1,30 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsListOrgBudgetEnforcements } from "../../funcs/llmModelsListOrgBudgetEnforcements.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+export const tool$llmModelsListOrgBudgetEnforcements: ToolDefinition = {
+  name: "llm-models-list-org-budget-enforcements",
+  description: `List every organization's budget-enforcement override
+
+Cluster-admin inventory of the organizations that hold a budget-enforcement override, with the cluster default and each one's effective mode — what to review before flipping the cluster default, and where to look when an organization is being refused with \`budget_check_unavailable\`. Organizations that inherit the default are not listed. API keys need \`read:budgets\`.`,
+  tool: async (client, ctx) => {
+    const [result, apiCall] = await llmModelsListOrgBudgetEnforcements(
+      client,
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/llmModelsSetOrgBudgetEnforcement.ts

@@ -0,0 +1,39 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsSetOrgBudgetEnforcement } from "../../funcs/llmModelsSetOrgBudgetEnforcement.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.SetOrgBudgetEnforcementRequest$inboundSchema,
+};
+
+export const tool$llmModelsSetOrgBudgetEnforcement: ToolDefinition<
+  typeof args
+> = {
+  name: "llm-models-set-org-budget-enforcement",
+  description: `Set an organization's budget-enforcement setting
+
+Choose \`block\` to refuse this organization's billable requests whenever the budget check is unavailable (fail closed), \`allow\` to admit them (fail open), or \`null\` to inherit the cluster default again. Applies to new admissions within 10 seconds on every gateway replica. Cluster admins may set any organization to anything; organization admins may only tighten their own — \`block\`, or \`null\` to inherit while that does not loosen an effective \`block\` — and never \`allow\`, so neither a cluster-wide nor a cluster-admin-pinned \`block\` can be switched off or pre-empted by a tenant (never through an exchanged end-user token). API keys need \`write:budgets\`. The cluster organization has no override: it is the operator's own identity and always fails open.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsSetOrgBudgetEnforcement(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/sandboxResourcePoliciesGet.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { sandboxResourcePoliciesGet } from "../../funcs/sandboxResourcePoliciesGet.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.GetTenantResourcePolicyRequest$inboundSchema,
+};
+
+export const tool$sandboxResourcePoliciesGet: ToolDefinition<typeof args> = {
+  name: "sandbox-resource-policies-get",
+  description: `Get Tenant Resource Policy
+
+Read organization or team resource limits for standard sandbox sessions.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await sandboxResourcePoliciesGet(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/sandboxResourcePoliciesSet.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { sandboxResourcePoliciesSet } from "../../funcs/sandboxResourcePoliciesSet.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.SetTenantResourcePolicyRequest$inboundSchema,
+};
+
+export const tool$sandboxResourcePoliciesSet: ToolDefinition<typeof args> = {
+  name: "sandbox-resource-policies-set",
+  description: `Set Tenant Resource Policy
+
+Update organization or team resource limits for standard sandbox sessions. The expected revision rejects stale writes.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await sandboxResourcePoliciesSet(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};