@lota-sdk/core 0.4.9 → 0.4.11

package/src/runtime/social-chat/social-chat-agent-runner.ts

@@ -2,7 +2,7 @@ import { stepCountIs } from 'ai'
 import type { ToolSet } from 'ai'
 import { Schema, Effect } from 'effect'
 
-import { getAgentRuntimeConfig, getResolvedAgentFactoryConfig } from '../../config/agent-defaults'
+import type { ResolvedAgentFactoryConfig } from '../../config/agent-defaults'
 import { aiLogger } from '../../config/logger'
 
 interface ExecutableToolDefinition {
@@ -46,13 +46,13 @@ export function withLoggedSocialToolSet(
           execute: (...args: unknown[]): Promise<unknown> =>
             Effect.runPromise(
               Effect.gen(function* () {
-                aiLogger.info`Slack social-chat tool start: agentId=${params.agentId}, tool=${toolName}, channelId=${params.channelId}, threadId=${params.threadId}`
+                aiLogger.debug`Slack social-chat tool start: agentId=${params.agentId}, tool=${toolName}, channelId=${params.channelId}, threadId=${params.threadId}`
                 const result: unknown = yield* Effect.tryPromise({
                   try: () => executableTool.execute(...args),
                   catch: (error: unknown) => new SocialChatAgentRunnerError({ message: String(error), cause: error }),
                 })
                 params.executedToolNames.push(toolName)
-                aiLogger.info`Slack social-chat tool finish: agentId=${params.agentId}, tool=${toolName}, channelId=${params.channelId}, threadId=${params.threadId}`
+                aiLogger.debug`Slack social-chat tool finish: agentId=${params.agentId}, tool=${toolName}, channelId=${params.channelId}, threadId=${params.threadId}`
                 return result
               }).pipe(
                 Effect.tapError((error) =>
@@ -69,6 +69,7 @@ export function withLoggedSocialToolSet(
 }
 
 function runSocialAgentTurnEffect(params: {
+  agentFactoryConfig: ResolvedAgentFactoryConfig
   agentId: string
   mode: 'fixedThreadMode' | 'threadMode'
   threadType: 'group'
@@ -85,7 +86,8 @@ function runSocialAgentTurnEffect(params: {
   abortSignal: AbortSignal
 }): Effect.Effect<{ text: string; displayName: string }, SocialChatAgentRunnerError> {
   return Effect.gen(function* () {
-    const runtimeConfig = getAgentRuntimeConfig({
+    const { agentFactoryConfig } = params
+    const runtimeConfig = agentFactoryConfig.getAgentRuntimeConfig({
       agentId: params.agentId,
       threadType: params.threadType,
       mode: params.mode,
@@ -103,7 +105,7 @@ function runSocialAgentTurnEffect(params: {
       additionalInstructionSections: params.additionalInstructionSections,
     })
 
-    const agentFactoryEntry = getResolvedAgentFactoryConfig().createAgent[runtimeConfig.id]
+    const agentFactoryEntry = agentFactoryConfig.createAgent[runtimeConfig.id]
     if (typeof agentFactoryEntry !== 'function') {
       return yield* new SocialChatAgentRunnerError({
         message: `Social chat agent factory is not configured for ${runtimeConfig.id}`,
@@ -117,9 +119,11 @@ function runSocialAgentTurnEffect(params: {
       extraInstructions: runtimeConfig.extraInstructions,
       stopWhen: [stepCountIs(runtimeConfig.maxSteps)],
     })
-    const response = yield* Effect.promise(() =>
-      agent.generate({ prompt: params.prompt, abortSignal: params.abortSignal }),
-    )
+    const response = yield* Effect.tryPromise({
+      try: () => agent.generate({ prompt: params.prompt, abortSignal: params.abortSignal }),
+      catch: (cause) =>
+        new SocialChatAgentRunnerError({ message: `Social chat agent ${params.agentId} generate() failed.`, cause }),
+    })
     const text = response.text.trim()
     if (!text) {
       return yield* new SocialChatAgentRunnerError({
@@ -138,6 +142,7 @@ function runSocialAgentTurnEffect(params: {
 }
 
 export function runSocialAgentTurn(params: {
+  agentFactoryConfig: ResolvedAgentFactoryConfig
   agentId: string
   mode: 'fixedThreadMode' | 'threadMode'
   threadType: 'group'

package/src/runtime/social-chat/social-chat-history.ts

@@ -2,6 +2,8 @@ import { stripSlackToolExecutionNoticeMarkdown } from '@lota-sdk/shared'
 import type { Message, Thread } from 'chat'
 import { Schema, Effect } from 'effect'
 
+import type { ResolvedAgentConfig } from '../../config/agent-defaults'
+import { iterateEffect } from '../../effect/helpers'
 import type {
   SocialChatHistoryMessage,
   SocialChatHistoryMetadata,
@@ -24,9 +26,10 @@ export function readSlackAuthorName(message: Message): string | undefined {
 }
 
 export function buildSocialChatThreadTranscript(
+  agentConfig: ResolvedAgentConfig,
   messages: ReadonlyArray<Pick<SocialChatHistoryMessage, 'role' | 'parts' | 'metadata'>>,
 ): string {
-  const historyMessages = toHistoryMessages([...messages])
+  const historyMessages = toHistoryMessages(agentConfig, [...messages])
   if (historyMessages.length === 0) return 'No prior thread history.'
 
   return historyMessages
@@ -107,21 +110,29 @@ export function collectThreadMessages(thread: Thread, incomingMessage: Message):
   return Effect.runPromise(
     Effect.catch(
       Effect.gen(function* () {
-        const collected: Message[] = []
         const iterator = thread.allMessages[Symbol.asyncIterator]()
 
-        for (;;) {
-          const next = yield* Effect.tryPromise({
-            try: () => iterator.next(),
-            catch: (error: unknown) =>
-              new SocialChatHistoryError({ message: 'Failed to read thread messages', cause: error }),
-          })
-          if (next.done) break
-          collected.push(next.value)
-        }
+        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.length > 0
-          ? collected
+        return collected.messages.length > 0
+          ? collected.messages
           : thread.recentMessages.length > 0
             ? [...thread.recentMessages]
             : [incomingMessage]