@axiom-lattice/react-sdk 2.1.90 → 2.1.92

package/dist/index.d.mts

@@ -726,6 +726,8 @@ interface PersonalAssistantInfo {
     assistantId: string;
     projectId: string;
     workspaceId: string;
+    /** tenantId for debugging cross-tenant leakage */
+    tenantId?: string | null;
 }
 interface AuthContextValue {
     user: User | null;

package/dist/index.d.ts

@@ -726,6 +726,8 @@ interface PersonalAssistantInfo {
     assistantId: string;
     projectId: string;
     workspaceId: string;
+    /** tenantId for debugging cross-tenant leakage */
+    tenantId?: string | null;
 }
 interface AuthContextValue {
     user: User | null;

package/dist/index.js

@@ -164,18 +164,24 @@ var AuthProvider = ({
   const [personalAssistant, setPersonalAssistant] = (0, import_react.useState)(null);
   const [isLoading, setIsLoading] = (0, import_react.useState)(false);
   const [error, setError] = (0, import_react.useState)(null);
+  const getPAKey = (tenantId) => tenantId ? `lattice_personal_assistant_${tenantId}` : "lattice_personal_assistant";
+  const currentTenantRef = (0, import_react.useRef)(currentTenant);
+  currentTenantRef.current = currentTenant;
   (0, import_react.useEffect)(() => {
     const storedUser = sessionStorage.getItem("lattice_user");
     const storedTenants = sessionStorage.getItem("lattice_tenants");
     const storedCurrentTenant = sessionStorage.getItem("lattice_current_tenant");
-    const storedPA = sessionStorage.getItem("lattice_personal_assistant");
     if (storedUser) {
       try {
         const userData = JSON.parse(storedUser);
         setUser(userData);
         if (storedTenants) setTenants(JSON.parse(storedTenants));
-        if (storedCurrentTenant) setCurrentTenant(JSON.parse(storedCurrentTenant));
-        if (storedPA) setPersonalAssistant(JSON.parse(storedPA));
+        if (storedCurrentTenant) {
+          const ct = JSON.parse(storedCurrentTenant);
+          setCurrentTenant(ct);
+          const storedPA = sessionStorage.getItem(getPAKey(ct.id));
+          if (storedPA) setPersonalAssistant(JSON.parse(storedPA));
+        }
       } catch {
         sessionStorage.removeItem("lattice_user");
         sessionStorage.removeItem("lattice_tenants");
@@ -281,13 +287,18 @@ var AuthProvider = ({
       }
       const result = await response.json();
       const tenantData = result.data.tenant;
+      const oldTenantId = currentTenantRef.current?.id;
+      if (oldTenantId) {
+        sessionStorage.removeItem(getPAKey(oldTenantId));
+      }
       setCurrentTenant(tenantData);
       sessionStorage.setItem("lattice_current_tenant", JSON.stringify(tenantData));
       setPersonalAssistant(result.data.personalAssistant || null);
       if (result.data.personalAssistant) {
-        sessionStorage.setItem("lattice_personal_assistant", JSON.stringify(result.data.personalAssistant));
+        const paWithTenant = { ...result.data.personalAssistant, tenantId: tenantData.id };
+        sessionStorage.setItem(getPAKey(tenantData.id), JSON.stringify(paWithTenant));
       } else {
-        sessionStorage.removeItem("lattice_personal_assistant");
+        sessionStorage.removeItem(getPAKey(tenantData.id));
       }
       if (result.data.token) {
         sessionStorage.setItem("lattice_token", result.data.token);
@@ -411,9 +422,14 @@ var AuthProvider = ({
   }, [baseURL]);
   const wrappedSetPersonalAssistant = (0, import_react.useCallback)((info) => {
     setPersonalAssistant(info);
+    const tenantId = currentTenantRef.current?.id;
     if (info) {
-      sessionStorage.setItem("lattice_personal_assistant", JSON.stringify(info));
+      const paWithTenant = { ...info, tenantId: tenantId || info.tenantId };
+      sessionStorage.setItem(getPAKey(tenantId), JSON.stringify(paWithTenant));
     } else {
+      if (tenantId) {
+        sessionStorage.removeItem(getPAKey(tenantId));
+      }
       sessionStorage.removeItem("lattice_personal_assistant");
     }
   }, []);
@@ -31599,7 +31615,7 @@ var PERSONALITY_MAP = {
 var PersonalAssistantPage = () => {
   const { styles } = useStyles20();
   const { get, post, del } = useApi();
-  const { personalAssistant, setPersonalAssistant } = useAuth();
+  const { personalAssistant, setPersonalAssistant, currentTenant } = useAuth();
   const auth = useAuthOptional();
   const userName = auth?.user?.name || auth?.user?.email?.split("@")[0];
   const [creating, setCreating] = (0, import_react102.useState)(false);
@@ -31626,7 +31642,7 @@ var PersonalAssistantPage = () => {
         import_antd91.message.error(b.message);
         return;
       }
-      const info = { assistantId: b.data.id, projectId: b.data.projectId, workspaceId: b.data.workspaceId || "default" };
+      const info = { assistantId: b.data.id, projectId: b.data.projectId, workspaceId: b.data.workspaceId || "default", tenantId: currentTenant?.id };
       setPersonalAssistant(info);
       import_antd91.message.success("Done!");
     } catch {