From 65c54411de971d7e088db33b7ce0e507532ab999 Mon Sep 17 00:00:00 2001
From: Isaac Sanchez-Hawkins <266845420+isanchez404@users.noreply.github.com>
Date: Tue, 12 May 2026 04:01:10 -0400
Subject: [PATCH] fix(sync): fetch sessions in active directory (#1231)

Co-authored-by: Isaac Sanchez <isanchez-hawkins@arize.com>
---
 packages/ui/src/sync/use-sync.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/packages/ui/src/sync/use-sync.ts b/packages/ui/src/sync/use-sync.ts
index 7ec8ca4e..fd8b9f41 100644
--- a/packages/ui/src/sync/use-sync.ts
+++ b/packages/ui/src/sync/use-sync.ts
@@ -189,7 +189,7 @@ export function useSync() {
   const fetchMessages = useCallback(
     async (sessionID: string, limit: number, before?: string) => {
       const result = await retry(() =>
-        sdk.session.messages({ sessionID, limit, before }),
+        sdk.session.messages({ sessionID, directory, limit, before }),
       )
       const items = (result.data ?? []).filter((x: { info?: { id?: string } }) => !!x?.info?.id)
       const session = items
@@ -202,7 +202,7 @@ export function useSync() {
       const cursor = result.response?.headers?.get?.("x-next-cursor") ?? undefined
       return { session, part, cursor, complete: !cursor }
     },
-    [sdk],
+    [sdk, directory],
   )
 
   // Load messages for a session
@@ -286,7 +286,7 @@ export function useSync() {
         // Fetch session info if needed
         if (!hasSession || force) {
           try {
-            const result = await retry(() => sdk.session.get({ sessionID }))
+            const result = await retry(() => sdk.session.get({ sessionID, directory }))
             if (result.data) {
               const s = store.getState()
               const sessions = [...s.session]