@axiom-lattice/react-sdk 2.1.91 → 2.1.93

package/dist/index.mjs

@@ -1,5 +1,5 @@
 // src/hooks/useChat.ts
-import { useState as useState3, useCallback as useCallback3, useEffect as useEffect3, useRef as useRef2 } from "react";
+import { useState as useState3, useCallback as useCallback3, useEffect as useEffect3, useRef as useRef3 } from "react";
 import { v4 as uuidv4 } from "uuid";
 
 // src/context.tsx
@@ -9,12 +9,12 @@ import {
   useContext as useContext2,
   useEffect as useEffect2,
   useMemo,
-  useRef
+  useRef as useRef2
 } from "react";
 import { Client } from "@axiom-lattice/client-sdk";
 
 // src/context/AuthContext.tsx
-import { createContext, useContext, useState, useCallback, useEffect } from "react";
+import { createContext, useContext, useState, useCallback, useEffect, useRef } from "react";
 import { jsx } from "react/jsx-runtime";
 var AuthContext = createContext(null);
 var useAuth = () => {
@@ -41,18 +41,24 @@ var AuthProvider = ({
   const [personalAssistant, setPersonalAssistant] = useState(null);
   const [isLoading, setIsLoading] = useState(false);
   const [error, setError] = useState(null);
+  const getPAKey = (tenantId) => tenantId ? `lattice_personal_assistant_${tenantId}` : "lattice_personal_assistant";
+  const currentTenantRef = useRef(currentTenant);
+  currentTenantRef.current = currentTenant;
   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");
@@ -158,13 +164,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);
@@ -288,9 +299,14 @@ var AuthProvider = ({
   }, [baseURL]);
   const wrappedSetPersonalAssistant = 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");
     }
   }, []);
@@ -335,7 +351,7 @@ function AxiomLatticeProvider({
     }
     return config.apiKey;
   }, [authContext?.isAuthenticated, authContext?.currentTenant?.id, config.apiKey]);
-  const clientCacheRef = useRef(/* @__PURE__ */ new Map());
+  const clientCacheRef = useRef2(/* @__PURE__ */ new Map());
   const baseConfig = useMemo(
     () => ({
       baseURL: config.baseURL,
@@ -424,8 +440,8 @@ function useChat(threadId, assistantId, options = {}) {
     interrupts: void 0,
     ...options.enableReturnStateWhenStreamCompleted ? { agentState: null } : {}
   });
-  const stopStreamingRef = useRef2(null);
-  const chunkMessageMerger = useRef2(createSimpleMessageMerger());
+  const stopStreamingRef = useRef3(null);
+  const chunkMessageMerger = useRef3(createSimpleMessageMerger());
   const fetchAndUpdateAgentState = useCallback3(
     async (threadId2) => {
       if (!options.enableReturnStateWhenStreamCompleted) return;
@@ -723,7 +739,7 @@ import {
   useState as useState7,
   useCallback as useCallback7,
   useEffect as useEffect6,
-  useRef as useRef5
+  useRef as useRef6
 } from "react";
 import {
   createSimpleMessageMerger as createSimpleMessageMerger2
@@ -737,7 +753,7 @@ import {
   useContext as useContext5,
   useEffect as useEffect5,
   useMemo as useMemo2,
-  useRef as useRef4,
+  useRef as useRef5,
   useState as useState6
 } from "react";
 
@@ -747,7 +763,7 @@ import {
   useCallback as useCallback4,
   useContext as useContext3,
   useEffect as useEffect4,
-  useRef as useRef3,
+  useRef as useRef4,
   useState as useState4
 } from "react";
 import { jsx as jsx3 } from "react/jsx-runtime";
@@ -793,7 +809,7 @@ var AssistantContextProvider = ({
     isLoading: false,
     error: null
   });
-  const assistantsRef = useRef3([]);
+  const assistantsRef = useRef4([]);
   useEffect4(() => {
     assistantsRef.current = state.assistants;
   }, [state.assistants]);
@@ -1562,10 +1578,10 @@ var ConversationContextProvider = ({
   const [isLoading, setIsLoading] = useState6(false);
   const [error, setError] = useState6(null);
   const [customRunConfig, setCustomRunConfig] = useState6({});
-  const loadedAssistantIdRef = useRef4(null);
-  const prevAssistantIdRef = useRef4(null);
-  const isLoadingRef = useRef4(false);
-  const clientRef = useRef4(client);
+  const loadedAssistantIdRef = useRef5(null);
+  const prevAssistantIdRef = useRef5(null);
+  const isLoadingRef = useRef5(false);
+  const clientRef = useRef5(client);
   useEffect5(() => {
     clientRef.current = client;
   }, [client]);
@@ -1880,14 +1896,14 @@ function AgentThreadProvider({
       threadId
     }));
   }, [clientAssistantId, tenantId, threadId]);
-  const stopStreamingRef = useRef5(null);
-  const sendMessageStreamsRef = useRef5(/* @__PURE__ */ new Map());
-  const chunkMessageMerger = useRef5(createSimpleMessageMerger2());
-  const lastAgentStateCreatedAtRef = useRef5(null);
-  const messageCountRef = useRef5(0);
-  const onToolCompletedRef = useRef5(onToolCompleted);
+  const stopStreamingRef = useRef6(null);
+  const sendMessageStreamsRef = useRef6(/* @__PURE__ */ new Map());
+  const chunkMessageMerger = useRef6(createSimpleMessageMerger2());
+  const lastAgentStateCreatedAtRef = useRef6(null);
+  const messageCountRef = useRef6(0);
+  const onToolCompletedRef = useRef6(onToolCompleted);
   onToolCompletedRef.current = onToolCompleted;
-  const onStreamCompletedRef = useRef5(onStreamCompleted);
+  const onStreamCompletedRef = useRef6(onStreamCompleted);
   onStreamCompletedRef.current = onStreamCompleted;
   useEffect6(() => {
     messageCountRef.current = state.messages.length;
@@ -2431,15 +2447,15 @@ function useAgentChat(options) {
 }
 
 // src/hooks/useAgentState.ts
-import { useState as useState8, useCallback as useCallback8, useEffect as useEffect7, useRef as useRef6 } from "react";
+import { useState as useState8, useCallback as useCallback8, useEffect as useEffect7, useRef as useRef7 } from "react";
 function useAgentState(threadId, assistantId, options = {}) {
   const client = useClient(assistantId);
   const [agentState, setAgentState] = useState8(null);
   const [isLoading, setIsLoading] = useState8(false);
   const [error, setError] = useState8(null);
-  const pollingIntervalRef = useRef6(null);
-  const pollingTimeoutRef = useRef6(null);
-  const hasFetchedRef = useRef6(false);
+  const pollingIntervalRef = useRef7(null);
+  const pollingTimeoutRef = useRef7(null);
+  const hasFetchedRef = useRef7(false);
   const {
     pollingInterval = 2e3,
     // 2 seconds by default
@@ -2761,7 +2777,7 @@ function useEvalRuns(projectId) {
 }
 
 // src/hooks/eval/useEvalRunStream.ts
-import { useState as useState15, useEffect as useEffect12, useRef as useRef7, useCallback as useCallback15 } from "react";
+import { useState as useState15, useEffect as useEffect12, useRef as useRef8, useCallback as useCallback15 } from "react";
 function useEvalRunStream(runId) {
   const client = useClient("__GLOBAL__");
   const [status, setStatus] = useState15("idle");
@@ -2772,7 +2788,7 @@ function useEvalRunStream(runId) {
     failed: 0
   });
   const [connected, setConnected] = useState15(false);
-  const abortRef = useRef7(null);
+  const abortRef = useRef8(null);
   const stopStreaming = useCallback15(() => {
     abortRef.current?.abort();
     abortRef.current = null;
@@ -4230,12 +4246,12 @@ import React71, {
   useState as useState78,
   useCallback as useCallback40,
   useEffect as useEffect54,
-  useRef as useRef29
+  useRef as useRef30
 } from "react";
 import { WorkspaceClient, Client as Client3 } from "@axiom-lattice/client-sdk";
 
 // src/components/Chat/WorkspaceResourceManager.tsx
-import { useMemo as useMemo35, useEffect as useEffect53, useRef as useRef28, useState as useState77 } from "react";
+import { useMemo as useMemo35, useEffect as useEffect53, useRef as useRef29, useState as useState77 } from "react";
 import { FolderOpen as FolderOpen3, Activity as Activity4, Database as Database7, Plug as Plug2, Bot as Bot5, Wrench as Wrench2, Zap as Zap2, LogOut as LogOut3, Building2 as Building23, Key, Share2, History, Inbox as Inbox3, FlaskConical as FlaskConical3 } from "lucide-react";
 import { Modal as Modal21, Avatar as Avatar12, Popover as Popover3, Button as Button56 } from "antd";
 
@@ -8012,7 +8028,7 @@ var McpConfigFormModal = ({
 };
 
 // src/components/Chat/AssistantFlow.tsx
-import { useMemo as useMemo6, useEffect as useEffect21, useState as useState31, useCallback as useCallback21, useRef as useRef9 } from "react";
+import { useMemo as useMemo6, useEffect as useEffect21, useState as useState31, useCallback as useCallback21, useRef as useRef10 } from "react";
 import {
   ReactFlow,
   Background,
@@ -8818,7 +8834,7 @@ var AssistantNode = ({
 var AssistantNode_default = AssistantNode;
 
 // src/components/Chat/AgentConfigPanel.tsx
-import { useState as useState28, useEffect as useEffect19, useMemo as useMemo5, useCallback as useCallback19, useRef as useRef8 } from "react";
+import { useState as useState28, useEffect as useEffect19, useMemo as useMemo5, useCallback as useCallback19, useRef as useRef9 } from "react";
 import {
   Form as Form4,
   Input as Input7,
@@ -9082,7 +9098,7 @@ var SkillListField = ({ propertyKey, label, value, onChange, token }) => {
   const [skills, setSkills] = useState28([]);
   const [loading, setLoading] = useState28(false);
   const { get } = useApi();
-  const fetchedRef = useRef8(false);
+  const fetchedRef = useRef9(false);
   useEffect19(() => {
     if (fetchedRef.current) return;
     fetchedRef.current = true;
@@ -9193,7 +9209,7 @@ var DatabaseListField = ({ propertyKey, label, value, onChange, token }) => {
   const [databases, setDatabases] = useState28([]);
   const [loading, setLoading] = useState28(false);
   const { get } = useApi();
-  const fetchedRef = useRef8(false);
+  const fetchedRef = useRef9(false);
   useEffect19(() => {
     if (fetchedRef.current) return;
     fetchedRef.current = true;
@@ -9308,7 +9324,7 @@ var MetricsListField = ({ propertyKey, label, value, onChange, token }) => {
   const [servers, setServers] = useState28([]);
   const [loading, setLoading] = useState28(false);
   const { get } = useApi();
-  const fetchedRef = useRef8(false);
+  const fetchedRef = useRef9(false);
   useEffect19(() => {
     if (fetchedRef.current) return;
     fetchedRef.current = true;
@@ -11306,7 +11322,7 @@ var AssistantFlowInner = ({ onChat }) => {
       }
     }
   }, [assistants, selectedAssistant, configPanelVisible]);
-  const lastCreatedAssistantRef = useRef9(null);
+  const lastCreatedAssistantRef = useRef10(null);
   useEffect21(() => {
     if (lastCreatedAssistantRef.current && !assistants.find((a) => a.id === lastCreatedAssistantRef.current?.id)) {
       lastCreatedAssistantRef.current = null;
@@ -13408,7 +13424,7 @@ var ToolsList = ({
 };
 
 // src/components/Chat/TopologyRuntimeView.tsx
-import { useEffect as useEffect45, useState as useState68, useMemo as useMemo27, useCallback as useCallback36, useRef as useRef24 } from "react";
+import { useEffect as useEffect45, useState as useState68, useMemo as useMemo27, useCallback as useCallback36, useRef as useRef25 } from "react";
 import {
   Spin as Spin16,
   Empty as Empty14,
@@ -13451,7 +13467,7 @@ import dagre2 from "dagre";
 
 // src/components/GenUI/MDResponse.tsx
 import XMarkdown from "@ant-design/x-markdown";
-import React54, { useRef as useRef23, useState as useState66, useMemo as useMemo24, useDeferredValue } from "react";
+import React54, { useRef as useRef24, useState as useState66, useMemo as useMemo24, useDeferredValue } from "react";
 import { createStyles as createStyles30 } from "antd-style";
 
 // src/components/GenUI/elements/confirm_feedback.tsx
@@ -14505,7 +14521,7 @@ import {
 import { createStyles as createStyles10 } from "antd-style";
 
 // src/components/GenUI/MarkdownViewer.tsx
-import React24, { useState as useState40, useEffect as useEffect26, useRef as useRef10 } from "react";
+import React24, { useState as useState40, useEffect as useEffect26, useRef as useRef11 } from "react";
 import { Button as Button21, Tooltip as Tooltip4, message as message9 } from "antd";
 import {
   CopyOutlined as CopyOutlined2,
@@ -14779,7 +14795,7 @@ var MarkdownViewer = ({
 }) => {
   const { styles, cx } = useStyles3();
   const [copied, setCopied] = useState40(false);
-  const contentRef = useRef10(null);
+  const contentRef = useRef11(null);
   useEffect26(() => {
     if (copied) {
       const timer = setTimeout(() => setCopied(false), 2e3);
@@ -15759,7 +15775,7 @@ function AttachmentsViewerSideApp({
 import { Button as Button26, Space as Space17, Typography as Typography22 } from "antd";
 
 // src/components/GenUI/elements/ContentPreviewCollapse.tsx
-import { useRef as useRef11, useState as useState45, useEffect as useEffect30, useCallback as useCallback23 } from "react";
+import { useRef as useRef12, useState as useState45, useEffect as useEffect30, useCallback as useCallback23 } from "react";
 import { Collapse as Collapse5 } from "antd";
 import { createStyles as createStyles12 } from "antd-style";
 import { DownOutlined as DownOutlined3, UpOutlined as UpOutlined2 } from "@ant-design/icons";
@@ -15831,7 +15847,7 @@ var ContentPreviewCollapse = ({
 }) => {
   const [showFullContent, setShowFullContent] = useState45(false);
   const [isOverflowing, setIsOverflowing] = useState45(false);
-  const contentRef = useRef11(null);
+  const contentRef = useRef12(null);
   const showShadow = isOverflowing && !showFullContent;
   const { styles, cx } = useStyle5({ showShadow });
   const checkOverflow = useCallback23(() => {
@@ -16336,7 +16352,7 @@ import {
   useCallback as useCallback24,
   useEffect as useEffect31,
   useMemo as useMemo11,
-  useRef as useRef12,
+  useRef as useRef13,
   useState as useState46
 } from "react";
 import { jsx as jsx52 } from "react/jsx-runtime";
@@ -16345,7 +16361,7 @@ var LazyBubble = ({
   renderContent,
   autoLoadRightPanel
 }) => {
-  const ref = useRef12(null);
+  const ref = useRef13(null);
   const [isVisible, setIsVisible] = useState46(false);
   const [wasEverVisible, setWasEverVisible] = useState46(false);
   useEffect31(() => {
@@ -16400,7 +16416,7 @@ var MessageList = ({
 }) => {
   const { styles } = useStyle();
   const openSideApp = useSideAppOpener();
-  const messageLengthRef = useRef12(messages?.length ?? 0);
+  const messageLengthRef = useRef13(messages?.length ?? 0);
   useEffect31(() => {
     if (messages?.length) {
       messageLengthRef.current = messages?.length;
@@ -16511,7 +16527,7 @@ import {
   Space as Space28,
   Typography as Typography33
 } from "antd";
-import React38, { useCallback as useCallback28, useContext as useContext10, useEffect as useEffect37, useRef as useRef18, useState as useState56 } from "react";
+import React38, { useCallback as useCallback28, useContext as useContext10, useEffect as useEffect37, useRef as useRef19, useState as useState56 } from "react";
 import { BrainCircuit as BrainCircuit3 } from "lucide-react";
 
 // src/components/GenUI/HITLContainer.tsx
@@ -17569,7 +17585,7 @@ var ThreadManagementButtons = () => {
 };
 
 // src/components/Chat/DatabasePicker.tsx
-import { useRef as useRef13, useState as useState49 } from "react";
+import { useRef as useRef14, useState as useState49 } from "react";
 import { Modal as Modal10, List as List7, Checkbox as Checkbox5, Spin as Spin8, Empty as Empty5, Typography as Typography27, Button as Button35, Space as Space23, Tooltip as Tooltip13 } from "antd";
 import { Database as Database4 } from "lucide-react";
 import { Fragment as Fragment12, jsx as jsx66, jsxs as jsxs45 } from "react/jsx-runtime";
@@ -17579,7 +17595,7 @@ var DatabasePicker = ({ senderRef, iconOnly }) => {
   const [loading, setLoading] = useState49(false);
   const [selectedDatabases, setSelectedDatabases] = useState49([]);
   const { get } = useApi();
-  const fetchedRef = useRef13(false);
+  const fetchedRef = useRef14(false);
   const loadDatabases = async () => {
     setLoading(true);
     try {
@@ -17719,7 +17735,7 @@ var DatabasePicker = ({ senderRef, iconOnly }) => {
 };
 
 // src/components/Chat/SkillPicker.tsx
-import { useRef as useRef14, useState as useState50 } from "react";
+import { useRef as useRef15, useState as useState50 } from "react";
 import { Modal as Modal11, List as List8, Checkbox as Checkbox6, Spin as Spin9, Empty as Empty6, Typography as Typography28, Button as Button36, Space as Space24, Tooltip as Tooltip14 } from "antd";
 import { BrainCircuit } from "lucide-react";
 import { Fragment as Fragment13, jsx as jsx67, jsxs as jsxs46 } from "react/jsx-runtime";
@@ -17729,7 +17745,7 @@ var SkillPicker = ({ senderRef, iconOnly }) => {
   const [loading, setLoading] = useState50(false);
   const [selectedSkills, setSelectedSkills] = useState50([]);
   const { get } = useApi();
-  const fetchedRef = useRef14(false);
+  const fetchedRef = useRef15(false);
   const loadSkills = async () => {
     setLoading(true);
     try {
@@ -17869,7 +17885,7 @@ var SkillPicker = ({ senderRef, iconOnly }) => {
 };
 
 // src/components/Chat/AgentPicker.tsx
-import { useRef as useRef15, useState as useState51 } from "react";
+import { useRef as useRef16, useState as useState51 } from "react";
 import { Modal as Modal12, List as List9, Empty as Empty7, Typography as Typography29, Button as Button37, Tooltip as Tooltip15 } from "antd";
 import { Bot as Bot2 } from "lucide-react";
 import { Fragment as Fragment14, jsx as jsx68, jsxs as jsxs47 } from "react/jsx-runtime";
@@ -17878,7 +17894,7 @@ var AgentPicker = ({ senderRef, iconOnly }) => {
   const [loading, setLoading] = useState51(false);
   const [selectedAgent, setSelectedAgent] = useState51(null);
   const { assistants, currentAssistant, selectAssistant } = useAssistantContext();
-  const fetchedRef = useRef15(false);
+  const fetchedRef = useRef16(false);
   const handleOpenModal = () => {
     setSelectedAgent(currentAssistant?.id || null);
     setModalOpen(true);
@@ -17985,7 +18001,7 @@ var AgentPicker = ({ senderRef, iconOnly }) => {
 };
 
 // src/components/Chat/MetricsDataSourcePicker.tsx
-import { useEffect as useEffect34, useState as useState52, useRef as useRef16 } from "react";
+import { useEffect as useEffect34, useState as useState52, useRef as useRef17 } from "react";
 import { Modal as Modal13, List as List10, Spin as Spin11, Empty as Empty8, Typography as Typography30, Button as Button38, Tag as Tag14, Tooltip as Tooltip16 } from "antd";
 import { Database as Database5, Check as Check4, Server as Server2 } from "lucide-react";
 import { Fragment as Fragment15, jsx as jsx69, jsxs as jsxs48 } from "react/jsx-runtime";
@@ -18000,7 +18016,7 @@ var MetricsDataSourcePicker = ({
   const { config } = useLatticeChatShellContext();
   const { customRunConfig, updateCustomRunConfig } = useConversationContext();
   const { get } = useApi();
-  const hasInitializedRef = useRef16(false);
+  const hasInitializedRef = useRef17(false);
   const loadDataSources = async () => {
     setLoading(true);
     try {
@@ -18325,7 +18341,7 @@ var MetricsDataSourcePicker = ({
 };
 
 // src/components/Chat/ModelSelector.tsx
-import { useState as useState53, useEffect as useEffect35, useCallback as useCallback27, useRef as useRef17 } from "react";
+import { useState as useState53, useEffect as useEffect35, useCallback as useCallback27, useRef as useRef18 } from "react";
 import { Select as Select6 } from "antd";
 import { jsx as jsx70 } from "react/jsx-runtime";
 var STORAGE_KEY = "axiom-lattice:selected-model";
@@ -18353,8 +18369,8 @@ var ModelSelector = ({
   const [internalValue, setInternalValue] = useState53(null);
   const [dropdownOpen, setDropdownOpen] = useState53(false);
   const [isHovered, setIsHovered] = useState53(false);
-  const hasFetchedRef = useRef17(false);
-  const hasSetDefaultRef = useRef17(false);
+  const hasFetchedRef = useRef18(false);
+  const hasSetDefaultRef = useRef18(false);
   const { get } = useApi();
   const selectedModelConfig = value !== void 0 ? value : internalValue;
   const setSelectedModelConfig = useCallback27((config) => {
@@ -19123,7 +19139,7 @@ var Chating = ({
   const [attachedFiles, setAttachedFiles] = useState56([]);
   const { styles } = useStyle();
   const [headerOpen, setHeaderOpen] = useState56(false);
-  const attachmentsRef = useRef18(null);
+  const attachmentsRef = useRef19(null);
   const senderRef = React38.useRef(null);
   const {
     assistantId,
@@ -19156,9 +19172,9 @@ var Chating = ({
     return () => clearInterval(interval);
   }, [isStreaming]);
   const conversationContext = useConversationContext();
-  const systemContextSentRef = useRef18(false);
-  const initialMessageSentRef = useRef18(false);
-  const prevLoadingRef = useRef18(false);
+  const systemContextSentRef = useRef19(false);
+  const initialMessageSentRef = useRef19(false);
+  const prevLoadingRef = useRef19(false);
   useEffect37(() => {
     systemContextSentRef.current = false;
   }, [threadId]);
@@ -23868,7 +23884,7 @@ var TaskBoard = ({
 };
 
 // src/components/GenUI/elements/Mailbox.tsx
-import { useState as useState63, useMemo as useMemo20, useRef as useRef19, useEffect as useEffect39 } from "react";
+import { useState as useState63, useMemo as useMemo20, useRef as useRef20, useEffect as useEffect39 } from "react";
 import { jsx as jsx91, jsxs as jsxs68 } from "react/jsx-runtime";
 var useStyle12 = () => {
   return {
@@ -24155,7 +24171,7 @@ var TeamChat = ({ data }) => {
   const styles = useStyle12();
   const { teamName, currentUser, teammates, messages, onSendMessage } = data || {};
   const [inputValue, setInputValue] = useState63("");
-  const messagesEndRef = useRef19(null);
+  const messagesEndRef = useRef20(null);
   const sortedMessages = useMemo20(() => {
     return [...messages || []].sort(
       (a, b) => new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime()
@@ -24338,7 +24354,7 @@ import { Button as Button46, Typography as Typography47 } from "antd";
 import { ExpandOutlined as ExpandOutlined2 } from "@ant-design/icons";
 
 // src/streaming-html/StreamingHTMLRenderer.tsx
-import React48, { useEffect as useEffect40, useRef as useRef20, useCallback as useCallback30, useState as useState64 } from "react";
+import React48, { useEffect as useEffect40, useRef as useRef21, useCallback as useCallback30, useState as useState64 } from "react";
 
 // src/streaming-html/show-widget-css-generator.ts
 function generateShowWidgetCSS(tokens) {
@@ -25029,14 +25045,14 @@ var StreamingHTMLRenderer = ({
   title,
   loadingMessages
 }) => {
-  const iframeRef = useRef20(null);
-  const containerRef = useRef20(null);
-  const resizeObserverRef = useRef20(null);
-  const prevHTMLRef = useRef20("");
-  const isReadyRef = useRef20(false);
-  const pendingChunksRef = useRef20([]);
-  const isCompleteRef = useRef20(isComplete);
-  const isScriptExecuted = useRef20(false);
+  const iframeRef = useRef21(null);
+  const containerRef = useRef21(null);
+  const resizeObserverRef = useRef21(null);
+  const prevHTMLRef = useRef21("");
+  const isReadyRef = useRef21(false);
+  const pendingChunksRef = useRef21([]);
+  const isCompleteRef = useRef21(isComplete);
+  const isScriptExecuted = useRef21(false);
   const [iframeHeight, setIframeHeight] = React48.useState(0);
   const [iframeWidth, setIframeWidth] = React48.useState(void 0);
   const [currentMessageIndex, setCurrentMessageIndex] = useState64(0);
@@ -25531,12 +25547,12 @@ import { CodeHighlighter, Mermaid } from "@ant-design/x";
 
 // src/components/GenUI/ReactInfographic.tsx
 import { Infographic } from "@antv/infographic";
-import { useRef as useRef21, useEffect as useEffect41 } from "react";
+import { useRef as useRef22, useEffect as useEffect41 } from "react";
 import { jsx as jsx95 } from "react/jsx-runtime";
 var ReactInfographic = (props) => {
   const { children } = props;
-  const $container = useRef21(null);
-  const infographicInstance = useRef21(null);
+  const $container = useRef22(null);
+  const infographicInstance = useRef22(null);
   useEffect41(() => {
     if ($container.current) {
       infographicInstance.current = new Infographic({
@@ -25568,7 +25584,7 @@ data
 };
 
 // src/components/GenUI/ReactChart.tsx
-import { useEffect as useEffect42, useRef as useRef22, useState as useState65 } from "react";
+import { useEffect as useEffect42, useRef as useRef23, useState as useState65 } from "react";
 import * as echarts from "echarts";
 import { Card as Card22 } from "antd";
 import { parse } from "best-effort-json-parser";
@@ -25881,9 +25897,9 @@ function parseAndValidateChartData(input) {
 }
 var ReactChart = (props) => {
   const { children } = props;
-  const chartRef = useRef22(null);
-  const chartInstance = useRef22(null);
-  const resizeObserverRef = useRef22(null);
+  const chartRef = useRef23(null);
+  const chartInstance = useRef23(null);
+  const resizeObserverRef = useRef23(null);
   const [tableData, setTableData] = useState65(null);
   useEffect42(() => {
     if (!chartRef.current) {
@@ -28154,8 +28170,8 @@ var RunDetail = ({ run, agentName, open, onClose, onRunUpdate, autoRefresh, onAu
   const [loading, setLoading] = useState68(false);
   const [graphDefinition, setGraphDefinition] = useState68(null);
   const [graphLoading, setGraphLoading] = useState68(false);
-  const stepsPollTimeoutRef = useRef24(null);
-  const stepsPollSessionRef = useRef24(0);
+  const stepsPollTimeoutRef = useRef25(null);
+  const stepsPollSessionRef = useRef25(0);
   const fetchSteps = useCallback36(async () => {
     try {
       const [stepsRes, runsRes] = await Promise.all([
@@ -28268,7 +28284,7 @@ var TopologyRuntimeView = () => {
   const [error, setError] = useState68(null);
   const [selectedRun, setSelectedRun] = useState68(null);
   const [autoRefresh, setAutoRefresh] = useState68(true);
-  const pollingSessionRef = useRef24(0);
+  const pollingSessionRef = useRef25(0);
   const handleRunUpdate = useCallback36((updated) => {
     setRuns(
       (prev) => prev.map((r) => r.id === updated.id ? updated : r)
@@ -28541,7 +28557,7 @@ var TopologyInboxView = () => {
 };
 
 // src/components/Chat/WorkflowAutomationView.tsx
-import { useEffect as useEffect48, useState as useState70, useCallback as useCallback37, useMemo as useMemo30, useRef as useRef25 } from "react";
+import { useEffect as useEffect48, useState as useState70, useCallback as useCallback37, useMemo as useMemo30, useRef as useRef26 } from "react";
 import {
   ReactFlowProvider as ReactFlowProvider6
 } from "@xyflow/react";
@@ -29559,7 +29575,7 @@ var WorkflowAutomationView = () => {
   }, [get]);
   const [copilotOpen, setCopilotOpen] = useState70(true);
   const [copilotWidth, setCopilotWidth] = useState70(420);
-  const resizingRef = useRef25(null);
+  const resizingRef = useRef26(null);
   const [initialLoading, setInitialLoading] = useState70(true);
   const [error, setError] = useState70(null);
   const [specModalOpen, setSpecModalOpen] = useState70(false);
@@ -30917,7 +30933,7 @@ var EvalPanel = () => {
 };
 
 // src/components/Chat/PersonalAssistantPage.tsx
-import { useEffect as useEffect51, useState as useState75, useCallback as useCallback39, useRef as useRef27 } from "react";
+import { useEffect as useEffect51, useState as useState75, useCallback as useCallback39, useRef as useRef28 } from "react";
 import { Card as Card29, Button as Button55, Input as Input17, Typography as Typography60, message as message19, Dropdown as Dropdown3, Modal as Modal20 } from "antd";
 import { createStyles as createStyles36 } from "antd-style";
 import { Bot as Bot4, Sparkles as Sparkles3, ArrowRight as ArrowRight4, ArrowLeft as ArrowLeft3, Smile, Zap, Heart, Lightbulb, Edit3, MoreHorizontal, Link2 as Link23, Trash2 as Trash29 } from "lucide-react";
@@ -30933,7 +30949,7 @@ var LatticeAgentWorkspace = ({
 }) => /* @__PURE__ */ jsx117(WorkspaceContextProvider, { workspaceId, projectId, children: /* @__PURE__ */ jsx117(AssistantContextProvider, { autoLoad: true, initialAssistantId: assistant_id, children: /* @__PURE__ */ jsx117(LatticeChat, { showProjectSelector: false, assistant_id, ...chatProps }) }) });
 
 // src/components/Chat/PersonalAssistantChannelModal.tsx
-import { useEffect as useEffect50, useState as useState74, useCallback as useCallback38, useMemo as useMemo32, useRef as useRef26 } from "react";
+import { useEffect as useEffect50, useState as useState74, useCallback as useCallback38, useMemo as useMemo32, useRef as useRef27 } from "react";
 import {
   Button as Button54,
   Form as Form9,
@@ -31180,25 +31196,30 @@ var PersonalAssistantChannelModal = ({
   const [qrLoading, setQrLoading] = useState74(false);
   const [qrCodeUrl, setQrCodeUrl] = useState74(null);
   const [qrStatus, setQrStatus] = useState74("idle");
-  const qrCodeRef = useRef26(null);
-  const qrPollRef = useRef26(null);
-  const clearQrPoll = useCallback38(() => {
-    if (qrPollRef.current) {
-      clearInterval(qrPollRef.current);
-      qrPollRef.current = null;
+  const [qrCountdown, setQrCountdown] = useState74(0);
+  const qrCodeRef = useRef27(null);
+  const qrCountdownRef = useRef27(null);
+  const clearQrCountdown = useCallback38(() => {
+    if (qrCountdownRef.current) {
+      clearInterval(qrCountdownRef.current);
+      qrCountdownRef.current = null;
     }
   }, []);
   const resetQrState = useCallback38(() => {
-    clearQrPoll();
+    clearQrCountdown();
     setQrLoading(false);
     setQrCodeUrl(null);
     setQrStatus("idle");
+    setQrCountdown(0);
     qrCodeRef.current = null;
-  }, [clearQrPoll]);
+  }, [clearQrCountdown]);
+  const QR_EXPIRY_SECONDS = 300;
   const fetchQrCode = useCallback38(async () => {
     setQrLoading(true);
     setQrStatus("loading");
     setQrCodeUrl(null);
+    clearQrCountdown();
+    setQrCountdown(0);
     try {
       const res = await get(
         "/api/channels/wechat/setup/qrcode"
@@ -31207,32 +31228,17 @@ var PersonalAssistantChannelModal = ({
         qrCodeRef.current = res.data.qrcode;
         setQrCodeUrl(res.data.qrcodeImgUrl);
         setQrStatus("wait");
-        clearQrPoll();
-        qrPollRef.current = setInterval(async () => {
-          try {
-            const statusRes = await get(`/api/channels/wechat/setup/status?qrcode=${encodeURIComponent(qrCodeRef.current)}`);
-            if (statusRes.success && statusRes.data) {
-              const s = statusRes.data.status;
-              if (s === "confirmed" && statusRes.data.botToken) {
-                setQrStatus("confirmed");
-                clearQrPoll();
-                form.setFieldsValue({
-                  botToken: statusRes.data.botToken,
-                  uin: statusRes.data.uin
-                });
-                message18.success("WeChat login confirmed!");
-              } else if (s === "expired") {
-                setQrStatus("expired");
-                clearQrPoll();
-              } else if (s === "scaned") {
-                setQrStatus("scaned");
-              } else if (s === "wait") {
-                setQrStatus("wait");
-              }
+        setQrCountdown(QR_EXPIRY_SECONDS);
+        qrCountdownRef.current = setInterval(() => {
+          setQrCountdown((prev) => {
+            if (prev <= 1) {
+              clearQrCountdown();
+              setQrStatus("expired");
+              return 0;
             }
-          } catch {
-          }
-        }, 2e3);
+            return prev - 1;
+          });
+        }, 1e3);
       } else {
         setQrStatus("error");
         message18.error("Failed to get QR code");
@@ -31243,10 +31249,36 @@ var PersonalAssistantChannelModal = ({
     } finally {
       setQrLoading(false);
     }
-  }, [get, clearQrPoll, form]);
+  }, [get, clearQrCountdown]);
+  const checkQrStatus = useCallback38(async () => {
+    if (!qrCodeRef.current) return;
+    try {
+      const statusRes = await get(`/api/channels/wechat/setup/status?qrcode=${encodeURIComponent(qrCodeRef.current)}`);
+      if (statusRes.success && statusRes.data) {
+        const s = statusRes.data.status;
+        if (s === "confirmed" && statusRes.data.botToken) {
+          setQrStatus("confirmed");
+          clearQrCountdown();
+          form.setFieldsValue({
+            botToken: statusRes.data.botToken,
+            uin: statusRes.data.uin
+          });
+          message18.success("WeChat login confirmed!");
+        } else if (s === "expired") {
+          setQrStatus("expired");
+          clearQrCountdown();
+          setQrCountdown(0);
+        } else if (s === "scaned") {
+          setQrStatus("scaned");
+        }
+      }
+    } catch {
+      message18.error("Failed to check QR status");
+    }
+  }, [get, clearQrCountdown, form]);
   useEffect50(() => {
     return () => {
-      if (qrPollRef.current) clearInterval(qrPollRef.current);
+      if (qrCountdownRef.current) clearInterval(qrCountdownRef.current);
     };
   }, []);
   const loadInstallations = useCallback38(async () => {
@@ -31624,7 +31656,7 @@ var PersonalAssistantChannelModal = ({
                   /* @__PURE__ */ jsx118(Spin20, {}),
                   /* @__PURE__ */ jsx118(Text48, { type: "secondary", style: { display: "block", marginTop: 8, fontSize: 12 }, children: "Generating QR code..." })
                 ] }),
-                (qrStatus === "wait" || qrStatus === "scaned") && qrCodeUrl && /* @__PURE__ */ jsxs85("div", { children: [
+                (qrStatus === "wait" || qrStatus === "scaned") && qrCodeUrl && /* @__PURE__ */ jsxs85("div", { style: { display: "flex", flexDirection: "column", alignItems: "center" }, children: [
                   /* @__PURE__ */ jsx118(QRCode, { value: qrCodeUrl, size: 180 }),
                   /* @__PURE__ */ jsx118(
                     Text48,
@@ -31639,7 +31671,16 @@ var PersonalAssistantChannelModal = ({
                       children: qrStatus === "scaned" ? "Scanned! Confirm on your phone..." : "Scan with WeChat"
                     }
                   ),
-                  /* @__PURE__ */ jsx118(Text48, { type: "secondary", style: { fontSize: 11, marginTop: 4, display: "block" }, children: "QR expires in 5 minutes" })
+                  /* @__PURE__ */ jsx118(Text48, { type: "secondary", style: { fontSize: 11, marginTop: 4, display: "block" }, children: qrCountdown > 0 ? `Expires in ${Math.floor(qrCountdown / 60)}:${String(qrCountdown % 60).padStart(2, "0")}` : "QR code expired" }),
+                  /* @__PURE__ */ jsx118(
+                    Button54,
+                    {
+                      size: "small",
+                      style: { marginTop: 8 },
+                      onClick: checkQrStatus,
+                      children: "Refresh Status"
+                    }
+                  )
                 ] }),
                 qrStatus === "confirmed" && /* @__PURE__ */ jsxs85("div", { style: { padding: "12px 0" }, children: [
                   /* @__PURE__ */ jsx118(Text48, { strong: true, style: { fontSize: 14, color: "var(--color-success, #22c55e)", display: "block" }, children: "Login confirmed!" }),
@@ -31887,7 +31928,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] = useState75(false);
@@ -31914,7 +31955,7 @@ var PersonalAssistantPage = () => {
         message19.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);
       message19.success("Done!");
     } catch {
@@ -31937,7 +31978,7 @@ var PersonalAssistantPage = () => {
     );
     return created.data.id;
   }, [get, post, workspaceId, projectId]);
-  const lastDate = useRef27("");
+  const lastDate = useRef28("");
   useEffect51(() => {
     if (!assistantId) return;
     let timer;
@@ -32727,8 +32768,8 @@ var WorkspaceResourceManager = ({
   logo
 }) => {
   const { openContentApp, menuCollapsed, setMenuCollapsed } = useChatUIContext();
-  const hasOpenedDefault = useRef28(false);
-  const hasRegistered = useRef28(false);
+  const hasOpenedDefault = useRef29(false);
+  const hasRegistered = useRef29(false);
   useEffect53(() => {
     if (!hasRegistered.current) {
       hasRegistered.current = true;
@@ -33158,7 +33199,7 @@ var WorkspaceContextProvider = ({
       setWorkspaceId(firstWorkspace.id);
     }
   }, [workspaces, workspaceId]);
-  const prevWorkspaceId = useRef29(workspaceId);
+  const prevWorkspaceId = useRef30(workspaceId);
   useEffect54(() => {
     if (workspaceId) {
       refreshProjects(workspaceId);
@@ -33533,7 +33574,7 @@ var PinToMenuButton = ({ assistantId }) => {
 import { createStyles as createStyles40 } from "antd-style";
 
 // src/components/Chat/ProjectSelector.tsx
-import { useState as useState80, useCallback as useCallback42, useMemo as useMemo36, useRef as useRef30 } from "react";
+import { useState as useState80, useCallback as useCallback42, useMemo as useMemo36, useRef as useRef31 } from "react";
 import { Modal as Modal23, Input as Input19, Button as Button58, message as message21 } from "antd";
 import { createStyles as createStyles38 } from "antd-style";
 import { Folder, ChevronDown as ChevronDown6, Building2 as Building24 } from "lucide-react";
@@ -33770,13 +33811,13 @@ var ProjectSelector = ({
     createProject
   } = useWorkspaceContext();
   const [isWorkspaceListOpen, setIsWorkspaceListOpen] = useState80(false);
-  const workspaceDropdownRef = useRef30(null);
+  const workspaceDropdownRef = useRef31(null);
   const [isProjectListOpen, setIsProjectListOpen] = useState80(false);
   const [isModalOpen, setIsModalOpen] = useState80(false);
   const [projectName, setProjectName] = useState80("");
   const [validationError, setValidationError] = useState80(null);
   const [isCreating, setIsCreating] = useState80(false);
-  const projectNameInputRef = useRef30(null);
+  const projectNameInputRef = useRef31(null);
   const currentProject = useMemo36(() => {
     return projects.find((p) => p.id === projectId);
   }, [projects, projectId]);
@@ -35015,7 +35056,7 @@ var LatticeChatView = (props) => {
 };
 
 // src/components/Chat/SettingsModal.tsx
-import { useState as useState83, useEffect as useEffect57, useRef as useRef31 } from "react";
+import { useState as useState83, useEffect as useEffect57, useRef as useRef32 } from "react";
 import {
   Modal as Modal25,
   Input as Input20,
@@ -35428,7 +35469,7 @@ var SettingsModal = ({
     return [];
   });
   const [serverConfigs, setServerConfigs] = useState83({});
-  const connectionsRef = useRef31(connections);
+  const connectionsRef = useRef32(connections);
   useEffect57(() => {
     connectionsRef.current = connections;
   }, [connections]);