@lota-sdk/core 0.1.31 → 0.1.32

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/core",
-  "version": "0.1.31",
+  "version": "0.1.32",
   "type": "module",
   "main": "./src/index.ts",
   "types": "./src/index.ts",
@@ -32,7 +32,7 @@
     "@chat-adapter/slack": "^4.23.0",
     "@chat-adapter/state-ioredis": "^4.23.0",
     "@logtape/logtape": "^2.0.5",
-    "@lota-sdk/shared": "0.1.31",
+    "@lota-sdk/shared": "0.1.32",
     "@mendable/firecrawl-js": "^4.18.0",
     "@surrealdb/node": "^3.0.3",
     "ai": "^6.0.141",

package/src/services/workstream-turn-preparation.service.ts

@@ -363,6 +363,7 @@ async function streamAgentResponse(
     mode: streamParams.mode,
     tools: streamParams.tools,
     extraInstructions: config.extraInstructions,
+    maxRetries: 3,
     stopWhen: (agentResolution?.stopWhen as StopCondition<ToolSet> | Array<StopCondition<ToolSet>> | undefined) ??
       streamParams.stopWhen ?? [stepCountIs(config.maxSteps as number)],
     prepareStep: (agentResolution?.prepareStep as PrepareStepFunction<ToolSet> | undefined) ?? streamParams.prepareStep,
@@ -370,19 +371,32 @@ async function streamAgentResponse(
   const agentAbortSignal = streamParams.abortSignal ?? ctx.runAbortSignal
   agentTimer.step('agent-construction')
 
+  const MAX_STREAM_RETRIES = 3
   let result: unknown
-  try {
-    result = await streamParams.observer.run(() =>
-      agent.stream({ messages: modelMessages, abortSignal: agentAbortSignal }),
-    )
-    agentTimer.step('agent.stream()-resolved')
-  } catch (error) {
-    if (agentAbortSignal.aborted) {
-      streamParams.observer.recordAbort(error)
-    } else {
-      streamParams.observer.recordError(error)
+  for (let attempt = 0; ; attempt++) {
+    try {
+      result = await streamParams.observer.run(() =>
+        agent.stream({ messages: modelMessages, abortSignal: agentAbortSignal }),
+      )
+      agentTimer.step('agent.stream()-resolved')
+      break
+    } catch (error) {
+      if (agentAbortSignal.aborted) {
+        streamParams.observer.recordAbort(error)
+        throw error
+      }
+      const errorMessage = error instanceof Error ? error.message : String(error)
+      const isTransient =
+        errorMessage.includes('client disconnected') ||
+        errorMessage.includes('ECONNRESET') ||
+        errorMessage.includes('socket hang up') ||
+        errorMessage.includes('fetch failed')
+      if (!isTransient || attempt >= MAX_STREAM_RETRIES - 1) {
+        streamParams.observer.recordError(error)
+        throw error
+      }
+      aiLogger.warn`Transient stream error (attempt ${attempt + 1}/${MAX_STREAM_RETRIES}): ${errorMessage} — retrying`
     }
-    throw error
   }
   if (!hasUIMessageStream(result)) {
     throw new Error(`Agent run for ${resolvedAgentId} did not expose a UI message stream.`)