@ensembleapp/client-sdk 0.0.49 → 0.0.50

package/dist/index.d.ts

@@ -1,128 +1,12 @@
 import React, { ReactElement, ReactNode } from 'react';
 import * as ai from 'ai';
 import { UIMessage } from 'ai';
-import z, { z as z$1 } from 'zod';
+import { CollectUserInputArgs, MessageFeedback, FeedbackRating, FeedbackNote, EnrichmentResult, WidgetEnrichConfig } from '@repo/agent-sdk/schemas';
+export { FeedbackNote, FeedbackRating, MessageFeedback, WidgetEnrichConfig } from '@repo/agent-sdk/schemas';
 import { FlexibleSchema, InferSchema } from '@ai-sdk/provider-utils';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ClassValue } from 'clsx';
 
-declare const VersionSelectorSchema: z.ZodUnion<readonly [z.ZodLiteral<"latest">, z.ZodLiteral<"published">, z.ZodNumber]>;
-type VersionSelector = z.infer<typeof VersionSelectorSchema>;
-
-declare const FeedbackRatingSchema: z.ZodEnum<{
-    positive: "positive";
-    negative: "negative";
-}>;
-type FeedbackRating = z.infer<typeof FeedbackRatingSchema>;
-interface FeedbackNote {
-    noteId?: string;
-    [key: string]: unknown;
-}
-interface MessageFeedback {
-    rating?: FeedbackRating;
-    improvementText?: string;
-    notes?: FeedbackNote[];
-}
-
-/**
- * Arguments for the collect_user_input tool.
- */
-declare const CollectUserInputArgsSchema: z$1.ZodObject<{
-    title: z$1.ZodOptional<z$1.ZodString>;
-    description: z$1.ZodOptional<z$1.ZodString>;
-    fields: z$1.ZodArray<z$1.ZodDiscriminatedUnion<[z$1.ZodObject<{
-        type: z$1.ZodLiteral<"choices">;
-        name: z$1.ZodString;
-        label: z$1.ZodString;
-        options: z$1.ZodArray<z$1.ZodObject<{
-            label: z$1.ZodString;
-            value: z$1.ZodString;
-            description: z$1.ZodOptional<z$1.ZodString>;
-        }, z$1.core.$strip>>;
-        multiSelect: z$1.ZodOptional<z$1.ZodBoolean>;
-        required: z$1.ZodOptional<z$1.ZodBoolean>;
-        defaultValue: z$1.ZodOptional<z$1.ZodUnion<readonly [z$1.ZodString, z$1.ZodArray<z$1.ZodString>]>>;
-    }, z$1.core.$strip>, z$1.ZodObject<{
-        type: z$1.ZodLiteral<"text">;
-        name: z$1.ZodString;
-        label: z$1.ZodString;
-        placeholder: z$1.ZodOptional<z$1.ZodString>;
-        multiline: z$1.ZodOptional<z$1.ZodBoolean>;
-        required: z$1.ZodOptional<z$1.ZodBoolean>;
-        defaultValue: z$1.ZodOptional<z$1.ZodString>;
-    }, z$1.core.$strip>, z$1.ZodObject<{
-        type: z$1.ZodLiteral<"number">;
-        name: z$1.ZodString;
-        label: z$1.ZodString;
-        min: z$1.ZodOptional<z$1.ZodNumber>;
-        max: z$1.ZodOptional<z$1.ZodNumber>;
-        required: z$1.ZodOptional<z$1.ZodBoolean>;
-        defaultValue: z$1.ZodOptional<z$1.ZodNumber>;
-    }, z$1.core.$strip>, z$1.ZodObject<{
-        type: z$1.ZodLiteral<"checkbox">;
-        name: z$1.ZodString;
-        label: z$1.ZodString;
-        defaultValue: z$1.ZodOptional<z$1.ZodBoolean>;
-    }, z$1.core.$strip>, z$1.ZodObject<{
-        type: z$1.ZodLiteral<"date">;
-        name: z$1.ZodString;
-        label: z$1.ZodString;
-        required: z$1.ZodOptional<z$1.ZodBoolean>;
-        defaultValue: z$1.ZodOptional<z$1.ZodString>;
-    }, z$1.core.$strip>], "type">>;
-    submitLabel: z$1.ZodOptional<z$1.ZodString>;
-    cancelLabel: z$1.ZodOptional<z$1.ZodString>;
-    allowFreeformResponse: z$1.ZodDefault<z$1.ZodOptional<z$1.ZodBoolean>>;
-}, z$1.core.$strip>;
-type CollectUserInputArgs = z$1.infer<typeof CollectUserInputArgsSchema>;
-
-/**
- * fetch data via tool call.
- */
-type ToolCallConfig = {
-    toolId: string;
-    /**
-     * Select the tool version (default to 'latest' if not provided).
-     * - 'latest': pick draft if exists, fallback to latest published version
-     * - 'published': pick latest published version, fallback to draft version
-     * - number: specific tool version (1+)
-     */
-    version?: VersionSelector;
-    /**
-     * Tool input can be literal or templated string with expressions
-     */
-    inputs?: Record<string, unknown>;
-};
-/**
- * Widget enrichment configuration.
- * Keys are used to access the enrichment results in the render function.
- *
- * Example:
- * ```typescript
- * enrich: {
- *   vendorDetails: { toolId: 'vendor-tool', inputs: { ids: "${vendors}" } },
- *   pricing: { toolId: 'pricing-tool', inputs: { ids: "${vendors}" } },
- * },
- * render: (payload, enriched) => {
- *   const vendors = enriched?.vendorDetails?.data;
- *   const pricing = enriched?.pricing?.data;
- * }
- * ```
- */
-type WidgetEnrichConfig = Record<string, ToolCallConfig>;
-/**
- * Enrichment result containing the full tool response.
- * This should mirror ToolResult on server-side without debug/metadata info.
- */
-type EnrichmentResult<TData = unknown> = {
-    success: boolean;
-    error?: {
-        message?: string;
-        code?: string;
-    };
-    data?: TData;
-};
-
 /** Derived part type from UIMessage */
 type UIMessagePart = UIMessage['parts'][number];
 /**
@@ -196,8 +80,17 @@ type UseChatConfig = DeprecatedChatConfig & {
     onFinish?: (message: any) => void;
     onData?: (data: any) => void;
     onMessage?: (message: UIMessage) => void;
+    /** Called when usage data is received (input/output token counts and percentages) */
+    onUsage?: (usage: {
+        inputTokens: number;
+        outputTokens: number;
+        maxInputTokens?: number;
+        maxOutputTokens?: number;
+        inputTokensPercent?: number;
+        outputTokensPercent?: number;
+    }) => void;
 };
-declare function useChat({ api, threadId, agentId, agentVersion, agentExecutionId, dataContext, onError, onAuthError, onFinish, onData, onMessage, }: UseChatConfig): {
+declare function useChat({ api, threadId, agentId, agentVersion, agentExecutionId, dataContext, onError, onAuthError, onFinish, onData, onMessage, onUsage, }: UseChatConfig): {
     messages: ChatMessage[];
     status: ai.ChatStatus;
     isLoadingInitial: boolean;
@@ -224,19 +117,7 @@ declare function useChat({ api, threadId, agentId, agentVersion, agentExecutionI
     sendStructuredMessage: (text: string, payload: StructuredMessagePayload) => Promise<void>;
     stop: () => Promise<void>;
     setMessages: (messages: UIMessage<unknown, ai.UIDataTypes, ai.UITools>[] | ((messages: UIMessage<unknown, ai.UIDataTypes, ai.UITools>[]) => UIMessage<unknown, ai.UIDataTypes, ai.UITools>[])) => void;
-    addToolOutput: <TOOL extends string>({ state, tool, toolCallId, output, errorText, }: {
-        state?: "output-available";
-        tool: TOOL;
-        toolCallId: string;
-        output: unknown;
-        errorText?: never;
-    } | {
-        state: "output-error";
-        tool: TOOL;
-        toolCallId: string;
-        output?: never;
-        errorText: string;
-    }) => Promise<void>;
+    addToolOutput: ai.ChatAddToolOutputFunction<UIMessage<unknown, ai.UIDataTypes, ai.UITools>>;
     addToolApprovalResponse: ai.ChatAddToolApproveResponseFunction;
     pendingToolOutput: PendingToolOutput | null;
     pendingToolApproval: PendingToolApproval | null;
@@ -756,4 +637,4 @@ declare const getVendorCardsWidget: (isProd?: boolean) => SDKUIWidgetDefinition[
 
 declare function cn(...inputs: ClassValue[]): string;
 
-export { type AgentVersion, type AnyUIWidgetDefinition, type ApiConfig, type ApprovalUIConfig, BriefDisplayMode, BriefDisplayMode as BriefRenderer, type ChatMessage, type ChatMessageSection, ChatWidget, type ChatWidgetFeedbackOptions, type ChatWidgetHandle, type ChatWidgetConfig as ChatWidgetProps, type ChatWidgetSpeechToTextOptions, type ChatWidgetStyles, type ChatWidgetVoiceOptions, type CustomerReactDOM, type DisplayMode, type DisplayModeProps, type EmbeddableChatWidgetConfig, type EnrichedResults, type FeedbackConfig, type FeedbackHandlers, type FeedbackNote, type FeedbackRating, type FetchApprovalConfigOptions, FullDisplayMode, FullDisplayMode as FullRenderer, type InvokeTool, type MessageContext, type MessageFeedback, MessageRenderer, type MessageRendererProps, type PendingToolApproval, type PendingToolOutput, type PopupAnchorConfig, PopupChatWidget, type PopupChatWidgetProps, type DisplayModeProps as RendererProps, type SDKUIWidgetDefinition, SDK_WIDGET_MARKER, type StructuredMessagePayload, type SubmitFeedbackParams, TagGroupDisplay, type TagGroupDisplayProps, ToolCallDisplay, type ToolCallDisplayProps, type ToolInvokeRequest, type ToolInvokeResult, type ToolPart, type UIMessagePart, type UIWidgetDefinition, type UseChatConfig, type UseFeedbackConfig, type WidgetEnrichConfig, type WidgetRenderOptions, cn, createSDKWidget, createWidget, defaultChatWidgets, fetchApprovalConfig, getVendorCardsWidget, isSDKWidget, registerChatWidgets, transformMessage, useChat, useFeedback };
+export { type AgentVersion, type AnyUIWidgetDefinition, type ApiConfig, type ApprovalUIConfig, BriefDisplayMode, BriefDisplayMode as BriefRenderer, type ChatMessage, type ChatMessageSection, ChatWidget, type ChatWidgetFeedbackOptions, type ChatWidgetHandle, type ChatWidgetConfig as ChatWidgetProps, type ChatWidgetSpeechToTextOptions, type ChatWidgetStyles, type ChatWidgetVoiceOptions, type CustomerReactDOM, type DisplayMode, type DisplayModeProps, type EmbeddableChatWidgetConfig, type EnrichedResults, type FeedbackConfig, type FeedbackHandlers, type FetchApprovalConfigOptions, FullDisplayMode, FullDisplayMode as FullRenderer, type InvokeTool, type MessageContext, MessageRenderer, type MessageRendererProps, type PendingToolApproval, type PendingToolOutput, type PopupAnchorConfig, PopupChatWidget, type PopupChatWidgetProps, type DisplayModeProps as RendererProps, type SDKUIWidgetDefinition, SDK_WIDGET_MARKER, type StructuredMessagePayload, type SubmitFeedbackParams, TagGroupDisplay, type TagGroupDisplayProps, ToolCallDisplay, type ToolCallDisplayProps, type ToolInvokeRequest, type ToolInvokeResult, type ToolPart, type UIMessagePart, type UIWidgetDefinition, type UseChatConfig, type UseFeedbackConfig, type WidgetRenderOptions, cn, createSDKWidget, createWidget, defaultChatWidgets, fetchApprovalConfig, getVendorCardsWidget, isSDKWidget, registerChatWidgets, transformMessage, useChat, useFeedback };