From 284c01018ed73a7509b6226399e51c2f6f54a58d Mon Sep 17 00:00:00 2001
From: Dax Raad <d@ironbay.co>
Date: Thu, 24 Jul 2025 17:34:15 -0400
Subject: wip: more snapshot stuff

---
 packages/sdk/src/resources/config.ts  | 10 +++++++++-
 packages/sdk/src/resources/event.ts   | 15 +++++++++++++++
 packages/sdk/src/resources/session.ts | 27 ++++++++++++++++++++++++++-
 3 files changed, 50 insertions(+), 2 deletions(-)

(limited to 'packages/sdk/src/resources')

diff --git a/packages/sdk/src/resources/config.ts b/packages/sdk/src/resources/config.ts
index 638af98f8..7637573d4 100644
--- a/packages/sdk/src/resources/config.ts
+++ b/packages/sdk/src/resources/config.ts
@@ -147,7 +147,7 @@ export namespace Config {
 
     npm?: string;
 
-    options?: { [key: string]: unknown };
+    options?: Provider.Options;
   }
 
   export namespace Provider {
@@ -190,6 +190,14 @@ export namespace Config {
         output: number;
       }
     }
+
+    export interface Options {
+      apiKey?: string;
+
+      baseURL?: string;
+
+      [k: string]: unknown;
+    }
   }
 }
 
diff --git a/packages/sdk/src/resources/event.ts b/packages/sdk/src/resources/event.ts
index a3dd9288c..eae023b44 100644
--- a/packages/sdk/src/resources/event.ts
+++ b/packages/sdk/src/resources/event.ts
@@ -24,6 +24,7 @@ export type EventListResponse =
   | EventListResponse.EventMessageUpdated
   | EventListResponse.EventMessageRemoved
   | EventListResponse.EventMessagePartUpdated
+  | EventListResponse.EventMessagePartRemoved
   | EventListResponse.EventStorageWrite
   | EventListResponse.EventSessionUpdated
   | EventListResponse.EventSessionDeleted
@@ -135,6 +136,20 @@ export namespace EventListResponse {
     }
   }
 
+  export interface EventMessagePartRemoved {
+    properties: EventMessagePartRemoved.Properties;
+
+    type: 'message.part.removed';
+  }
+
+  export namespace EventMessagePartRemoved {
+    export interface Properties {
+      messageID: string;
+
+      partID: string;
+    }
+  }
+
   export interface EventStorageWrite {
     properties: EventStorageWrite.Properties;
 
diff --git a/packages/sdk/src/resources/session.ts b/packages/sdk/src/resources/session.ts
index bb761033a..322c9e9ae 100644
--- a/packages/sdk/src/resources/session.ts
+++ b/packages/sdk/src/resources/session.ts
@@ -102,6 +102,8 @@ export interface AssistantMessage {
 
   cost: number;
 
+  mode: string;
+
   modelID: string;
 
   path: AssistantMessage.Path;
@@ -217,7 +219,30 @@ export interface FileSource {
 
 export type Message = UserMessage | AssistantMessage;
 
-export type Part = TextPart | FilePart | ToolPart | StepStartPart | StepFinishPart | SnapshotPart;
+export type Part =
+  | TextPart
+  | FilePart
+  | ToolPart
+  | StepStartPart
+  | StepFinishPart
+  | SnapshotPart
+  | Part.PatchPart;
+
+export namespace Part {
+  export interface PatchPart {
+    id: string;
+
+    files: Array<string>;
+
+    hash: string;
+
+    messageID: string;
+
+    sessionID: string;
+
+    type: 'patch';
+  }
+}
 
 export interface Session {
   id: string;
-- 
cgit v1.2.3