@kmmao/happy-wire 0.2.3 → 0.2.5

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")
@@ -133,6 +145,9 @@ const sessionPromptSuggestionEventSchema = z__namespace.object({
   t: z__namespace.literal("prompt-suggestion"),
   suggestion: z__namespace.string()
 });
+const sessionNeedsContinueEventSchema = z__namespace.object({
+  t: z__namespace.literal("needs-continue")
+});
 const sessionEventSchema = z__namespace.discriminatedUnion("t", [
   sessionTextEventSchema,
   sessionServiceMessageEventSchema,
@@ -148,7 +163,8 @@ const sessionEventSchema = z__namespace.discriminatedUnion("t", [
   sessionTaskProgressEventSchema,
   sessionTaskEndEventSchema,
   sessionToolProgressEventSchema,
-  sessionPromptSuggestionEventSchema
+  sessionPromptSuggestionEventSchema,
+  sessionNeedsContinueEventSchema
 ]);
 const sessionEnvelopeSchema = z__namespace.object({
   id: z__namespace.string(),
@@ -167,7 +183,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.ev.t === "prompt-suggestion") && 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.ev.t === "needs-continue") && envelope.role !== "agent") {
     ctx.addIssue({
       code: z__namespace.ZodIssueCode.custom,
       message: `${envelope.ev.t} events must use role "agent"`,
@@ -313,6 +329,8 @@ exports.createEnvelope = createEnvelope;
 exports.sessionEnvelopeSchema = sessionEnvelopeSchema;
 exports.sessionEventSchema = sessionEventSchema;
 exports.sessionFileEventSchema = sessionFileEventSchema;
+exports.sessionModelUsageSchema = sessionModelUsageSchema;
+exports.sessionNeedsContinueEventSchema = sessionNeedsContinueEventSchema;
 exports.sessionPromptSuggestionEventSchema = sessionPromptSuggestionEventSchema;
 exports.sessionRoleSchema = sessionRoleSchema;
 exports.sessionServiceMessageEventSchema = sessionServiceMessageEventSchema;