@axiom-lattice/react-sdk 2.1.68 → 2.1.70

package/dist/index.mjs

@@ -307,16 +307,22 @@ function AxiomLatticeProvider({
 }) {
   const authContext = useContext2(AuthContext);
   const tenantId = authContext?.currentTenant?.id;
+  const effectiveApiKey = useMemo(() => {
+    if (authContext?.isAuthenticated) {
+      return sessionStorage.getItem("lattice_token") || config.apiKey;
+    }
+    return config.apiKey;
+  }, [authContext?.isAuthenticated, authContext?.currentTenant?.id, config.apiKey]);
   const clientCacheRef = useRef(/* @__PURE__ */ new Map());
   const baseConfig = useMemo(
     () => ({
       baseURL: config.baseURL,
-      apiKey: config.apiKey,
+      apiKey: effectiveApiKey,
       assistantId: "__BASE__",
       // 占位符，不会被直接使用
       transport: config.transport
     }),
-    [config.baseURL, config.apiKey, config.transport]
+    [config.baseURL, effectiveApiKey, config.transport]
   );
   const baseClient = useMemo(() => {
     const client = new Client(baseConfig);
@@ -19083,7 +19089,7 @@ var useWorkspaceContext = () => {
 var WorkspaceContextProvider = ({
   children
 }) => {
-  const { currentTenant } = useAuth();
+  const { currentTenant, isAuthenticated } = useAuth();
   const tenantId = currentTenant?.id || "";
   const { config } = useLatticeChatShellContext();
   const axiomClient = useClient("__GLOBAL__");
@@ -19116,13 +19122,14 @@ var WorkspaceContextProvider = ({
   const [loading, setLoading] = useState49(false);
   const [error, setError] = useState49(null);
   const client = React38.useMemo(() => {
+    const authApiKey = isAuthenticated ? sessionStorage.getItem("lattice_token") || config.apiKey || "" : config.apiKey || "";
     return new WorkspaceClient({
       baseURL: config.baseURL,
-      apiKey: config.apiKey || "",
+      apiKey: authApiKey,
       assistantId: "",
       transport: "sse"
     }, tenantId);
-  }, [config.baseURL, config.apiKey, tenantId]);
+  }, [config.baseURL, config.apiKey, tenantId, isAuthenticated]);
   const resetSelectedWorkspace = useCallback26(() => {
     setProjects([]);
     setWorkspaceId(null);