326 lines · 10.9 KB · TypeScript
RawDownload
import type { ThreadId } from "@t3tools/contracts";import * as Crypto from "effect/Crypto";import * as Effect from "effect/Effect";import * as Option from "effect/Option";import { Atom } from "effect/unstable/reactivity";import { WS_METHODS } from "@t3tools/contracts"; import {  createAtomCommandScheduler,  createEnvironmentCommand,  createEnvironmentRpcCommand,} from "./runtime.ts";import {  type ArchiveThreadInput,  type CancelQueuedRunInput,  type CreateThreadInput,  type DeleteThreadInput,  type EditQueuedRunInput,  type InterruptThreadTurnInput,  type MarkThreadUnreadInput,  type ForkThreadFromRunInput,  type MergeThreadBackInput,  type PromoteQueuedRunInput,  type ReorderQueuedRunInput,  type RespondToThreadApprovalInput,  type RespondToThreadUserInputInput,  type RevertThreadCheckpointInput,  type SetThreadInteractionModeInput,  type SetThreadRuntimeModeInput,  type PinThreadInput,  type ReorderPinnedThreadInput,  type SettleThreadInput,  type SnoozeThreadInput,  type StartThreadTurnInput,  type StopThreadSessionInput,  type UnarchiveThreadInput,  type UnpinThreadInput,  type UnsettleThreadInput,  type UnsnoozeThreadInput,  type UpdateThreadMetadataInput,  type VisitThreadInput,  archiveThread,  cancelQueuedRun,  createThread,  deleteThread,  editQueuedRun,  interruptThreadTurn,  forkThreadFromRun,  markThreadUnread,  mergeThreadBack,  promoteQueuedRun,  reorderQueuedRun,  respondToThreadApproval,  respondToThreadUserInput,  revertThreadCheckpoint,  setThreadInteractionMode,  setThreadRuntimeMode,  pinThread,  reorderPinnedThread,  settleThread,  snoozeThread,  startThreadTurn,  stopThreadSession,  unarchiveThread,  unpinThread,  unsettleThread,  unsnoozeThread,  updateThreadMetadata,  visitThread,} from "../operations/commands.ts";import type { EnvironmentRegistry } from "../connection/registry.ts";import { EnvironmentSupervisor } from "../connection/supervisor.ts";import {  ThreadHistoryController,  type ThreadHistoryLoadEarlierResult,} from "./threadHistoryController.ts"; export type LoadEarlierThreadHistoryInput = {  readonly threadId: ThreadId;}; export type {  ArchiveThreadInput,  CancelQueuedRunInput,  CreateThreadInput,  DeleteThreadInput,  EditQueuedRunInput,  InterruptThreadTurnInput,  MarkThreadUnreadInput,  ForkThreadFromRunInput,  MergeThreadBackInput,  PromoteQueuedRunInput,  ReorderQueuedRunInput,  RespondToThreadApprovalInput,  RespondToThreadUserInputInput,  RevertThreadCheckpointInput,  SetThreadInteractionModeInput,  SetThreadRuntimeModeInput,  PinThreadInput,  ReorderPinnedThreadInput,  SettleThreadInput,  SnoozeThreadInput,  StartThreadTurnInput,  StopThreadSessionInput,  ThreadCommandInput,  UnarchiveThreadInput,  UnpinThreadInput,  UnsettleThreadInput,  UnsnoozeThreadInput,  UpdateThreadMetadataInput,  VisitThreadInput,} from "../operations/commands.ts"; export function createThreadEnvironmentAtoms<R, E>(  runtime: Atom.AtomRuntime<EnvironmentRegistry | Crypto.Crypto | R, E>,) {  const scheduler = createAtomCommandScheduler();  const concurrency = {    mode: "serial" as const,    key: ({ environmentId, input }: { environmentId: string; input: { threadId: string } }) =>      JSON.stringify([environmentId, input.threadId]),  };  return {    create: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:create",      execute: (input: CreateThreadInput) => createThread(input),      scheduler,      concurrency,    }),    delete: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:delete",      execute: (input: DeleteThreadInput) => deleteThread(input),      scheduler,      concurrency,    }),    archive: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:archive",      execute: (input: ArchiveThreadInput) => archiveThread(input),      scheduler,      concurrency,    }),    unarchive: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:unarchive",      execute: (input: UnarchiveThreadInput) => unarchiveThread(input),      scheduler,      concurrency,    }),    settle: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:settle",      execute: (input: SettleThreadInput) => settleThread(input),      scheduler,      concurrency,    }),    unsettle: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:unsettle",      execute: (input: UnsettleThreadInput) => unsettleThread(input),      scheduler,      concurrency,    }),    snooze: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:snooze",      execute: (input: SnoozeThreadInput) => snoozeThread(input),      scheduler,      concurrency,    }),    unsnooze: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:unsnooze",      execute: (input: UnsnoozeThreadInput) => unsnoozeThread(input),      scheduler,      concurrency,    }),    pin: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:pin",      execute: (input: PinThreadInput) => pinThread(input),      scheduler,      concurrency,    }),    unpin: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:unpin",      execute: (input: UnpinThreadInput) => unpinThread(input),      scheduler,      concurrency,    }),    reorderPin: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:reorder-pin",      execute: (input: ReorderPinnedThreadInput) => reorderPinnedThread(input),      scheduler,      concurrency,    }),    visit: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:visit",      execute: (input: VisitThreadInput) => visitThread(input),      scheduler,      concurrency,    }),    markUnread: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:mark-unread",      execute: (input: MarkThreadUnreadInput) => markThreadUnread(input),      scheduler,      concurrency,    }),    updateMetadata: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:update-metadata",      execute: (input: UpdateThreadMetadataInput) => updateThreadMetadata(input),      scheduler,      concurrency,    }),    setRuntimeMode: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:set-runtime-mode",      execute: (input: SetThreadRuntimeModeInput) => setThreadRuntimeMode(input),      scheduler,      concurrency,    }),    setInteractionMode: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:set-interaction-mode",      execute: (input: SetThreadInteractionModeInput) => setThreadInteractionMode(input),      scheduler,      concurrency,    }),    startTurn: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:start-turn",      execute: (input: StartThreadTurnInput) => startThreadTurn(input),      scheduler,      concurrency,    }),    interruptTurn: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:interrupt-turn",      execute: (input: InterruptThreadTurnInput) => interruptThreadTurn(input),      scheduler,      concurrency,    }),    respondToApproval: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:respond-to-approval",      execute: (input: RespondToThreadApprovalInput) => respondToThreadApproval(input),      scheduler,      concurrency,    }),    respondToUserInput: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:respond-to-user-input",      execute: (input: RespondToThreadUserInputInput) => respondToThreadUserInput(input),      scheduler,      concurrency,    }),    revertCheckpoint: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:revert-checkpoint",      execute: (input: RevertThreadCheckpointInput) => revertThreadCheckpoint(input),      scheduler,      concurrency,    }),    stopSession: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:stop-session",      execute: (input: StopThreadSessionInput) => stopThreadSession(input),      scheduler,      concurrency,    }),    forkFromRun: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:fork-from-run",      execute: (input: ForkThreadFromRunInput) => forkThreadFromRun(input),      scheduler,      concurrency: {        mode: "serial",        key: ({ environmentId, input }) => JSON.stringify([environmentId, input.sourceThreadId]),      },    }),    mergeBack: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:merge-back",      execute: (input: MergeThreadBackInput) => mergeThreadBack(input),      scheduler,      concurrency: {        mode: "serial",        key: ({ environmentId, input }) =>          JSON.stringify([environmentId, input.sourceThreadId, input.targetThreadId]),      },    }),    reorderQueuedRun: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:reorder-queued-run",      execute: (input: ReorderQueuedRunInput) => reorderQueuedRun(input),      scheduler,      concurrency,    }),    promoteQueuedRun: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:promote-queued-run",      execute: (input: PromoteQueuedRunInput) => promoteQueuedRun(input),      scheduler,      concurrency,    }),    cancelQueuedRun: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:cancel-queued-run",      execute: (input: CancelQueuedRunInput) => cancelQueuedRun(input),      scheduler,      concurrency,    }),    editQueuedRun: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:edit-queued-run",      execute: (input: EditQueuedRunInput) => editQueuedRun(input),      scheduler,      concurrency,    }),    loadEarlierHistory: createEnvironmentCommand(runtime, {      label: "environment-data:commands:thread:load-earlier-history",      execute: (input: LoadEarlierThreadHistoryInput) =>        Effect.gen(function* () {          const supervisor = yield* EnvironmentSupervisor;          const controller = yield* Effect.serviceOption(ThreadHistoryController);          if (Option.isNone(controller)) {            return { _tag: "noop" } satisfies ThreadHistoryLoadEarlierResult;          }          return yield* controller.value.loadEarlier(            supervisor.target.environmentId,            input.threadId,          );        }),      scheduler,      concurrency: {        mode: "serial",        key: ({ environmentId, input }) => JSON.stringify([environmentId, input.threadId]),      },    }),    uploadFeedback: createEnvironmentRpcCommand(runtime, {      label: "environment-data:commands:thread:upload-feedback",      tag: WS_METHODS.providerUploadFeedback,      scheduler,      concurrency,    }),  };}