feat(tui): improve session fork handling

packages/core/src/session/info.ts

@@ -17,6 +17,12 @@ export function fromRow(row: typeof SessionTable.$inferSelect): SessionSchema.In
     projectID: ProjectV2.ID.make(row.project_id),
     title: row.title,
     parentID: row.parent_id ? SessionSchema.ID.make(row.parent_id) : undefined,
+    fork: row.fork_session_id
+      ? {
+          sessionID: SessionSchema.ID.make(row.fork_session_id),
+          messageID: row.fork_message_id ? SessionMessage.ID.make(row.fork_message_id) : undefined,
+        }
+      : undefined,
     agent: row.agent ? AgentV2.ID.make(row.agent) : undefined,
     model: row.model
       ? {

packages/core/src/session/projector.ts

@@ -190,7 +190,9 @@ const projectFork = Effect.fn("SessionProjector.projectFork")(function* (
     .insert(SessionTable)
     .values({
       id: event.data.sessionID,
-      parent_id: event.data.parentID,
+      parent_id: null,
+      fork_session_id: event.data.parentID,
+      fork_message_id: event.data.from,
       project_id: parent.project_id,
       workspace_id: parent.workspace_id,
       slug: Slug.create(),

packages/core/src/session/runner/to-llm-message.ts

@@ -41,6 +41,26 @@ const textAttachment = (file: FileAttachment) =>
     },
   })
 
+const directoryAttachment = (file: FileAttachment) =>
+  Message.make({
+    role: "user",
+    content: [
+      `Attached directory: ${file.name ?? (file.source.type === "uri" ? file.source.uri : "directory")}`,
+      file.description === undefined ? undefined : `Description: ${file.description}`,
+      file.data.length === 0 ? undefined : "",
+      file.data.length === 0 ? undefined : Buffer.from(file.data, "base64").toString("utf8"),
+    ]
+      .filter((line): line is string => line !== undefined)
+      .join("\n"),
+    metadata: {
+      attachment: {
+        source: file.source,
+        name: file.name,
+        description: file.description,
+      },
+    },
+  })
+
 const decodeToolInput = Schema.decodeUnknownOption(Schema.UnknownFromJsonString)
 
 const providerMetadata = (
@@ -157,6 +177,7 @@ function toLLMMessage(message: SessionMessage.Message, model: ModelV2.Ref): Mess
       const files = message.files ?? []
       return [
         ...files.filter((file) => file.mime === "text/plain").map(textAttachment),
+        ...files.filter((file) => file.mime === "application/x-directory").map(directoryAttachment),
         Message.make({
           id: message.id,
           role: "user",

packages/core/src/session/sql.ts

@@ -29,6 +29,8 @@ export const SessionTable = sqliteTable(
       .references(() => ProjectTable.id, { onDelete: "cascade" }),
     workspace_id: text().$type<WorkspaceV2.ID>(),
     parent_id: text().$type<SessionSchema.ID>(),
+    fork_session_id: text().$type<SessionSchema.ID>(),
+    fork_message_id: text().$type<SessionMessage.ID>(),
     slug: text().notNull(),
     directory: directoryColumn().notNull(),
     path: pathColumn(),