@axiom-lattice/react-sdk 2.1.68 → 2.1.70

package/dist/index.js

@@ -408,16 +408,22 @@ function AxiomLatticeProvider({
 }) {
   const authContext = (0, import_react2.useContext)(AuthContext);
   const tenantId = authContext?.currentTenant?.id;
+  const effectiveApiKey = (0, import_react2.useMemo)(() => {
+    if (authContext?.isAuthenticated) {
+      return sessionStorage.getItem("lattice_token") || config.apiKey;
+    }
+    return config.apiKey;
+  }, [authContext?.isAuthenticated, authContext?.currentTenant?.id, config.apiKey]);
   const clientCacheRef = (0, import_react2.useRef)(/* @__PURE__ */ new Map());
   const baseConfig = (0, import_react2.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 = (0, import_react2.useMemo)(() => {
     const client = new import_client_sdk.Client(baseConfig);
@@ -18919,7 +18925,7 @@ var useWorkspaceContext = () => {
 var WorkspaceContextProvider = ({
   children
 }) => {
-  const { currentTenant } = useAuth();
+  const { currentTenant, isAuthenticated } = useAuth();
   const tenantId = currentTenant?.id || "";
   const { config } = useLatticeChatShellContext();
   const axiomClient = useClient("__GLOBAL__");
@@ -18952,13 +18958,14 @@ var WorkspaceContextProvider = ({
   const [loading, setLoading] = (0, import_react61.useState)(false);
   const [error, setError] = (0, import_react61.useState)(null);
   const client = import_react61.default.useMemo(() => {
+    const authApiKey = isAuthenticated ? sessionStorage.getItem("lattice_token") || config.apiKey || "" : config.apiKey || "";
     return new import_client_sdk6.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 = (0, import_react61.useCallback)(() => {
     setProjects([]);
     setWorkspaceId(null);