@kmmao/happy-wire 0.2.2 → 0.2.4

package/dist/index.cjs

@@ -67,6 +67,15 @@ const sessionTurnEndStatusSchema = z__namespace.enum([
   "failed",
   "cancelled"
 ]);
+const sessionModelUsageSchema = z__namespace.object({
+  inputTokens: z__namespace.number(),
+  outputTokens: z__namespace.number(),
+  cacheReadInputTokens: z__namespace.number(),
+  cacheCreationInputTokens: z__namespace.number(),
+  costUSD: z__namespace.number(),
+  contextWindow: z__namespace.number(),
+  maxOutputTokens: z__namespace.number()
+});
 const sessionTurnEndEventSchema = z__namespace.object({
   t: z__namespace.literal("turn-end"),
   status: sessionTurnEndStatusSchema,
@@ -77,7 +86,10 @@ const sessionTurnEndEventSchema = z__namespace.object({
     cache_creation_input_tokens: z__namespace.number().optional(),
     cache_read_input_tokens: z__namespace.number().optional()
   }).optional(),
-  durationMs: z__namespace.number().optional()
+  durationMs: z__namespace.number().optional(),
+  totalCostUsd: z__namespace.number().optional(),
+  numTurns: z__namespace.number().optional(),
+  modelUsage: z__namespace.record(z__namespace.string(), sessionModelUsageSchema).optional()
 });
 const sessionStopEventSchema = z__namespace.object({
   t: z__namespace.literal("stop")
@@ -129,6 +141,10 @@ const sessionToolProgressEventSchema = z__namespace.object({
   elapsedSeconds: z__namespace.number(),
   taskId: z__namespace.string().optional()
 });
+const sessionPromptSuggestionEventSchema = z__namespace.object({
+  t: z__namespace.literal("prompt-suggestion"),
+  suggestion: z__namespace.string()
+});
 const sessionEventSchema = z__namespace.discriminatedUnion("t", [
   sessionTextEventSchema,
   sessionServiceMessageEventSchema,
@@ -143,7 +159,8 @@ const sessionEventSchema = z__namespace.discriminatedUnion("t", [
   sessionTaskStartEventSchema,
   sessionTaskProgressEventSchema,
   sessionTaskEndEventSchema,
-  sessionToolProgressEventSchema
+  sessionToolProgressEventSchema,
+  sessionPromptSuggestionEventSchema
 ]);
 const sessionEnvelopeSchema = z__namespace.object({
   id: z__namespace.string(),
@@ -162,7 +179,7 @@ const sessionEnvelopeSchema = z__namespace.object({
       path: ["role"]
     });
   }
-  if ((envelope.ev.t === "start" || envelope.ev.t === "stop" || envelope.ev.t === "usage-update" || envelope.ev.t === "task-start" || envelope.ev.t === "task-progress" || envelope.ev.t === "task-end" || envelope.ev.t === "tool-progress") && envelope.role !== "agent") {
+  if ((envelope.ev.t === "start" || envelope.ev.t === "stop" || envelope.ev.t === "usage-update" || envelope.ev.t === "task-start" || envelope.ev.t === "task-progress" || envelope.ev.t === "task-end" || envelope.ev.t === "tool-progress" || envelope.ev.t === "prompt-suggestion") && envelope.role !== "agent") {
     ctx.addIssue({
       code: z__namespace.ZodIssueCode.custom,
       message: `${envelope.ev.t} events must use role "agent"`,
@@ -308,6 +325,8 @@ exports.createEnvelope = createEnvelope;
 exports.sessionEnvelopeSchema = sessionEnvelopeSchema;
 exports.sessionEventSchema = sessionEventSchema;
 exports.sessionFileEventSchema = sessionFileEventSchema;
+exports.sessionModelUsageSchema = sessionModelUsageSchema;
+exports.sessionPromptSuggestionEventSchema = sessionPromptSuggestionEventSchema;
 exports.sessionRoleSchema = sessionRoleSchema;
 exports.sessionServiceMessageEventSchema = sessionServiceMessageEventSchema;
 exports.sessionStartEventSchema = sessionStartEventSchema;