@meetkai/mka1 0.48.32 → 0.48.33

package/src/funcs/budgetsSetOrg.ts

@@ -0,0 +1,178 @@
+/*
+ * 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 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 budgetsSetOrg(
+  client: SDKCore,
+  request: operations.SetOrgBudgetRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    operations.SetOrgBudgetResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.SetOrgBudgetRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      operations.SetOrgBudgetResponseBody,
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) => operations.SetOrgBudgetRequest$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 = {
+    orgId: encodeSimple("orgId", payload.orgId, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc("/api/v1/budgeting/budgets/org/{orgId}")(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);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "setOrgBudget",
+    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.SetOrgBudgetResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, operations.SetOrgBudgetResponseBody$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/usageCosts.ts

@@ -0,0 +1,179 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { SDKCore } from "../core.js";
+import { encodeFormQuery } 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";
+
+/**
+ * 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 usageCosts(
+  client: SDKCore,
+  request: operations.GetCostsRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    operations.GetCostsResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.GetCostsRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      operations.GetCostsResponseBody,
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) => operations.GetCostsRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = null;
+
+  const path = pathToFunc("/api/v1/budgeting/usage/costs")();
+
+  const query = encodeFormQuery({
+    "end_time": payload.end_time,
+    "group_by": payload.group_by,
+    "org_id": payload.org_id,
+    "start_time": payload.start_time,
+  });
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.bearerAuth);
+  const securityInput = secConfig == null ? {} : { bearerAuth: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "getCosts",
+    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: "GET",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    query: query,
+    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.GetCostsResponseBody,
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, operations.GetCostsResponseBody$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/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.48.32",
+  sdkVersion: "0.48.33",
   genVersion: "2.916.2",
-  userAgent: "speakeasy-sdk/typescript 0.48.32 2.916.2 1.1.0 @meetkai/mka1",
+  userAgent: "speakeasy-sdk/typescript 0.48.33 2.916.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.48.32",
+    currentVersion: "0.48.33",
   },
 });
 

package/src/mcp-server/server.ts

@@ -70,6 +70,17 @@ import { tool$authTeamsList } from "./tools/authTeamsList.js";
 import { tool$authTeamsListMembers } from "./tools/authTeamsListMembers.js";
 import { tool$authTeamsRemoveMember } from "./tools/authTeamsRemoveMember.js";
 import { tool$authTeamsUpdate } from "./tools/authTeamsUpdate.js";
+import { tool$budgetsApiKeyEvents } from "./tools/budgetsApiKeyEvents.js";
+import { tool$budgetsDeleteApiKey } from "./tools/budgetsDeleteApiKey.js";
+import { tool$budgetsDeleteOrg } from "./tools/budgetsDeleteOrg.js";
+import { tool$budgetsGetApiKey } from "./tools/budgetsGetApiKey.js";
+import { tool$budgetsGetCurrency } from "./tools/budgetsGetCurrency.js";
+import { tool$budgetsGetOrg } from "./tools/budgetsGetOrg.js";
+import { tool$budgetsList } from "./tools/budgetsList.js";
+import { tool$budgetsOrgEvents } from "./tools/budgetsOrgEvents.js";
+import { tool$budgetsSetApiKey } from "./tools/budgetsSetApiKey.js";
+import { tool$budgetsSetCurrency } from "./tools/budgetsSetCurrency.js";
+import { tool$budgetsSetOrg } from "./tools/budgetsSetOrg.js";
 import { tool$guardrailsDeleteGuardrails } from "./tools/guardrailsDeleteGuardrails.js";
 import { tool$guardrailsGetGuardrails } from "./tools/guardrailsGetGuardrails.js";
 import { tool$guardrailsListGuardrailsPolicies } from "./tools/guardrailsListGuardrailsPolicies.js";
@@ -292,6 +303,7 @@ import { tool$searchTextStoreCreateTextStore } from "./tools/searchTextStoreCrea
 import { tool$searchTextStoreDeleteTexts } from "./tools/searchTextStoreDeleteTexts.js";
 import { tool$searchTextStoreDeleteTextStore } from "./tools/searchTextStoreDeleteTextStore.js";
 import { tool$searchTextStoreSearchTexts } from "./tools/searchTextStoreSearchTexts.js";
+import { tool$usageCosts } from "./tools/usageCosts.js";
 
 export function createMCPServer(deps: {
   logger: ConsoleLogger;
@@ -303,7 +315,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.48.32",
+    version: "0.48.33",
   });
 
   const client = new SDKCore({
@@ -372,6 +384,18 @@ export function createMCPServer(deps: {
   tool(tool$agentRunsStreamAgentRunEvents);
   tool(tool$agentRunsWakeAgentRun);
   tool(tool$agentRunsGetAgentRun);
+  tool(tool$budgetsList);
+  tool(tool$budgetsGetOrg);
+  tool(tool$budgetsSetOrg);
+  tool(tool$budgetsDeleteOrg);
+  tool(tool$budgetsOrgEvents);
+  tool(tool$budgetsGetApiKey);
+  tool(tool$budgetsSetApiKey);
+  tool(tool$budgetsDeleteApiKey);
+  tool(tool$budgetsApiKeyEvents);
+  tool(tool$budgetsGetCurrency);
+  tool(tool$budgetsSetCurrency);
+  tool(tool$usageCosts);
   tool(tool$permissionsLlmGrant);
   tool(tool$permissionsLlmRevoke);
   tool(tool$permissionsLlmCheck);

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

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsApiKeyEvents } from "../../funcs/budgetsApiKeyEvents.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.ListApiKeyBudgetEventsRequest$inboundSchema,
+};
+
+export const tool$budgetsApiKeyEvents: ToolDefinition<typeof args> = {
+  name: "budgets-api-key-events",
+  description: `List apikey budget events
+
+Returns recent threshold-alert and block events for the scope's budgets.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await budgetsApiKeyEvents(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.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/budgetsDeleteApiKey.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsDeleteApiKey } from "../../funcs/budgetsDeleteApiKey.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.DeleteApiKeyBudgetRequest$inboundSchema,
+};
+
+export const tool$budgetsDeleteApiKey: ToolDefinition<typeof args> = {
+  name: "budgets-delete-api-key",
+  description: `Delete a apikey budget
+
+Removes the apikey's budget for the given owner; enforcement and alerts stop.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await budgetsDeleteApiKey(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.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/budgetsDeleteOrg.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsDeleteOrg } from "../../funcs/budgetsDeleteOrg.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.DeleteOrgBudgetRequest$inboundSchema,
+};
+
+export const tool$budgetsDeleteOrg: ToolDefinition<typeof args> = {
+  name: "budgets-delete-org",
+  description: `Delete a org budget
+
+Removes the org's budget for the given owner; enforcement and alerts stop.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await budgetsDeleteOrg(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.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/budgetsGetApiKey.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsGetApiKey } from "../../funcs/budgetsGetApiKey.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.GetApiKeyBudgetsRequest$inboundSchema,
+};
+
+export const tool$budgetsGetApiKey: ToolDefinition<typeof args> = {
+  name: "budgets-get-api-key",
+  description: `List apikey budgets
+
+Returns the apikey's budgets (self-budget and/or ceiling) with each one's live spend gauge.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await budgetsGetApiKey(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.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/budgetsGetCurrency.ts

@@ -0,0 +1,28 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsGetCurrency } from "../../funcs/budgetsGetCurrency.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+export const tool$budgetsGetCurrency: ToolDefinition = {
+  name: "budgets-get-currency",
+  description: `Get display currency`,
+  tool: async (client, ctx) => {
+    const [result, apiCall] = await budgetsGetCurrency(
+      client,
+      { fetchOptions: { signal: ctx.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/budgetsGetOrg.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsGetOrg } from "../../funcs/budgetsGetOrg.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.GetOrgBudgetsRequest$inboundSchema,
+};
+
+export const tool$budgetsGetOrg: ToolDefinition<typeof args> = {
+  name: "budgets-get-org",
+  description: `List org budgets
+
+Returns the org's budgets (self-budget and/or ceiling) with each one's live spend gauge.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await budgetsGetOrg(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.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/budgetsList.ts

@@ -0,0 +1,30 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { budgetsList } from "../../funcs/budgetsList.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+export const tool$budgetsList: ToolDefinition = {
+  name: "budgets-list",
+  description: `List budgets
+
+Cluster admins see all budgets; org admins see their org's budgets.`,
+  tool: async (client, ctx) => {
+    const [result, apiCall] = await budgetsList(
+      client,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};