@meetkai/mka1 0.48.25 → 0.48.26

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

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsSetClusterPrice } from "../../funcs/llmModelsSetClusterPrice.js";
+import * as components from "../../models/components/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: components.PutModelPriceBody$inboundSchema,
+};
+
+export const tool$llmModelsSetClusterPrice: ToolDefinition<typeof args> = {
+  name: "llm-models-set-cluster-price",
+  description: `Set a cluster-default model price
+
+Upserts the cluster-default price for a model as a new dated version. \`rates\` is a flexible metric→rate map; omitted metrics are not charged.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsSetClusterPrice(
+      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/llmModelsSetOrgPrice.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmModelsSetOrgPrice } from "../../funcs/llmModelsSetOrgPrice.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.SetOrgPriceRequest$inboundSchema,
+};
+
+export const tool$llmModelsSetOrgPrice: ToolDefinition<typeof args> = {
+  name: "llm-models-set-org-price",
+  description: `Set an org price override
+
+Upserts a per-org price override for a model (new dated version). Resolution at spend time is org override → cluster default → $0.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmModelsSetOrgPrice(
+      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/llmUsageCosts.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { llmUsageCosts } from "../../funcs/llmUsageCosts.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.GetCostsUsageRequest$inboundSchema,
+};
+
+export const tool$llmUsageCosts: ToolDefinition<typeof args> = {
+  name: "llm-usage-costs",
+  description: `Get cost by model
+
+Aggregates spend by model over a time range, priced in the cluster currency from the model price book. Org-scoped: org admins see their org; cluster admins may pass \`org_id\`. Includes billable work with no usage row of its own (store=false responses, agent tool calls), so totals reconcile with the live budget gauge.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await llmUsageCosts(
+      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/models/components/budgetevent.ts

@@ -0,0 +1,108 @@
+/*
+ * 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 BudgetEventType = {
+  ThresholdAlert: "threshold_alert",
+  Blocked: "blocked",
+} as const;
+export type BudgetEventType = ClosedEnum<typeof BudgetEventType>;
+
+export type BudgetEvent = {
+  id: string;
+  budgetId: string;
+  type: BudgetEventType;
+  pct: number | null;
+  periodKey: string;
+  spendSnapshot: string;
+  limitSnapshot: string;
+  createdAt: string;
+};
+
+/** @internal */
+export const BudgetEventType$inboundSchema: z.ZodNativeEnum<
+  typeof BudgetEventType
+> = z.nativeEnum(BudgetEventType);
+/** @internal */
+export const BudgetEventType$outboundSchema: z.ZodNativeEnum<
+  typeof BudgetEventType
+> = BudgetEventType$inboundSchema;
+
+/** @internal */
+export const BudgetEvent$inboundSchema: z.ZodType<
+  BudgetEvent,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  id: z.string(),
+  budget_id: z.string(),
+  type: BudgetEventType$inboundSchema,
+  pct: z.nullable(z.number()),
+  period_key: z.string(),
+  spend_snapshot: z.string(),
+  limit_snapshot: z.string(),
+  created_at: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "budget_id": "budgetId",
+    "period_key": "periodKey",
+    "spend_snapshot": "spendSnapshot",
+    "limit_snapshot": "limitSnapshot",
+    "created_at": "createdAt",
+  });
+});
+/** @internal */
+export type BudgetEvent$Outbound = {
+  id: string;
+  budget_id: string;
+  type: string;
+  pct: number | null;
+  period_key: string;
+  spend_snapshot: string;
+  limit_snapshot: string;
+  created_at: string;
+};
+
+/** @internal */
+export const BudgetEvent$outboundSchema: z.ZodType<
+  BudgetEvent$Outbound,
+  z.ZodTypeDef,
+  BudgetEvent
+> = z.object({
+  id: z.string(),
+  budgetId: z.string(),
+  type: BudgetEventType$outboundSchema,
+  pct: z.nullable(z.number()),
+  periodKey: z.string(),
+  spendSnapshot: z.string(),
+  limitSnapshot: z.string(),
+  createdAt: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    budgetId: "budget_id",
+    periodKey: "period_key",
+    spendSnapshot: "spend_snapshot",
+    limitSnapshot: "limit_snapshot",
+    createdAt: "created_at",
+  });
+});
+
+export function budgetEventToJSON(budgetEvent: BudgetEvent): string {
+  return JSON.stringify(BudgetEvent$outboundSchema.parse(budgetEvent));
+}
+export function budgetEventFromJSON(
+  jsonString: string,
+): SafeParseResult<BudgetEvent, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => BudgetEvent$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'BudgetEvent' from JSON`,
+  );
+}

package/src/models/components/budgetlimitkind.ts

@@ -0,0 +1,21 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { ClosedEnum } from "../../types/enums.js";
+
+export const BudgetLimitKind = {
+  Cost: "cost",
+  Tokens: "tokens",
+} as const;
+export type BudgetLimitKind = ClosedEnum<typeof BudgetLimitKind>;
+
+/** @internal */
+export const BudgetLimitKind$inboundSchema: z.ZodNativeEnum<
+  typeof BudgetLimitKind
+> = z.nativeEnum(BudgetLimitKind);
+/** @internal */
+export const BudgetLimitKind$outboundSchema: z.ZodNativeEnum<
+  typeof BudgetLimitKind
+> = BudgetLimitKind$inboundSchema;

package/src/models/components/budgetowner.ts

@@ -0,0 +1,19 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { ClosedEnum } from "../../types/enums.js";
+
+export const BudgetOwner = {
+  Cluster: "cluster",
+  Org: "org",
+} as const;
+export type BudgetOwner = ClosedEnum<typeof BudgetOwner>;
+
+/** @internal */
+export const BudgetOwner$inboundSchema: z.ZodNativeEnum<typeof BudgetOwner> = z
+  .nativeEnum(BudgetOwner);
+/** @internal */
+export const BudgetOwner$outboundSchema: z.ZodNativeEnum<typeof BudgetOwner> =
+  BudgetOwner$inboundSchema;

package/src/models/components/budgetperiod.ts

@@ -0,0 +1,20 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { ClosedEnum } from "../../types/enums.js";
+
+export const BudgetPeriod = {
+  Daily: "daily",
+  Weekly: "weekly",
+  Monthly: "monthly",
+} as const;
+export type BudgetPeriod = ClosedEnum<typeof BudgetPeriod>;
+
+/** @internal */
+export const BudgetPeriod$inboundSchema: z.ZodNativeEnum<typeof BudgetPeriod> =
+  z.nativeEnum(BudgetPeriod);
+/** @internal */
+export const BudgetPeriod$outboundSchema: z.ZodNativeEnum<typeof BudgetPeriod> =
+  BudgetPeriod$inboundSchema;

package/src/models/components/budgetresponse.ts

@@ -0,0 +1,128 @@
+/*
+ * 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 { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  BudgetLimitKind,
+  BudgetLimitKind$inboundSchema,
+  BudgetLimitKind$outboundSchema,
+} from "./budgetlimitkind.js";
+import {
+  BudgetOwner,
+  BudgetOwner$inboundSchema,
+  BudgetOwner$outboundSchema,
+} from "./budgetowner.js";
+import {
+  BudgetPeriod,
+  BudgetPeriod$inboundSchema,
+  BudgetPeriod$outboundSchema,
+} from "./budgetperiod.js";
+import {
+  BudgetScope,
+  BudgetScope$inboundSchema,
+  BudgetScope$outboundSchema,
+} from "./budgetscope.js";
+import {
+  BudgetThresholdInput,
+  BudgetThresholdInput$inboundSchema,
+  BudgetThresholdInput$Outbound,
+  BudgetThresholdInput$outboundSchema,
+} from "./budgetthresholdinput.js";
+import {
+  SpendGauge,
+  SpendGauge$inboundSchema,
+  SpendGauge$Outbound,
+  SpendGauge$outboundSchema,
+} from "./spendgauge.js";
+
+export type BudgetResponse = {
+  id: string;
+  scope: BudgetScope;
+  scopeId: string;
+  owner: BudgetOwner;
+  period: BudgetPeriod;
+  limitKind: BudgetLimitKind;
+  limit: number;
+  thresholds: Array<BudgetThresholdInput>;
+  alertWebhookUrl: string | null;
+  spend: SpendGauge;
+};
+
+/** @internal */
+export const BudgetResponse$inboundSchema: z.ZodType<
+  BudgetResponse,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  id: z.string(),
+  scope: BudgetScope$inboundSchema,
+  scope_id: z.string(),
+  owner: BudgetOwner$inboundSchema,
+  period: BudgetPeriod$inboundSchema,
+  limit_kind: BudgetLimitKind$inboundSchema,
+  limit: z.number(),
+  thresholds: z.array(BudgetThresholdInput$inboundSchema),
+  alert_webhook_url: z.nullable(z.string()),
+  spend: SpendGauge$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "scope_id": "scopeId",
+    "limit_kind": "limitKind",
+    "alert_webhook_url": "alertWebhookUrl",
+  });
+});
+/** @internal */
+export type BudgetResponse$Outbound = {
+  id: string;
+  scope: string;
+  scope_id: string;
+  owner: string;
+  period: string;
+  limit_kind: string;
+  limit: number;
+  thresholds: Array<BudgetThresholdInput$Outbound>;
+  alert_webhook_url: string | null;
+  spend: SpendGauge$Outbound;
+};
+
+/** @internal */
+export const BudgetResponse$outboundSchema: z.ZodType<
+  BudgetResponse$Outbound,
+  z.ZodTypeDef,
+  BudgetResponse
+> = z.object({
+  id: z.string(),
+  scope: BudgetScope$outboundSchema,
+  scopeId: z.string(),
+  owner: BudgetOwner$outboundSchema,
+  period: BudgetPeriod$outboundSchema,
+  limitKind: BudgetLimitKind$outboundSchema,
+  limit: z.number(),
+  thresholds: z.array(BudgetThresholdInput$outboundSchema),
+  alertWebhookUrl: z.nullable(z.string()),
+  spend: SpendGauge$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    scopeId: "scope_id",
+    limitKind: "limit_kind",
+    alertWebhookUrl: "alert_webhook_url",
+  });
+});
+
+export function budgetResponseToJSON(budgetResponse: BudgetResponse): string {
+  return JSON.stringify(BudgetResponse$outboundSchema.parse(budgetResponse));
+}
+export function budgetResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<BudgetResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => BudgetResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'BudgetResponse' from JSON`,
+  );
+}

package/src/models/components/budgetscope.ts

@@ -0,0 +1,19 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { ClosedEnum } from "../../types/enums.js";
+
+export const BudgetScope = {
+  Org: "org",
+  Apikey: "apikey",
+} as const;
+export type BudgetScope = ClosedEnum<typeof BudgetScope>;
+
+/** @internal */
+export const BudgetScope$inboundSchema: z.ZodNativeEnum<typeof BudgetScope> = z
+  .nativeEnum(BudgetScope);
+/** @internal */
+export const BudgetScope$outboundSchema: z.ZodNativeEnum<typeof BudgetScope> =
+  BudgetScope$inboundSchema;

package/src/models/components/budgetthresholdinput.ts

@@ -0,0 +1,69 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+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 Action = {
+  Alert: "alert",
+  Block: "block",
+} as const;
+export type Action = ClosedEnum<typeof Action>;
+
+export type BudgetThresholdInput = {
+  pct: number;
+  action: Action;
+};
+
+/** @internal */
+export const Action$inboundSchema: z.ZodNativeEnum<typeof Action> = z
+  .nativeEnum(Action);
+/** @internal */
+export const Action$outboundSchema: z.ZodNativeEnum<typeof Action> =
+  Action$inboundSchema;
+
+/** @internal */
+export const BudgetThresholdInput$inboundSchema: z.ZodType<
+  BudgetThresholdInput,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  pct: z.number().int(),
+  action: Action$inboundSchema,
+});
+/** @internal */
+export type BudgetThresholdInput$Outbound = {
+  pct: number;
+  action: string;
+};
+
+/** @internal */
+export const BudgetThresholdInput$outboundSchema: z.ZodType<
+  BudgetThresholdInput$Outbound,
+  z.ZodTypeDef,
+  BudgetThresholdInput
+> = z.object({
+  pct: z.number().int(),
+  action: Action$outboundSchema,
+});
+
+export function budgetThresholdInputToJSON(
+  budgetThresholdInput: BudgetThresholdInput,
+): string {
+  return JSON.stringify(
+    BudgetThresholdInput$outboundSchema.parse(budgetThresholdInput),
+  );
+}
+export function budgetThresholdInputFromJSON(
+  jsonString: string,
+): SafeParseResult<BudgetThresholdInput, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => BudgetThresholdInput$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'BudgetThresholdInput' from JSON`,
+  );
+}

package/src/models/components/clustercurrencyresponse.ts

@@ -0,0 +1,51 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type ClusterCurrencyResponse = {
+  currency: string | null;
+};
+
+/** @internal */
+export const ClusterCurrencyResponse$inboundSchema: z.ZodType<
+  ClusterCurrencyResponse,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  currency: z.nullable(z.string()),
+});
+/** @internal */
+export type ClusterCurrencyResponse$Outbound = {
+  currency: string | null;
+};
+
+/** @internal */
+export const ClusterCurrencyResponse$outboundSchema: z.ZodType<
+  ClusterCurrencyResponse$Outbound,
+  z.ZodTypeDef,
+  ClusterCurrencyResponse
+> = z.object({
+  currency: z.nullable(z.string()),
+});
+
+export function clusterCurrencyResponseToJSON(
+  clusterCurrencyResponse: ClusterCurrencyResponse,
+): string {
+  return JSON.stringify(
+    ClusterCurrencyResponse$outboundSchema.parse(clusterCurrencyResponse),
+  );
+}
+export function clusterCurrencyResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<ClusterCurrencyResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ClusterCurrencyResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ClusterCurrencyResponse' from JSON`,
+  );
+}

package/src/models/components/costgrouprow.ts

@@ -0,0 +1,94 @@
+/*
+ * 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 { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type CostGroupRow = {
+  model: string | null;
+  apiKeyId: string | null;
+  teamId: string | null;
+  externalUserId: string | null;
+  orgId: string | null;
+  /**
+   * Cost in major currency units (e.g. dollars)
+   */
+  cost: number;
+  /**
+   * Cost in integer NANO-units — 1e9 per major currency unit (NOT cents). Divide by 1e9 for the major amount; use `cost` for the ready-made major value.
+   */
+  costNano: string;
+};
+
+/** @internal */
+export const CostGroupRow$inboundSchema: z.ZodType<
+  CostGroupRow,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  model: z.nullable(z.string()),
+  api_key_id: z.nullable(z.string()),
+  team_id: z.nullable(z.string()),
+  external_user_id: z.nullable(z.string()),
+  org_id: z.nullable(z.string()),
+  cost: z.number(),
+  cost_nano: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "api_key_id": "apiKeyId",
+    "team_id": "teamId",
+    "external_user_id": "externalUserId",
+    "org_id": "orgId",
+    "cost_nano": "costNano",
+  });
+});
+/** @internal */
+export type CostGroupRow$Outbound = {
+  model: string | null;
+  api_key_id: string | null;
+  team_id: string | null;
+  external_user_id: string | null;
+  org_id: string | null;
+  cost: number;
+  cost_nano: string;
+};
+
+/** @internal */
+export const CostGroupRow$outboundSchema: z.ZodType<
+  CostGroupRow$Outbound,
+  z.ZodTypeDef,
+  CostGroupRow
+> = z.object({
+  model: z.nullable(z.string()),
+  apiKeyId: z.nullable(z.string()),
+  teamId: z.nullable(z.string()),
+  externalUserId: z.nullable(z.string()),
+  orgId: z.nullable(z.string()),
+  cost: z.number(),
+  costNano: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    apiKeyId: "api_key_id",
+    teamId: "team_id",
+    externalUserId: "external_user_id",
+    orgId: "org_id",
+    costNano: "cost_nano",
+  });
+});
+
+export function costGroupRowToJSON(costGroupRow: CostGroupRow): string {
+  return JSON.stringify(CostGroupRow$outboundSchema.parse(costGroupRow));
+}
+export function costGroupRowFromJSON(
+  jsonString: string,
+): SafeParseResult<CostGroupRow, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => CostGroupRow$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'CostGroupRow' from JSON`,
+  );
+}

package/src/models/components/costsusagegroupby.ts

@@ -0,0 +1,24 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { ClosedEnum } from "../../types/enums.js";
+
+export const CostsUsageGroupBy = {
+  Model: "model",
+  ApiKeyId: "api_key_id",
+  TeamId: "team_id",
+  ExternalUserId: "external_user_id",
+  OrgId: "org_id",
+} as const;
+export type CostsUsageGroupBy = ClosedEnum<typeof CostsUsageGroupBy>;
+
+/** @internal */
+export const CostsUsageGroupBy$inboundSchema: z.ZodNativeEnum<
+  typeof CostsUsageGroupBy
+> = z.nativeEnum(CostsUsageGroupBy);
+/** @internal */
+export const CostsUsageGroupBy$outboundSchema: z.ZodNativeEnum<
+  typeof CostsUsageGroupBy
+> = CostsUsageGroupBy$inboundSchema;