fix(search): retry native read socket timeouts and mark excluded Confluence matches partial

Author: waleedlatif1

apps/sim/lib/sim-search/live/application.test.ts

@@ -796,6 +796,16 @@ describe('authorized live retrieval', () => {
       new McpError(ErrorCode.RequestTimeout, 'TimeoutError'),
       { retryable: true, message: expect.stringContaining('took too long') },
     ],
+    [
+      'a native request socket timeout',
+      Object.assign(new Error('Request timed out after 10000ms'), { code: 'ETIMEDOUT' }),
+      { retryable: true, message: expect.stringContaining('Try again') },
+    ],
+    [
+      'a native request dispatcher timeout',
+      Object.assign(new Error('Headers Timeout Error'), { code: 'UND_ERR_HEADERS_TIMEOUT' }),
+      { retryable: true, message: expect.stringContaining('Try again') },
+    ],
   ])('classifies %s during a read so the caller can act on it', async (_, failure, expected) => {
     const search = await searchLiveKnowledge.execute({ principal, input })
     mocks.read.mockRejectedValueOnce(failure)

apps/sim/lib/sim-search/live/application.ts

@@ -18,6 +18,7 @@ import {
 } from '@/lib/api/contracts/mothership-assistant-tools'
 import { canonicalJson, fingerprint, instantScopePart } from '@/lib/api/cursor-binding'
 import { env } from '@/lib/core/config/env'
+import { isRetryableNetworkError } from '@/lib/core/errors/retryable-infrastructure'
 import { OrchestrationError } from '@/lib/core/orchestration/types'
 import {
   type ResourceOwner,
@@ -795,6 +796,11 @@ function liveReadFailure(error: unknown, provider: string, deadline?: AbortSigna
       `${name} took too long to return this document. Try again, or read a different result.`,
       true
     )
+  if (isRetryableNetworkError(error))
+    return new LiveReadError(
+      `${name} did not respond while reading this document. Try again shortly.`,
+      true
+    )
   return error
 }
 

apps/sim/lib/sim-search/live/atlassian.test.ts

@@ -179,5 +179,6 @@ describe('Confluence live documents', () => {
       { id: '123', kind: 'page' },
     ])
     expect(page.message).toContain('cannot be read')
+    expect(page.partial).toBe(true)
   })
 })

apps/sim/lib/sim-search/live/atlassian.ts

@@ -212,7 +212,7 @@ export async function searchAtlassian(
   const excluded = pages.some((result) => result.excluded)
   return {
     documents,
-    partial: !input.native?.project && allSites.length > selected.length,
+    partial: excluded || (!input.native?.project && allSites.length > selected.length),
     hasMore: pages.some((result) => Boolean(result.next)),
     nextCursor: single ? pages[0]?.next : undefined,
     message: