@lota-sdk/core 0.1.5

package/src/services/workstream-turn.ts

@@ -0,0 +1,37 @@
+import type { ChatMessage } from '@lota-sdk/shared/schemas/chat-message'
+import { createUIMessageStream } from 'ai'
+
+import { hasApprovalRespondedParts, isApprovalContinuationRequest } from '../runtime/approval-continuation'
+import { prepareWorkstreamRunCore } from './workstream-turn-preparation'
+import type { WorkstreamTurnParams, WorkstreamApprovalContinuationParams } from './workstream-turn-preparation'
+
+export { hasApprovalRespondedParts, isApprovalContinuationRequest }
+
+export async function createWorkstreamApprovalContinuationStream(params: WorkstreamApprovalContinuationParams) {
+  const prepared = await prepareWorkstreamRunCore({ ...params, kind: 'approvalContinuation' })
+
+  return createUIMessageStream<ChatMessage>({
+    originalMessages: prepared.originalMessages,
+    onError: (error) => (error instanceof Error ? error.message : 'Approval continuation stream failed.'),
+    execute: async ({ writer }) => {
+      await prepared.run(writer)
+    },
+  })
+}
+
+export async function createWorkstreamTurnStream(params: WorkstreamTurnParams) {
+  const prepared = await prepareWorkstreamRunCore({ ...params, kind: 'userTurn' })
+
+  return createUIMessageStream<ChatMessage>({
+    originalMessages: prepared.originalMessages,
+    onError: (error) => (error instanceof Error ? error.message : 'Chat stream failed.'),
+    execute: async ({ writer }) => {
+      await prepared.run(writer)
+    },
+  })
+}
+
+export async function runWorkstreamTurnInBackground(params: WorkstreamTurnParams): Promise<void> {
+  const prepared = await prepareWorkstreamRunCore({ ...params, kind: 'userTurn' })
+  await prepared.run()
+}