@lota-sdk/core 0.4.8 → 0.4.10

package/src/runtime/{social-chat-history.ts → social-chat/social-chat-history.ts}

@@ -1,8 +1,19 @@
 import { stripSlackToolExecutionNoticeMarkdown } from '@lota-sdk/shared'
 import type { Message, Thread } from 'chat'
+import { Schema, Effect } from 'effect'
 
-import type { SocialChatHistoryMessage } from '../services/social-chat-history.service'
-import { toHistoryMessages, toOptionalTrimmedString } from './thread-chat-helpers'
+import { iterateEffect } from '../../effect/helpers'
+import type {
+  SocialChatHistoryMessage,
+  SocialChatHistoryMetadata,
+  SocialChatHistoryPart,
+} from '../../services/social-chat-history.service'
+import { toHistoryMessages, toOptionalTrimmedString } from '../thread-chat-helpers'
+
+class SocialChatHistoryError extends Schema.TaggedErrorClass<SocialChatHistoryError>()('SocialChatHistoryError', {
+  message: Schema.String,
+  cause: Schema.optional(Schema.Defect),
+}) {}
 
 export function readSlackAuthorName(message: Message): string | undefined {
   return (
@@ -14,13 +25,9 @@ export function readSlackAuthorName(message: Message): string | undefined {
 }
 
 export function buildSocialChatThreadTranscript(
-  messages: Array<{
-    role: 'user' | 'assistant'
-    parts: Array<Record<string, unknown>>
-    metadata?: Record<string, unknown>
-  }>,
+  messages: ReadonlyArray<Pick<SocialChatHistoryMessage, 'role' | 'parts' | 'metadata'>>,
 ): string {
-  const historyMessages = toHistoryMessages(messages)
+  const historyMessages = toHistoryMessages([...messages])
   if (historyMessages.length === 0) return 'No prior thread history.'
 
   return historyMessages
@@ -58,11 +65,11 @@ export function normalizeSocialHistoryMessage(params: {
     mediaType: attachment.mimeType ?? 'application/octet-stream',
     sizeBytes: typeof attachment.size === 'number' ? attachment.size : null,
     storageKey: attachment.url ?? 'external',
-  }))
+  })) satisfies SocialChatHistoryPart[]
   const parts = [...(text ? [{ type: 'text', text }] : []), ...fileParts]
   if (parts.length === 0) return null
 
-  const metadata: Record<string, unknown> = {
+  const metadata: SocialChatHistoryMetadata = {
     platform: 'slack',
     channelId: params.channelId,
     threadId: params.message.threadId,
@@ -97,14 +104,38 @@ export function normalizeSocialHistoryMessage(params: {
   }
 }
 
-export async function collectThreadMessages(thread: Thread, incomingMessage: Message): Promise<Message[]> {
-  try {
-    const messages: Message[] = []
-    for await (const message of thread.allMessages) {
-      messages.push(message)
-    }
-    return messages.length > 0 ? messages : [incomingMessage]
-  } catch {
-    return thread.recentMessages.length > 0 ? [...thread.recentMessages] : [incomingMessage]
-  }
+export function collectThreadMessages(thread: Thread, incomingMessage: Message): Promise<Message[]> {
+  return Effect.runPromise(
+    Effect.catch(
+      Effect.gen(function* () {
+        const iterator = thread.allMessages[Symbol.asyncIterator]()
+
+        const collected = yield* iterateEffect<{ messages: Message[]; done: boolean }, SocialChatHistoryError, never>(
+          { messages: [], done: false },
+          {
+            while: (state) => !state.done,
+            body: (state) =>
+              Effect.tryPromise({
+                try: () => iterator.next(),
+                catch: (error: unknown) =>
+                  new SocialChatHistoryError({ message: 'Failed to read thread messages', cause: error }),
+              }).pipe(
+                Effect.map((next) =>
+                  next.done
+                    ? { messages: state.messages, done: true }
+                    : { messages: [...state.messages, next.value], done: false },
+                ),
+              ),
+          },
+        )
+
+        return collected.messages.length > 0
+          ? collected.messages
+          : thread.recentMessages.length > 0
+            ? [...thread.recentMessages]
+            : [incomingMessage]
+      }),
+      () => Effect.succeed(thread.recentMessages.length > 0 ? [...thread.recentMessages] : [incomingMessage]),
+    ),
+  )
 }