wip: more snapshot stuff

1 files changed, 26 insertions, 1 deletions

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;