@chaoslabs/ai-sdk 0.0.10 → 1.0.0

package/dist/schemas.js

@@ -2,6 +2,48 @@
 // Updated to match actual server response structure
 import { z } from 'zod';
 // ============================================================================
+// Client Configuration Schemas
+// ============================================================================
+export const ChaosConfigSchema = z.object({
+    apiKey: z.string(),
+    baseUrl: z.string().nullish(),
+    timeout: z.number().nullish(),
+    useNativeHttp: z.boolean().nullish(),
+});
+// ============================================================================
+// Request Schemas
+// ============================================================================
+export const InputItemSchema = z.object({
+    type: z.literal('message'),
+    role: z.enum(['user', 'system', 'assistant']),
+    content: z.string(),
+});
+export const WalletInfoSchema = z.object({
+    address: z.string(),
+    chain: z.string(),
+});
+export const RequestMetadataSchema = z.object({
+    user_id: z.string(),
+    session_id: z.string(),
+    wallets: z.array(WalletInfoSchema).nullish(),
+});
+// Note: CreateResponseParams schema omitted as it contains a function callback (onStreamEvent)
+// which cannot be validated with Zod
+// ============================================================================
+// Response Schemas
+// ============================================================================
+export const ResponseErrorSchema = z.object({
+    message: z.string(),
+    type: z.string(),
+    code: z.string(),
+});
+export const OutputTextSchema = z.object({
+    type: z.literal('output_text'),
+    text: z.string(),
+});
+// Forward declaration for ChaosBlockSchema (defined after BlockSchema)
+// ChaosBlockSchema will be defined after all block schemas
+// ============================================================================
 // Block Schemas - Matching Actual Server Response Structure
 // ============================================================================
 // --- Markdown Block ---
@@ -11,39 +53,12 @@ export const MarkdownBlockSchema = z.object({
 });
 // --- Table Block ---
 // Server sends: { blockType: "table", ... }
-export const TableColumnTypeSchema = z.enum([
-    'text',
-    'number',
-    'currency',
-    'percentage',
-    'token',
-    'date',
-    'link',
-    'boolean',
-]);
-export const TableSortDirectionSchema = z.enum(['asc', 'desc']);
-export const TableColumnConfigSchema = z.object({
-    type: TableColumnTypeSchema.optional(),
-    sortable: z.boolean().optional(),
-    align: z.enum(['left', 'center', 'right']).optional(),
-    width: z.string().optional(),
-});
 export const TableBlockSchema = z.object({
     blockType: z.literal('table'),
-    title: z.string(),
+    title: z.string().nullish(),
     tableHeaders: z.array(z.string()),
-    tableRows: z.array(z.array(z.union([z.string(), z.number(), z.boolean(), z.null()]))),
-    tableHeadersTypes: z.record(z.string(), z.union([TableColumnTypeSchema, z.literal('currency'), z.literal('percentage')])).optional().nullable(),
-    tableColumnConfigs: z.record(z.string(), TableColumnConfigSchema).optional().nullable(),
-    tableHeadersMetadata: z.record(z.string(), z.object({ type: z.string().optional() }).passthrough()).optional().nullable(),
-    sourceName: z.string().optional().nullable(),
-    defaultSortColumn: z.string().optional().nullable(),
-    defaultSortDirection: TableSortDirectionSchema.optional().nullable(),
-    maxRows: z.number().optional().nullable(),
-    searchable: z.boolean().optional().nullable(),
-    exportable: z.boolean().optional().nullable(),
-    tool_params: z.unknown().optional().nullable(),
-    tool_name: z.string().optional().nullable(),
+    tableRows: z.array(z.array(z.unknown())),
+    tableHeadersMetadata: z.record(z.string(), z.record(z.string(), z.unknown())).nullish(),
 });
 // --- Chart Block ---
 // Server sends: { blockType: "chart", data: [[label, value], ...], ... }
@@ -56,7 +71,7 @@ export const ChartDataPointSchema = z.union([
 export const ChartSeriesSchema = z.object({
     name: z.string(),
     data: z.array(ChartDataPointSchema),
-    color: z.string().optional(),
+    color: z.string().nullish(),
 });
 // Timeseries format: server sends series as { "24H": [{ label, data }, ...] }
 export const TimeseriesSeriesItemSchema = z.object({
@@ -73,35 +88,44 @@ export const ChartSeriesFieldSchema = z.union([
 export const ChartSegmentSchema = z.object({
     label: z.string(),
     value: z.number(),
-    color: z.string().optional(),
+    color: z.string().nullish(),
 });
 export const ChartAxisSchema = z.object({
-    title: z.string().optional(),
-    min: z.number().optional(),
-    max: z.number().optional(),
-    type: z.string().optional(),
+    title: z.string().nullish(),
+    min: z.number().nullish(),
+    max: z.number().nullish(),
+    type: z.string().nullish(),
 });
 // Legacy data format for pie/donut charts: [[label, value], ...]
 export const ChartLegacyDataSchema = z.array(z.tuple([z.string(), z.number()]));
-export const ChartBlockSchema = z.object({
+// --- Pie Chart Block ---
+// Server sends: { blockType: "chart", chartType: "pie", data: [[label, value], ...] }
+export const PieChartBlockSchema = z.object({
     blockType: z.literal('chart'),
-    title: z.string(),
-    chartType: z.enum(['pie', 'donut', 'line', 'area', 'bar', 'timeseries']).optional().nullable(),
-    series: ChartSeriesFieldSchema.nullish(),
-    segments: z.array(ChartSegmentSchema).optional().nullable(),
-    // Legacy data format from server: [[label, value], ...]
-    data: ChartLegacyDataSchema.optional().nullable(),
-    categories: z.array(z.string()).optional().nullable(),
-    xAxis: ChartAxisSchema.optional().nullable(),
-    yAxis: ChartAxisSchema.optional().nullable(),
-    isCurrency: z.boolean().optional().nullable(),
-    sourceName: z.string().optional().nullable(),
-    timeframe: z.string().optional().nullable(),
-    captions: z.array(z.object({ label: z.string(), value: z.string() })).nullish(),
-    titleCryptIcons: z.array(z.object({ name: z.string(), entity: z.string().optional() })).nullish(),
-    tool_params: z.unknown().optional().nullable(),
-    tool_name: z.string().optional().nullable(),
+    chartType: z.literal('pie'),
+    title: z.string().nullish(),
+    data: z.array(z.tuple([z.string(), z.number()])),
+    isCurrency: z.boolean().nullish(),
 });
+// --- Timeseries Chart Block ---
+// Server sends: { blockType: "chart", chartType: "timeseries", series: { ... } }
+export const TimeseriesDataPointSchema = z.object({
+    timestamp: z.number(),
+    value: z.number(),
+});
+export const TimeseriesChartBlockSchema = z.object({
+    blockType: z.literal('chart'),
+    chartType: z.literal('timeseries'),
+    title: z.string().nullish(),
+    series: z.record(z.string(), z.array(TimeseriesDataPointSchema)),
+    captions: z.array(z.record(z.string(), z.unknown())).nullish(),
+    titleCryptoIcons: z.array(z.record(z.string(), z.unknown())).nullish(),
+    isCurrency: z.boolean().nullish(),
+});
+export const ChartBlockSchema = z.discriminatedUnion('chartType', [
+    PieChartBlockSchema,
+    TimeseriesChartBlockSchema,
+]);
 // --- Transaction Action Block ---
 // Server sends: { blockType: "transaction_action", primitives: [...], transactions: [...], risks: {...}, ... }
 // Primitive display icon
@@ -113,39 +137,40 @@ export const PrimitiveIconSchema = z.object({
 export const PrimitiveLineItemSchema = z.object({
     label: z.string(),
     value: z.string(),
-    icon: PrimitiveIconSchema.optional(),
+    icon: PrimitiveIconSchema.nullish(),
 });
 // Primitive display info
 export const PrimitiveDisplaySchema = z.object({
-    headline: z.string().optional(),
-    action_verb: z.string().optional(),
-    primary_icon: PrimitiveIconSchema.optional(),
-    secondary_icon: PrimitiveIconSchema.optional(),
-    line_items: z.array(PrimitiveLineItemSchema).optional(),
+    headline: z.string().nullish(),
+    action_verb: z.string().nullish(),
+    primary_icon: PrimitiveIconSchema.nullish(),
+    secondary_icon: PrimitiveIconSchema.nullish(),
+    line_items: z.array(PrimitiveLineItemSchema).nullish(),
 });
 // Primitive (the actual action like swap, supply, etc.)
 export const PrimitiveSchema = z.object({
     primitive: z.string(),
-    params: z.record(z.string(), z.unknown()).optional(),
-    display: PrimitiveDisplaySchema.optional(),
+    params: z.record(z.string(), z.unknown()).nullish(),
+    display: PrimitiveDisplaySchema.nullish(),
 });
 // Raw transaction data from server
 export const RawTransactionSchema = z.object({
-    to: z.string().optional(),
-    data: z.string().optional(),
-    value: z.string().optional(),
-    chainId: z.number().optional(),
-    description: z.string().optional(),
-    contractName: z.string().optional(),
-    contractVerified: z.boolean().optional(),
-    protocolName: z.string().optional(),
-    contractIcon: z.string().optional(),
+    chainId: z.number(),
+    contractIcon: z.string().nullish(),
+    contractName: z.string().nullish(),
+    contractVerified: z.boolean().nullish(),
+    data: z.string(),
+    description: z.string(),
+    gasEstimate: z.number().optional(),
+    protocolName: z.string().nullish(),
+    to: z.string(),
+    value: z.string(),
 });
 // Transaction group (contains multiple raw transactions)
 export const TransactionGroupSchema = z.object({
     transactions: z.array(RawTransactionSchema).optional(),
-    requiresApproval: z.boolean().optional(),
-    verificationUnavailable: z.boolean().optional(),
+    requiresApproval: z.boolean().nullish(),
+    verificationUnavailable: z.boolean().nullish(),
 });
 // Risk impact (quantified risk details)
 export const RiskImpactSchema = z.object({
@@ -175,19 +200,17 @@ export const RisksSchema = z.object({
 });
 export const TransactionActionBlockSchema = z.object({
     blockType: z.literal('transaction_action'),
-    value: z.record(z.string(), z.unknown()).optional(),
-    sequence: z.boolean().optional(),
-    primitives: z.array(PrimitiveSchema).optional(),
+    value: z.record(z.string(), z.unknown()).nullish(),
+    sequence: z.boolean(),
+    primitives: z.array(PrimitiveSchema).min(1),
     transactions: z.array(TransactionGroupSchema).optional(),
-    needs_confirmation: z.boolean().optional(),
-    notes: z.string().optional(),
-    risks: RisksSchema.optional(),
-    metadata: z.record(z.string(), z.unknown()).optional(),
-    tool_params: z.unknown().optional().nullable(),
-    tool_name: z.string().optional().nullable(),
+    needs_confirmation: z.boolean(),
+    notes: z.string(),
+    risks: RisksSchema,
+    metadata: z.record(z.string(), z.unknown()),
 });
 // --- Interactive Block ---
-// Server sends: { blockType: "interactive", title, body?, context?, style?, options? }
+// Server sends: { blockType: "interactive", title, body?, context?, style, options? }
 // Note: Using .nullish() to accept both null and undefined (Python sends null for None)
 export const InteractiveOptionSchema = z.object({
     id: z.string(),
@@ -200,9 +223,23 @@ export const InteractiveBlockSchema = z.object({
     title: z.string(), // Required, matches Python
     body: z.string().nullish(), // Optional, matches Python
     context: z.string().nullish(), // Optional, matches Python
-    style: z.enum(['options', 'confirm_cancel']).nullish(),
+    style: z.enum(['options', 'confirm_cancel']), // Required, matches Python
     options: z.array(InteractiveOptionSchema).nullish(),
 });
+// --- Info Block ---
+// Server sends: { blockType: "info", content, error? }
+export const InfoBlockSchema = z.object({
+    blockType: z.literal('info'),
+    content: z.string(),
+    error: z.string().nullish(),
+});
+// --- Code Block ---
+// Server sends: { blockType: "code", content, language }
+export const CodeBlockSchema = z.object({
+    blockType: z.literal('code'),
+    content: z.string(),
+    language: z.string(),
+});
 // ============================================================================
 // Block Detection and Parsing
 // ============================================================================
@@ -217,9 +254,13 @@ export function detectBlockType(raw) {
     if (obj.blockType === 'chart')
         return 'chart';
     if (obj.blockType === 'transaction_action')
-        return 'transaction_action';
+        return 'action';
     if (obj.blockType === 'interactive')
         return 'interactive';
+    if (obj.blockType === 'info')
+        return 'info';
+    if (obj.blockType === 'code')
+        return 'code';
     // Markdown blocks have no blockType, just content
     if ('content' in obj && typeof obj.content === 'string' && !('blockType' in obj)) {
         return 'markdown';
@@ -251,20 +292,38 @@ export function parseRawBlock(raw) {
             }
             return { success: false, error: `Chart parse error: ${result.error.message}` };
         }
-        case 'transaction_action': {
+        case 'action': {
             const result = TransactionActionBlockSchema.safeParse(raw);
             if (result.success) {
-                return { success: true, data: { type: 'transaction_action', ...result.data }, type: 'transaction_action' };
+                return { success: true, data: { type: 'action', ...result.data }, type: 'action' };
             }
             return { success: false, error: `TransactionAction parse error: ${result.error.message}` };
         }
         case 'interactive': {
             const result = InteractiveBlockSchema.safeParse(raw);
             if (result.success) {
-                return { success: true, data: { type: 'interactive', ...result.data }, type: 'interactive' };
+                return {
+                    success: true,
+                    data: { type: 'interactive', ...result.data },
+                    type: 'interactive',
+                };
             }
             return { success: false, error: `Interactive parse error: ${result.error.message}` };
         }
+        case 'info': {
+            const result = InfoBlockSchema.safeParse(raw);
+            if (result.success) {
+                return { success: true, data: { type: 'info', ...result.data }, type: 'info' };
+            }
+            return { success: false, error: `Info parse error: ${result.error.message}` };
+        }
+        case 'code': {
+            const result = CodeBlockSchema.safeParse(raw);
+            if (result.success) {
+                return { success: true, data: { type: 'code', ...result.data }, type: 'code' };
+            }
+            return { success: false, error: `Code parse error: ${result.error.message}` };
+        }
         default:
             return { success: false, error: `Unknown block type` };
     }
@@ -275,9 +334,441 @@ export function parseRawBlock(raw) {
 // Create a discriminated union schema that adds 'type' field
 // This is for SDK consumers who expect 'type' field
 export const BlockSchema = z.union([
-    MarkdownBlockSchema.transform(data => ({ type: 'markdown', ...data })),
-    TableBlockSchema.transform(data => ({ type: 'table', ...data })),
-    ChartBlockSchema.transform(data => ({ type: 'chart', ...data })),
-    TransactionActionBlockSchema.transform(data => ({ type: 'transaction_action', ...data })),
-    InteractiveBlockSchema.transform(data => ({ type: 'interactive', ...data })),
+    MarkdownBlockSchema.transform((data) => ({ type: 'markdown', ...data })),
+    TableBlockSchema.transform((data) => ({ type: 'table', ...data })),
+    ChartBlockSchema.transform((data) => ({ type: 'chart', ...data })),
+    PieChartBlockSchema.transform((data) => ({ type: 'chart', ...data })),
+    TimeseriesChartBlockSchema.transform((data) => ({ type: 'chart', ...data })),
+    TransactionActionBlockSchema.transform((data) => ({ type: 'action', ...data })),
+    InteractiveBlockSchema.transform((data) => ({ type: 'interactive', ...data })),
+    InfoBlockSchema.transform((data) => ({ type: 'info', ...data })),
+    CodeBlockSchema.transform((data) => ({ type: 'code', ...data })),
+]);
+// ============================================================================
+// Response Content Schemas (depend on BlockSchema)
+// ============================================================================
+export const ChaosBlockSchema = z.object({
+    type: z.literal('chaos.block'),
+    block: BlockSchema,
+});
+export const ContentPartSchema = z.union([OutputTextSchema, ChaosBlockSchema]);
+export const OutputItemSchema = z.object({
+    type: z.literal('message'),
+    role: z.literal('assistant'),
+    content: z.array(ContentPartSchema),
+});
+export const ResponseSchema = z.object({
+    id: z.string(),
+    object: z.literal('response'),
+    model: z.string(),
+    status: z.enum(['completed', 'failed']),
+    output: z.array(OutputItemSchema),
+    error: ResponseErrorSchema.nullish(),
+});
+// ============================================================================
+// V1 API Request Schemas (from request.ts)
+// ============================================================================
+export const PreviousMessageSchema = z.object({
+    type: z.literal('message'),
+    role: z.enum(['user', 'assistant']),
+    content: z.string(),
+});
+export const V1WalletRequestSchema = z.object({
+    model: z.string(),
+    query: z.string(),
+    user_id: z.string(),
+    session_id: z.string(),
+    wallets: z.array(WalletInfoSchema).nullish(),
+    previous_messages: z.array(PreviousMessageSchema).nullish(),
+});
+export const V1AskRequestSchema = z.object({
+    model: z.string(),
+    query: z.string(),
+    user_id: z.string(),
+    session_id: z.string(),
+    previous_messages: z.array(PreviousMessageSchema).nullish(),
+});
+// ============================================================================
+// V1 API Response Schemas (from response.ts)
+// ============================================================================
+export const V1StreamEventSchema = z
+    .object({
+    type: z.string().nullish(),
+    content: z.record(z.string(), z.unknown()).nullish(),
+})
+    .passthrough();
+export const V1FinalStateSchema = z.object({
+    blocks: z.array(z.unknown()),
+});
+// ============================================================================
+// Stream Message Schemas (from stream.ts)
+// ============================================================================
+export const MessageTypeSchema = z.enum([
+    'agent_status_change',
+    'agent_message',
+    'report',
+    'follow_up_suggestions',
+    'user_input',
+]);
+export const AgentStatusSchema = z.enum(['processing', 'done', 'error', 'cancelled']);
+export const StreamMessageContextSchema = z.object({
+    sessionId: z.string(),
+    artifactId: z.string(),
+    query: z.string().nullish(),
+});
+export const AgentStatusContentSchema = z.object({
+    status: AgentStatusSchema,
+});
+export const AgentTextDataSchema = z.object({
+    message: z.string(),
+});
+export const AgentMessageContentSchema = z.object({
+    messageId: z.string(),
+    artifactId: z.string(),
+    type: z.literal('agent_message'),
+    data: AgentTextDataSchema,
+    order: z.number().int(),
+    createdAt: z.number(),
+    updatedAt: z.number(),
+    metadata: z.record(z.string(), z.unknown()).nullish(),
+});
+export const AgentMessageSchema = z.object({
+    id: z.string(),
+    type: z.literal('agent_message'),
+    timestamp: z.number(),
+    content: AgentMessageContentSchema,
+    context: StreamMessageContextSchema,
+});
+export const AgentStatusChangeMessageScheam = z.object({
+    id: z.string(),
+    type: z.literal('agent_status_change'),
+    timestamp: z.number().int(),
+    content: {
+        status: z.string(),
+    },
+    context: z.object({
+        sessionId: z.string(),
+        artifactId: z.string(),
+        query: z.string(),
+    }),
+});
+export const CaptionSchema = z.object({
+    label: z.string(),
+    value: z.unknown(),
+    group: z.string().nullish(),
+    collapsable: z.boolean().nullish(),
+});
+export const ReferenceSchema = z.object({
+    title: z.string(),
+    url: z.string(),
+    content: z.string(),
+    hiddenUrl: z.string().nullish(),
+    textSpans: z.array(z.array(z.number())).nullish(),
+    engineId: z.string().nullish(),
+});
+export const ReportContentSchema = z.object({
+    id: z.string(),
+    artifactId: z.string(),
+    type: z.string(),
+    data: BlockSchema,
+    order: z.number(),
+    createdAt: z.number(),
+    updatedAt: z.number(),
+    createdBy: z.string().nullish(),
+    metadata: z.record(z.string(), z.unknown()).nullish(),
+    captions: z.array(CaptionSchema).nullish(),
+    references: z.array(ReferenceSchema).nullish(),
+    version: z.number().nullish(),
+});
+export const FollowUpDataSchema = z.object({
+    followUpQueries: z.array(z.string()),
+});
+export const FollowUpSuggestionsContentSchema = z.object({
+    messageId: z.string(),
+    artifactId: z.string(),
+    type: z.string(),
+    data: FollowUpDataSchema,
+    order: z.number(),
+    createdAt: z.number(),
+    updatedAt: z.number(),
+    metadata: z.record(z.string(), z.unknown()).nullish(),
+});
+export const UserInputCandidateSchema = z.object({
+    symbol: z.string(),
+    name: z.string(),
+    market_cap: z.number(),
+    token_id: z.string(),
+    score: z.number(),
+});
+export const UserInputContentSchema = z.object({
+    type: z.string(),
+    candidates: z.record(z.string(), z.array(UserInputCandidateSchema)),
+});
+// Stream message base fields
+const StreamMessageBaseSchema = z.object({
+    id: z.string(),
+    timestamp: z.number(),
+    context: StreamMessageContextSchema,
+});
+export const AgentStatusMessageSchema = StreamMessageBaseSchema.extend({
+    type: z.literal('agent_status_change'),
+    content: AgentStatusContentSchema,
+});
+export const AgentTextMessageSchema = StreamMessageBaseSchema.extend({
+    type: z.literal('agent_message'),
+    content: AgentMessageContentSchema,
+});
+export const ReportMessageSchema = StreamMessageBaseSchema.extend({
+    type: z.literal('report'),
+    content: ReportContentSchema,
+});
+export const FollowUpSuggestionsMessageSchema = StreamMessageBaseSchema.extend({
+    type: z.literal('follow_up_suggestions'),
+    content: FollowUpSuggestionsContentSchema,
+});
+export const UserInputMessageSchema = StreamMessageBaseSchema.extend({
+    type: z.literal('user_input'),
+    content: UserInputContentSchema,
+});
+export const StreamMessageSchema = z.discriminatedUnion('type', [
+    AgentStatusMessageSchema,
+    AgentTextMessageSchema,
+    ReportMessageSchema,
+    FollowUpSuggestionsMessageSchema,
+    UserInputMessageSchema,
 ]);
+// ============================================================================
+// Conversation Schemas (from conversation.ts)
+// ============================================================================
+export const ConversationOptionsSchema = z.object({
+    model: z.string().nullish(),
+    maxHistoryLength: z.number().nullish(),
+    userId: z.string(),
+    wallets: z.array(z.object({ address: z.string(), chain: z.string() })).nullish(),
+    sessionId: z.string().nullish(),
+});
+export const ConversationStatsSchema = z.object({
+    userTurns: z.number(),
+    assistantTurns: z.number(),
+    totalMessages: z.number(),
+    sessionId: z.string(),
+    startedAt: z.date(),
+    lastMessageAt: z.date().nullable(),
+});
+// ============================================================================
+// Primitive Param Schemas (from primitives.ts)
+// ============================================================================
+export const PrimitiveTypeSchema = z.enum([
+    'swap',
+    'supply',
+    'withdraw',
+    'borrow',
+    'repay',
+    'stake',
+    'unstake',
+    'claim',
+    'bridge',
+    'add_liquidity',
+    'remove_liquidity',
+    'open_position',
+    'close_position',
+    'deposit',
+    'mint',
+    'burn',
+    'restake',
+    'approve',
+    'transfer',
+    'wrap',
+    'unwrap',
+]);
+export const SwapParamsSchema = z.object({
+    token_in: z.string(),
+    token_out: z.string(),
+    amount_in: z.string().nullish(),
+    amount_out: z.string().nullish(),
+    slippage: z.number().nullish(),
+    protocol: z.string().nullish(),
+});
+export const BridgeParamsSchema = z.object({
+    asset: z.string(),
+    amount: z.string(),
+    from_chain: z.string(),
+    to_chain: z.string(),
+    recipient: z.string().nullish(),
+    protocol: z.string().nullish(),
+});
+export const LendingParamsSchema = z.object({
+    asset: z.string(),
+    amount: z.string(),
+    protocol: z.string().nullish(),
+    interest_rate_mode: z.number().nullish(),
+});
+export const StakingParamsSchema = z.object({
+    amount: z.string(),
+    protocol: z.string().nullish(),
+    operator: z.string().nullish(),
+});
+export const ClaimParamsSchema = z.object({
+    protocol: z.string().nullish(),
+    assets: z.array(z.string()).nullish(),
+});
+export const LiquidityParamsSchema = z.object({
+    token_a: z.string(),
+    token_b: z.string(),
+    amount_a: z.string().nullish(),
+    amount_b: z.string().nullish(),
+    percentage: z.number().nullish(),
+    fee_tier: z.number().nullish(),
+    protocol: z.string().nullish(),
+});
+export const PositionParamsSchema = z.object({
+    market: z.string(),
+    side: z.enum(['long', 'short']).nullish(),
+    size: z.string().nullish(),
+    leverage: z.number().nullish(),
+    percentage: z.number().nullish(),
+    protocol: z.string().nullish(),
+});
+export const DepositParamsSchema = z.object({
+    vault: z.string(),
+    amount: z.string(),
+    protocol: z.string().nullish(),
+});
+export const MintParamsSchema = z.object({
+    asset: z.string(),
+    amount: z.string(),
+    collateral: z.string().nullish(),
+    protocol: z.string().nullish(),
+});
+export const ApproveParamsSchema = z.object({
+    token: z.string(),
+    spender: z.string(),
+    amount: z.string().nullish(),
+});
+export const TransferParamsSchema = z.object({
+    token: z.string(),
+    to: z.string(),
+    amount: z.string(),
+});
+export const WrapParamsSchema = z.object({
+    amount: z.string(),
+});
+// Typed primitive schemas
+const TypedPrimitiveBaseSchema = z.object({
+    display: PrimitiveDisplaySchema.nullish(),
+});
+export const SwapPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('swap'),
+    params: SwapParamsSchema,
+});
+export const BridgePrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('bridge'),
+    params: BridgeParamsSchema,
+});
+export const SupplyPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('supply'),
+    params: LendingParamsSchema,
+});
+export const WithdrawPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('withdraw'),
+    params: LendingParamsSchema,
+});
+export const BorrowPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('borrow'),
+    params: LendingParamsSchema,
+});
+export const RepayPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('repay'),
+    params: LendingParamsSchema,
+});
+export const StakePrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('stake'),
+    params: StakingParamsSchema,
+});
+export const UnstakePrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('unstake'),
+    params: StakingParamsSchema,
+});
+export const ClaimPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('claim'),
+    params: ClaimParamsSchema,
+});
+export const AddLiquidityPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('add_liquidity'),
+    params: LiquidityParamsSchema,
+});
+export const RemoveLiquidityPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('remove_liquidity'),
+    params: LiquidityParamsSchema,
+});
+export const OpenPositionPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('open_position'),
+    params: PositionParamsSchema,
+});
+export const ClosePositionPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('close_position'),
+    params: PositionParamsSchema,
+});
+export const DepositPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('deposit'),
+    params: DepositParamsSchema,
+});
+export const MintPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('mint'),
+    params: MintParamsSchema,
+});
+export const BurnPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('burn'),
+    params: MintParamsSchema,
+});
+export const RestakePrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('restake'),
+    params: StakingParamsSchema,
+});
+export const ApprovePrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('approve'),
+    params: ApproveParamsSchema,
+});
+export const TransferPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('transfer'),
+    params: TransferParamsSchema,
+});
+export const WrapPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('wrap'),
+    params: WrapParamsSchema,
+});
+export const UnwrapPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.literal('unwrap'),
+    params: WrapParamsSchema,
+});
+export const UnknownPrimitiveSchema = TypedPrimitiveBaseSchema.extend({
+    primitive: z.string(),
+    params: z.record(z.string(), z.unknown()),
+});
+export const TypedPrimitiveSchema = z.union([
+    SwapPrimitiveSchema,
+    BridgePrimitiveSchema,
+    SupplyPrimitiveSchema,
+    WithdrawPrimitiveSchema,
+    BorrowPrimitiveSchema,
+    RepayPrimitiveSchema,
+    StakePrimitiveSchema,
+    UnstakePrimitiveSchema,
+    ClaimPrimitiveSchema,
+    AddLiquidityPrimitiveSchema,
+    RemoveLiquidityPrimitiveSchema,
+    OpenPositionPrimitiveSchema,
+    ClosePositionPrimitiveSchema,
+    DepositPrimitiveSchema,
+    MintPrimitiveSchema,
+    BurnPrimitiveSchema,
+    RestakePrimitiveSchema,
+    ApprovePrimitiveSchema,
+    TransferPrimitiveSchema,
+    WrapPrimitiveSchema,
+    UnwrapPrimitiveSchema,
+    UnknownPrimitiveSchema,
+]);
+// ============================================================================
+// Chart Subtype Schemas (from type-guards.ts)
+// ============================================================================