ignore: v2 experiments

packages/opencode/src/v2/session-entry.ts

@@ -1,114 +1,51 @@
-import { Identifier } from "@/id/id"
-import { Database } from "@/node"
-import type { SessionID } from "@/session/schema"
-import { SessionEntryTable } from "@/session/session.sql"
-import { withStatics } from "@/util/schema"
-import { Context, DateTime, Effect, Layer, Schema } from "effect"
-import { eq } from "../storage/db"
+import { Schema } from "effect"
+import { SessionEvent } from "./session-event"
+import { produce } from "immer"
 
 export namespace SessionEntry {
-  export const ID = Schema.String.pipe(Schema.brand("Session.Entry.ID")).pipe(
-    withStatics((s) => ({
-      create: () => s.make(Identifier.ascending("entry")),
-      prefix: "ent",
-    })),
-  )
+  export const ID = SessionEvent.ID
   export type ID = Schema.Schema.Type<typeof ID>
 
   const Base = {
-    id: ID,
+    id: SessionEvent.ID,
     metadata: Schema.Record(Schema.String, Schema.Unknown).pipe(Schema.optional),
     time: Schema.Struct({
       created: Schema.DateTimeUtc,
     }),
   }
 
-  export class Source extends Schema.Class<Source>("Session.Entry.Source")({
-    start: Schema.Number,
-    end: Schema.Number,
-    text: Schema.String,
-  }) {}
-
-  export class FileAttachment extends Schema.Class<FileAttachment>("Session.Entry.File.Attachment")({
-    uri: Schema.String,
-    mime: Schema.String,
-    name: Schema.String.pipe(Schema.optional),
-    description: Schema.String.pipe(Schema.optional),
-    source: Source.pipe(Schema.optional),
-  }) {
-    static create(url: string) {
-      return new FileAttachment({
-        uri: url,
-        mime: "text/plain",
-      })
-    }
-  }
-
-  export class AgentAttachment extends Schema.Class<AgentAttachment>("Session.Entry.Agent.Attachment")({
-    name: Schema.String,
-    source: Source.pipe(Schema.optional),
-  }) {}
-
   export class User extends Schema.Class<User>("Session.Entry.User")({
     ...Base,
+    text: SessionEvent.Prompt.fields.text,
+    files: SessionEvent.Prompt.fields.files,
+    agents: SessionEvent.Prompt.fields.agents,
     type: Schema.Literal("user"),
-    text: Schema.String,
-    files: Schema.Array(FileAttachment).pipe(Schema.optional),
-    agents: Schema.Array(AgentAttachment).pipe(Schema.optional),
+    time: Schema.Struct({
+      created: Schema.DateTimeUtc,
+    }),
   }) {
-    static create(input: { text: User["text"]; files?: User["files"]; agents?: User["agents"] }) {
-      const msg = new User({
-        id: ID.create(),
+    static fromEvent(event: SessionEvent.Prompt) {
+      return new User({
+        id: event.id,
         type: "user",
-        ...input,
-        time: {
-          created: Effect.runSync(DateTime.now),
-        },
+        metadata: event.metadata,
+        text: event.text,
+        files: event.files,
+        agents: event.agents,
+        time: { created: event.timestamp },
       })
-      return msg
     }
   }
 
   export class Synthetic extends Schema.Class<Synthetic>("Session.Entry.Synthetic")({
+    ...SessionEvent.Synthetic.fields,
     ...Base,
     type: Schema.Literal("synthetic"),
-    text: Schema.String,
-  }) {}
-
-  export class Request extends Schema.Class<Request>("Session.Entry.Request")({
-    ...Base,
-    type: Schema.Literal("start"),
-    model: Schema.Struct({
-      id: Schema.String,
-      providerID: Schema.String,
-      variant: Schema.String.pipe(Schema.optional),
-    }),
-  }) {}
-
-  export class Text extends Schema.Class<Text>("Session.Entry.Text")({
-    ...Base,
-    type: Schema.Literal("text"),
-    text: Schema.String,
-    time: Schema.Struct({
-      ...Base.time.fields,
-      completed: Schema.DateTimeUtc.pipe(Schema.optional),
-    }),
-  }) {}
-
-  export class Reasoning extends Schema.Class<Reasoning>("Session.Entry.Reasoning")({
-    ...Base,
-    type: Schema.Literal("reasoning"),
-    text: Schema.String,
-    time: Schema.Struct({
-      ...Base.time.fields,
-      completed: Schema.DateTimeUtc.pipe(Schema.optional),
-    }),
   }) {}
 
   export class ToolStatePending extends Schema.Class<ToolStatePending>("Session.Entry.ToolState.Pending")({
     status: Schema.Literal("pending"),
-    input: Schema.Record(Schema.String, Schema.Unknown),
-    raw: Schema.String,
+    input: Schema.String,
   }) {}
 
   export class ToolStateRunning extends Schema.Class<ToolStateRunning>("Session.Entry.ToolState.Running")({
@@ -124,7 +61,7 @@ export namespace SessionEntry {
     output: Schema.String,
     title: Schema.String,
     metadata: Schema.Record(Schema.String, Schema.Unknown),
-    attachments: Schema.Array(FileAttachment).pipe(Schema.optional),
+    attachments: SessionEvent.FileAttachment.pipe(Schema.Array, Schema.optional),
   }) {}
 
   export class ToolStateError extends Schema.Class<ToolStateError>("Session.Entry.ToolState.Error")({
@@ -132,34 +69,42 @@ export namespace SessionEntry {
     input: Schema.Record(Schema.String, Schema.Unknown),
     error: Schema.String,
     metadata: Schema.Record(Schema.String, Schema.Unknown).pipe(Schema.optional),
-    time: Schema.Struct({
-      start: Schema.Number,
-      end: Schema.Number,
-    }),
   }) {}
 
   export const ToolState = Schema.Union([ToolStatePending, ToolStateRunning, ToolStateCompleted, ToolStateError])
   export type ToolState = Schema.Schema.Type<typeof ToolState>
 
-  export class Tool extends Schema.Class<Tool>("Session.Entry.Tool")({
-    ...Base,
+  export class AssistantTool extends Schema.Class<AssistantTool>("Session.Entry.Assistant.Tool")({
     type: Schema.Literal("tool"),
     callID: Schema.String,
     name: Schema.String,
     state: ToolState,
     time: Schema.Struct({
-      ...Base.time.fields,
+      created: Schema.DateTimeUtc,
       ran: Schema.DateTimeUtc.pipe(Schema.optional),
       completed: Schema.DateTimeUtc.pipe(Schema.optional),
       pruned: Schema.DateTimeUtc.pipe(Schema.optional),
     }),
   }) {}
 
-  export class Complete extends Schema.Class<Complete>("Session.Entry.Complete")({
+  export class AssistantText extends Schema.Class<AssistantText>("Session.Entry.Assistant.Text")({
+    type: Schema.Literal("text"),
+    text: Schema.String,
+  }) {}
+
+  export class AssistantReasoning extends Schema.Class<AssistantReasoning>("Session.Entry.Assistant.Reasoning")({
+    type: Schema.Literal("reasoning"),
+    text: Schema.String,
+  }) {}
+
+  export const AssistantContent = Schema.Union([AssistantText, AssistantReasoning, AssistantTool])
+  export type AssistantContent = Schema.Schema.Type<typeof AssistantContent>
+
+  export class Assistant extends Schema.Class<Assistant>("Session.Entry.Assistant")({
     ...Base,
-    type: Schema.Literal("complete"),
-    cost: Schema.Number,
-    reason: Schema.String,
+    type: Schema.Literal("assistant"),
+    content: AssistantContent.pipe(Schema.Array),
+    cost: Schema.Number.pipe(Schema.optional),
     tokens: Schema.Struct({
       input: Schema.Number,
       output: Schema.Number,
@@ -168,30 +113,174 @@ export namespace SessionEntry {
         read: Schema.Number,
         write: Schema.Number,
       }),
+    }).pipe(Schema.optional),
+    error: Schema.String.pipe(Schema.optional),
+    time: Schema.Struct({
+      created: Schema.DateTimeUtc,
+      completed: Schema.DateTimeUtc.pipe(Schema.optional),
     }),
   }) {}
 
-  export class Retry extends Schema.Class<Retry>("Session.Entry.Retry")({
-    ...Base,
-    type: Schema.Literal("retry"),
-    attempt: Schema.Number,
-    error: Schema.String,
-  }) {}
-
   export class Compaction extends Schema.Class<Compaction>("Session.Entry.Compaction")({
-    ...Base,
+    ...SessionEvent.Compacted.fields,
     type: Schema.Literal("compaction"),
-    auto: Schema.Boolean,
-    overflow: Schema.Boolean.pipe(Schema.optional),
+    ...Base,
   }) {}
 
-  export const Entry = Schema.Union([User, Synthetic, Request, Tool, Text, Reasoning, Complete, Retry, Compaction], {
-    mode: "oneOf",
-  })
+  export const Entry = Schema.Union([User, Synthetic, Assistant, Compaction])
+
   export type Entry = Schema.Schema.Type<typeof Entry>
 
   export type Type = Entry["type"]
 
+  export type History = {
+    entries: Entry[]
+    pending: Entry[]
+  }
+
+  export function step(old: History, event: SessionEvent.Event): History {
+    return produce(old, (draft) => {
+      const lastAssistant = draft.entries.findLast((x) => x.type === "assistant")
+      const pendingAssistant = lastAssistant && !lastAssistant.time.completed ? lastAssistant : undefined
+
+      switch (event.type) {
+        case "prompt": {
+          if (pendingAssistant) {
+            // @ts-expect-error
+            draft.pending.push(User.fromEvent(event))
+            break
+          }
+          // @ts-expect-error
+          draft.entries.push(User.fromEvent(event))
+          break
+        }
+        case "step.started": {
+          if (pendingAssistant) pendingAssistant.time.completed = event.timestamp
+          draft.entries.push({
+            id: event.id,
+            type: "assistant",
+            time: {
+              created: event.timestamp,
+            },
+            content: [],
+          })
+          break
+        }
+        case "text.started": {
+          if (!pendingAssistant) break
+          pendingAssistant.content.push({
+            type: "text",
+            text: "",
+          })
+          break
+        }
+        case "text.delta": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "text")
+          if (match) match.text += event.delta
+          break
+        }
+        case "text.ended": {
+          break
+        }
+        case "tool.input.started": {
+          if (!pendingAssistant) break
+          pendingAssistant.content.push({
+            type: "tool",
+            callID: event.callID,
+            name: event.name,
+            time: {
+              created: event.timestamp,
+            },
+            state: {
+              status: "pending",
+              input: "",
+            },
+          })
+          break
+        }
+        case "tool.input.delta": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "tool")
+          if (match) match.state.input += event.delta
+          break
+        }
+        case "tool.input.ended": {
+          break
+        }
+        case "tool.called": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "tool")
+          if (match) {
+            match.time.ran = event.timestamp
+            match.state = {
+              status: "running",
+              input: event.input,
+            }
+          }
+          break
+        }
+        case "tool.success": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "tool")
+          if (match && match.state.status === "running") {
+            match.state = {
+              status: "completed",
+              input: match.state.input,
+              output: event.output ?? "",
+              title: event.title,
+              metadata: event.metadata ?? {},
+              // @ts-expect-error
+              attachments: event.attachments ?? [],
+            }
+          }
+          break
+        }
+        case "tool.error": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "tool")
+          if (match && match.state.status === "running") {
+            match.state = {
+              status: "error",
+              error: event.error,
+              input: match.state.input,
+              metadata: event.metadata ?? {},
+            }
+          }
+          break
+        }
+        case "reasoning.started": {
+          if (!pendingAssistant) break
+          pendingAssistant.content.push({
+            type: "reasoning",
+            text: "",
+          })
+          break
+        }
+        case "reasoning.delta": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "reasoning")
+          if (match) match.text += event.delta
+          break
+        }
+        case "reasoning.ended": {
+          if (!pendingAssistant) break
+          const match = pendingAssistant.content.findLast((x) => x.type === "reasoning")
+          if (match) match.text = event.text
+          break
+        }
+        case "step.ended": {
+          if (!pendingAssistant) break
+          pendingAssistant.time.completed = event.timestamp
+          pendingAssistant.cost = event.cost
+          pendingAssistant.tokens = event.tokens
+          break
+        }
+      }
+    })
+  }
+
+  /*
   export interface Interface {
     readonly decode: (row: typeof SessionEntryTable.$inferSelect) => Entry
     readonly fromSession: (sessionID: SessionID) => Effect.Effect<Entry[], never>
@@ -224,4 +313,5 @@ export namespace SessionEntry {
       })
     }),
   )
+  */
 }