@blazingagents/sdk 0.21.0 → 0.23.0

package/dist/chat-stream-transport.d.ts

@@ -0,0 +1,5 @@
+import { DefaultChatTransport, type UIMessage } from "ai";
+export declare class ChatStreamTransport<UI_MESSAGE extends UIMessage = UIMessage> extends DefaultChatTransport<UI_MESSAGE> {
+    /** Native Chat rejects an update job for error chunks, so fail the stream before that job runs. */
+    protected processResponseStream(stream: ReadableStream<Uint8Array>): ReadableStream<any>;
+}

package/dist/chat-stream-transport.js

@@ -0,0 +1,15 @@
+import { DefaultChatTransport } from "ai";
+export class ChatStreamTransport extends DefaultChatTransport {
+    /** Native Chat rejects an update job for error chunks, so fail the stream before that job runs. */
+    processResponseStream(stream) {
+        return super.processResponseStream(stream).pipeThrough(new TransformStream({
+            transform(chunk, controller) {
+                if (chunk.type === "error") {
+                    controller.error(new Error(chunk.errorText));
+                    return;
+                }
+                controller.enqueue(chunk);
+            },
+        }));
+    }
+}

package/dist/chat-transport.js

@@ -1,4 +1,5 @@
-import { DefaultChatTransport, lastAssistantMessageIsCompleteWithApprovalResponses, } from "ai";
+import { lastAssistantMessageIsCompleteWithApprovalResponses, } from "ai";
+import { ChatStreamTransport } from "./chat-stream-transport.js";
 import { sessionIdSchema } from "./contracts/ids.js";
 /**
  * Adapts AI SDK chat requests to the Blazing Agents backend-relay shape and
@@ -16,7 +17,7 @@ export class BlazingAgentsChatTransport {
         const transportFetch = transportOptions.fetch ?? globalThis.fetch;
         this.#sessionId =
             sessionId === undefined ? undefined : sessionIdSchema.parse(sessionId);
-        this.#transport = new DefaultChatTransport({
+        this.#transport = new ChatStreamTransport({
             ...transportOptions,
             /** Reads the first Session ID from Location before returning the response. */
             fetch: async (input, init) => {

package/dist/contracts/api.d.ts

@@ -60,6 +60,7 @@ export declare const apiErrorCodeSchema: z.ZodEnum<{
     message_not_found: "message_not_found";
     model_discovery_unsupported: "model_discovery_unsupported";
     model_not_found: "model_not_found";
+    model_spending_limit_exceeded: "model_spending_limit_exceeded";
     model_validation_unavailable: "model_validation_unavailable";
     not_found: "not_found";
     prompt_variable_missing: "prompt_variable_missing";
@@ -143,6 +144,7 @@ export declare const apiErrorSchema: z.ZodObject<{
         message_not_found: "message_not_found";
         model_discovery_unsupported: "model_discovery_unsupported";
         model_not_found: "model_not_found";
+        model_spending_limit_exceeded: "model_spending_limit_exceeded";
         model_validation_unavailable: "model_validation_unavailable";
         not_found: "not_found";
         prompt_variable_missing: "prompt_variable_missing";
@@ -231,6 +233,7 @@ export declare const apiErrorResponseSchema: z.ZodObject<{
             message_not_found: "message_not_found";
             model_discovery_unsupported: "model_discovery_unsupported";
             model_not_found: "model_not_found";
+            model_spending_limit_exceeded: "model_spending_limit_exceeded";
             model_validation_unavailable: "model_validation_unavailable";
             not_found: "not_found";
             prompt_variable_missing: "prompt_variable_missing";

package/dist/contracts/api.js

@@ -28,6 +28,7 @@ export const apiErrorCodeSchema = z.enum([
     "forbidden",
     "not_found",
     "quota_exceeded",
+    "model_spending_limit_exceeded",
     "subscription_required",
     "usage_credit_required",
     "rate_limited",

package/dist/contracts/entities/spending-limits.d.ts

@@ -0,0 +1,101 @@
+import { z } from "zod";
+export declare const spendingLimitResetIntervalSchema: z.ZodEnum<{
+    biweekly: "biweekly";
+    daily: "daily";
+    monthly: "monthly";
+    weekly: "weekly";
+}>;
+export declare const spendingLimitSchema: z.ZodObject<{
+    amountUsd: z.ZodNumber;
+    resetStartDate: z.ZodISODate;
+    resetInterval: z.ZodEnum<{
+        biweekly: "biweekly";
+        daily: "daily";
+        monthly: "monthly";
+        weekly: "weekly";
+    }>;
+}, z.core.$strict>;
+export declare const updateSpendingLimitBodySchema: z.ZodObject<{
+    spendingLimit: z.ZodNullable<z.ZodObject<{
+        amountUsd: z.ZodNumber;
+        resetStartDate: z.ZodISODate;
+        resetInterval: z.ZodEnum<{
+            biweekly: "biweekly";
+            daily: "daily";
+            monthly: "monthly";
+            weekly: "weekly";
+        }>;
+    }, z.core.$strict>>;
+}, z.core.$strict>;
+export declare const spendingLimitPeriodSchema: z.ZodObject<{
+    startsAt: z.ZodISODateTime;
+    endsAt: z.ZodISODateTime;
+    spentUsd: z.ZodNumber;
+    reservedUsd: z.ZodNumber;
+    availableUsd: z.ZodNumber;
+}, z.core.$strict>;
+export declare const spendingLimitResponseSchema: z.ZodObject<{
+    spendingLimit: z.ZodNullable<z.ZodObject<{
+        amountUsd: z.ZodNumber;
+        resetStartDate: z.ZodISODate;
+        resetInterval: z.ZodEnum<{
+            biweekly: "biweekly";
+            daily: "daily";
+            monthly: "monthly";
+            weekly: "weekly";
+        }>;
+    }, z.core.$strict>>;
+    period: z.ZodNullable<z.ZodObject<{
+        startsAt: z.ZodISODateTime;
+        endsAt: z.ZodISODateTime;
+        spentUsd: z.ZodNumber;
+        reservedUsd: z.ZodNumber;
+        availableUsd: z.ZodNumber;
+    }, z.core.$strict>>;
+    nextResetAt: z.ZodNullable<z.ZodISODateTime>;
+}, z.core.$strict>;
+export declare const spendingLimitStopDetailsSchema: z.ZodObject<{
+    scope: z.ZodEnum<{
+        agent: "agent";
+        both: "both";
+        tenant: "tenant";
+    }>;
+    reason: z.ZodEnum<{
+        exhausted: "exhausted";
+        reserved: "reserved";
+        unknown_usage: "unknown_usage";
+        unpriced: "unpriced";
+    }>;
+    spentUsd: z.ZodNumber;
+    reservedUsd: z.ZodNumber;
+    availableUsd: z.ZodNumber;
+    nextResetAt: z.ZodNullable<z.ZodISODateTime>;
+}, z.core.$strict>;
+export type SpendingLimit = z.infer<typeof spendingLimitSchema>;
+export type SpendingLimitResponse = z.infer<typeof spendingLimitResponseSchema>;
+export type SpendingLimitStopDetails = z.infer<typeof spendingLimitStopDetailsSchema>;
+export type UpdateSpendingLimitBody = z.infer<typeof updateSpendingLimitBodySchema>;
+export declare function nextSpendingLimitReset(schedule: Pick<SpendingLimit, "resetStartDate" | "resetInterval">, after: string | Date): string;
+export declare const spendingLimitStopEventSchema: z.ZodObject<{
+    type: z.ZodLiteral<"data-model-spending-limit">;
+    data: z.ZodObject<{
+        scope: z.ZodEnum<{
+            agent: "agent";
+            both: "both";
+            tenant: "tenant";
+        }>;
+        reason: z.ZodEnum<{
+            exhausted: "exhausted";
+            reserved: "reserved";
+            unknown_usage: "unknown_usage";
+            unpriced: "unpriced";
+        }>;
+        spentUsd: z.ZodNumber;
+        reservedUsd: z.ZodNumber;
+        availableUsd: z.ZodNumber;
+        nextResetAt: z.ZodNullable<z.ZodISODateTime>;
+        code: z.ZodLiteral<"model_spending_limit_exceeded">;
+    }, z.core.$strict>;
+    transient: z.ZodLiteral<true>;
+}, z.core.$strict>;
+export type SpendingLimitStopEvent = z.infer<typeof spendingLimitStopEventSchema>;

package/dist/contracts/entities/spending-limits.js

@@ -0,0 +1,79 @@
+import { z } from "zod";
+export const spendingLimitResetIntervalSchema = z.enum([
+    "daily",
+    "weekly",
+    "biweekly",
+    "monthly",
+]);
+export const spendingLimitSchema = z
+    .object({
+    amountUsd: z.number().positive().max(1_000_000_000).multipleOf(0.000_001),
+    resetStartDate: z.iso.date(),
+    resetInterval: spendingLimitResetIntervalSchema,
+})
+    .strict()
+    .meta({ id: "ModelSpendingLimit" });
+export const updateSpendingLimitBodySchema = z
+    .object({
+    spendingLimit: spendingLimitSchema
+        .nullable()
+        .describe("Model-only USD allowance and UTC reset schedule. Set null to disable the limit."),
+})
+    .strict();
+export const spendingLimitPeriodSchema = z
+    .object({
+    startsAt: z.iso.datetime({ offset: true }),
+    endsAt: z.iso.datetime({ offset: true }),
+    spentUsd: z.number().nonnegative(),
+    reservedUsd: z.number().nonnegative(),
+    availableUsd: z.number().nonnegative(),
+})
+    .strict();
+export const spendingLimitResponseSchema = z
+    .object({
+    spendingLimit: spendingLimitSchema.nullable(),
+    period: spendingLimitPeriodSchema.nullable(),
+    nextResetAt: z.iso.datetime({ offset: true }).nullable(),
+})
+    .strict()
+    .meta({ id: "ModelSpendingLimitStatus" });
+export const spendingLimitStopDetailsSchema = z
+    .object({
+    scope: z.enum(["agent", "tenant", "both"]),
+    reason: z.enum(["exhausted", "reserved", "unpriced", "unknown_usage"]),
+    spentUsd: z.number().nonnegative(),
+    reservedUsd: z.number().nonnegative(),
+    availableUsd: z.number().nonnegative(),
+    nextResetAt: z.iso.datetime({ offset: true }).nullable(),
+})
+    .strict();
+export function nextSpendingLimitReset(schedule, after) {
+    const anchor = new Date(`${schedule.resetStartDate}T00:00:00.000Z`);
+    const instant = new Date(after).getTime();
+    if (anchor.getTime() > instant) {
+        return anchor.toISOString();
+    }
+    if (schedule.resetInterval !== "monthly") {
+        const days = { daily: 1, weekly: 7, biweekly: 14 }[schedule.resetInterval];
+        const interval = days * 86_400_000;
+        return new Date(anchor.getTime() +
+            (Math.floor((instant - anchor.getTime()) / interval) + 1) * interval).toISOString();
+    }
+    const current = new Date(instant);
+    let month = current.getUTCMonth();
+    const year = current.getUTCFullYear();
+    const boundary = () => new Date(Date.UTC(year, month, Math.min(anchor.getUTCDate(), new Date(Date.UTC(year, month + 1, 0)).getUTCDate())));
+    if (boundary().getTime() <= instant) {
+        month += 1;
+    }
+    return boundary().toISOString();
+}
+export const spendingLimitStopEventSchema = z
+    .object({
+    type: z.literal("data-model-spending-limit"),
+    data: spendingLimitStopDetailsSchema.extend({
+        code: z.literal("model_spending_limit_exceeded"),
+    }),
+    transient: z.literal(true),
+})
+    .strict();

package/dist/contracts/index.d.ts

@@ -19,6 +19,8 @@ export { chatSteerConsumedEventSchema, sessionActivitySchema, sessionInputReques
 export type { ContinueToolApprovalsBody, SessionForkedFrom, ToolApprovalContinuationState, ToolApprovalDecision, ToolApprovalsResponse, } from "./entities/sessions.ts";
 export { continueToolApprovalsBodySchema, latestSessionListItemSchema, latestSessionsListResponseSchema, sessionForkedFromSchema, sessionListItemSchema, sessionMessageSchema, sessionMessagesResponseSchema, sessionResponseSchema, sessionsListResponseSchema, toolApprovalContinuationStateSchema, toolApprovalDecisionSchema, toolApprovalStateSchema, toolApprovalsResponseSchema, } from "./entities/sessions.ts";
 export { copySkillBodySchema, createSkillBodySchema, skillCopyResultsSchema, skillDetailSchema, skillsListResponseSchema, } from "./entities/skills.ts";
+export type { SpendingLimit, SpendingLimitResponse, SpendingLimitStopDetails, SpendingLimitStopEvent, UpdateSpendingLimitBody, } from "./entities/spending-limits.ts";
+export { nextSpendingLimitReset, spendingLimitPeriodSchema, spendingLimitResetIntervalSchema, spendingLimitResponseSchema, spendingLimitSchema, spendingLimitStopDetailsSchema, spendingLimitStopEventSchema, updateSpendingLimitBodySchema, } from "./entities/spending-limits.ts";
 export { createTaskBodySchema, createTaskResponseSchema, createTaskRunBodySchema, createTaskRunResponseSchema, taskCronConfigSchema, taskIntervalConfigSchema, taskListItemSchema, taskOnceConfigSchema, taskResponseSchema, taskRunMessagesResponseSchema, taskRunResponseSchema, taskRunSchema, taskRunStatusSchema, taskRunsListResponseSchema, taskScheduleInputSchema, taskScheduleKindSchema, taskSchema, tasksListResponseSchema, updateTaskBodySchema, } from "./entities/tasks.ts";
 export { quotaSchema, subscriptionStatusSchema, tenantDeletionSchema, tenantResponseSchema, tenantSchema, tenantSettingsResponseSchema, updateTenantSettingsBodySchema, } from "./entities/tenants.ts";
 export type { SessionUsageQuery, SessionUsageResponse, UsageBucket, UsageOverviewQuery, UsageOverviewResponse, UsageResponse, UsageTotals, } from "./entities/usage.ts";

package/dist/contracts/index.js

@@ -16,6 +16,7 @@ export { createProviderBodySchema, providerListItemSchema, providerModelsRespons
 export { chatSteerConsumedEventSchema, sessionActivitySchema, sessionInputRequestIdSchema, sessionInputResponseSchema, sessionInputSchema, sessionInputStateSchema, sessionInputsQuerySchema, sessionInputsResponseSchema, stopSessionBodySchema, stopSessionResponseSchema, submitSessionInputBodySchema, } from "./entities/session-inputs.js";
 export { continueToolApprovalsBodySchema, latestSessionListItemSchema, latestSessionsListResponseSchema, sessionForkedFromSchema, sessionListItemSchema, sessionMessageSchema, sessionMessagesResponseSchema, sessionResponseSchema, sessionsListResponseSchema, toolApprovalContinuationStateSchema, toolApprovalDecisionSchema, toolApprovalStateSchema, toolApprovalsResponseSchema, } from "./entities/sessions.js";
 export { copySkillBodySchema, createSkillBodySchema, skillCopyResultsSchema, skillDetailSchema, skillsListResponseSchema, } from "./entities/skills.js";
+export { nextSpendingLimitReset, spendingLimitPeriodSchema, spendingLimitResetIntervalSchema, spendingLimitResponseSchema, spendingLimitSchema, spendingLimitStopDetailsSchema, spendingLimitStopEventSchema, updateSpendingLimitBodySchema, } from "./entities/spending-limits.js";
 export { createTaskBodySchema, createTaskResponseSchema, createTaskRunBodySchema, createTaskRunResponseSchema, taskCronConfigSchema, taskIntervalConfigSchema, taskListItemSchema, taskOnceConfigSchema, taskResponseSchema, taskRunMessagesResponseSchema, taskRunResponseSchema, taskRunSchema, taskRunStatusSchema, taskRunsListResponseSchema, taskScheduleInputSchema, taskScheduleKindSchema, taskSchema, tasksListResponseSchema, updateTaskBodySchema, } from "./entities/tasks.js";
 export { quotaSchema, subscriptionStatusSchema, tenantDeletionSchema, tenantResponseSchema, tenantSchema, tenantSettingsResponseSchema, updateTenantSettingsBodySchema, } from "./entities/tenants.js";
 export { sessionUsageQuerySchema, sessionUsageResponseSchema, usageOverviewQuerySchema, usageOverviewResponseSchema, usageResponseSchema, } from "./entities/usage.js";

package/dist/direct-chat-transport.d.ts

@@ -1,4 +1,5 @@
-import { type ChatTransport, DefaultChatTransport, type UIMessage } from "ai";
+import { type ChatTransport, type UIMessage } from "ai";
+import { ChatStreamTransport } from "./chat-stream-transport.ts";
 import type { BlazingAgents } from "./client.ts";
 import type { ChatFunctions } from "./functions.ts";
 export type BlazingAgentsDirectChatTransportOptions = {
@@ -26,7 +27,7 @@ export type BlazingAgentsDirectChatTransportOptions = {
     variables?: never;
 });
 /** Connects useChat directly to the SDK, including native streaming fetch clients. */
-export declare class BlazingAgentsDirectChatTransport<UI_MESSAGE extends UIMessage = UIMessage> extends DefaultChatTransport<UI_MESSAGE> {
+export declare class BlazingAgentsDirectChatTransport<UI_MESSAGE extends UIMessage = UIMessage> extends ChatStreamTransport<UI_MESSAGE> {
     #private;
     /**
      * Creates a chat transport and validates any supplied Session ID.
@@ -41,13 +42,13 @@ export declare class BlazingAgentsDirectChatTransport<UI_MESSAGE extends UIMessa
     sendUserMessages(input: {
         messages: UI_MESSAGE[];
         abortSignal?: AbortSignal;
-    }): Promise<ReadableStream<import("ai").UIMessageChunk<unknown, import("ai").UIDataTypes>>>;
+    }): Promise<ReadableStream<any>>;
     /**
      * Submits the latest user message or complete assistant approval round to the chat.
      * @param input - Operation input and optional cancellation signal.
      * @returns The decoded AI SDK message-chunk stream.
      */
-    sendMessages(input: Parameters<ChatTransport<UI_MESSAGE>["sendMessages"]>[0]): Promise<ReadableStream<import("ai").UIMessageChunk<unknown, import("ai").UIDataTypes>>>;
+    sendMessages(input: Parameters<ChatTransport<UI_MESSAGE>["sendMessages"]>[0]): Promise<ReadableStream<any>>;
     /**
      * Returns null because reconnecting to an existing stream is unsupported.
      * @param _input - Unused reconnect request.

package/dist/direct-chat-transport.js

@@ -1,8 +1,9 @@
-import { DefaultChatTransport, lastAssistantMessageIsCompleteWithApprovalResponses, } from "ai";
+import { lastAssistantMessageIsCompleteWithApprovalResponses, } from "ai";
 import { extractApprovalDecisions } from "./approvals.js";
+import { ChatStreamTransport } from "./chat-stream-transport.js";
 import { sessionIdSchema } from "./contracts/ids.js";
 /** Connects useChat directly to the SDK, including native streaming fetch clients. */
-export class BlazingAgentsDirectChatTransport extends DefaultChatTransport {
+export class BlazingAgentsDirectChatTransport extends ChatStreamTransport {
     #options;
     #sessionId;
     /**

package/dist/index.d.ts

@@ -19,10 +19,12 @@ export type { ChatSteerConsumedEvent, SessionActivity, SessionInput, SessionInpu
 export type { ContinueToolApprovalsBody, SessionForkedFrom, SessionResponse, ToolApprovalDecision, ToolApprovalState, ToolApprovalsResponse, } from "./contracts/entities/sessions.ts";
 export { sessionResponseSchema } from "./contracts/entities/sessions.ts";
 export type { CreateSkillBody, Skill, SkillArchiveType, SkillCopyResult, SkillCopyResults, SkillDetail, SkillFile, SkillsListResponse, } from "./contracts/entities/skills.ts";
+export type { SpendingLimit, SpendingLimitResponse, SpendingLimitStopDetails, SpendingLimitStopEvent, UpdateSpendingLimitBody, } from "./contracts/entities/spending-limits.ts";
+export { nextSpendingLimitReset } from "./contracts/entities/spending-limits.ts";
 export type { SessionUsageQuery, SessionUsageResponse, UsageBucket, UsageOverviewQuery, UsageOverviewResponse, UsageTotals, } from "./contracts/entities/usage.ts";
 export type { CreateWorkspaceBody, UpdateWorkspaceBody, Workspace, WorkspaceNetworkPolicy, WorkspacesListResponse, WorkspaceTier, } from "./contracts/entities/workspaces.ts";
 export { BlazingAgentsDirectChatTransport, type BlazingAgentsDirectChatTransportOptions, } from "./direct-chat-transport.ts";
 export { BlazingAgentsError } from "./errors.ts";
 export { type ChatFunction, type ChatFunctionContext, type ChatFunctions, defineFunction, } from "./functions.ts";
 export { createChatRelay, createCompletionRelay, type RelayContext, type SessionOwnershipStore, } from "./relay.ts";
-export type { AgentClient, AgentSkillsResource, AgentsListOptions, AgentsResource, ArtifactsResource, AttributionInput, BlazingAgentsErrorCode, BlazingAgentsOptions, BlazingAgentsRequestOptions, BlazingAgentsUIMessage, BlazingAgentsUIMessageChunk, ChatConnectionsResource, ChatDeliveriesListOptions, ChatDeliveriesResource, ChatInput, ChatMessageInput, ChatMessagesInput, ChatPromptInput, ChatResult, ChatTrigger, CompletionInput, CompletionPromptIdInput, CompletionPromptInput, CompletionResult, ContinueChatInput, KnownBlazingAgentsErrorCode, LatestSessionsListOptions, McpConnectionsResource, MemoriesListOptions, MemoriesResource, MerchantBindingsListOptions, MerchantBindingsResource, MerchantConnectionResource, MerchantUsageEventsListOptions, MerchantUsageEventsResource, ObjectInput, ObjectPromptIdInput, ObjectPromptInput, ObjectResult, PromptsResource, ProvidersResource, ResourceRequestOptions, ResponseObservation, SessionsResource, SkillsListOptions, TasksResource, TenantResource, TerminalStreamResult, UsageResource, UserClient, WorkspacesListOptions, WorkspacesResource, } from "./types.ts";
+export type { AgentClient, AgentSkillsResource, AgentsListOptions, AgentsResource, ArtifactsResource, AttributionInput, BlazingAgentsErrorCode, BlazingAgentsOptions, BlazingAgentsRequestOptions, BlazingAgentsUIMessage, BlazingAgentsUIMessageChunk, BlazingAgentsUIMessageData, ChatConnectionsResource, ChatDeliveriesListOptions, ChatDeliveriesResource, ChatInput, ChatMessageInput, ChatMessagesInput, ChatPromptInput, ChatResult, ChatTrigger, CompletionInput, CompletionPromptIdInput, CompletionPromptInput, CompletionResult, ContinueChatInput, KnownBlazingAgentsErrorCode, LatestSessionsListOptions, McpConnectionsResource, MemoriesListOptions, MemoriesResource, MerchantBindingsListOptions, MerchantBindingsResource, MerchantConnectionResource, MerchantUsageEventsListOptions, MerchantUsageEventsResource, ObjectInput, ObjectPromptIdInput, ObjectPromptInput, ObjectResult, PromptsResource, ProvidersResource, ResourceRequestOptions, ResponseObservation, SessionsResource, SkillsListOptions, TasksResource, TenantResource, TerminalStreamResult, UsageResource, UserClient, WorkspacesListOptions, WorkspacesResource, } from "./types.ts";

package/dist/index.js

@@ -8,6 +8,7 @@ export { BlazingAgentsChatTransport, } from "./chat-transport.js";
 export { BlazingAgents } from "./client.js";
 export { agentConfigSchema } from "./contracts/entities/agents.js";
 export { sessionResponseSchema } from "./contracts/entities/sessions.js";
+export { nextSpendingLimitReset } from "./contracts/entities/spending-limits.js";
 export { BlazingAgentsDirectChatTransport, } from "./direct-chat-transport.js";
 export { BlazingAgentsError } from "./errors.js";
 export { defineFunction, } from "./functions.js";

package/dist/resources/agents.js

@@ -1,5 +1,6 @@
 import { agentResponseSchema, agentsResponseSchema, } from "../contracts/entities/agents.js";
 import { mcpAttachmentResponseSchema, mcpAttachmentsResponseSchema, } from "../contracts/entities/mcp-connections.js";
+import { spendingLimitResponseSchema } from "../contracts/entities/spending-limits.js";
 import { requestJson } from "../http.js";
 /**
  * Builds the Agents operations using the shared HTTP configuration.
@@ -16,6 +17,14 @@ export function createAgentsResource(config) {
         method: "PUT",
     }, agentResponseSchema);
     return {
+        /** Retrieves the model spending limit using tenant authority. */
+        async getSpendingLimit({ agentId, abortSignal }) {
+            return await requestJson(config, `/v1/agents/${agentId}/spending-limit`, { signal: abortSignal }, spendingLimitResponseSchema);
+        },
+        /** Sets the model spending limit. Null disables it. Requires tenant authority. */
+        async updateSpendingLimit({ agentId, abortSignal, ...body }) {
+            return await requestJson(config, `/v1/agents/${agentId}/spending-limit`, { method: "PUT", json: body, signal: abortSignal }, spendingLimitResponseSchema);
+        },
         /**
          * Creates agents.
          */

package/dist/resources/tenant.js

@@ -1,3 +1,4 @@
+import { spendingLimitResponseSchema } from "../contracts/entities/spending-limits.js";
 import { tenantSettingsResponseSchema } from "../contracts/entities/tenants.js";
 import { requestJson } from "../http.js";
 /**
@@ -8,6 +9,14 @@ import { requestJson } from "../http.js";
  */
 export function createTenantResource(config) {
     return {
+        /** Retrieves the model spending limit using tenant authority. */
+        async getSpendingLimit({ abortSignal } = {}) {
+            return await requestJson(config, "/v1/tenant/spending-limit", { signal: abortSignal }, spendingLimitResponseSchema);
+        },
+        /** Sets the model spending limit. Null disables it. Requires tenant authority. */
+        async updateSpendingLimit({ abortSignal, ...body }) {
+            return await requestJson(config, "/v1/tenant/spending-limit", { method: "PUT", json: body, signal: abortSignal }, spendingLimitResponseSchema);
+        },
         /**
          * Retrieves tenant.
          */

package/dist/types.d.ts

@@ -9,9 +9,10 @@ import type { CreateMemoryBody, MemoriesListResponse, MemoryResponse, UpdateMemo
 import type { CreateMerchantConnectionBody, MerchantBindingResponse, MerchantBindingsResponse, MerchantConnectionResponse, MerchantUsageEventResponse, MerchantUsageEventStatus, MerchantUsageEventsResponse, MerchantUsageSummaryResponse, UpdateMerchantConnectionBody, UpsertMerchantBindingBody } from "./contracts/entities/merchant.ts";
 import type { CreatePromptBody, PromptResponse, PromptsResponse, UpdatePromptBody } from "./contracts/entities/prompts.ts";
 import type { CreateProviderBody, DeleteProviderOptions, ProviderModelsResponse, ProviderResponse, ProvidersResponse, ThinkingLevelsResponse, UpdateProviderBody } from "./contracts/entities/providers.ts";
-import type { SessionInputResponse, SessionInputsQuery, SessionInputsResponse, StopSessionBody, StopSessionResponse, SubmitSessionInputBody } from "./contracts/entities/session-inputs.ts";
+import type { ChatSteerConsumedEvent, SessionInputResponse, SessionInputsQuery, SessionInputsResponse, StopSessionBody, StopSessionResponse, SubmitSessionInputBody } from "./contracts/entities/session-inputs.ts";
 import type { LatestSessionsListResponse, SessionMessagesResponse, SessionResponse, SessionsListResponse, ToolApprovalDecision, ToolApprovalsResponse } from "./contracts/entities/sessions.ts";
 import type { CreateSkillBody, SkillArchiveType, SkillCopyResults, SkillDetail, SkillsListResponse } from "./contracts/entities/skills.ts";
+import type { SpendingLimitResponse, SpendingLimitStopEvent, UpdateSpendingLimitBody } from "./contracts/entities/spending-limits.ts";
 import type { CreateTaskBody, CreateTaskResponse, CreateTaskRunBody, CreateTaskRunResponse, TaskResponse, TaskRunMessagesResponse, TaskRunResponse, TaskRunsListResponse, TasksListResponse, UpdateTaskBody } from "./contracts/entities/tasks.ts";
 import type { TenantSettingsResponse, UpdateTenantSettingsBody } from "./contracts/entities/tenants.ts";
 import type { SessionUsageQuery, SessionUsageResponse, UsageOverviewQuery, UsageOverviewResponse, UsageQuery, UsageResponse } from "./contracts/entities/usage.ts";
@@ -117,8 +118,9 @@ export interface ResponseObservation {
     status: number;
 }
 export type ChatTrigger = "submit-message" | "regenerate-message";
-export type BlazingAgentsUIMessage = UIMessage<BlazingAgentsChatMessageMetadata>;
-export type BlazingAgentsUIMessageChunk = UIMessageChunk<BlazingAgentsChatMessageMetadata>;
+export type BlazingAgentsUIMessageData = Record<"ba-steer-consumed", ChatSteerConsumedEvent["data"]> & Record<"model-spending-limit", SpendingLimitStopEvent["data"]>;
+export type BlazingAgentsUIMessage = UIMessage<BlazingAgentsChatMessageMetadata, BlazingAgentsUIMessageData>;
+export type BlazingAgentsUIMessageChunk = UIMessageChunk<BlazingAgentsChatMessageMetadata, BlazingAgentsUIMessageData>;
 /**
  * End-user attribution (ADR-0001) carried on every generation request.
  * `userId` defaults to `''` (tenant-level) server-side; `metadata` is an
@@ -319,6 +321,10 @@ export interface AgentsResource {
     get(input: {
         agentId: string;
     } & ResourceRequestOptions): Promise<Agent>;
+    /** Retrieves the Agent model spending limit. Requires tenant authority. */
+    getSpendingLimit(input: {
+        agentId: string;
+    } & ResourceRequestOptions): Promise<SpendingLimitResponse>;
     /**
      * Lists available Agent records.
      * @param input - filters and pagination options.
@@ -363,6 +369,10 @@ export interface AgentsResource {
         agentId: string;
         mcpConnectionId: string;
     } & ResourceRequestOptions): Promise<McpAttachmentResponse>;
+    /** Sets the Agent model spending limit. Pass null to disable it. Requires tenant authority. */
+    updateSpendingLimit(input: UpdateSpendingLimitBody & {
+        agentId: string;
+    } & ResourceRequestOptions): Promise<SpendingLimitResponse>;
     /**
      * Uploads an Agent avatar as multipart form data.
      * @param input - Target `agentId`, image `file`, and optional abortSignal.
@@ -1033,6 +1043,8 @@ export interface TenantResource {
      * @throws BlazingAgentsError - If the request fails, is aborted, or returns an invalid response.
      */
     get(input?: ResourceRequestOptions): Promise<TenantSettingsResponse>;
+    /** Retrieves the tenant model spending limit. Requires tenant authority. */
+    getSpendingLimit(input?: ResourceRequestOptions): Promise<SpendingLimitResponse>;
     /**
      * Updates Tenant settings.
      * @param input - fields to change; optional abortSignal.
@@ -1040,6 +1052,8 @@ export interface TenantResource {
      * @throws BlazingAgentsError - If the request fails, is aborted, or returns an invalid response.
      */
     patch(input: UpdateTenantSettingsBody & ResourceRequestOptions): Promise<TenantSettingsResponse>;
+    /** Sets the tenant model spending limit. Pass null to disable it. Requires tenant authority. */
+    updateSpendingLimit(input: UpdateSpendingLimitBody & ResourceRequestOptions): Promise<SpendingLimitResponse>;
 }
 export interface UsageResource {
     /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blazingagents/sdk",
-  "version": "0.21.0",
+  "version": "0.23.0",
   "description": "TypeScript SDK for the Blazing Agents API",
   "homepage": "https://docs.blazingagents.com",
   "bugs": {