@factory/droid-sdk 0.2.0 → 0.4.0

package/dist/index.js

@@ -3,6 +3,7 @@ var DroidServerMethod = /* @__PURE__ */ ((DroidServerMethod2) => {
   DroidServerMethod2["INITIALIZE_SESSION"] = "droid.initialize_session";
   DroidServerMethod2["LOAD_SESSION"] = "droid.load_session";
   DroidServerMethod2["ADD_USER_MESSAGE"] = "droid.add_user_message";
+  DroidServerMethod2["CLOSE_SESSION"] = "droid.close_session";
   DroidServerMethod2["INTERRUPT_SESSION"] = "droid.interrupt_session";
   DroidServerMethod2["KILL_WORKER_SESSION"] = "droid.kill_worker_session";
   DroidServerMethod2["UPDATE_SESSION_SETTINGS"] = "droid.update_session_settings";
@@ -45,7 +46,10 @@ var SessionNotificationType = /* @__PURE__ */ ((SessionNotificationType2) => {
   SessionNotificationType2["SESSION_TITLE_UPDATED"] = "session_title_updated";
   SessionNotificationType2["MCP_STATUS_CHANGED"] = "mcp_status_changed";
   SessionNotificationType2["ASSISTANT_TEXT_DELTA"] = "assistant_text_delta";
+  SessionNotificationType2["ASSISTANT_TEXT_COMPLETE"] = "assistant_text_complete";
   SessionNotificationType2["THINKING_TEXT_DELTA"] = "thinking_text_delta";
+  SessionNotificationType2["THINKING_TEXT_COMPLETE"] = "thinking_text_complete";
+  SessionNotificationType2["TOOL_CALL"] = "tool_call";
   SessionNotificationType2["SESSION_TOKEN_USAGE_CHANGED"] = "session_token_usage_changed";
   SessionNotificationType2["MISSION_STATE_CHANGED"] = "mission_state_changed";
   SessionNotificationType2["MISSION_FEATURES_CHANGED"] = "mission_features_changed";
@@ -55,6 +59,9 @@ var SessionNotificationType = /* @__PURE__ */ ((SessionNotificationType2) => {
   SessionNotificationType2["MISSION_WORKER_COMPLETED"] = "mission_worker_completed";
   SessionNotificationType2["MCP_AUTH_REQUIRED"] = "mcp_auth_required";
   SessionNotificationType2["MCP_AUTH_COMPLETED"] = "mcp_auth_completed";
+  SessionNotificationType2["HOOK_EXECUTION_STARTED"] = "hook_execution_started";
+  SessionNotificationType2["HOOK_EXECUTION_COMPLETED"] = "hook_execution_completed";
+  SessionNotificationType2["STRUCTURED_OUTPUT"] = "structured_output";
   return SessionNotificationType2;
 })(SessionNotificationType || {});
 var ToolConfirmationOutcome = /* @__PURE__ */ ((ToolConfirmationOutcome3) => {
@@ -248,6 +255,11 @@ var JsonRpcMessageType = /* @__PURE__ */ ((JsonRpcMessageType2) => {
   JsonRpcMessageType2["Notification"] = "notification";
   return JsonRpcMessageType2;
 })(JsonRpcMessageType || {});
+var ServerRequestHandlerType = /* @__PURE__ */ ((ServerRequestHandlerType2) => {
+  ServerRequestHandlerType2["Permission"] = "permission";
+  ServerRequestHandlerType2["AskUser"] = "askUser";
+  return ServerRequestHandlerType2;
+})(ServerRequestHandlerType || {});
 var SettingsLevel = /* @__PURE__ */ ((SettingsLevel2) => {
   SettingsLevel2["Org"] = "org";
   SettingsLevel2["Runtime"] = "runtime";
@@ -268,7 +280,7 @@ var SkillLocation = /* @__PURE__ */ ((SkillLocation2) => {
 // src/schemas/constants.ts
 var JSONRPC_VERSION = "2.0";
 var LEGACY_FACTORY_API_VERSION = "1.0.0";
-var FACTORY_PROTOCOL_VERSION = "1.3.0";
+var FACTORY_PROTOCOL_VERSION = "1.51.0";
 var FACTORY_CLIENT_HEADER = "X-Factory-Client";
 var FACTORY_CLIENT_VERSION = "X-Client-Version";
 var DEFAULT_REQUEST_TIMEOUT = 3e4;
@@ -359,6 +371,12 @@ var JsonRpcMessageSchema = z.discriminatedUnion("type", [
   }),
   JsonRpcNotificationSchema
 ]);
+function createResponseSchema(resultSchema) {
+  return z.union([
+    JsonRpcResponseSuccessSchema.extend({ result: resultSchema }),
+    JsonRpcResponseFailureSchema
+  ]);
+}
 
 // src/schemas/session-metadata.ts
 import { z as z6 } from "zod";
@@ -806,6 +824,9 @@ var AddUserMessageRequestParamsSchema = z5.object({
   outputFormat: OutputFormatSchema.optional()
 }).strict();
 var InterruptSessionRequestParamsSchema = z5.object({}).strict();
+var CloseSessionRequestParamsSchema = z5.object({
+  reason: z5.enum(["clear", "logout", "prompt_input_exit", "other"]).optional()
+}).strict();
 var KillWorkerSessionRequestParamsSchema = z5.object({
   workerSessionId: z5.string()
 }).strict();
@@ -922,6 +943,10 @@ var InterruptSessionRequestSchema = JsonRpcRequestSchema.extend({
   method: z5.literal("droid.interrupt_session" /* INTERRUPT_SESSION */),
   params: InterruptSessionRequestParamsSchema
 });
+var CloseSessionRequestSchema = JsonRpcRequestSchema.extend({
+  method: z5.literal("droid.close_session" /* CLOSE_SESSION */),
+  params: CloseSessionRequestParamsSchema
+});
 var KillWorkerSessionRequestSchema = JsonRpcRequestSchema.extend({
   method: z5.literal("droid.kill_worker_session" /* KILL_WORKER_SESSION */),
   params: KillWorkerSessionRequestParamsSchema
@@ -1014,6 +1039,7 @@ var ClientRequestSchema = z5.discriminatedUnion("method", [
   InitializeSessionRequestSchema,
   LoadSessionRequestSchema,
   AddUserMessageRequestSchema,
+  CloseSessionRequestSchema,
   InterruptSessionRequestSchema,
   KillWorkerSessionRequestSchema,
   UpdateSessionSettingsRequestSchema,
@@ -1074,6 +1100,7 @@ var LoadSessionResultSchema = z5.object({
 }).passthrough();
 var AddUserMessageResultSchema = EmptyResultSchema;
 var InterruptSessionResultSchema = EmptyResultSchema;
+var CloseSessionResultSchema = EmptyResultSchema;
 var KillWorkerSessionResultSchema = EmptyResultSchema;
 var UpdateSessionSettingsResultSchema = EmptyResultSchema;
 var ToggleMcpServerResultSchema = SuccessResultSchema;
@@ -1113,128 +1140,102 @@ var ForkSessionResultSchema = z5.object({
   newSessionId: z5.string()
 }).passthrough();
 var GetContextStatsResultSchema = ContextStatsSchema;
+var ContextBreakdownCategorySchema = z5.object({
+  name: z5.string(),
+  tokens: z5.number(),
+  colorKey: z5.string()
+}).passthrough();
+var ContextBreakdownResultSchema = z5.object({
+  modelId: z5.string(),
+  modelDisplayName: z5.string(),
+  contextBudget: z5.number(),
+  usedTokens: z5.number(),
+  freeTokens: z5.number(),
+  categories: z5.array(ContextBreakdownCategorySchema)
+}).passthrough();
 var RenameSessionResultSchema = SuccessResultSchema;
-var InitializeSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: InitializeSessionResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var _LoadSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: LoadSessionResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
+var InitializeSessionResponseSchema = createResponseSchema(
+  InitializeSessionResultSchema
+);
+var _LoadSessionResponseSchema = createResponseSchema(
+  LoadSessionResultSchema
+);
 var LoadSessionResponseSchema = _LoadSessionResponseSchema;
-var AddUserMessageResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: AddUserMessageResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var InterruptSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: InterruptSessionResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var KillWorkerSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: KillWorkerSessionResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var UpdateSessionSettingsResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: UpdateSessionSettingsResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var ToggleMcpServerResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: ToggleMcpServerResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var AuthenticateMcpServerResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: AuthenticateMcpServerResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var CancelMcpAuthResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: CancelMcpAuthResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ClearMcpAuthResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ClearMcpAuthResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var SubmitMcpAuthCodeResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: SubmitMcpAuthCodeResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var AddMcpServerResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: AddMcpServerResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var RemoveMcpServerResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: RemoveMcpServerResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ListMcpRegistryResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: ListMcpRegistryResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var ListMcpToolsResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ListMcpToolsResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ListToolsResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ListToolsResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ListMcpServersResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ListMcpServersResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ToggleMcpToolResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ToggleMcpToolResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ListSkillsResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ListSkillsResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var SubmitBugReportResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: SubmitBugReportResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var GetRewindInfoResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: GetRewindInfoResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ExecuteRewindResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ExecuteRewindResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var CompactSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: CompactSessionResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var ForkSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: ForkSessionResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var GetContextStatsResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: GetContextStatsResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var RenameSessionResponseSchema = z5.union([
-  JsonRpcResponseSuccessSchema.extend({ result: RenameSessionResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
+var AddUserMessageResponseSchema = createResponseSchema(
+  AddUserMessageResultSchema
+);
+var InterruptSessionResponseSchema = createResponseSchema(
+  InterruptSessionResultSchema
+);
+var CloseSessionResponseSchema = createResponseSchema(
+  CloseSessionResultSchema
+);
+var KillWorkerSessionResponseSchema = createResponseSchema(
+  KillWorkerSessionResultSchema
+);
+var UpdateSessionSettingsResponseSchema = createResponseSchema(
+  UpdateSessionSettingsResultSchema
+);
+var ToggleMcpServerResponseSchema = createResponseSchema(
+  ToggleMcpServerResultSchema
+);
+var AuthenticateMcpServerResponseSchema = createResponseSchema(
+  AuthenticateMcpServerResultSchema
+);
+var CancelMcpAuthResponseSchema = createResponseSchema(
+  CancelMcpAuthResultSchema
+);
+var ClearMcpAuthResponseSchema = createResponseSchema(
+  ClearMcpAuthResultSchema
+);
+var SubmitMcpAuthCodeResponseSchema = createResponseSchema(
+  SubmitMcpAuthCodeResultSchema
+);
+var AddMcpServerResponseSchema = createResponseSchema(
+  AddMcpServerResultSchema
+);
+var RemoveMcpServerResponseSchema = createResponseSchema(
+  RemoveMcpServerResultSchema
+);
+var ListMcpRegistryResponseSchema = createResponseSchema(
+  ListMcpRegistryResultSchema
+);
+var ListMcpToolsResponseSchema = createResponseSchema(
+  ListMcpToolsResultSchema
+);
+var ListToolsResponseSchema = createResponseSchema(
+  ListToolsResultSchema
+);
+var ListMcpServersResponseSchema = createResponseSchema(
+  ListMcpServersResultSchema
+);
+var ToggleMcpToolResponseSchema = createResponseSchema(
+  ToggleMcpToolResultSchema
+);
+var ListSkillsResponseSchema = createResponseSchema(
+  ListSkillsResultSchema
+);
+var SubmitBugReportResponseSchema = createResponseSchema(
+  SubmitBugReportResultSchema
+);
+var GetRewindInfoResponseSchema = createResponseSchema(
+  GetRewindInfoResultSchema
+);
+var ExecuteRewindResponseSchema = createResponseSchema(
+  ExecuteRewindResultSchema
+);
+var CompactSessionResponseSchema = createResponseSchema(
+  CompactSessionResultSchema
+);
+var ForkSessionResponseSchema = createResponseSchema(
+  ForkSessionResultSchema
+);
+var GetContextStatsResponseSchema = createResponseSchema(
+  GetContextStatsResultSchema
+);
+var RenameSessionResponseSchema = createResponseSchema(
+  RenameSessionResultSchema
+);
 
 // src/schemas/session-metadata.ts
 var SessionMetadataSchema = z6.object({
@@ -1265,140 +1266,231 @@ var SessionSettingsFileSchema = z6.object({
   tags: z6.array(SessionTagSchema).optional()
 }).passthrough();
 
-// src/schemas/server.ts
+// src/schemas/hooks.ts
 import { z as z7 } from "zod";
+var DroidHookEventSchema = z7.enum([
+  "PreToolUse",
+  "PostToolUse",
+  "Notification",
+  "UserPromptSubmit",
+  "Stop",
+  "SubagentStop",
+  "PreCompact",
+  "SessionStart",
+  "SessionEnd"
+]);
+var DroidHookSpecificOutputSchema = z7.object({
+  hookEventName: DroidHookEventSchema.optional(),
+  permissionDecision: z7.enum(["allow", "deny", "ask"]).optional(),
+  permissionDecisionReason: z7.string().optional(),
+  additionalContext: z7.string().optional(),
+  updatedInput: z7.record(z7.unknown()).optional()
+}).passthrough();
+var DroidHookOutputSchema = z7.object({
+  continue: z7.boolean().optional(),
+  stopReason: z7.string().optional(),
+  suppressOutput: z7.boolean().optional(),
+  systemMessage: z7.string().optional(),
+  decision: z7.enum(["block", "approve"]).optional(),
+  reason: z7.string().optional(),
+  hookSpecificOutput: DroidHookSpecificOutputSchema.optional()
+}).passthrough();
+var DroidHookExecutionResultSchema = DroidHookOutputSchema.extend({
+  exitCode: z7.number(),
+  stdout: z7.string(),
+  stderr: z7.string()
+});
+
+// src/schemas/server.ts
+import { z as z8 } from "zod";
 var ToolUseBlockSchema2 = ToolUseBlockSchema;
 var ToolUseSchema = ToolUseBlockSchema2;
-var ErrorDetailSchema = z7.object({
-  name: z7.string(),
-  message: z7.string()
-}).passthrough();
-var ToolProgressUpdateSchema = z7.object({
-  type: z7.enum(["tool_call", "tool_result", "error", "status", "message"]),
-  toolName: z7.string().optional(),
-  status: z7.string().optional(),
-  details: z7.string().optional(),
-  text: z7.string().optional(),
-  error: z7.string().optional(),
-  timestamp: z7.number().optional(),
+var ErrorDetailSchema = z8.object({
+  name: z8.string(),
+  message: z8.string()
+}).passthrough();
+var ToolProgressUpdateSchema = z8.object({
+  type: z8.enum(["tool_call", "tool_result", "error", "status", "message"]),
+  toolName: z8.string().optional(),
+  status: z8.string().optional(),
+  details: z8.string().optional(),
+  text: z8.string().optional(),
+  error: z8.string().optional(),
+  timestamp: z8.number().optional(),
   parameters: JsonObjectSchema.optional(),
-  valueSnippet: z7.string().optional(),
-  subagentSessionId: z7.string().optional()
-}).passthrough();
-var SettingsUpdatedPayloadSchema = z7.object({
-  autonomyMode: z7.nativeEnum(AutonomyMode).optional(),
-  interactionMode: z7.nativeEnum(DroidInteractionMode).optional().catch(void 0),
-  autonomyLevel: z7.nativeEnum(AutonomyLevel).optional().catch(void 0),
-  modelId: z7.string().optional(),
-  reasoningEffort: z7.nativeEnum(ReasoningEffort).optional(),
-  specModeModelId: z7.string().optional(),
-  specModeReasoningEffort: z7.nativeEnum(ReasoningEffort).optional(),
+  valueSnippet: z8.string().optional(),
+  subagentSessionId: z8.string().optional()
+}).passthrough();
+var SettingsUpdatedPayloadSchema = z8.object({
+  autonomyMode: z8.nativeEnum(AutonomyMode).optional(),
+  interactionMode: z8.nativeEnum(DroidInteractionMode).optional().catch(void 0),
+  autonomyLevel: z8.nativeEnum(AutonomyLevel).optional().catch(void 0),
+  modelId: z8.string().optional(),
+  reasoningEffort: z8.nativeEnum(ReasoningEffort).optional(),
+  specModeModelId: z8.string().optional(),
+  specModeReasoningEffort: z8.nativeEnum(ReasoningEffort).optional(),
   ...ToolSelectionOverridesSchema.shape
 }).passthrough();
-var ToolResultNotificationSchema = z7.object({
-  type: z7.literal("tool_result" /* TOOL_RESULT */),
-  messageId: z7.string(),
-  toolUseId: z7.string(),
+var ToolResultNotificationSchema = z8.object({
+  type: z8.literal("tool_result" /* TOOL_RESULT */),
+  messageId: z8.string(),
+  toolUseId: z8.string(),
   content: JsonValueSchema.optional(),
-  isError: z7.boolean().optional(),
-  id: z7.string().optional()
+  isError: z8.boolean().optional(),
+  id: z8.string().optional()
 }).passthrough();
-var ToolProgressUpdateNotificationSchema = z7.object({
-  type: z7.literal("tool_progress_update" /* TOOL_PROGRESS_UPDATE */),
-  toolUseId: z7.string(),
-  toolName: z7.string(),
+var ToolProgressUpdateNotificationSchema = z8.object({
+  type: z8.literal("tool_progress_update" /* TOOL_PROGRESS_UPDATE */),
+  toolUseId: z8.string(),
+  toolName: z8.string(),
   update: ToolProgressUpdateSchema
 }).passthrough();
-var CreateMessageNotificationSchema = z7.object({
-  type: z7.literal("create_message" /* CREATE_MESSAGE */),
+var CreateMessageNotificationSchema = z8.object({
+  type: z8.literal("create_message" /* CREATE_MESSAGE */),
   message: FactoryDroidMessageSchema,
-  parentId: z7.string().optional(),
-  requestId: z7.string().optional()
-}).passthrough();
-var ErrorNotificationSchema = z7.object({
-  type: z7.literal("error" /* ERROR */),
-  message: z7.string(),
-  errorType: z7.nativeEnum(DroidErrorType),
-  timestamp: z7.string(),
+  parentId: z8.string().optional(),
+  requestId: z8.string().optional()
+}).passthrough();
+var ErrorNotificationSchema = z8.object({
+  type: z8.literal("error" /* ERROR */),
+  message: z8.string(),
+  errorType: z8.nativeEnum(DroidErrorType),
+  timestamp: z8.string(),
   error: ErrorDetailSchema.optional()
 }).passthrough();
-var DroidWorkingStateChangedNotificationSchema = z7.object({
-  type: z7.literal("droid_working_state_changed" /* DROID_WORKING_STATE_CHANGED */),
-  newState: z7.nativeEnum(DroidWorkingState)
+var DroidWorkingStateChangedNotificationSchema = z8.object({
+  type: z8.literal("droid_working_state_changed" /* DROID_WORKING_STATE_CHANGED */),
+  newState: z8.nativeEnum(DroidWorkingState)
 }).passthrough();
-var PermissionResolvedNotificationSchema = z7.object({
-  type: z7.literal("permission_resolved" /* PERMISSION_RESOLVED */),
-  requestId: z7.string(),
-  toolUseIds: z7.array(z7.string()),
-  selectedOption: z7.nativeEnum(ToolConfirmationOutcome)
+var PermissionResolvedNotificationSchema = z8.object({
+  type: z8.literal("permission_resolved" /* PERMISSION_RESOLVED */),
+  requestId: z8.string(),
+  toolUseIds: z8.array(z8.string()),
+  selectedOption: z8.nativeEnum(ToolConfirmationOutcome)
 }).passthrough();
-var SettingsUpdatedNotificationSchema = z7.object({
-  type: z7.literal("settings_updated" /* SETTINGS_UPDATED */),
+var SettingsUpdatedNotificationSchema = z8.object({
+  type: z8.literal("settings_updated" /* SETTINGS_UPDATED */),
   settings: SettingsUpdatedPayloadSchema
 }).passthrough();
-var SessionTitleUpdatedNotificationSchema = z7.object({
-  type: z7.literal("session_title_updated" /* SESSION_TITLE_UPDATED */),
-  title: z7.string()
+var SessionTitleUpdatedNotificationSchema = z8.object({
+  type: z8.literal("session_title_updated" /* SESSION_TITLE_UPDATED */),
+  title: z8.string()
 }).passthrough();
-var McpStatusChangedNotificationSchema = z7.object({
-  type: z7.literal("mcp_status_changed" /* MCP_STATUS_CHANGED */),
-  servers: z7.array(McpServerStatusInfoSchema),
+var McpStatusChangedNotificationSchema = z8.object({
+  type: z8.literal("mcp_status_changed" /* MCP_STATUS_CHANGED */),
+  servers: z8.array(McpServerStatusInfoSchema),
   summary: McpStatusSummarySchema
 }).passthrough();
-var AssistantTextDeltaNotificationSchema = z7.object({
-  type: z7.literal("assistant_text_delta" /* ASSISTANT_TEXT_DELTA */),
-  messageId: z7.string(),
-  blockIndex: z7.number(),
-  textDelta: z7.string()
-}).passthrough();
-var ThinkingTextDeltaNotificationSchema = z7.object({
-  type: z7.literal("thinking_text_delta" /* THINKING_TEXT_DELTA */),
-  messageId: z7.string(),
-  blockIndex: z7.number(),
-  textDelta: z7.string()
-}).passthrough();
-var SessionTokenUsageChangedNotificationSchema = z7.object({
-  type: z7.literal("session_token_usage_changed" /* SESSION_TOKEN_USAGE_CHANGED */),
-  sessionId: z7.string(),
+var AssistantTextDeltaNotificationSchema = z8.object({
+  type: z8.literal("assistant_text_delta" /* ASSISTANT_TEXT_DELTA */),
+  messageId: z8.string(),
+  blockIndex: z8.number(),
+  textDelta: z8.string()
+}).passthrough();
+var AssistantTextCompleteNotificationSchema = z8.object({
+  type: z8.literal("assistant_text_complete" /* ASSISTANT_TEXT_COMPLETE */),
+  messageId: z8.string(),
+  blockIndex: z8.number()
+}).passthrough();
+var ThinkingTextDeltaNotificationSchema = z8.object({
+  type: z8.literal("thinking_text_delta" /* THINKING_TEXT_DELTA */),
+  messageId: z8.string(),
+  blockIndex: z8.number(),
+  textDelta: z8.string()
+}).passthrough();
+var ThinkingTextCompleteNotificationSchema = z8.object({
+  type: z8.literal("thinking_text_complete" /* THINKING_TEXT_COMPLETE */),
+  messageId: z8.string(),
+  blockIndex: z8.number(),
+  durationMs: z8.number().optional()
+}).passthrough();
+var ToolCallNotificationSchema = z8.object({
+  type: z8.literal("tool_call" /* TOOL_CALL */),
+  toolUse: ToolUseBlockSchema2
+}).passthrough();
+var SessionTokenUsageChangedNotificationSchema = z8.object({
+  type: z8.literal("session_token_usage_changed" /* SESSION_TOKEN_USAGE_CHANGED */),
+  sessionId: z8.string(),
   tokenUsage: TokenUsageSchema
 }).passthrough();
-var MissionStateChangedNotificationSchema = z7.object({
-  type: z7.literal("mission_state_changed" /* MISSION_STATE_CHANGED */),
-  state: z7.nativeEnum(MissionState)
+var MissionStateChangedNotificationSchema = z8.object({
+  type: z8.literal("mission_state_changed" /* MISSION_STATE_CHANGED */),
+  state: z8.nativeEnum(MissionState)
+}).passthrough();
+var MissionFeaturesChangedNotificationSchema = z8.object({
+  type: z8.literal("mission_features_changed" /* MISSION_FEATURES_CHANGED */),
+  features: z8.array(MissionFeatureSchema)
+}).passthrough();
+var MissionProgressEntryNotificationSchema = z8.object({
+  type: z8.literal("mission_progress_entry" /* MISSION_PROGRESS_ENTRY */),
+  progressLog: z8.array(ProgressLogEntrySchema)
+}).passthrough();
+var MissionHeartbeatNotificationSchema = z8.object({
+  type: z8.literal("mission_heartbeat" /* MISSION_HEARTBEAT */),
+  timestamp: z8.string()
+}).passthrough();
+var MissionWorkerStartedNotificationSchema = z8.object({
+  type: z8.literal("mission_worker_started" /* MISSION_WORKER_STARTED */),
+  workerSessionId: z8.string()
 }).passthrough();
-var MissionFeaturesChangedNotificationSchema = z7.object({
-  type: z7.literal("mission_features_changed" /* MISSION_FEATURES_CHANGED */),
-  features: z7.array(MissionFeatureSchema)
+var MissionWorkerCompletedNotificationSchema = z8.object({
+  type: z8.literal("mission_worker_completed" /* MISSION_WORKER_COMPLETED */),
+  workerSessionId: z8.string(),
+  exitCode: z8.number()
 }).passthrough();
-var MissionProgressEntryNotificationSchema = z7.object({
-  type: z7.literal("mission_progress_entry" /* MISSION_PROGRESS_ENTRY */),
-  progressLog: z7.array(ProgressLogEntrySchema)
+var McpAuthRequiredNotificationSchema = z8.object({
+  type: z8.literal("mcp_auth_required" /* MCP_AUTH_REQUIRED */),
+  serverName: z8.string(),
+  authUrl: z8.string(),
+  message: z8.string(),
+  state: z8.string()
 }).passthrough();
-var MissionHeartbeatNotificationSchema = z7.object({
-  type: z7.literal("mission_heartbeat" /* MISSION_HEARTBEAT */),
-  timestamp: z7.string()
+var McpAuthCompletedNotificationSchema = z8.object({
+  type: z8.literal("mcp_auth_completed" /* MCP_AUTH_COMPLETED */),
+  serverName: z8.string(),
+  outcome: z8.nativeEnum(McpAuthOutcome),
+  message: z8.string()
 }).passthrough();
-var MissionWorkerStartedNotificationSchema = z7.object({
-  type: z7.literal("mission_worker_started" /* MISSION_WORKER_STARTED */),
-  workerSessionId: z7.string()
+var HookCommandSchema = z8.object({
+  command: z8.string(),
+  timeout: z8.number().optional()
 }).passthrough();
-var MissionWorkerCompletedNotificationSchema = z7.object({
-  type: z7.literal("mission_worker_completed" /* MISSION_WORKER_COMPLETED */),
-  workerSessionId: z7.string(),
-  exitCode: z7.number()
+var HookResultSchema = z8.object({
+  exitCode: z8.number(),
+  stdout: z8.string(),
+  stderr: z8.string(),
+  command: z8.string().optional(),
+  timeout: z8.number().optional()
 }).passthrough();
-var McpAuthRequiredNotificationSchema = z7.object({
-  type: z7.literal("mcp_auth_required" /* MCP_AUTH_REQUIRED */),
-  serverName: z7.string(),
-  authUrl: z7.string(),
-  message: z7.string(),
-  state: z7.string()
+var HookExecutionStartedNotificationSchema = z8.object({
+  type: z8.literal("hook_execution_started" /* HOOK_EXECUTION_STARTED */),
+  hookId: z8.string(),
+  hookEventName: DroidHookEventSchema,
+  hookMatcher: z8.string().optional(),
+  hookCommands: z8.array(HookCommandSchema),
+  hookToolCallId: z8.string().optional(),
+  isParallelExecution: z8.boolean().optional(),
+  parallelGroupId: z8.string().optional()
 }).passthrough();
-var McpAuthCompletedNotificationSchema = z7.object({
-  type: z7.literal("mcp_auth_completed" /* MCP_AUTH_COMPLETED */),
-  serverName: z7.string(),
-  outcome: z7.nativeEnum(McpAuthOutcome),
-  message: z7.string()
+var HookExecutionCompletedNotificationSchema = z8.object({
+  type: z8.literal("hook_execution_completed" /* HOOK_EXECUTION_COMPLETED */),
+  hookId: z8.string(),
+  hookEventName: DroidHookEventSchema.optional(),
+  hookMatcher: z8.string().optional(),
+  hookToolCallId: z8.string().optional(),
+  hookStatus: z8.enum(["completed", "error"]),
+  hookResults: z8.array(HookResultSchema).optional()
+}).passthrough();
+var StructuredOutputErrorSchema = z8.object({
+  code: z8.string(),
+  message: z8.string(),
+  details: z8.unknown().optional()
+}).passthrough();
+var StructuredOutputNotificationSchema = z8.object({
+  type: z8.literal("structured_output" /* STRUCTURED_OUTPUT */),
+  messageId: z8.string(),
+  structuredOutput: JsonObjectSchema.nullable(),
+  structuredOutputError: StructuredOutputErrorSchema.nullable()
 }).passthrough();
 var SessionNotificationSchemaList = [
   ToolResultNotificationSchema,
@@ -1411,7 +1503,10 @@ var SessionNotificationSchemaList = [
   SessionTitleUpdatedNotificationSchema,
   McpStatusChangedNotificationSchema,
   AssistantTextDeltaNotificationSchema,
+  AssistantTextCompleteNotificationSchema,
   ThinkingTextDeltaNotificationSchema,
+  ThinkingTextCompleteNotificationSchema,
+  ToolCallNotificationSchema,
   SessionTokenUsageChangedNotificationSchema,
   MissionStateChangedNotificationSchema,
   MissionFeaturesChangedNotificationSchema,
@@ -1420,88 +1515,91 @@ var SessionNotificationSchemaList = [
   MissionWorkerStartedNotificationSchema,
   MissionWorkerCompletedNotificationSchema,
   McpAuthRequiredNotificationSchema,
-  McpAuthCompletedNotificationSchema
+  McpAuthCompletedNotificationSchema,
+  HookExecutionStartedNotificationSchema,
+  HookExecutionCompletedNotificationSchema,
+  StructuredOutputNotificationSchema
 ];
-var SessionNotificationPayloadSchema = z7.discriminatedUnion(
+var SessionNotificationPayloadSchema = z8.discriminatedUnion(
   "type",
   SessionNotificationSchemaList
 );
-var _SessionNotificationParamsSchema = z7.object({
+var _SessionNotificationParamsSchema = z8.object({
   notification: SessionNotificationPayloadSchema
 }).passthrough();
 var SessionNotificationParamsSchema = _SessionNotificationParamsSchema;
 var _SessionNotificationSchema = JsonRpcNotificationSchema.extend({
-  method: z7.literal("droid.session_notification" /* SESSION_NOTIFICATION */),
+  method: z8.literal("droid.session_notification" /* SESSION_NOTIFICATION */),
   params: SessionNotificationParamsSchema
 });
 var SessionNotificationSchema = _SessionNotificationSchema;
-var EditToolConfirmationDetailsSchema = z7.object({
-  type: z7.literal("edit" /* Edit */),
-  filePath: z7.string(),
-  fileName: z7.string(),
-  oldContent: z7.string().optional(),
-  newContent: z7.string().optional()
-}).passthrough();
-var ExecuteToolConfirmationDetailsSchema = z7.object({
-  type: z7.literal("exec" /* Execute */),
-  fullCommand: z7.string(),
-  command: z7.string(),
-  extractedCommands: z7.array(z7.string()).optional(),
-  impactLevel: z7.string().optional()
-}).passthrough();
-var CreateToolConfirmationDetailsSchema = z7.object({
-  type: z7.literal("create" /* Create */),
-  filePath: z7.string(),
-  fileName: z7.string(),
-  content: z7.string()
-}).passthrough();
-var AskUserQuestionSchema = z7.object({
-  index: z7.number(),
-  topic: z7.string(),
-  question: z7.string(),
-  options: z7.array(z7.string())
-}).passthrough();
-var AskUserConfirmationDetailsSchema = z7.object({
-  type: z7.literal("ask_user" /* AskUser */),
-  questionnaire: z7.string(),
-  parsed: z7.object({
-    questions: z7.array(AskUserQuestionSchema)
+var EditToolConfirmationDetailsSchema = z8.object({
+  type: z8.literal("edit" /* Edit */),
+  filePath: z8.string(),
+  fileName: z8.string(),
+  oldContent: z8.string().optional(),
+  newContent: z8.string().optional()
+}).passthrough();
+var ExecuteToolConfirmationDetailsSchema = z8.object({
+  type: z8.literal("exec" /* Execute */),
+  fullCommand: z8.string(),
+  command: z8.string(),
+  extractedCommands: z8.array(z8.string()).optional(),
+  impactLevel: z8.string().optional()
+}).passthrough();
+var CreateToolConfirmationDetailsSchema = z8.object({
+  type: z8.literal("create" /* Create */),
+  filePath: z8.string(),
+  fileName: z8.string(),
+  content: z8.string()
+}).passthrough();
+var AskUserQuestionSchema = z8.object({
+  index: z8.number(),
+  topic: z8.string(),
+  question: z8.string(),
+  options: z8.array(z8.string())
+}).passthrough();
+var AskUserConfirmationDetailsSchema = z8.object({
+  type: z8.literal("ask_user" /* AskUser */),
+  questionnaire: z8.string(),
+  parsed: z8.object({
+    questions: z8.array(AskUserQuestionSchema)
   }).optional(),
-  parseError: z7.object({
-    message: z7.string(),
-    line: z7.number().optional()
+  parseError: z8.object({
+    message: z8.string(),
+    line: z8.number().optional()
   }).optional()
 }).passthrough();
-var ExitSpecModeConfirmationDetailsSchema = z7.object({
-  type: z7.literal("exit_spec_mode" /* ExitSpecMode */),
-  plan: z7.string(),
-  title: z7.string().optional(),
-  optionNames: z7.array(z7.string()).optional()
-}).passthrough();
-var ProposeMissionConfirmationDetailsSchema = z7.object({
-  type: z7.literal("propose_mission" /* ProposeMission */),
-  proposal: z7.string(),
-  title: z7.string().optional()
-}).passthrough();
-var StartMissionRunConfirmationDetailsSchema = z7.object({
-  type: z7.literal("start_mission_run" /* StartMissionRun */),
-  runningMissionCount: z7.number(),
-  runningMissionSessionIds: z7.array(z7.string())
-}).passthrough();
-var ApplyPatchToolConfirmationDetailsSchema = z7.object({
-  type: z7.literal("apply_patch" /* ApplyPatch */),
-  filePath: z7.string(),
-  fileName: z7.string(),
-  patchContent: z7.string(),
-  oldContent: z7.string().optional(),
-  newContent: z7.string().optional()
-}).passthrough();
-var McpToolConfirmationDetailsSchema = z7.object({
-  type: z7.literal("mcp_tool" /* McpTool */),
-  toolName: z7.string(),
-  impactLevel: z7.string()
-}).passthrough();
-var ToolConfirmationDetailsSchema = z7.discriminatedUnion("type", [
+var ExitSpecModeConfirmationDetailsSchema = z8.object({
+  type: z8.literal("exit_spec_mode" /* ExitSpecMode */),
+  plan: z8.string(),
+  title: z8.string().optional(),
+  optionNames: z8.array(z8.string()).optional()
+}).passthrough();
+var ProposeMissionConfirmationDetailsSchema = z8.object({
+  type: z8.literal("propose_mission" /* ProposeMission */),
+  proposal: z8.string(),
+  title: z8.string().optional()
+}).passthrough();
+var StartMissionRunConfirmationDetailsSchema = z8.object({
+  type: z8.literal("start_mission_run" /* StartMissionRun */),
+  runningMissionCount: z8.number(),
+  runningMissionSessionIds: z8.array(z8.string())
+}).passthrough();
+var ApplyPatchToolConfirmationDetailsSchema = z8.object({
+  type: z8.literal("apply_patch" /* ApplyPatch */),
+  filePath: z8.string(),
+  fileName: z8.string(),
+  patchContent: z8.string(),
+  oldContent: z8.string().optional(),
+  newContent: z8.string().optional()
+}).passthrough();
+var McpToolConfirmationDetailsSchema = z8.object({
+  type: z8.literal("mcp_tool" /* McpTool */),
+  toolName: z8.string(),
+  impactLevel: z8.string()
+}).passthrough();
+var ToolConfirmationDetailsSchema = z8.discriminatedUnion("type", [
   EditToolConfirmationDetailsSchema,
   ExecuteToolConfirmationDetailsSchema,
   CreateToolConfirmationDetailsSchema,
@@ -1512,51 +1610,45 @@ var ToolConfirmationDetailsSchema = z7.discriminatedUnion("type", [
   ApplyPatchToolConfirmationDetailsSchema,
   McpToolConfirmationDetailsSchema
 ]);
-var ToolConfirmationInfoSchema = z7.object({
+var ToolConfirmationInfoSchema = z8.object({
   toolUse: ToolUseSchema,
-  confirmationType: z7.nativeEnum(ToolConfirmationType),
+  confirmationType: z8.nativeEnum(ToolConfirmationType),
   details: ToolConfirmationDetailsSchema
 }).passthrough();
-var RequestPermissionRequestParamsSchema = z7.object({
-  toolUses: z7.array(ToolConfirmationInfoSchema),
-  options: z7.array(ToolConfirmationListItemSchema)
+var RequestPermissionRequestParamsSchema = z8.object({
+  toolUses: z8.array(ToolConfirmationInfoSchema),
+  options: z8.array(ToolConfirmationListItemSchema)
 }).passthrough();
 var RequestPermissionRequestSchema = JsonRpcRequestSchema.extend({
-  method: z7.literal("droid.request_permission" /* REQUEST_PERMISSION */),
+  method: z8.literal("droid.request_permission" /* REQUEST_PERMISSION */),
   params: RequestPermissionRequestParamsSchema
 });
-var RequestPermissionResultSchema = z7.object({
-  selectedOption: z7.nativeEnum(ToolConfirmationOutcome),
-  comment: z7.string().optional()
+var RequestPermissionResultSchema = z8.object({
+  selectedOption: z8.nativeEnum(ToolConfirmationOutcome),
+  comment: z8.string().optional()
 }).strict();
-var RequestPermissionResponseSchema = z7.union([
-  JsonRpcResponseSuccessSchema.extend({
-    result: RequestPermissionResultSchema
-  }),
-  JsonRpcResponseFailureSchema
-]);
-var AskUserRequestParamsSchema = z7.object({
-  toolCallId: z7.string(),
-  questions: z7.array(AskUserQuestionSchema)
+var RequestPermissionResponseSchema = createResponseSchema(
+  RequestPermissionResultSchema
+);
+var AskUserRequestParamsSchema = z8.object({
+  toolCallId: z8.string(),
+  questions: z8.array(AskUserQuestionSchema)
 }).passthrough();
 var AskUserRequestSchema = JsonRpcRequestSchema.extend({
-  method: z7.literal("droid.ask_user" /* ASK_USER */),
+  method: z8.literal("droid.ask_user" /* ASK_USER */),
   params: AskUserRequestParamsSchema
 });
-var AskUserCollectedAnswerSchema = z7.object({
-  index: z7.number(),
-  question: z7.string(),
-  answer: z7.string()
+var AskUserCollectedAnswerSchema = z8.object({
+  index: z8.number(),
+  question: z8.string(),
+  answer: z8.string()
 }).strict();
-var AskUserResultSchema = z7.object({
-  cancelled: z7.boolean().optional(),
-  answers: z7.array(AskUserCollectedAnswerSchema)
+var AskUserResultSchema = z8.object({
+  cancelled: z8.boolean().optional(),
+  answers: z8.array(AskUserCollectedAnswerSchema)
 }).strict();
-var AskUserResponseSchema = z7.union([
-  JsonRpcResponseSuccessSchema.extend({ result: AskUserResultSchema }),
-  JsonRpcResponseFailureSchema
-]);
-var _CliRequestOrNotificationSchema = z7.union([
+var AskUserResponseSchema = createResponseSchema(AskUserResultSchema);
+var _CliRequestOrNotificationSchema = z8.union([
   SessionNotificationSchema,
   RequestPermissionRequestSchema,
   AskUserRequestSchema
@@ -1564,7 +1656,7 @@ var _CliRequestOrNotificationSchema = z7.union([
 var CliRequestOrNotificationSchema = _CliRequestOrNotificationSchema;
 
 // src/constants.ts
-var SDK_VERSION = "0.2.0";
+var SDK_VERSION = "0.4.0";
 var SDK_TAG = {
   name: "sdk",
   metadata: { language: "typescript", version: SDK_VERSION }
@@ -1880,9 +1972,32 @@ var ProcessTransport = class {
 
 // src/protocol.ts
 import { v4 as uuidv4 } from "uuid";
+function dispatchNotification(notification, listeners) {
+  let notificationType;
+  const parsed = SessionNotificationParamsSchema.safeParse(
+    notification["params"]
+  );
+  if (parsed.success) {
+    notificationType = parsed.data.notification.type;
+  }
+  for (const listener of listeners) {
+    if (listener.filter?.type != null && listener.filter.type !== notificationType) {
+      continue;
+    }
+    try {
+      listener.callback(notification);
+    } catch {
+    }
+  }
+}
+var DEFAULT_SERVER_REQUEST_METHOD_MAP = {
+  ["droid.request_permission" /* REQUEST_PERMISSION */]: "permission" /* Permission */,
+  ["droid.ask_user" /* ASK_USER */]: "askUser" /* AskUser */
+};
 var ProtocolEngine = class {
   _transport;
   _defaultTimeout;
+  _serverRequestMethodMap;
   _pendingRequests = /* @__PURE__ */ new Map();
   _notificationListeners = /* @__PURE__ */ new Set();
   _permissionHandler = null;
@@ -1892,6 +2007,7 @@ var ProtocolEngine = class {
   constructor(options) {
     this._transport = options.transport;
     this._defaultTimeout = options.defaultTimeout ?? DEFAULT_REQUEST_TIMEOUT;
+    this._serverRequestMethodMap = options.serverRequestMethodMap ?? DEFAULT_SERVER_REQUEST_METHOD_MAP;
     this._transport.onMessage((message) => {
       this._handleMessage(message);
     });
@@ -2044,27 +2160,13 @@ var ProtocolEngine = class {
     pending.resolve(result);
   }
   _handleNotification(notification) {
-    let notificationType;
-    const parsed = SessionNotificationParamsSchema.safeParse(
-      notification["params"]
-    );
-    if (parsed.success) {
-      notificationType = parsed.data.notification.type;
-    }
-    for (const listener of this._notificationListeners) {
-      if (listener.filter?.type != null && listener.filter.type !== notificationType) {
-        continue;
-      }
-      try {
-        listener.callback(notification);
-      } catch {
-      }
-    }
+    dispatchNotification(notification, this._notificationListeners);
   }
   async _handleServerRequest(method, requestId, params) {
-    if (method === "droid.request_permission" /* REQUEST_PERMISSION */) {
+    const handlerType = this._serverRequestMethodMap[method];
+    if (handlerType === "permission" /* Permission */) {
       await this._handlePermissionRequest(requestId, params);
-    } else if (method === "droid.ask_user" /* ASK_USER */) {
+    } else if (handlerType === "askUser" /* AskUser */) {
       await this._handleAskUserRequest(requestId, params);
     }
   }
@@ -2180,10 +2282,6 @@ var DroidClient = class {
   _engine;
   _sessionId = null;
   _closed = false;
-  /**
-   * Client-level notification listeners.
-   * Each entry is [callback, optional notification filter].
-   */
   _notificationListeners = [];
   /** Client-level permission handler. */
   _permissionHandler = null;
@@ -2257,6 +2355,13 @@ var DroidClient = class {
       InterruptSessionResultSchema
     );
   }
+  async closeSession(params = {}) {
+    return this._sessionRpc(
+      "droid.close_session" /* CLOSE_SESSION */,
+      params,
+      CloseSessionResultSchema
+    );
+  }
   async killWorkerSession(params) {
     return this._sessionRpc(
       "droid.kill_worker_session" /* KILL_WORKER_SESSION */,
@@ -2445,23 +2550,7 @@ var DroidClient = class {
     await this._engine.close();
   }
   _dispatchNotification(notification) {
-    let notificationType;
-    const parsed = SessionNotificationParamsSchema.safeParse(
-      notification["params"]
-    );
-    if (parsed.success) {
-      notificationType = parsed.data.notification.type;
-    }
-    const listeners = [...this._notificationListeners];
-    for (const listener of listeners) {
-      if (listener.filter?.type != null && listener.filter.type !== notificationType) {
-        continue;
-      }
-      try {
-        listener.callback(notification);
-      } catch {
-      }
-    }
+    dispatchNotification(notification, [...this._notificationListeners]);
   }
   _dispatchPermissionRequest(params) {
     const handler = this._permissionHandler;
@@ -2495,14 +2584,18 @@ var DroidClient = class {
 
 // src/stream.ts
 var DroidMessageType = {
+  Assistant: "assistant",
+  User: "user",
+  ToolCall: "tool_call",
   AssistantTextDelta: "assistant_text_delta",
+  AssistantTextComplete: "assistant_text_complete",
   ThinkingTextDelta: "thinking_text_delta",
-  ToolUse: "tool_use",
+  ThinkingTextComplete: "thinking_text_complete",
+  ToolCallDelta: "tool_call_delta",
   ToolResult: "tool_result",
   ToolProgress: "tool_progress",
   WorkingStateChanged: "working_state_changed",
   TokenUsageUpdate: "token_usage_update",
-  CreateMessage: "create_message",
   PermissionResolved: "permission_resolved",
   SettingsUpdated: "settings_updated",
   SessionTitleUpdated: "session_title_updated",
@@ -2515,8 +2608,9 @@ var DroidMessageType = {
   MissionWorkerCompleted: "mission_worker_completed",
   McpAuthRequired: "mcp_auth_required",
   McpAuthCompleted: "mcp_auth_completed",
+  Hook: "hook",
   Error: "error",
-  TurnComplete: "turn_complete"
+  Result: "result"
 };
 function convertNotificationToStreamMessage(raw) {
   const parsed = SessionNotificationPayloadSchema.safeParse(raw);
@@ -2532,6 +2626,12 @@ function convertNotificationToStreamMessage(raw) {
         blockIndex: notification.blockIndex,
         text: notification.textDelta
       };
+    case "assistant_text_complete" /* ASSISTANT_TEXT_COMPLETE */:
+      return {
+        type: DroidMessageType.AssistantTextComplete,
+        messageId: notification.messageId,
+        blockIndex: notification.blockIndex
+      };
     case "thinking_text_delta" /* THINKING_TEXT_DELTA */:
       return {
         type: DroidMessageType.ThinkingTextDelta,
@@ -2539,6 +2639,18 @@ function convertNotificationToStreamMessage(raw) {
         blockIndex: notification.blockIndex,
         text: notification.textDelta
       };
+    case "thinking_text_complete" /* THINKING_TEXT_COMPLETE */:
+      return {
+        type: DroidMessageType.ThinkingTextComplete,
+        messageId: notification.messageId,
+        blockIndex: notification.blockIndex,
+        durationMs: notification.durationMs
+      };
+    case "tool_call" /* TOOL_CALL */:
+      return {
+        type: DroidMessageType.ToolCallDelta,
+        toolUse: notification.toolUse
+      };
     case "tool_result" /* TOOL_RESULT */:
       return {
         type: DroidMessageType.ToolResult,
@@ -2578,22 +2690,33 @@ function convertNotificationToStreamMessage(raw) {
       const msg = notification.message;
       const messages = [];
       for (const block of msg.content) {
-        if (block.type === DroidMessageType.ToolUse) {
+        if (block.type === "tool_use") {
           messages.push({
-            type: DroidMessageType.ToolUse,
-            toolName: block.name,
-            toolInput: block.input,
-            toolUseId: block.id
+            type: DroidMessageType.ToolCall,
+            toolUse: block
           });
         }
       }
-      messages.push({
-        type: DroidMessageType.CreateMessage,
-        messageId: msg.id,
-        role: msg.role,
-        content: msg.content,
-        parentId: notification.parentId
-      });
+      if (msg.role === FactoryDroidMessageRole.Assistant) {
+        messages.push({
+          type: DroidMessageType.Assistant,
+          message: msg,
+          text: extractTextFromMessage(msg)
+        });
+      } else if (msg.role === FactoryDroidMessageRole.User) {
+        messages.push({
+          type: DroidMessageType.User,
+          message: msg
+        });
+      } else {
+        messages.push({
+          type: "create_message",
+          messageId: msg.id,
+          role: msg.role,
+          content: msg.content,
+          parentId: notification.parentId
+        });
+      }
       return messages;
     }
     case "error" /* ERROR */:
@@ -2672,42 +2795,205 @@ function convertNotificationToStreamMessage(raw) {
         outcome: notification.outcome,
         message: notification.message
       };
+    case "hook_execution_started" /* HOOK_EXECUTION_STARTED */:
+      return notification.hookCommands.map((hookCommand) => ({
+        type: DroidMessageType.Hook,
+        hookId: notification.hookId,
+        eventName: notification.hookEventName,
+        matcher: notification.hookMatcher,
+        toolCallId: notification.hookToolCallId,
+        command: hookCommand.command,
+        timeout: hookCommand.timeout,
+        status: "started"
+      }));
+    case "hook_execution_completed" /* HOOK_EXECUTION_COMPLETED */:
+      return (notification.hookResults ?? []).map((hookResult) => ({
+        type: DroidMessageType.Hook,
+        hookId: notification.hookId,
+        eventName: notification.hookEventName,
+        matcher: notification.hookMatcher,
+        toolCallId: notification.hookToolCallId,
+        command: hookResult.command,
+        timeout: hookResult.timeout,
+        status: notification.hookStatus,
+        exitCode: hookResult.exitCode,
+        stdout: hookResult.stdout,
+        stderr: hookResult.stderr
+      }));
+    case "structured_output" /* STRUCTURED_OUTPUT */:
+      return {
+        type: "structured_output",
+        messageId: notification.messageId,
+        structuredOutput: notification.structuredOutput,
+        structuredOutputError: notification.structuredOutputError
+      };
     default:
       return null;
   }
 }
 var StreamStateTracker = class {
+  constructor(options = {}) {
+    this.options = options;
+  }
   hasBeenNonIdle = false;
   lastTokenUsage = null;
+  fullText = "";
+  finalAssistantText = "";
+  structuredOutput = null;
+  structuredOutputError = null;
+  errors = [];
+  numTurns = 0;
+  emittedMessages = [];
   toolNameMap = /* @__PURE__ */ new Map();
   getToolName(toolUseId) {
     return this.toolNameMap.get(toolUseId) ?? "";
   }
   processMessage(message) {
     const additional = [];
-    if (message.type === DroidMessageType.ToolUse) {
+    if (message.type === "tool_use") {
       this.toolNameMap.set(message.toolUseId, message.toolName);
     }
+    if (message.type === DroidMessageType.ToolCall) {
+      this.toolNameMap.set(message.toolUse.id, message.toolUse.name);
+    }
+    if (message.type === DroidMessageType.ToolCallDelta) {
+      this.toolNameMap.set(message.toolUse.id, message.toolUse.name);
+    }
     if (message.type === DroidMessageType.ToolResult) {
       message = { ...message, toolName: this.getToolName(message.toolUseId) };
     }
+    if (message.type === DroidMessageType.AssistantTextDelta) {
+      this.fullText += message.text;
+    }
+    if (message.type === DroidMessageType.Assistant) {
+      this.finalAssistantText = message.text;
+      if (this.fullText.length === 0) {
+        this.fullText = message.text;
+      }
+    }
     if (message.type === DroidMessageType.TokenUsageUpdate) {
       this.lastTokenUsage = message;
     }
+    if (message.type === DroidMessageType.Error) {
+      this.errors.push(message);
+    }
+    if (message.type === "structured_output") {
+      this.structuredOutput = message.structuredOutput;
+      this.structuredOutputError = message.structuredOutputError;
+      return { message: null, additional };
+    }
     if (message.type === DroidMessageType.WorkingStateChanged) {
       if (message.state !== "idle" /* Idle */) {
         this.hasBeenNonIdle = true;
       } else if (this.hasBeenNonIdle) {
-        additional.push({
-          type: DroidMessageType.TurnComplete,
-          tokenUsage: this.lastTokenUsage
-        });
+        this.numTurns++;
+        additional.push(this.createResultMessage());
         this.hasBeenNonIdle = false;
+        this.resetTurnState();
       }
     }
+    if (message.type !== DroidMessageType.WorkingStateChanged) {
+      this.trackEmittedMessage(message);
+    }
     return { message, additional };
   }
+  createResultMessage() {
+    if (this.options.hasOutputFormat && this.structuredOutput === null && this.structuredOutputError === null) {
+      this.structuredOutput = parseJsonObject(this.finalAssistantText || this.fullText) ?? null;
+    }
+    const tokenUsage = this.lastTokenUsage ? stripTokenUsageType(this.lastTokenUsage) : null;
+    const result = this.finalAssistantText || this.fullText;
+    const base = {
+      type: DroidMessageType.Result,
+      sessionId: this.options.sessionId ?? "",
+      durationMs: Date.now() - (this.options.startedAt ?? Date.now()),
+      numTurns: this.numTurns,
+      result,
+      tokenUsage,
+      messages: [...this.emittedMessages],
+      text: result,
+      turnCount: this.numTurns
+    };
+    if (this.structuredOutputError) {
+      return {
+        ...base,
+        subtype: "error_structured_output",
+        isError: true,
+        success: false,
+        errors: [this.structuredOutputError.message],
+        structuredOutput: this.structuredOutput,
+        structuredOutputError: this.structuredOutputError,
+        error: null
+      };
+    }
+    if (this.errors.length > 0) {
+      return {
+        ...base,
+        subtype: "error_during_execution",
+        isError: true,
+        success: false,
+        errors: this.errors.map((error) => error.message),
+        structuredOutput: this.structuredOutput,
+        structuredOutputError: null,
+        error: this.errors[0] ?? null
+      };
+    }
+    return {
+      ...base,
+      subtype: "success",
+      isError: false,
+      success: true,
+      structuredOutput: this.structuredOutput,
+      structuredOutputError: null,
+      error: null
+    };
+  }
+  trackEmittedMessage(message) {
+    if (isDefaultStreamMessage(message)) {
+      this.emittedMessages.push(message);
+    }
+  }
+  resetTurnState() {
+    this.lastTokenUsage = null;
+    this.fullText = "";
+    this.finalAssistantText = "";
+    this.structuredOutput = null;
+    this.structuredOutputError = null;
+    this.errors = [];
+    this.emittedMessages = [];
+  }
 };
+function isInternalMessage(message) {
+  return message.type === "structured_output" || message.type === "tool_use" || message.type === "create_message";
+}
+function isDefaultStreamMessage(message) {
+  return message.type === DroidMessageType.Assistant || message.type === DroidMessageType.User || message.type === DroidMessageType.ToolCall || message.type === DroidMessageType.ToolResult || message.type === DroidMessageType.Hook || message.type === DroidMessageType.Error || message.type === DroidMessageType.Result;
+}
+function stripTokenUsageType(message) {
+  const {
+    inputTokens,
+    outputTokens,
+    cacheCreationTokens,
+    cacheReadTokens,
+    thinkingTokens
+  } = message;
+  return {
+    inputTokens,
+    outputTokens,
+    cacheCreationTokens,
+    cacheReadTokens,
+    thinkingTokens
+  };
+}
+function parseJsonObject(text) {
+  try {
+    const parsed = JSON.parse(text);
+    const result = JsonObjectSchema.safeParse(parsed);
+    return result.success ? result.data : null;
+  } catch {
+    return null;
+  }
+}
 function normalizeToolResultContent(content) {
   if (content == null) {
     return "";
@@ -2720,6 +3006,9 @@ function normalizeToolResultContent(content) {
   }
   return String(content);
 }
+function extractTextFromMessage(message) {
+  return message.content.filter((block) => block.type === "text").map((block) => block.text).join("");
+}
 
 // src/helpers.ts
 function wireAbortSignal(signal, onAbort) {
@@ -2735,15 +3024,88 @@ function wireAbortSignal(signal, onAbort) {
     return () => signal.removeEventListener("abort", listener);
   }
 }
+function getAbortError(signal) {
+  if (signal.reason instanceof Error) {
+    return signal.reason;
+  }
+  return new Error(
+    typeof signal.reason === "string" ? signal.reason : "Operation aborted"
+  );
+}
+function throwIfAborted(signal) {
+  if (signal?.aborted) {
+    throw getAbortError(signal);
+  }
+}
+async function* streamFromClient(client, sessionId, activeBridges, prompt, options) {
+  throwIfAborted(options?.abortSignal);
+  const startedAt = Date.now();
+  let resolveDone = () => {
+  };
+  const donePromise = new Promise((resolve2) => {
+    resolveDone = resolve2;
+  });
+  const bridge = new MessageBridge(resolveDone, {
+    includePartialMessages: options?.includePartialMessages,
+    sessionId,
+    startedAt,
+    outputFormat: options?.outputFormat
+  });
+  activeBridges.add(bridge);
+  const unsubscribe = client.onNotification(bridge.notificationHandler);
+  let resolveAbort = () => {
+  };
+  const abortPromise = new Promise((resolve2) => {
+    resolveAbort = resolve2;
+  });
+  const cleanupAbortSignal = wireAbortSignal(options?.abortSignal, () => {
+    bridge.signalDone();
+    resolveAbort();
+    void client.interruptSession().catch(() => {
+    });
+  });
+  try {
+    await Promise.race([
+      client.addUserMessage({
+        text: prompt,
+        images: options?.images,
+        files: options?.files,
+        outputFormat: options?.outputFormat
+      }),
+      donePromise,
+      abortPromise
+    ]);
+    throwIfAborted(options?.abortSignal);
+    for await (const msg of bridge.messages()) {
+      throwIfAborted(options?.abortSignal);
+      yield msg;
+    }
+    throwIfAborted(options?.abortSignal);
+  } finally {
+    cleanupAbortSignal();
+    unsubscribe();
+    activeBridges.delete(bridge);
+  }
+}
 function extractInnerNotification(notification) {
   const parsed = SessionNotificationSchema.safeParse(notification);
   return parsed.success ? parsed.data.params.notification : null;
 }
 var MessageBridge = class {
+  constructor(onDone, _options = {}) {
+    this._options = _options;
+    this._onDone = onDone;
+    this._stateTracker = new StreamStateTracker({
+      sessionId: _options.sessionId,
+      startedAt: _options.startedAt,
+      hasOutputFormat: _options.outputFormat !== void 0
+    });
+  }
   _queue = [];
+  _onDone;
   _resolveWaiting = null;
   _done = false;
-  _stateTracker = new StreamStateTracker();
+  _stateTracker;
   notificationHandler = (notification) => {
     const inner = extractInnerNotification(notification);
     if (!inner) return;
@@ -2752,10 +3114,14 @@ var MessageBridge = class {
     const messages = Array.isArray(converted) ? converted : [converted];
     for (const msg of messages) {
       const { message, additional } = this._stateTracker.processMessage(msg);
-      this._enqueue(message);
+      if (message && this._shouldYield(message)) {
+        this._enqueue(message);
+      }
       for (const extra of additional) {
-        this._enqueue(extra);
-        if (extra.type === DroidMessageType.TurnComplete) {
+        if (this._shouldYield(extra)) {
+          this._enqueue(extra);
+        }
+        if (extra.type === DroidMessageType.Result) {
           this._signalDone();
         }
       }
@@ -2769,7 +3135,7 @@ var MessageBridge = class {
       while (this._queue.length > 0) {
         const msg = this._queue.shift();
         yield msg;
-        if (msg.type === DroidMessageType.TurnComplete) {
+        if (msg.type === DroidMessageType.Result) {
           return;
         }
       }
@@ -2781,6 +3147,12 @@ var MessageBridge = class {
       });
     }
   }
+  _shouldYield(message) {
+    if (isInternalMessage(message)) {
+      return false;
+    }
+    return this._options.includePartialMessages ? true : isDefaultStreamMessage(message);
+  }
   _enqueue(msg) {
     this._queue.push(msg);
     if (this._resolveWaiting) {
@@ -2790,7 +3162,11 @@ var MessageBridge = class {
     }
   }
   _signalDone() {
+    const wasDone = this._done;
     this._done = true;
+    if (!wasDone) {
+      this._onDone?.();
+    }
     if (this._resolveWaiting) {
       const resolve2 = this._resolveWaiting;
       this._resolveWaiting = null;
@@ -2806,7 +3182,7 @@ async function createTransport(options) {
     execPath: options.execPath,
     execArgs: options.execArgs,
     cwd: options.cwd,
-    env: options.env
+    env: { ...options.env, FACTORY_API_KEY: options.apiKey }
   };
   const processTransport = new ProcessTransport(transportOptions);
   await processTransport.connect();
@@ -2864,6 +3240,9 @@ function buildInitParams(options) {
     ...options.disabledToolIds !== void 0 && {
       disabledToolIds: options.disabledToolIds
     },
+    ...options.sessionSource !== void 0 && {
+      sessionSource: options.sessionSource
+    },
     tags: [...options.tags ?? [], SDK_TAG]
   };
 }
@@ -2873,7 +3252,7 @@ import { once } from "events";
 import http from "http";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
-import { z as z8 } from "zod";
+import { z as z9 } from "zod";
 function tool(name, description, inputSchemaOrHandler, handler) {
   if (typeof inputSchemaOrHandler === "function") {
     const actualHandler = inputSchemaOrHandler;
@@ -2886,7 +3265,7 @@ function tool(name, description, inputSchemaOrHandler, handler) {
   if (!handler) {
     throw new Error(`Missing handler for SDK MCP tool "${name}"`);
   }
-  const inputObjectSchema = z8.object(inputSchemaOrHandler);
+  const inputObjectSchema = z9.object(inputSchemaOrHandler);
   return {
     name,
     description,
@@ -3063,312 +3442,44 @@ async function startSdkMcpServers(servers) {
   };
 }
 
-// src/query.ts
-var DroidQueryImpl = class {
-  _generator;
-  _getSessionId;
-  _getInitResult;
-  _getInitialized;
-  _interruptFn;
-  _abortFn;
-  constructor(generator, getSessionId, getInitResult, getInitialized, interruptFn, abortFn) {
-    this._generator = generator;
-    this._getSessionId = getSessionId;
-    this._getInitResult = getInitResult;
-    this._getInitialized = getInitialized;
-    this._interruptFn = interruptFn;
-    this._abortFn = abortFn;
+// src/session.ts
+var DroidSession = class {
+  _client;
+  _sessionId;
+  _initResult;
+  _closed = false;
+  _cleanupAbortSignal = null;
+  _cleanupCallbacks = [];
+  _activeBridges = /* @__PURE__ */ new Set();
+  /** @internal */
+  constructor(client, sessionId, initResult) {
+    this._client = client;
+    this._sessionId = sessionId;
+    this._initResult = initResult;
   }
   get sessionId() {
-    return this._getSessionId();
+    return this._sessionId;
   }
   get initResult() {
-    return this._getInitResult();
+    return this._initResult;
   }
-  get initialized() {
-    return this._getInitialized();
+  /** @internal */
+  setAbortSignalCleanup(cleanup) {
+    this._cleanupAbortSignal = cleanup;
   }
-  async interrupt() {
-    return this._interruptFn();
+  /** @internal */
+  addCleanup(cleanup) {
+    this._cleanupCallbacks.push(cleanup);
   }
-  abort() {
-    this._abortFn();
-  }
-  next(...args) {
-    return this._generator.next(...args);
-  }
-  return(value) {
-    return this._generator.return(value);
-  }
-  throw(e) {
-    return this._generator.throw(e);
-  }
-  [Symbol.asyncIterator]() {
-    return this;
-  }
-};
-function query(options) {
-  let transport = null;
-  let client = null;
-  let sessionId = null;
-  let initResult = null;
-  let aborted = false;
-  let initializationPromise = null;
-  let promptPromise = null;
-  let cleanupAbortSignal = () => {
-  };
-  let cleanupSdkMcpServers = null;
-  const bridge = new MessageBridge();
-  const closeActiveResources = async () => {
-    const closer = client ?? transport;
-    const cleanupMcp = cleanupSdkMcpServers;
-    client = null;
-    transport = null;
-    cleanupSdkMcpServers = null;
-    await Promise.all([closeQuietly(closer), cleanupMcp?.()]);
-  };
-  const ensureInitialized = () => {
-    if (initResult) {
-      return Promise.resolve(initResult);
-    }
-    if (initializationPromise) {
-      return initializationPromise;
-    }
-    initializationPromise = (async () => {
-      try {
-        if (aborted) {
-          throw new Error("Query aborted before initialization");
-        }
-        const configuredClient = await createConfiguredClient(options);
-        transport = configuredClient.transport;
-        client = configuredClient.client;
-        client.onNotification(bridge.notificationHandler);
-        if (aborted) {
-          throw new Error("Query aborted before initialization");
-        }
-        const sdkMcpServers = await startSdkMcpServers(options.mcpServers);
-        cleanupSdkMcpServers = sdkMcpServers.cleanup;
-        const result = await client.initializeSession(
-          buildInitParams({
-            ...options,
-            mcpServers: sdkMcpServers.mcpServers
-          })
-        );
-        sessionId = result.sessionId;
-        initResult = result;
-        return result;
-      } catch (error) {
-        await closeActiveResources();
-        throw error;
-      }
-    })();
-    return initializationPromise;
-  };
-  const ensurePromptSent = () => {
-    if (promptPromise) {
-      return promptPromise;
-    }
-    promptPromise = (async () => {
-      await ensureInitialized();
-      if (aborted || client === null) {
-        return;
-      }
-      await client.addUserMessage({ text: options.prompt });
-    })();
-    return promptPromise;
-  };
-  async function* generateMessages() {
-    if (aborted) return;
-    await ensurePromptSent();
-    if (aborted) return;
-    for await (const msg of bridge.messages()) {
-      if (aborted) return;
-      yield msg;
-    }
-  }
-  const generator = generateMessages();
-  async function* wrappedGenerator() {
-    try {
-      yield* generator;
-    } finally {
-      cleanupAbortSignal();
-      await closeActiveResources();
-    }
-  }
-  const droidQuery = new DroidQueryImpl(
-    wrappedGenerator(),
-    () => sessionId,
-    () => initResult,
-    () => ensureInitialized(),
-    async () => {
-      if (client && !aborted) {
-        await client.interruptSession();
-      }
-    },
-    () => {
-      aborted = true;
-      bridge.signalDone();
-      cleanupAbortSignal();
-      void closeActiveResources();
-    }
-  );
-  cleanupAbortSignal = wireAbortSignal(
-    options.abortSignal,
-    () => droidQuery.abort()
-  );
-  return droidQuery;
-}
-
-// src/session.ts
-function getAbortError(signal) {
-  if (signal.reason instanceof Error) {
-    return signal.reason;
-  }
-  return new Error(
-    typeof signal.reason === "string" ? signal.reason : "Operation aborted"
-  );
-}
-function throwIfAborted(signal) {
-  if (signal?.aborted) {
-    throw getAbortError(signal);
-  }
-}
-function parseJsonObject(text) {
-  try {
-    const parsed = JSON.parse(text);
-    const result = JsonObjectSchema.safeParse(parsed);
-    return result.success ? result.data : null;
-  } catch {
-    return null;
-  }
-}
-function extractAssistantText(message) {
-  if (message.type !== DroidMessageType.CreateMessage) {
-    return "";
-  }
-  if (message.role !== FactoryDroidMessageRole.Assistant) {
-    return "";
-  }
-  return message.content.filter((block) => block.type === "text").map((block) => block.text).join("");
-}
-var DroidSession = class {
-  _client;
-  _sessionId;
-  _initResult;
-  _closed = false;
-  _cleanupAbortSignal = null;
-  _cleanupCallbacks = [];
-  /** @internal */
-  constructor(client, sessionId, initResult) {
-    this._client = client;
-    this._sessionId = sessionId;
-    this._initResult = initResult;
-  }
-  get sessionId() {
-    return this._sessionId;
-  }
-  get initResult() {
-    return this._initResult;
-  }
-  /** @internal */
-  setAbortSignalCleanup(cleanup) {
-    this._cleanupAbortSignal = cleanup;
-  }
-  /** @internal */
-  addCleanup(cleanup) {
-    this._cleanupCallbacks.push(cleanup);
-  }
-  /** Yields {@link DroidMessage} events until `turn_complete`. */
-  async *stream(text, options) {
-    this._ensureNotClosed();
-    throwIfAborted(options?.abortSignal);
-    const bridge = new MessageBridge();
-    const unsubscribe = this._client.onNotification(bridge.notificationHandler);
-    let resolveAbort = () => {
-    };
-    const abortPromise = new Promise((resolve2) => {
-      resolveAbort = resolve2;
-    });
-    const cleanupAbortSignal = wireAbortSignal(options?.abortSignal, () => {
-      bridge.signalDone();
-      resolveAbort();
-      void this._client.interruptSession().catch(() => {
-      });
-    });
-    try {
-      await Promise.race([
-        this._client.addUserMessage({
-          text,
-          images: options?.images,
-          files: options?.files,
-          outputFormat: options?.outputFormat
-        }),
-        abortPromise
-      ]);
-      throwIfAborted(options?.abortSignal);
-      for await (const msg of bridge.messages()) {
-        throwIfAborted(options?.abortSignal);
-        yield msg;
-      }
-      throwIfAborted(options?.abortSignal);
-    } finally {
-      cleanupAbortSignal();
-      unsubscribe();
-    }
-  }
-  /** Consumes the stream and returns an aggregated {@link DroidResult}. */
-  async send(text, options) {
+  async *stream(prompt, options) {
     this._ensureNotClosed();
-    const messages = [];
-    let fullText = "";
-    let lastTokenUsage = null;
-    let firstError = null;
-    let structuredOutput = null;
-    let finalAssistantText = "";
-    let turnCount = 0;
-    const startedAt = Date.now();
-    for await (const msg of this.stream(text, options)) {
-      messages.push(msg);
-      if (msg.type === DroidMessageType.AssistantTextDelta) {
-        fullText += msg.text;
-      }
-      const assistantText = extractAssistantText(msg);
-      if (assistantText) {
-        finalAssistantText = assistantText;
-        if (options?.outputFormat && fullText.length === 0) {
-          fullText = assistantText;
-        }
-      }
-      if (msg.type === DroidMessageType.TokenUsageUpdate) {
-        lastTokenUsage = msg;
-      }
-      if (msg.type === DroidMessageType.Error && firstError === null) {
-        firstError = msg;
-      }
-      if (options?.outputFormat && structuredOutput === null) {
-        const textToParse = finalAssistantText || fullText;
-        if (textToParse) {
-          structuredOutput = parseJsonObject(textToParse);
-        }
-      }
-      if (msg.type === DroidMessageType.TurnComplete) {
-        turnCount++;
-        if (msg.tokenUsage) {
-          lastTokenUsage = msg.tokenUsage;
-        }
-      }
-    }
-    return {
-      sessionId: this._sessionId,
-      text: fullText,
-      messages,
-      tokenUsage: lastTokenUsage,
-      durationMs: Date.now() - startedAt,
-      turnCount,
-      error: firstError,
-      structuredOutput,
-      success: firstError === null
-    };
+    yield* streamFromClient(
+      this._client,
+      this._sessionId,
+      this._activeBridges,
+      prompt,
+      options
+    );
   }
   async interrupt() {
     this._ensureNotClosed();
@@ -3381,7 +3492,12 @@ var DroidSession = class {
     this._closed = true;
     this._cleanupAbortSignal?.();
     this._cleanupAbortSignal = null;
+    for (const bridge of this._activeBridges) {
+      bridge.signalDone();
+    }
     try {
+      await this._client.closeSession({ reason: "other" }).catch(() => {
+      });
       await this._client.close();
     } finally {
       const cleanups = this._cleanupCallbacks.splice(0);
@@ -3466,7 +3582,7 @@ var DroidSession = class {
     }
   }
 };
-async function createSession(options = {}) {
+async function createSession(options) {
   const { client } = await createConfiguredClient(options);
   let cleanupInitAbortSignal = options.abortSignal?.aborted ? () => {
   } : wireAbortSignal(options.abortSignal, () => {
@@ -3496,7 +3612,7 @@ async function createSession(options = {}) {
     throw error;
   }
 }
-async function resumeSession(sessionId, options = {}) {
+async function resumeSession(sessionId, options) {
   const { client } = await createConfiguredClient(options);
   let sdkMcpServers;
   try {
@@ -3520,10 +3636,18 @@ async function resumeSession(sessionId, options = {}) {
 }
 
 // src/run.ts
-async function run(text, options = {}) {
+async function run(prompt, options) {
   const session = await createSession(options);
   try {
-    return await session.send(text, options);
+    for await (const msg of session.stream(prompt, {
+      ...options,
+      includePartialMessages: false
+    })) {
+      if (msg.type === DroidMessageType.Result) {
+        return msg;
+      }
+    }
+    throw new Error("Stream completed without a result message");
   } finally {
     await session.close();
   }
@@ -3752,6 +3876,1431 @@ async function listSessions(options = {}) {
   }
   return sorted;
 }
+
+// src/api-types.ts
+import { z as z10 } from "zod";
+var EnvironmentVariableSchema = z10.object({
+  key: z10.string(),
+  value: z10.string()
+});
+var MachineTemplateBuildStatusSchema = z10.object({
+  status: z10.enum(["building", "success", "failed"]),
+  failureReason: z10.enum(["setup_script_error", "system_error"]).optional(),
+  buildStartedAt: z10.number().int().optional(),
+  builtAt: z10.number().int().optional(),
+  logs: z10.string().optional()
+});
+var MachineTemplateSchema = z10.object({
+  templateId: z10.string(),
+  repoUrl: z10.string(),
+  templateName: z10.string(),
+  defaultBranch: z10.string(),
+  createdBy: z10.string(),
+  createdAt: z10.number().int().optional(),
+  buildStatus: MachineTemplateBuildStatusSchema.optional(),
+  lastUpdatedAt: z10.number().int().nullable().optional(),
+  environmentVariables: z10.array(EnvironmentVariableSchema).optional(),
+  userEnvironmentVariablesByUser: z10.array(EnvironmentVariableSchema).optional(),
+  setupScript: z10.string().optional()
+});
+var MachineTemplateListResponseSchema = z10.object({
+  templates: z10.array(MachineTemplateSchema),
+  pagination: z10.object({
+    hasMore: z10.boolean(),
+    nextCursor: z10.string().nullable()
+  })
+});
+var ComputerSchema = z10.object({
+  id: z10.string(),
+  name: z10.string(),
+  hostname: z10.string().optional(),
+  providerType: z10.enum(["byom", "e2b"]),
+  status: z10.enum(["provisioning", "active", "error"]).optional(),
+  createdAt: z10.number().int(),
+  relayClientUrl: z10.string().optional(),
+  remoteUser: z10.string().optional()
+}).passthrough();
+var ComputerListResponseSchema = z10.object({
+  computers: z10.array(ComputerSchema)
+});
+var RefreshComputerResponseSchema = z10.object({
+  configured: z10.number().int()
+});
+var ComputerMetricSchema = z10.object({
+  timestamp: z10.string(),
+  cpuUsedPct: z10.number(),
+  cpuCount: z10.number(),
+  memUsed: z10.number(),
+  memTotal: z10.number(),
+  diskUsed: z10.number(),
+  diskTotal: z10.number()
+});
+var ComputerMetricsResponseSchema = z10.array(ComputerMetricSchema);
+var RemoteSessionSchema = z10.object({
+  sessionId: z10.string(),
+  title: z10.string().optional(),
+  status: z10.enum(["idle", "pending", "running"]),
+  messageCount: z10.number().int(),
+  createdAt: z10.number().int(),
+  updatedAt: z10.number().int(),
+  completedAt: z10.number().int().optional(),
+  computerId: z10.string().optional()
+});
+var RemoteSessionListResponseSchema = z10.object({
+  sessions: z10.array(RemoteSessionSchema),
+  pagination: z10.object({
+    hasMore: z10.boolean(),
+    nextCursor: z10.string().nullable()
+  })
+});
+
+// src/api.ts
+var DEFAULT_BASE_URL = "https://api.factory.ai";
+function extractErrorMessage(body, status) {
+  if (isRecord(body) && typeof body["error"] === "string") {
+    return body["error"];
+  }
+  return `Factory API error (status ${status})`;
+}
+async function factoryFetch(path3, apiKey, baseUrl, options) {
+  const url = `${baseUrl ?? DEFAULT_BASE_URL}${path3}`;
+  const method = options?.method ?? "GET";
+  const headers = {
+    Authorization: `Bearer ${apiKey}`,
+    Accept: "application/json"
+  };
+  if (options?.body) {
+    headers["Content-Type"] = "application/json";
+  }
+  let response;
+  try {
+    response = await fetch(url, {
+      method,
+      headers,
+      ...options?.body ? { body: JSON.stringify(options.body) } : {}
+    });
+  } catch (error) {
+    throw new ConnectionError(
+      `Failed to reach Factory API: ${error instanceof Error ? error.message : String(error)}`
+    );
+  }
+  if (response.status === 204) {
+    return void 0;
+  }
+  let body;
+  try {
+    body = await response.json();
+  } catch {
+    throw new ProtocolError(
+      `Factory API returned non-JSON response (status ${response.status})`
+    );
+  }
+  if (!response.ok) {
+    const message = extractErrorMessage(body, response.status);
+    if (response.status === 401 || response.status === 403) {
+      throw new ProtocolError(`Invalid or expired API key: ${message}`, {
+        code: response.status
+      });
+    }
+    throw new ProtocolError(message, { code: response.status });
+  }
+  return body;
+}
+function parseResponse(body, schema) {
+  const parsed = schema.safeParse(body);
+  if (!parsed.success) {
+    throw new ProtocolError(
+      `Unexpected response format from Factory API: ${parsed.error.message}`
+    );
+  }
+  return parsed.data;
+}
+async function listMachineTemplates(options) {
+  const params = new URLSearchParams();
+  if (options.limit != null) {
+    params.set("limit", String(options.limit));
+  }
+  if (options.cursor != null) {
+    params.set("cursor", options.cursor);
+  }
+  const query = params.toString();
+  const path3 = `/api/v0/machines/templates${query ? `?${query}` : ""}`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, MachineTemplateListResponseSchema);
+}
+async function getMachineTemplate(options) {
+  const path3 = `/api/v0/machines/templates/${encodeURIComponent(options.templateId)}`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, MachineTemplateSchema);
+}
+async function listComputers(options) {
+  const path3 = "/api/v0/computers";
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, ComputerListResponseSchema);
+}
+async function getComputer(options) {
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, ComputerSchema);
+}
+async function createComputer(options) {
+  const path3 = "/api/v0/computers";
+  const reqBody = {
+    name: options.name,
+    remoteUser: options.remoteUser
+  };
+  if (options.provider != null) reqBody["provider"] = options.provider;
+  if (options.hostId != null) reqBody["hostId"] = options.hostId;
+  if (options.repos != null) reqBody["repos"] = options.repos;
+  if (options.autoInstallDeps != null)
+    reqBody["autoInstallDeps"] = options.autoInstallDeps;
+  if (options.serviceAccountId != null)
+    reqBody["serviceAccountId"] = options.serviceAccountId;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl, {
+    method: "POST",
+    body: reqBody
+  });
+  return parseResponse(body, ComputerSchema);
+}
+async function getComputerByName(options) {
+  const path3 = `/api/v0/computers/name/${encodeURIComponent(options.name)}`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, ComputerSchema);
+}
+async function updateComputer(options) {
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}`;
+  const reqBody = {};
+  if (options.name != null) reqBody["name"] = options.name;
+  if (options.remoteUser != null) reqBody["remoteUser"] = options.remoteUser;
+  if (options.hostId != null) reqBody["hostId"] = options.hostId;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl, {
+    method: "PATCH",
+    body: reqBody
+  });
+  return parseResponse(body, ComputerSchema);
+}
+async function deleteComputer(options) {
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}`;
+  await factoryFetch(path3, options.apiKey, options.baseUrl, {
+    method: "DELETE"
+  });
+}
+async function restartComputer(options) {
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}/restart`;
+  await factoryFetch(path3, options.apiKey, options.baseUrl, {
+    method: "POST"
+  });
+}
+async function refreshComputer(options) {
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}/refresh`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl, {
+    method: "POST"
+  });
+  return parseResponse(body, RefreshComputerResponseSchema);
+}
+async function getComputerMetrics(options) {
+  const params = new URLSearchParams();
+  if (options.start != null) {
+    params.set("start", options.start);
+  }
+  const query = params.toString();
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}/metrics${query ? `?${query}` : ""}`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, ComputerMetricsResponseSchema);
+}
+async function retryInstallDeps(options) {
+  const path3 = `/api/v0/computers/${encodeURIComponent(options.computerId)}/install-deps`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl, {
+    method: "POST"
+  });
+  return parseResponse(body, ComputerSchema);
+}
+async function listRemoteSessions(options) {
+  const params = new URLSearchParams();
+  if (options.computerId != null) {
+    params.set("computerId", options.computerId);
+  }
+  if (options.limit != null) {
+    params.set("limit", String(options.limit));
+  }
+  if (options.cursor != null) {
+    params.set("cursor", options.cursor);
+  }
+  const query = params.toString();
+  const path3 = `/api/v0/sessions${query ? `?${query}` : ""}`;
+  const body = await factoryFetch(path3, options.apiKey, options.baseUrl);
+  return parseResponse(body, RemoteSessionListResponseSchema);
+}
+
+// src/daemon/client.ts
+var DAEMON_SERVER_REQUEST_METHODS = {
+  "daemon.request_permission": "permission" /* Permission */,
+  "daemon.ask_user": "askUser" /* AskUser */
+};
+var DaemonClient = class {
+  _engine;
+  _apiKey;
+  _sessionId = null;
+  _closed = false;
+  _notificationListeners = [];
+  _permissionHandler = null;
+  _askUserHandler = null;
+  constructor(options) {
+    this._apiKey = options.apiKey;
+    this._engine = new ProtocolEngine({
+      transport: options.transport,
+      serverRequestMethodMap: DAEMON_SERVER_REQUEST_METHODS
+    });
+    this._engine.onNotification((notification) => {
+      const normalized = notification["method"] === "daemon.session_notification" ? { ...notification, method: "droid.session_notification" } : notification;
+      dispatchNotification(normalized, [...this._notificationListeners]);
+    });
+    this._engine.setPermissionHandler(
+      (params) => this._dispatchPermissionRequest(params)
+    );
+    this._engine.setAskUserHandler(
+      (params) => this._dispatchAskUserRequest(params)
+    );
+  }
+  get sessionId() {
+    return this._sessionId;
+  }
+  get isConnected() {
+    return !this._closed && this._engine.isHealthy;
+  }
+  async initializeSession(params) {
+    this._ensureNotClosed();
+    const result = await this._rpc(
+      "daemon.initialize_session" /* INITIALIZE_SESSION */,
+      { ...params, token: this._apiKey },
+      InitializeSessionResultSchema,
+      SESSION_INIT_TIMEOUT
+    );
+    this._sessionId = result.sessionId;
+    return result;
+  }
+  async loadSession(params) {
+    this._ensureNotClosed();
+    const result = await this._rpc(
+      "daemon.load_session" /* LOAD_SESSION */,
+      { ...params, token: this._apiKey },
+      LoadSessionResultSchema,
+      SESSION_INIT_TIMEOUT
+    );
+    this._sessionId = params.sessionId;
+    return result;
+  }
+  async addUserMessage(params) {
+    return this._sessionRpc(
+      "daemon.add_user_message" /* ADD_USER_MESSAGE */,
+      params,
+      AddUserMessageResultSchema
+    );
+  }
+  async interruptSession() {
+    return this._sessionRpc(
+      "daemon.interrupt_session" /* INTERRUPT_SESSION */,
+      {},
+      InterruptSessionResultSchema
+    );
+  }
+  async closeSession(params = {}) {
+    return this._sessionRpc(
+      "daemon.close_session" /* CLOSE_SESSION */,
+      params,
+      CloseSessionResultSchema
+    );
+  }
+  async updateSessionSettings(params) {
+    return this._sessionRpc(
+      "daemon.update_session_settings" /* UPDATE_SESSION_SETTINGS */,
+      params,
+      UpdateSessionSettingsResultSchema
+    );
+  }
+  async compactSession(params = {}) {
+    return this._sessionRpc(
+      "daemon.compact_session" /* COMPACT_SESSION */,
+      params,
+      CompactSessionResultSchema,
+      COMPACTION_TIMEOUT
+    );
+  }
+  async forkSession() {
+    return this._sessionRpc(
+      "daemon.fork_session" /* FORK_SESSION */,
+      {},
+      ForkSessionResultSchema
+    );
+  }
+  async getContextBreakdown() {
+    return this._sessionRpc(
+      "daemon.get_context_breakdown" /* GET_CONTEXT_STATS */,
+      {},
+      ContextBreakdownResultSchema
+    );
+  }
+  async renameSession(params) {
+    return this._sessionRpc(
+      "daemon.rename_session" /* RENAME_SESSION */,
+      params,
+      RenameSessionResultSchema
+    );
+  }
+  async getRewindInfo(params) {
+    return this._sessionRpc(
+      "daemon.get_rewind_info" /* GET_REWIND_INFO */,
+      params,
+      GetRewindInfoResultSchema,
+      REWIND_TIMEOUT
+    );
+  }
+  async executeRewind(params) {
+    return this._sessionRpc(
+      "daemon.execute_rewind" /* EXECUTE_REWIND */,
+      params,
+      ExecuteRewindResultSchema,
+      REWIND_TIMEOUT
+    );
+  }
+  async addMcpServer(params) {
+    return this._sessionRpc(
+      "daemon.add_mcp_server" /* ADD_MCP_SERVER */,
+      params,
+      AddMcpServerResultSchema
+    );
+  }
+  async removeMcpServer(params) {
+    return this._sessionRpc(
+      "daemon.remove_mcp_server" /* REMOVE_MCP_SERVER */,
+      params,
+      RemoveMcpServerResultSchema
+    );
+  }
+  async toggleMcpServer(params) {
+    return this._sessionRpc(
+      "daemon.toggle_mcp_server" /* TOGGLE_MCP_SERVER */,
+      params,
+      ToggleMcpServerResultSchema
+    );
+  }
+  async listMcpServers() {
+    return this._sessionRpc(
+      "daemon.list_mcp_servers" /* LIST_MCP_SERVERS */,
+      {},
+      ListMcpServersResultSchema
+    );
+  }
+  async listMcpTools() {
+    return this._sessionRpc(
+      "daemon.list_mcp_tools" /* LIST_MCP_TOOLS */,
+      {},
+      ListMcpToolsResultSchema
+    );
+  }
+  async authenticateMcpServer(params) {
+    return this._sessionRpc(
+      "daemon.authenticate_mcp_server" /* AUTHENTICATE_MCP_SERVER */,
+      params,
+      AuthenticateMcpServerResultSchema
+    );
+  }
+  async listSkills() {
+    return this._sessionRpc(
+      "daemon.list_skills" /* LIST_SKILLS */,
+      {},
+      ListSkillsResultSchema
+    );
+  }
+  onNotification(callback, filter) {
+    const entry = { callback, filter };
+    this._notificationListeners.push(entry);
+    let unsubscribed = false;
+    return () => {
+      if (!unsubscribed) {
+        unsubscribed = true;
+        const idx = this._notificationListeners.indexOf(entry);
+        if (idx !== -1) {
+          this._notificationListeners.splice(idx, 1);
+        }
+      }
+    };
+  }
+  setPermissionHandler(handler) {
+    this._permissionHandler = handler;
+  }
+  setAskUserHandler(handler) {
+    this._askUserHandler = handler;
+  }
+  async close() {
+    if (this._closed) {
+      return;
+    }
+    this._closed = true;
+    this._notificationListeners.length = 0;
+    this._permissionHandler = null;
+    this._askUserHandler = null;
+    await this._engine.close();
+  }
+  async _rpc(method, params, schema, timeout) {
+    const raw = await this._engine.sendRequest(method, params, timeout);
+    return schema.parse(raw);
+  }
+  async _sessionRpc(method, params, schema, timeout) {
+    this._ensureNotClosed();
+    this._ensureSession();
+    return this._rpc(
+      method,
+      { sessionId: this._sessionId, ...params },
+      schema,
+      timeout
+    );
+  }
+  _dispatchPermissionRequest(params) {
+    const handler = this._permissionHandler;
+    if (handler == null) {
+      return "cancel" /* Cancel */;
+    }
+    return handler(params);
+  }
+  _dispatchAskUserRequest(params) {
+    const handler = this._askUserHandler;
+    if (handler == null) {
+      return { cancelled: true, answers: [] };
+    }
+    return handler(params);
+  }
+  _ensureNotClosed() {
+    if (this._closed) {
+      throw new ConnectionError(
+        "Daemon client has been closed. Create a new DaemonClient instance."
+      );
+    }
+  }
+  _ensureSession() {
+    if (this._sessionId == null) {
+      throw new SessionError(
+        "No active session. Call initializeSession or loadSession first."
+      );
+    }
+  }
+};
+
+// src/daemon/local.ts
+import { spawn as spawn2 } from "child_process";
+import * as fs2 from "fs";
+import * as net from "net";
+import * as os2 from "os";
+import * as path2 from "path";
+var SOCKET_TIMEOUT_MS = 2e3;
+var STARTUP_POLL_INTERVAL_MS = 250;
+var STARTUP_TIMEOUT_MS = 3e4;
+var MAX_STARTUP_ATTEMPTS = 3;
+var FACTORY_DIR = ".factory";
+var DEFAULT_PORT = 37643;
+var DAEMON_PORT_FILE = "daemon.port";
+var DEFAULT_HOST = "127.0.0.1";
+var spawnedDaemonProcess = null;
+var daemonTarget = null;
+var daemonEnsureInflight = null;
+function getFactoryHome() {
+  return process.env.FACTORY_HOME_OVERRIDE || os2.homedir();
+}
+function getFactoryDirName() {
+  return FACTORY_DIR;
+}
+function getFactoryDir() {
+  return path2.join(getFactoryHome(), getFactoryDirName());
+}
+function getSdkDir() {
+  return path2.join(getFactoryDir(), "sdk");
+}
+function getDefaultDaemonPort() {
+  return DEFAULT_PORT;
+}
+function resolveExecPath() {
+  const override = process.env.FACTORY_DROID_BINARY?.trim();
+  if (override && override.length > 0) {
+    if (override.includes("/") || override.includes("\\")) {
+      try {
+        fs2.accessSync(override, fs2.constants.X_OK);
+        return override;
+      } catch {
+      }
+    } else {
+      return override;
+    }
+  }
+  return "droid";
+}
+async function allocatePort() {
+  return new Promise((resolve2, reject) => {
+    const server = net.createServer();
+    server.once("error", reject);
+    server.listen(0, DEFAULT_HOST, () => {
+      const address = server.address();
+      if (!address || typeof address === "string") {
+        server.close();
+        reject(new Error("Failed to determine dynamic port"));
+        return;
+      }
+      const port = address.port;
+      server.close((err) => {
+        if (err) reject(err);
+        else resolve2(port);
+      });
+    });
+  });
+}
+async function isDaemonReachable(port) {
+  return new Promise((resolve2) => {
+    const socket = net.createConnection({ host: DEFAULT_HOST, port });
+    const cleanup = () => {
+      socket.removeAllListeners();
+      socket.destroy();
+    };
+    const fail = () => {
+      cleanup();
+      resolve2(false);
+    };
+    socket.setTimeout(SOCKET_TIMEOUT_MS, fail);
+    socket.once("error", fail);
+    socket.once("connect", () => {
+      cleanup();
+      resolve2(true);
+    });
+  });
+}
+async function waitForDaemonReady(child, port, timeoutMs = STARTUP_TIMEOUT_MS) {
+  return new Promise((resolve2) => {
+    let settled = false;
+    let exited = false;
+    const deadline = Date.now() + timeoutMs;
+    const settle = (result) => {
+      if (settled) return;
+      settled = true;
+      child.removeListener("exit", onExit);
+      child.removeListener("error", onExit);
+      resolve2(result);
+    };
+    const onExit = () => {
+      exited = true;
+      settle("exited");
+    };
+    child.once("exit", onExit);
+    child.once("error", onExit);
+    const poll = async () => {
+      while (!exited && !settled) {
+        if (await isDaemonReachable(port)) {
+          settle("ready");
+          return;
+        }
+        if (Date.now() >= deadline) {
+          settle("timeout");
+          return;
+        }
+        await new Promise((r) => setTimeout(r, STARTUP_POLL_INTERVAL_MS));
+      }
+    };
+    void poll();
+  });
+}
+function readPortFile() {
+  try {
+    const portPath = path2.join(getSdkDir(), DAEMON_PORT_FILE);
+    const content = fs2.readFileSync(portPath, "utf-8").trim();
+    const port = parseInt(content, 10);
+    if (Number.isFinite(port) && port > 0 && port < 65536) {
+      return port;
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
+function writePortFile(port) {
+  try {
+    const sdkDir = getSdkDir();
+    fs2.mkdirSync(sdkDir, { recursive: true });
+    fs2.writeFileSync(path2.join(sdkDir, DAEMON_PORT_FILE), String(port), {
+      mode: 384
+    });
+  } catch {
+  }
+}
+function cacheTarget(port) {
+  daemonTarget = { host: DEFAULT_HOST, port };
+  return { port };
+}
+function clearSpawnedDaemonProcess(child) {
+  if (spawnedDaemonProcess === child) {
+    spawnedDaemonProcess = null;
+  }
+}
+async function spawnDaemon(execPath, port) {
+  const args = ["daemon", "--host", DEFAULT_HOST, "--port", String(port)];
+  let stderrFd;
+  try {
+    const logsDir = path2.join(getSdkDir(), "logs");
+    fs2.mkdirSync(logsDir, { recursive: true });
+    stderrFd = fs2.openSync(path2.join(logsDir, "daemon-stderr.log"), "a");
+  } catch {
+  }
+  const child = spawn2(execPath, args, {
+    detached: true,
+    stdio: ["ignore", "ignore", stderrFd ?? "ignore"],
+    cwd: os2.homedir(),
+    env: { ...process.env }
+  });
+  if (stderrFd !== void 0) {
+    try {
+      fs2.closeSync(stderrFd);
+    } catch {
+    }
+  }
+  spawnedDaemonProcess = child;
+  child.unref();
+  child.once("exit", () => {
+    clearSpawnedDaemonProcess(child);
+  });
+  child.on("error", () => {
+    clearSpawnedDaemonProcess(child);
+  });
+  const result = await waitForDaemonReady(child, port);
+  if (result !== "ready") {
+    clearSpawnedDaemonProcess(child);
+    if (result === "timeout") {
+      try {
+        child.kill("SIGTERM");
+      } catch {
+      }
+    }
+  }
+  return result;
+}
+async function _ensureLocalDaemon() {
+  const execPath = resolveExecPath();
+  const wellKnownPort = getDefaultDaemonPort();
+  if (await isDaemonReachable(wellKnownPort)) {
+    return cacheTarget(wellKnownPort);
+  }
+  const portFilePort = readPortFile();
+  if (portFilePort !== null && portFilePort !== wellKnownPort) {
+    if (await isDaemonReachable(portFilePort)) {
+      return cacheTarget(portFilePort);
+    }
+  }
+  const firstResult = await spawnDaemon(execPath, wellKnownPort);
+  if (firstResult === "ready") {
+    writePortFile(wellKnownPort);
+    return cacheTarget(wellKnownPort);
+  }
+  for (let attempt = 2; attempt <= MAX_STARTUP_ATTEMPTS; attempt++) {
+    const port = await allocatePort();
+    const result = await spawnDaemon(execPath, port);
+    if (result === "ready") {
+      writePortFile(port);
+      return cacheTarget(port);
+    }
+  }
+  throw new ConnectionError(
+    `Failed to start local droid daemon after ${MAX_STARTUP_ATTEMPTS} attempts. Ensure the \`droid\` CLI is installed and \`droid auth login\` has been run.`
+  );
+}
+async function ensureLocalDaemon() {
+  if (daemonTarget) {
+    if (await isDaemonReachable(daemonTarget.port)) {
+      return { port: daemonTarget.port };
+    }
+    daemonTarget = null;
+  }
+  if (!daemonEnsureInflight) {
+    daemonEnsureInflight = _ensureLocalDaemon().finally(() => {
+      daemonEnsureInflight = null;
+    });
+  }
+  return daemonEnsureInflight;
+}
+
+// src/daemon/session.ts
+var DaemonSession = class {
+  _client;
+  _sessionId;
+  _closed = false;
+  _activeBridges = /* @__PURE__ */ new Set();
+  _cleanupCallbacks = [];
+  /** @internal */
+  constructor(client, sessionId) {
+    this._client = client;
+    this._sessionId = sessionId;
+  }
+  /** @internal */
+  addCleanup(cleanup) {
+    this._cleanupCallbacks.push(cleanup);
+  }
+  get sessionId() {
+    return this._sessionId;
+  }
+  async *stream(prompt, options) {
+    this._ensureNotClosed();
+    yield* streamFromClient(
+      this._client,
+      this._sessionId,
+      this._activeBridges,
+      prompt,
+      options
+    );
+  }
+  async send(prompt, options) {
+    this._ensureNotClosed();
+    await this._client.addUserMessage({
+      text: prompt,
+      images: options?.images,
+      files: options?.files,
+      outputFormat: options?.outputFormat
+    });
+  }
+  async interrupt() {
+    this._ensureNotClosed();
+    await this._client.interruptSession();
+  }
+  async close() {
+    if (this._closed) {
+      return;
+    }
+    this._closed = true;
+    for (const bridge of this._activeBridges) {
+      bridge.signalDone();
+    }
+    try {
+      await this._client.closeSession({ reason: "other" }).catch(() => {
+      });
+    } finally {
+      await this._client.close();
+      for (const cleanup of this._cleanupCallbacks) {
+        try {
+          await cleanup();
+        } catch {
+        }
+      }
+      this._cleanupCallbacks.length = 0;
+    }
+  }
+  async updateSettings(params) {
+    this._ensureNotClosed();
+    return this._client.updateSessionSettings(params);
+  }
+  async enterSpecMode(params = {}) {
+    this._ensureNotClosed();
+    return this._client.updateSessionSettings({
+      interactionMode: "spec" /* Spec */,
+      ...params
+    });
+  }
+  async compactSession(params) {
+    this._ensureNotClosed();
+    return this._client.compactSession(params ?? {});
+  }
+  async forkSession() {
+    this._ensureNotClosed();
+    return this._client.forkSession();
+  }
+  async getContextBreakdown() {
+    this._ensureNotClosed();
+    return this._client.getContextBreakdown();
+  }
+  async renameSession(params) {
+    this._ensureNotClosed();
+    return this._client.renameSession(params);
+  }
+  async getRewindInfo(params) {
+    this._ensureNotClosed();
+    return this._client.getRewindInfo(params);
+  }
+  async executeRewind(params) {
+    this._ensureNotClosed();
+    return this._client.executeRewind(params);
+  }
+  async addMcpServer(params) {
+    this._ensureNotClosed();
+    return this._client.addMcpServer(params);
+  }
+  async removeMcpServer(params) {
+    this._ensureNotClosed();
+    return this._client.removeMcpServer(params);
+  }
+  async toggleMcpServer(params) {
+    this._ensureNotClosed();
+    return this._client.toggleMcpServer(params);
+  }
+  async listMcpServers() {
+    this._ensureNotClosed();
+    return this._client.listMcpServers();
+  }
+  async listMcpTools() {
+    this._ensureNotClosed();
+    return this._client.listMcpTools();
+  }
+  async authenticateMcpServer(params) {
+    this._ensureNotClosed();
+    return this._client.authenticateMcpServer(params);
+  }
+  async listSkills() {
+    this._ensureNotClosed();
+    return this._client.listSkills();
+  }
+  onNotification(callback, filter) {
+    return this._client.onNotification(callback, filter);
+  }
+  _ensureNotClosed() {
+    if (this._closed) {
+      throw new ConnectionError(
+        "Daemon session has been closed. Create a new session to continue."
+      );
+    }
+  }
+};
+
+// src/daemon/transport.ts
+import WebSocket from "ws";
+
+// src/daemon/types.ts
+var MachineType = /* @__PURE__ */ ((MachineType2) => {
+  MachineType2["Local"] = "local";
+  MachineType2["Computer"] = "computer";
+  return MachineType2;
+})(MachineType || {});
+var DEFAULT_DAEMON_PORT = 37643;
+var DEFAULT_RELAY_BASE_URL = "wss://relay.factory.ai";
+var DEFAULT_WS_CONFIG = {
+  maxConnectRetries: 5,
+  initialRetryDelayMs: 500,
+  maxRetryDelayMs: 5e3,
+  connectionTimeoutMs: 5e3
+};
+var COMPUTER_WS_CONFIG = {
+  connectionTimeoutMs: 45e3,
+  maxConnectRetries: 10,
+  initialRetryDelayMs: 2e3,
+  maxRetryDelayMs: 1e4
+};
+
+// src/daemon/transport.ts
+var WebSocketTransport = class {
+  maxConnectRetries;
+  initialRetryDelayMs;
+  maxRetryDelayMs;
+  connectionTimeoutMs;
+  ws = null;
+  messageHandler = null;
+  errorHandler = null;
+  _isConnected = false;
+  isClosing = false;
+  constructor(options = {}) {
+    this.maxConnectRetries = options.maxConnectRetries ?? DEFAULT_WS_CONFIG.maxConnectRetries;
+    this.initialRetryDelayMs = options.initialRetryDelayMs ?? DEFAULT_WS_CONFIG.initialRetryDelayMs;
+    this.maxRetryDelayMs = options.maxRetryDelayMs ?? DEFAULT_WS_CONFIG.maxRetryDelayMs;
+    this.connectionTimeoutMs = options.connectionTimeoutMs ?? DEFAULT_WS_CONFIG.connectionTimeoutMs;
+  }
+  get isConnected() {
+    return this._isConnected;
+  }
+  async connect(url) {
+    if (!url) {
+      throw new ConnectionError("WebSocket URL is required");
+    }
+    if (this._isConnected) {
+      throw new ConnectionError("Transport already connected");
+    }
+    this.isClosing = false;
+    let lastError;
+    for (let attempt = 0; attempt <= this.maxConnectRetries; attempt++) {
+      try {
+        await this._doConnect(url);
+        return;
+      } catch (error) {
+        lastError = error instanceof Error ? error : new Error(String(error));
+        if (attempt < this.maxConnectRetries) {
+          const delay = Math.min(
+            this.initialRetryDelayMs * 2 ** attempt,
+            this.maxRetryDelayMs
+          );
+          await new Promise((resolve2) => setTimeout(resolve2, delay));
+        }
+      }
+    }
+    throw lastError;
+  }
+  send(message) {
+    if (!this._isConnected || !this.ws) {
+      throw new ConnectionError("WebSocket transport not connected");
+    }
+    try {
+      this.ws.send(JSON.stringify(message));
+    } catch (error) {
+      const msg = error instanceof Error ? error.message : String(error);
+      throw new ConnectionError(`Failed to send WebSocket message: ${msg}`);
+    }
+  }
+  onMessage(callback) {
+    this.messageHandler = callback;
+  }
+  onError(callback) {
+    this.errorHandler = callback;
+  }
+  async close() {
+    if (this.isClosing) {
+      return;
+    }
+    this.isClosing = true;
+    this._isConnected = false;
+    const ws = this.ws;
+    this.ws = null;
+    if (ws && ws.readyState === WebSocket.OPEN) {
+      await new Promise((resolve2) => {
+        ws.once("close", () => resolve2());
+        ws.close(1e3, "Client disconnect");
+        setTimeout(() => resolve2(), 5e3);
+      });
+    }
+    this.isClosing = false;
+  }
+  async _doConnect(url) {
+    return new Promise((resolve2, reject) => {
+      let settled = false;
+      let timeoutId = null;
+      const cleanup = () => {
+        if (timeoutId) {
+          clearTimeout(timeoutId);
+          timeoutId = null;
+        }
+      };
+      const ws = new WebSocket(url);
+      const handleOpen = () => {
+        if (settled) return;
+        settled = true;
+        cleanup();
+        this.ws = ws;
+        this._isConnected = true;
+        this._setupOngoingHandlers(ws);
+        resolve2();
+      };
+      const handleError = (error) => {
+        if (settled) return;
+        settled = true;
+        cleanup();
+        try {
+          ws.close();
+        } catch {
+        }
+        reject(
+          new ConnectionError(`WebSocket connection failed: ${error.message}`)
+        );
+      };
+      const handleClose = (code, reason) => {
+        if (settled) return;
+        settled = true;
+        cleanup();
+        reject(
+          new ConnectionError(
+            `WebSocket closed before opening (code: ${code}, reason: ${reason.toString()})`
+          )
+        );
+      };
+      ws.once("open", handleOpen);
+      ws.once("error", handleError);
+      ws.once("close", handleClose);
+      timeoutId = setTimeout(() => {
+        if (settled) return;
+        settled = true;
+        try {
+          ws.close();
+        } catch {
+        }
+        reject(
+          new ConnectionError(
+            `WebSocket connection timeout after ${this.connectionTimeoutMs}ms: ${url}`
+          )
+        );
+      }, this.connectionTimeoutMs);
+    });
+  }
+  _setupOngoingHandlers(ws) {
+    ws.on("message", (data) => {
+      if (!this.messageHandler) return;
+      const text = typeof data === "string" ? data : data.toString();
+      try {
+        const parsed = JSON.parse(text);
+        if (isRecord(parsed)) {
+          this.messageHandler(parsed);
+        }
+      } catch {
+      }
+    });
+    ws.on("error", (error) => {
+      if (!this.isClosing && this.errorHandler) {
+        this.errorHandler(
+          new ConnectionError(`WebSocket error: ${error.message}`)
+        );
+      }
+    });
+    ws.on("close", (code, reason) => {
+      this._isConnected = false;
+      this.ws = null;
+      if (!this.isClosing && this.errorHandler) {
+        this.errorHandler(
+          new ConnectionError(
+            `WebSocket closed (code: ${code}, reason: ${reason.toString()})`
+          )
+        );
+      }
+    });
+  }
+};
+
+// src/daemon/connection.ts
+var DAEMON_AUTHENTICATE_METHOD = "daemon.authenticate";
+var DAEMON_AUTHENTICATE_TIMEOUT = 3e4;
+var SDK_CALLER = "droid-sdk";
+function resolveWebSocketUrl(options) {
+  if (options.url) {
+    return options.url;
+  }
+  const machine = options.machine;
+  if (!machine || machine.type === "local" /* Local */) {
+    const port = options._localPort ?? options.daemonPort ?? DEFAULT_DAEMON_PORT;
+    return `ws://127.0.0.1:${port}`;
+  }
+  if (machine.type === "computer" /* Computer */) {
+    const relayBase = options.relayBaseUrl ?? DEFAULT_RELAY_BASE_URL;
+    return `${relayBase}/v0/computer/${machine.computerId}/client`;
+  }
+  const _exhaustive = machine;
+  throw new ConnectionError(`Unsupported machine type: ${String(_exhaustive)}`);
+}
+function getWebSocketConfig(machine) {
+  if (machine?.type === "computer" /* Computer */) {
+    return { ...DEFAULT_WS_CONFIG, ...COMPUTER_WS_CONFIG };
+  }
+  if (!machine || machine.type === "local" /* Local */) {
+    return { ...DEFAULT_WS_CONFIG, connectionTimeoutMs: 1e4 };
+  }
+  return { ...DEFAULT_WS_CONFIG };
+}
+var SharedTransportMultiplexer = class {
+  constructor(_inner) {
+    this._inner = _inner;
+    this._inner.onMessage((message) => {
+      this._routeMessage(message);
+    });
+    this._inner.onError((error) => {
+      for (const view of this._views) {
+        if (view.errorCallback) {
+          try {
+            view.errorCallback(error);
+          } catch {
+          }
+        }
+      }
+    });
+  }
+  _views = /* @__PURE__ */ new Set();
+  createView() {
+    const inner = this._inner;
+    const viewState = {
+      messageCallback: null,
+      errorCallback: null,
+      pendingRequestIds: /* @__PURE__ */ new Set(),
+      sessionId: null
+    };
+    this._views.add(viewState);
+    const view = {
+      send: (message) => {
+        if (message["type"] === "request" && typeof message["id"] === "string") {
+          viewState.pendingRequestIds.add(message["id"]);
+        }
+        inner.send(message);
+      },
+      onMessage: (callback) => {
+        viewState.messageCallback = callback;
+      },
+      onError: (callback) => {
+        viewState.errorCallback = callback;
+      },
+      close: async () => {
+        viewState.messageCallback = null;
+        viewState.errorCallback = null;
+        viewState.pendingRequestIds.clear();
+        this._views.delete(viewState);
+      },
+      get isConnected() {
+        return inner.isConnected;
+      }
+    };
+    return view;
+  }
+  _routeMessage(message) {
+    const messageType = message["type"];
+    const messageId = message["id"];
+    if (messageType === "response" && typeof messageId === "string") {
+      for (const view of this._views) {
+        if (view.pendingRequestIds.has(messageId)) {
+          view.pendingRequestIds.delete(messageId);
+          const result = message["result"];
+          if (isRecord(result) && typeof result["sessionId"] === "string") {
+            view.sessionId = result["sessionId"];
+          }
+          if (view.messageCallback) {
+            try {
+              view.messageCallback(message);
+            } catch {
+            }
+          }
+          return;
+        }
+      }
+      return;
+    }
+    if (messageType === "notification") {
+      const params = message["params"];
+      const sessionId = isRecord(params) && typeof params["sessionId"] === "string" ? params["sessionId"] : null;
+      if (sessionId) {
+        for (const view of this._views) {
+          if (view.sessionId === sessionId && view.messageCallback) {
+            try {
+              view.messageCallback(message);
+            } catch {
+            }
+            return;
+          }
+        }
+      }
+      for (const view of this._views) {
+        if (view.messageCallback) {
+          try {
+            view.messageCallback(message);
+          } catch {
+          }
+        }
+      }
+      return;
+    }
+    if (messageType === "request") {
+      const params = message["params"];
+      const sessionId = isRecord(params) && typeof params["sessionId"] === "string" ? params["sessionId"] : null;
+      if (sessionId) {
+        for (const view of this._views) {
+          if (view.sessionId === sessionId && view.messageCallback) {
+            try {
+              view.messageCallback(message);
+            } catch {
+            }
+            return;
+          }
+        }
+      }
+      for (const view of this._views) {
+        if (view.messageCallback) {
+          try {
+            view.messageCallback(message);
+          } catch {
+          }
+        }
+      }
+    }
+  }
+};
+async function authenticate(transport, options) {
+  const requestId = crypto.randomUUID();
+  const envelope = {
+    jsonrpc: JSONRPC_VERSION,
+    factoryApiVersion: LEGACY_FACTORY_API_VERSION,
+    factoryProtocolVersion: FACTORY_PROTOCOL_VERSION,
+    type: "request" /* Request */,
+    id: requestId,
+    method: DAEMON_AUTHENTICATE_METHOD,
+    params: {
+      apiKey: options.apiKey,
+      caller: SDK_CALLER
+    }
+  };
+  return new Promise((resolve2, reject) => {
+    const timer = setTimeout(() => {
+      transport.onMessage(() => {
+      });
+      reject(
+        new ConnectionError(
+          `Daemon authentication timed out after ${DAEMON_AUTHENTICATE_TIMEOUT}ms`
+        )
+      );
+    }, DAEMON_AUTHENTICATE_TIMEOUT);
+    transport.onMessage((message) => {
+      if (message["id"] !== requestId) {
+        return;
+      }
+      clearTimeout(timer);
+      transport.onMessage(() => {
+      });
+      if (message["error"]) {
+        const error = message["error"];
+        const errorMessage = error && typeof error === "object" && "message" in error ? String(error.message) : "unknown error";
+        reject(
+          new ConnectionError(`Daemon authentication failed: ${errorMessage}`)
+        );
+        return;
+      }
+      resolve2();
+    });
+    transport.send(envelope);
+  });
+}
+var DaemonConnection = class {
+  _transport;
+  _multiplexer;
+  _apiKey;
+  _closed = false;
+  /** @internal */
+  constructor(transport, apiKey) {
+    this._transport = transport;
+    this._multiplexer = new SharedTransportMultiplexer(transport);
+    this._apiKey = apiKey;
+  }
+  async createSession(options = {}) {
+    this._ensureNotClosed();
+    const view = this._multiplexer.createView();
+    const client = new DaemonClient({
+      transport: view,
+      apiKey: this._apiKey
+    });
+    if (options.permissionHandler) {
+      client.setPermissionHandler(options.permissionHandler);
+    }
+    if (options.askUserHandler) {
+      client.setAskUserHandler(options.askUserHandler);
+    }
+    let sdkMcpServers;
+    try {
+      sdkMcpServers = await startSdkMcpServers(options.mcpServers);
+      const initParams = buildInitParams({
+        ...options,
+        mcpServers: sdkMcpServers.mcpServers
+      });
+      const initResult = await client.initializeSession(initParams);
+      const session = new DaemonSession(client, initResult.sessionId);
+      if (sdkMcpServers) {
+        session.addCleanup(sdkMcpServers.cleanup);
+      }
+      return session;
+    } catch (error) {
+      await sdkMcpServers?.cleanup();
+      await client.close();
+      throw error;
+    }
+  }
+  async resumeSession(sessionId, options = {}) {
+    this._ensureNotClosed();
+    const view = this._multiplexer.createView();
+    const client = new DaemonClient({
+      transport: view,
+      apiKey: this._apiKey
+    });
+    if (options.permissionHandler) {
+      client.setPermissionHandler(options.permissionHandler);
+    }
+    if (options.askUserHandler) {
+      client.setAskUserHandler(options.askUserHandler);
+    }
+    let sdkMcpServers;
+    try {
+      sdkMcpServers = await startSdkMcpServers(options.mcpServers);
+      await client.loadSession({
+        sessionId,
+        mcpServers: sdkMcpServers.mcpServers
+      });
+      const session = new DaemonSession(client, sessionId);
+      if (sdkMcpServers) {
+        session.addCleanup(sdkMcpServers.cleanup);
+      }
+      return session;
+    } catch (error) {
+      await sdkMcpServers?.cleanup();
+      await client.close();
+      throw error;
+    }
+  }
+  async interruptSession(sessionId) {
+    this._ensureNotClosed();
+    const view = this._multiplexer.createView();
+    const client = new DaemonClient({
+      transport: view,
+      apiKey: this._apiKey
+    });
+    try {
+      await client.loadSession({ sessionId });
+      await client.interruptSession();
+    } finally {
+      await client.close();
+    }
+  }
+  async close() {
+    if (this._closed) {
+      return;
+    }
+    this._closed = true;
+    await this._transport.close();
+  }
+  _ensureNotClosed() {
+    if (this._closed) {
+      throw new ConnectionError(
+        "Daemon connection has been closed. Call connectDaemon() to create a new connection."
+      );
+    }
+  }
+};
+async function connectDaemon(options) {
+  const isLocal = !options.url && (!options.machine || options.machine.type === "local" /* Local */);
+  let resolvedOptions = options;
+  if (isLocal) {
+    const { port } = await ensureLocalDaemon();
+    resolvedOptions = { ...options, _localPort: port };
+  }
+  const url = resolveWebSocketUrl(resolvedOptions);
+  const wsConfig = getWebSocketConfig(resolvedOptions.machine);
+  const transport = new WebSocketTransport(wsConfig);
+  const apiKey = resolvedOptions.apiKey;
+  const maxRetries = resolvedOptions.maxRetries ?? 0;
+  try {
+    if (maxRetries > 0) {
+      let lastError;
+      for (let attempt = 0; attempt <= maxRetries; attempt++) {
+        try {
+          await transport.connect(url);
+          await authenticate(transport, resolvedOptions);
+          return new DaemonConnection(transport, apiKey);
+        } catch (error) {
+          lastError = error instanceof Error ? error : new Error(String(error));
+          try {
+            await transport.close();
+          } catch {
+          }
+          if (attempt < maxRetries) {
+            await new Promise((resolve2) => setTimeout(resolve2, 2e3));
+          }
+        }
+      }
+      throw lastError;
+    }
+    await transport.connect(url);
+    await authenticate(transport, resolvedOptions);
+    return new DaemonConnection(transport, apiKey);
+  } catch (error) {
+    try {
+      await transport.close();
+    } catch {
+    }
+    throw error;
+  }
+}
 export {
   AddMcpServerRequestParamsSchema,
   AddMcpServerRequestSchema,
@@ -3769,6 +5318,7 @@ export {
   AskUserRequestSchema,
   AskUserResponseSchema,
   AskUserResultSchema,
+  AssistantTextCompleteNotificationSchema,
   AssistantTextDeltaNotificationSchema,
   AuthenticateMcpServerRequestParamsSchema,
   AuthenticateMcpServerRequestSchema,
@@ -3793,17 +5343,30 @@ export {
   ClearMcpAuthResultSchema,
   CliRequestOrNotificationSchema,
   ClientRequestSchema,
+  CloseSessionRequestParamsSchema,
+  CloseSessionRequestSchema,
+  CloseSessionResponseSchema,
+  CloseSessionResultSchema,
   CompactSessionRequestParamsSchema,
   CompactSessionRequestSchema,
   CompactSessionResponseSchema,
   CompactSessionResultSchema,
+  ComputerListResponseSchema,
+  ComputerMetricSchema,
+  ComputerMetricsResponseSchema,
+  ComputerSchema,
   ConnectionError,
   ContentBlockSchema,
+  ContextBreakdownCategorySchema,
+  ContextBreakdownResultSchema,
   ContextStatsAccuracy,
   ContextStatsSchema,
   CreateMessageNotificationSchema,
   CreateToolConfirmationDetailsSchema,
   DEFAULT_REQUEST_TIMEOUT,
+  DaemonClient,
+  DaemonConnection,
+  DaemonSession,
   DecompSessionType,
   DiscoveredIssueSchema,
   DismissalRecordSchema,
@@ -3813,6 +5376,10 @@ export {
   DroidClient,
   DroidClientMethod,
   DroidErrorType,
+  DroidHookEventSchema,
+  DroidHookExecutionResultSchema,
+  DroidHookOutputSchema,
+  DroidHookSpecificOutputSchema,
   DroidInteractionMode,
   DroidMessageType,
   DroidServerMethod,
@@ -3853,6 +5420,10 @@ export {
   GitRepoInfoSchema,
   HandoffItemsDismissedEntrySchema,
   HandoffSchema,
+  HookCommandSchema,
+  HookExecutionCompletedNotificationSchema,
+  HookExecutionStartedNotificationSchema,
+  HookResultSchema,
   HttpHeaderSchema,
   HttpMcpConfigSchema,
   ImageBlockSchema,
@@ -3910,6 +5481,10 @@ export {
   LoadSessionResponseSchema,
   LoadSessionResultSchema,
   MCP_AUTH_TIMEOUT,
+  MachineTemplateBuildStatusSchema,
+  MachineTemplateListResponseSchema,
+  MachineTemplateSchema,
+  MachineType,
   McpAuthCompletedNotificationSchema,
   McpAuthOutcome,
   McpAuthRequiredNotificationSchema,
@@ -3955,6 +5530,9 @@ export {
   REWIND_TIMEOUT,
   ReasoningEffort,
   RedactedThinkingBlockSchema,
+  RefreshComputerResponseSchema,
+  RemoteSessionListResponseSchema,
+  RemoteSessionSchema,
   RemoveMcpServerRequestParamsSchema,
   RemoveMcpServerRequestSchema,
   RemoveMcpServerResponseSchema,
@@ -3974,6 +5552,7 @@ export {
   SDK_VERSION,
   SESSION_INIT_TIMEOUT,
   SdkMcpServer,
+  ServerRequestHandlerType,
   SessionError,
   SessionMetadataSchema,
   SessionNotFoundError,
@@ -4001,6 +5580,8 @@ export {
   StartMissionRunConfirmationDetailsSchema,
   StdioMcpConfigSchema,
   StreamStateTracker,
+  StructuredOutputErrorSchema,
+  StructuredOutputNotificationSchema,
   SubmitBugReportRequestParamsSchema,
   SubmitBugReportRequestSchema,
   SubmitBugReportResponseSchema,
@@ -4015,6 +5596,7 @@ export {
   TestsSchema,
   TextBlockSchema,
   ThinkingBlockSchema,
+  ThinkingTextCompleteNotificationSchema,
   ThinkingTextDeltaNotificationSchema,
   TimeoutError,
   ToggleMcpServerRequestParamsSchema,
@@ -4026,6 +5608,7 @@ export {
   ToggleMcpToolResponseSchema,
   ToggleMcpToolResultSchema,
   TokenUsageSchema,
+  ToolCallNotificationSchema,
   ToolConfirmationDetailsSchema,
   ToolConfirmationInfoSchema,
   ToolConfirmationListItemSchema,
@@ -4045,19 +5628,37 @@ export {
   UpdateSessionSettingsResultSchema,
   VerificationCommandSchema,
   VerificationSchema,
+  WebSocketTransport,
   WorkerCompletedEntrySchema,
   WorkerFailedEntrySchema,
   WorkerPausedEntrySchema,
   WorkerSelectedFeatureEntrySchema,
   WorkerStartedEntrySchema,
   WorkerStateInfoSchema,
+  connectDaemon,
   convertNotificationToStreamMessage,
+  createComputer,
+  createResponseSchema,
   createSdkMcpServer,
   createSession,
+  deleteComputer,
+  dispatchNotification,
+  ensureLocalDaemon,
+  getComputer,
+  getComputerByName,
+  getComputerMetrics,
+  getMachineTemplate,
+  listComputers,
+  listMachineTemplates,
+  listRemoteSessions,
   listSessions,
-  query,
+  refreshComputer,
+  resolveWebSocketUrl,
+  restartComputer,
   resumeSession,
+  retryInstallDeps,
   run,
-  tool
+  tool,
+  updateComputer
 };
 //# sourceMappingURL=index.js.map