@townco/ui 0.1.14 → 0.1.16

package/dist/sdk/schemas/session.js

@@ -1,54 +1,99 @@
 import { z } from "zod";
+import { ToolCallSchema, ToolCallUpdateSchema, } from "../../core/schemas/tool-call.js";
 import { Message } from "./message.js";
 /**
  * Session status
  */
 export const SessionStatus = z.enum([
-	"idle",
-	"connecting",
-	"connected",
-	"active",
-	"streaming",
-	"error",
-	"disconnected",
+    "idle",
+    "connecting",
+    "connected",
+    "active",
+    "streaming",
+    "error",
+    "disconnected",
 ]);
 /**
  * Session configuration
  */
 export const SessionConfig = z.object({
-	agentPath: z.string(),
-	agentArgs: z.array(z.string()).optional(),
-	environment: z.record(z.string(), z.string()).optional(),
-	workingDirectory: z.string().optional(),
-	timeout: z.number().optional(),
+    agentPath: z.string(),
+    agentArgs: z.array(z.string()).optional(),
+    environment: z.record(z.string(), z.string()).optional(),
+    workingDirectory: z.string().optional(),
+    timeout: z.number().optional(),
 });
 /**
  * Session metadata
  */
 export const SessionMetadata = z.object({
-	agentName: z.string().optional(),
-	agentVersion: z.string().optional(),
-	capabilities: z.array(z.string()).optional(),
-	startedAt: z.iso.datetime(),
-	lastActivityAt: z.iso.datetime().optional(),
+    agentName: z.string().optional(),
+    agentVersion: z.string().optional(),
+    capabilities: z.array(z.string()).optional(),
+    startedAt: z.iso.datetime(),
+    lastActivityAt: z.iso.datetime().optional(),
 });
 /**
  * Session schema
  */
 export const Session = z.object({
-	id: z.string(),
-	status: SessionStatus,
-	config: SessionConfig,
-	metadata: SessionMetadata.optional(),
-	messages: z.array(Message),
-	error: z.string().optional(),
+    id: z.string(),
+    status: SessionStatus,
+    config: SessionConfig,
+    metadata: SessionMetadata.optional(),
+    messages: z.array(Message),
+    error: z.string().optional(),
 });
 /**
- * Session update event
+ * Session update event (base type)
  */
-export const SessionUpdate = z.object({
-	sessionId: z.string(),
-	status: SessionStatus.optional(),
-	message: Message.optional(),
-	error: z.string().optional(),
+const BaseSessionUpdate = z.object({
+    sessionId: z.string(),
+    status: SessionStatus.optional(),
+    message: Message.optional(),
+    error: z.string().optional(),
 });
+/**
+ * Session update with tool call (sessionUpdate: "tool_call")
+ */
+const ToolCallSessionUpdate = BaseSessionUpdate.extend({
+    type: z.literal("tool_call"),
+    toolCall: ToolCallSchema,
+    messageId: z.string().optional(),
+});
+/**
+ * Session update with tool call update (sessionUpdate: "tool_call_update")
+ */
+const ToolCallUpdateSessionUpdate = BaseSessionUpdate.extend({
+    type: z.literal("tool_call_update"),
+    toolCallUpdate: ToolCallUpdateSchema,
+    messageId: z.string().optional(),
+});
+/**
+ * Session update with tool output (sessionUpdate: "tool_output")
+ * Sent separately from tool_call_update to avoid PostgreSQL NOTIFY size limits
+ */
+const ToolOutputSessionUpdate = BaseSessionUpdate.extend({
+    type: z.literal("tool_output"),
+    toolOutput: z.object({
+        id: z.string(),
+        content: z.array(z.any()).optional(), // ToolCallContentBlock[] but use any to avoid circular import
+        rawOutput: z.record(z.string(), z.unknown()).optional(),
+    }),
+    messageId: z.string().optional(),
+});
+/**
+ * Generic session update
+ */
+const GenericSessionUpdate = BaseSessionUpdate.extend({
+    type: z.literal("generic").optional(),
+});
+/**
+ * Session update event (union type)
+ */
+export const SessionUpdate = z.union([
+    ToolCallSessionUpdate,
+    ToolCallUpdateSessionUpdate,
+    ToolOutputSessionUpdate,
+    GenericSessionUpdate,
+]);