@sanity/sdk 2.3.1 → 2.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/sdk",
-  "version": "2.3.1",
+  "version": "2.5.0",
   "private": false,
   "description": "Sanity SDK",
   "keywords": [
@@ -18,7 +18,8 @@
   },
   "repository": {
     "type": "git",
-    "url": "git+ssh://git@github.com/sanity-io/sdk.git"
+    "url": "git+https://github.com/sanity-io/sdk.git",
+    "directory": "packages/core"
   },
   "license": "MIT",
   "author": "Sanity <developers@sanity.io>",
@@ -48,7 +49,7 @@
     "@sanity/diff-match-patch": "^3.2.0",
     "@sanity/diff-patch": "^6.0.0",
     "@sanity/json-match": "^1.0.5",
-    "@sanity/message-protocol": "^0.12.0",
+    "@sanity/message-protocol": "^0.18.0",
     "@sanity/mutate": "^0.12.4",
     "@sanity/types": "^3.83.0",
     "groq": "3.88.1-typegen-experimental.0",
@@ -60,24 +61,24 @@
   },
   "devDependencies": {
     "@sanity/browserslist-config": "^1.0.5",
-    "@sanity/pkg-utils": "^7.9.6",
-    "@sanity/prettier-config": "^1.0.3",
+    "@sanity/pkg-utils": "^8.1.29",
+    "@sanity/prettier-config": "^1.0.6",
     "@types/lodash-es": "^4.17.12",
-    "@vitest/coverage-v8": "3.1.2",
+    "@vitest/coverage-v8": "3.2.4",
     "eslint": "^9.22.0",
-    "prettier": "^3.5.3",
+    "prettier": "^3.7.3",
     "rollup-plugin-visualizer": "^5.14.0",
     "typescript": "^5.8.3",
     "vite": "^6.3.4",
-    "vitest": "^3.1.2",
+    "vitest": "^3.2.4",
     "@repo/config-eslint": "0.0.0",
+    "@repo/config-test": "0.0.1",
     "@repo/package.bundle": "3.82.0",
     "@repo/package.config": "0.0.1",
-    "@repo/tsconfig": "0.0.1",
-    "@repo/config-test": "0.0.1"
+    "@repo/tsconfig": "0.0.1"
   },
   "engines": {
-    "node": ">=20.0.0"
+    "node": ">=20.19"
   },
   "publishConfig": {
     "access": "public"

package/src/_exports/index.ts

@@ -5,6 +5,25 @@ import {type SanityProject as _SanityProject} from '@sanity/client'
  */
 export type SanityProject = _SanityProject
 
+export type {
+  AgentGenerateOptions,
+  AgentGenerateResult,
+  AgentPatchOptions,
+  AgentPatchResult,
+  AgentPromptOptions,
+  AgentPromptResult,
+  AgentTransformOptions,
+  AgentTransformResult,
+  AgentTranslateOptions,
+  AgentTranslateResult,
+} from '../agent/agentActions'
+export {
+  agentGenerate,
+  agentPatch,
+  agentPrompt,
+  agentTransform,
+  agentTranslate,
+} from '../agent/agentActions'
 export {AuthStateType} from '../auth/authStateType'
 export {
   type AuthState,
@@ -24,6 +43,13 @@ export {
 export {observeOrganizationVerificationState} from '../auth/getOrganizationVerificationState'
 export {handleAuthCallback} from '../auth/handleAuthCallback'
 export {logout} from '../auth/logout'
+export {
+  type ApiErrorBody,
+  getClientErrorApiBody,
+  getClientErrorApiDescription,
+  getClientErrorApiType,
+  isProjectUserNotFoundClientError,
+} from '../auth/utils'
 export type {ClientStoreState as ClientState} from '../client/clientStore'
 export {type ClientOptions, getClient, getClientState} from '../client/clientStore'
 export {
@@ -50,9 +76,13 @@ export {
   createProjectHandle,
 } from '../config/handles'
 export {
+  canvasSource,
   type DatasetHandle,
+  datasetSource,
   type DocumentHandle,
+  type DocumentSource,
   type DocumentTypeHandle,
+  mediaLibrarySource,
   type PerspectiveHandle,
   type ProjectHandle,
   type ReleasePerspective,

package/src/agent/agentActions.test.ts

@@ -0,0 +1,81 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import {firstValueFrom, of} from 'rxjs'
+import {beforeEach, describe, expect, it, vi} from 'vitest'
+
+import {
+  agentGenerate,
+  agentPatch,
+  agentPrompt,
+  agentTransform,
+  agentTranslate,
+} from './agentActions'
+
+let mockClient: any
+
+vi.mock('../client/clientStore', () => {
+  return {
+    getClientState: () => ({observable: of(mockClient)}),
+  }
+})
+
+describe('agent actions', () => {
+  beforeEach(() => {
+    mockClient = {
+      observable: {
+        agent: {
+          action: {
+            generate: vi.fn(),
+            transform: vi.fn(),
+            translate: vi.fn(),
+          },
+        },
+      },
+      agent: {
+        action: {
+          prompt: vi.fn(),
+          patch: vi.fn(),
+        },
+      },
+    }
+  })
+
+  it('agentGenerate returns observable from client', async () => {
+    mockClient.observable.agent.action.generate.mockReturnValue(of('gen'))
+    const instance = {config: {projectId: 'p', dataset: 'd'}} as any
+    const value = await firstValueFrom(agentGenerate(instance, {foo: 'bar'} as any))
+    expect(value).toBe('gen')
+    expect(mockClient.observable.agent.action.generate).toHaveBeenCalledWith({foo: 'bar'})
+  })
+
+  it('agentTransform returns observable from client', async () => {
+    mockClient.observable.agent.action.transform.mockReturnValue(of('xform'))
+    const instance = {config: {projectId: 'p', dataset: 'd'}} as any
+    const value = await firstValueFrom(agentTransform(instance, {a: 1} as any))
+    expect(value).toBe('xform')
+    expect(mockClient.observable.agent.action.transform).toHaveBeenCalledWith({a: 1})
+  })
+
+  it('agentTranslate returns observable from client', async () => {
+    mockClient.observable.agent.action.translate.mockReturnValue(of('xlate'))
+    const instance = {config: {projectId: 'p', dataset: 'd'}} as any
+    const value = await firstValueFrom(agentTranslate(instance, {b: 2} as any))
+    expect(value).toBe('xlate')
+    expect(mockClient.observable.agent.action.translate).toHaveBeenCalledWith({b: 2})
+  })
+
+  it('agentPrompt wraps promise into observable', async () => {
+    mockClient.agent.action.prompt.mockResolvedValue('prompted')
+    const instance = {config: {projectId: 'p', dataset: 'd'}} as any
+    const value = await firstValueFrom(agentPrompt(instance, {p: true} as any))
+    expect(value).toBe('prompted')
+    expect(mockClient.agent.action.prompt).toHaveBeenCalledWith({p: true})
+  })
+
+  it('agentPatch wraps promise into observable', async () => {
+    mockClient.agent.action.patch.mockResolvedValue('patched')
+    const instance = {config: {projectId: 'p', dataset: 'd'}} as any
+    const value = await firstValueFrom(agentPatch(instance, {q: false} as any))
+    expect(value).toBe('patched')
+    expect(mockClient.agent.action.patch).toHaveBeenCalledWith({q: false})
+  })
+})

package/src/agent/agentActions.ts

@@ -0,0 +1,139 @@
+import {type SanityClient} from '@sanity/client'
+import {from, Observable, switchMap} from 'rxjs'
+
+import {getClientState} from '../client/clientStore'
+import {type SanityInstance} from '../store/createSanityInstance'
+
+const API_VERSION = 'vX'
+
+/** @alpha */
+export type AgentGenerateOptions = Parameters<
+  SanityClient['observable']['agent']['action']['generate']
+>[0]
+
+/** @alpha */
+export type AgentTransformOptions = Parameters<
+  SanityClient['observable']['agent']['action']['transform']
+>[0]
+
+/** @alpha */
+export type AgentTranslateOptions = Parameters<
+  SanityClient['observable']['agent']['action']['translate']
+>[0]
+
+/** @alpha */
+export type AgentPromptOptions = Parameters<SanityClient['agent']['action']['prompt']>[0]
+
+/** @alpha */
+export type AgentPatchOptions = Parameters<SanityClient['agent']['action']['patch']>[0]
+
+/** @alpha */
+export type AgentGenerateResult = Awaited<
+  ReturnType<SanityClient['observable']['agent']['action']['generate']>
+>
+
+/** @alpha */
+export type AgentTransformResult = Awaited<
+  ReturnType<SanityClient['observable']['agent']['action']['transform']>
+>
+
+/** @alpha */
+export type AgentTranslateResult = Awaited<
+  ReturnType<SanityClient['observable']['agent']['action']['translate']>
+>
+
+/** @alpha */
+export type AgentPromptResult = Awaited<ReturnType<SanityClient['agent']['action']['prompt']>>
+
+/** @alpha */
+export type AgentPatchResult = Awaited<ReturnType<SanityClient['agent']['action']['patch']>>
+
+/**
+ * Generates a new document using the agent.
+ * @param instance - The Sanity instance.
+ * @param options - The options for the agent generate action. See the [Agent Actions API](https://www.sanity.io/docs/agent-actions/introduction) for more details.
+ * @returns An Observable emitting the result of the agent generate action.
+ * @alpha
+ */
+export function agentGenerate(
+  instance: SanityInstance,
+  options: AgentGenerateOptions,
+): AgentGenerateResult {
+  return getClientState(instance, {
+    apiVersion: API_VERSION,
+    projectId: instance.config.projectId,
+    dataset: instance.config.dataset,
+  }).observable.pipe(switchMap((client) => client.observable.agent.action.generate(options)))
+}
+
+/**
+ * Transforms a document using the agent.
+ * @param instance - The Sanity instance.
+ * @param options - The options for the agent transform action. See the [Agent Actions API](https://www.sanity.io/docs/agent-actions/introduction) for more details.
+ * @returns An Observable emitting the result of the agent transform action.
+ * @alpha
+ */
+export function agentTransform(
+  instance: SanityInstance,
+  options: AgentTransformOptions,
+): AgentTransformResult {
+  return getClientState(instance, {
+    apiVersion: API_VERSION,
+    projectId: instance.config.projectId,
+    dataset: instance.config.dataset,
+  }).observable.pipe(switchMap((client) => client.observable.agent.action.transform(options)))
+}
+
+/**
+ * Translates a document using the agent.
+ * @param instance - The Sanity instance.
+ * @param options - The options for the agent translate action. See the [Agent Actions API](https://www.sanity.io/docs/agent-actions/introduction) for more details.
+ * @returns An Observable emitting the result of the agent translate action.
+ * @alpha
+ */
+export function agentTranslate(
+  instance: SanityInstance,
+  options: AgentTranslateOptions,
+): AgentTranslateResult {
+  return getClientState(instance, {
+    apiVersion: API_VERSION,
+    projectId: instance.config.projectId,
+    dataset: instance.config.dataset,
+  }).observable.pipe(switchMap((client) => client.observable.agent.action.translate(options)))
+}
+
+/**
+ * Prompts the agent using the same instruction template format as the other actions, but returns text or json instead of acting on a document.
+ * @param instance - The Sanity instance.
+ * @param options - The options for the agent prompt action. See the [Agent Actions API](https://www.sanity.io/docs/agent-actions/introduction) for more details.
+ * @returns An Observable emitting the result of the agent prompt action.
+ * @alpha
+ */
+export function agentPrompt(
+  instance: SanityInstance,
+  options: AgentPromptOptions,
+): Observable<AgentPromptResult> {
+  return getClientState(instance, {
+    apiVersion: API_VERSION,
+    projectId: instance.config.projectId,
+    dataset: instance.config.dataset,
+  }).observable.pipe(switchMap((client) => from(client.agent.action.prompt(options))))
+}
+
+/**
+ * Patches a document using the agent.
+ * @param instance - The Sanity instance.
+ * @param options - The options for the agent patch action. See the [Agent Actions API](https://www.sanity.io/docs/agent-actions/introduction) for more details.
+ * @returns An Observable emitting the result of the agent patch action.
+ * @alpha
+ */
+export function agentPatch(
+  instance: SanityInstance,
+  options: AgentPatchOptions,
+): Observable<AgentPatchResult> {
+  return getClientState(instance, {
+    apiVersion: API_VERSION,
+    projectId: instance.config.projectId,
+    dataset: instance.config.dataset,
+  }).observable.pipe(switchMap((client) => from(client.agent.action.patch(options))))
+}

package/src/auth/authStore.test.ts

@@ -92,7 +92,7 @@ describe('authStore', () => {
         },
       })
 
-      const {options, dashboardContext} = authStore.getInitialState(instance)
+      const {options, dashboardContext} = authStore.getInitialState(instance, null)
 
       expect(options.apiHost).toBe(apiHost)
       expect(options.callbackUrl).toBe(callbackUrl)
@@ -114,7 +114,7 @@ describe('authStore', () => {
         auth: {initialLocationHref},
       })
 
-      const {dashboardContext, authState} = authStore.getInitialState(instance)
+      const {dashboardContext, authState} = authStore.getInitialState(instance, null)
       expect(dashboardContext).toEqual(context)
       expect(authState.type).toBe(AuthStateType.LOGGED_OUT)
     })
@@ -129,7 +129,7 @@ describe('authStore', () => {
         auth: {initialLocationHref},
       })
 
-      const {dashboardContext, authState} = authStore.getInitialState(instance)
+      const {dashboardContext, authState} = authStore.getInitialState(instance, null)
       expect(dashboardContext).toEqual(expectedContext)
       expect(authState.type).toBe(AuthStateType.LOGGED_OUT)
     })
@@ -143,7 +143,7 @@ describe('authStore', () => {
         auth: {initialLocationHref},
       })
 
-      const {dashboardContext, authState} = authStore.getInitialState(instance)
+      const {dashboardContext, authState} = authStore.getInitialState(instance, null)
       expect(dashboardContext).toStrictEqual({})
       expect(authState.type).toBe(AuthStateType.LOGGED_OUT)
       expect(errorSpy).toHaveBeenCalledWith(
@@ -166,7 +166,7 @@ describe('authStore', () => {
         },
       })
 
-      const {authState, dashboardContext} = authStore.getInitialState(instance)
+      const {authState, dashboardContext} = authStore.getInitialState(instance, null)
       expect(authState).toMatchObject({type: AuthStateType.LOGGED_IN, token})
       expect(dashboardContext).toEqual(context)
     })
@@ -182,7 +182,7 @@ describe('authStore', () => {
 
       vi.mocked(getAuthCode).mockReturnValue('auth-code')
 
-      const {authState, dashboardContext} = authStore.getInitialState(instance)
+      const {authState, dashboardContext} = authStore.getInitialState(instance, null)
       expect(authState).toMatchObject({type: AuthStateType.LOGGING_IN})
       expect(dashboardContext).toEqual(context)
     })
@@ -197,7 +197,7 @@ describe('authStore', () => {
       vi.mocked(getAuthCode).mockReturnValue(null)
       vi.mocked(getTokenFromStorage).mockReturnValue(storageToken)
 
-      const {authState, dashboardContext} = authStore.getInitialState(instance)
+      const {authState, dashboardContext} = authStore.getInitialState(instance, null)
       expect(authState).toMatchObject({type: AuthStateType.LOGGED_IN, token: storageToken})
       expect(dashboardContext).toStrictEqual({})
     })
@@ -215,7 +215,7 @@ describe('authStore', () => {
       vi.mocked(getAuthCode).mockReturnValue(null)
       vi.mocked(getTokenFromStorage).mockReturnValue(storageToken)
 
-      const {authState, dashboardContext} = authStore.getInitialState(instance)
+      const {authState, dashboardContext} = authStore.getInitialState(instance, null)
       expect(authState).toMatchObject({type: AuthStateType.LOGGED_OUT})
       expect(dashboardContext).toEqual(context)
     })
@@ -229,7 +229,7 @@ describe('authStore', () => {
       vi.mocked(getAuthCode).mockReturnValue(null)
       vi.mocked(getTokenFromStorage).mockReturnValue(null)
 
-      const {authState, dashboardContext} = authStore.getInitialState(instance)
+      const {authState, dashboardContext} = authStore.getInitialState(instance, null)
       expect(authState).toMatchObject({type: AuthStateType.LOGGED_OUT})
       expect(dashboardContext).toStrictEqual({})
     })
@@ -252,7 +252,7 @@ describe('authStore', () => {
         auth: {storageArea: mockStorage}, // Provide mock storage
       })
 
-      const {authState, options} = authStore.getInitialState(instance)
+      const {authState, options} = authStore.getInitialState(instance, null)
       expect(getStudioTokenFromLocalStorage).toHaveBeenCalledWith(mockStorage, studioStorageKey)
       expect(authState).toMatchObject({type: AuthStateType.LOGGED_IN, token: studioToken})
       expect(options.authMethod).toBe('localstorage')
@@ -277,7 +277,7 @@ describe('authStore', () => {
       })
 
       // Verify initial state without async cookie probe
-      const {authState: initialAuthState} = authStore.getInitialState(instance)
+      const {authState: initialAuthState} = authStore.getInitialState(instance, null)
       expect(initialAuthState.type).toBe(AuthStateType.LOGGED_OUT)
       expect(getStudioTokenFromLocalStorage).toHaveBeenCalledWith(mockStorage, studioStorageKey)
 
@@ -296,7 +296,7 @@ describe('authStore', () => {
         dataset: 'd',
       })
 
-      const {authState, options} = authStore.getInitialState(instance)
+      const {authState, options} = authStore.getInitialState(instance, null)
       expect(getStudioTokenFromLocalStorage).not.toHaveBeenCalled()
       expect(checkForCookieAuth).not.toHaveBeenCalled()
       expect(getTokenFromStorage).toHaveBeenCalled()
@@ -314,7 +314,7 @@ describe('authStore', () => {
       vi.mocked(getAuthCode).mockReturnValue(null)
       vi.mocked(getTokenFromLocation).mockReturnValue('hash-token')
 
-      const {authState} = authStore.getInitialState(instance)
+      const {authState} = authStore.getInitialState(instance, null)
       expect(authState).toMatchObject({
         type: AuthStateType.LOGGING_IN,
         isExchangingToken: false,

package/src/auth/refreshStampedToken.test.ts

@@ -122,7 +122,7 @@ describe('refreshStampedToken', () => {
         dataset: 'd',
         auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
       })
-      const initialState = authStore.getInitialState(instance)
+      const initialState = authStore.getInitialState(instance, null)
       initialState.authState = {
         type: AuthStateType.LOGGED_IN,
         token: 'sk-initial-token-st123',
@@ -131,7 +131,7 @@ describe('refreshStampedToken', () => {
       initialState.dashboardContext = {mode: 'test'}
       const state = createStoreState(initialState)
 
-      const subscription = refreshStampedToken({state, instance})
+      const subscription = refreshStampedToken({state, instance, key: null})
       subscriptions.push(subscription)
 
       await vi.advanceTimersToNextTimerAsync()
@@ -168,7 +168,7 @@ describe('refreshStampedToken', () => {
         dataset: 'd',
         auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
       })
-      const initialState = authStore.getInitialState(instance)
+      const initialState = authStore.getInitialState(instance, null)
       initialState.authState = {
         type: AuthStateType.LOGGED_IN,
         token: 'sk-initial-token-st123',
@@ -177,7 +177,7 @@ describe('refreshStampedToken', () => {
       initialState.dashboardContext = {mode: 'test'}
       const state = createStoreState(initialState)
 
-      const subscription = refreshStampedToken({state, instance})
+      const subscription = refreshStampedToken({state, instance, key: null})
       subscriptions.push(subscription)
 
       await vi.advanceTimersToNextTimerAsync()
@@ -202,7 +202,7 @@ describe('refreshStampedToken', () => {
         dataset: 'd',
         auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
       })
-      const initialState = authStore.getInitialState(instance)
+      const initialState = authStore.getInitialState(instance, null)
       initialState.authState = {
         type: AuthStateType.LOGGED_IN,
         token: 'sk-initial-token-st123',
@@ -213,7 +213,7 @@ describe('refreshStampedToken', () => {
       let subscription: Subscription | undefined
       // We expect this NOT to throw, but accept we can't easily test the lock call or outcome
       expect(() => {
-        subscription = refreshStampedToken({state, instance})
+        subscription = refreshStampedToken({state, instance, key: null})
         subscriptions.push(subscription!)
       }).not.toThrow()
 
@@ -253,7 +253,7 @@ describe('refreshStampedToken', () => {
           dataset: 'd',
           auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
         })
-        const initialState = authStore.getInitialState(instance)
+        const initialState = authStore.getInitialState(instance, null)
         initialState.authState = {
           type: AuthStateType.LOGGED_IN,
           token: 'sk-initial-token-st123',
@@ -261,7 +261,7 @@ describe('refreshStampedToken', () => {
         }
         const state = createStoreState(initialState)
 
-        const subscription = refreshStampedToken({state, instance})
+        const subscription = refreshStampedToken({state, instance, key: null})
         subscriptions.push(subscription)
 
         // DO NOT advance timers or yield here - focus on immediate observable logic
@@ -303,7 +303,7 @@ describe('refreshStampedToken', () => {
           dataset: 'd',
           auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
         })
-        const initialState = authStore.getInitialState(instance)
+        const initialState = authStore.getInitialState(instance, null)
         initialState.authState = {
           type: AuthStateType.LOGGED_IN,
           token: 'sk-initial-token-st123',
@@ -311,7 +311,7 @@ describe('refreshStampedToken', () => {
         }
         const state = createStoreState(initialState)
 
-        const subscription = refreshStampedToken({state, instance})
+        const subscription = refreshStampedToken({state, instance, key: null})
         subscriptions.push(subscription)
 
         // Advance timers to allow the async `performRefresh` to execute
@@ -349,7 +349,7 @@ describe('refreshStampedToken', () => {
       dataset: 'd',
       auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
     })
-    const initialState = authStore.getInitialState(instance)
+    const initialState = authStore.getInitialState(instance, null)
     initialState.authState = {
       type: AuthStateType.LOGGED_IN,
       token: 'sk-initial-token-st123',
@@ -358,7 +358,7 @@ describe('refreshStampedToken', () => {
     initialState.dashboardContext = {mode: 'test'}
     const state = createStoreState(initialState)
 
-    const subscription = refreshStampedToken({state, instance})
+    const subscription = refreshStampedToken({state, instance, key: null})
     subscriptions.push(subscription)
 
     await vi.advanceTimersToNextTimerAsync()
@@ -378,14 +378,14 @@ describe('refreshStampedToken', () => {
       dataset: 'd',
       auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
     })
-    const initialState = authStore.getInitialState(instance)
+    const initialState = authStore.getInitialState(instance, null)
     initialState.authState = {
       type: AuthStateType.LOGGED_OUT,
       isDestroyingSession: false,
     } as AuthState
     const state = createStoreState(initialState)
 
-    const subscription = refreshStampedToken({state, instance})
+    const subscription = refreshStampedToken({state, instance, key: null})
     subscriptions.push(subscription)
 
     await vi.advanceTimersByTimeAsync(0)
@@ -404,7 +404,7 @@ describe('refreshStampedToken', () => {
       dataset: 'd',
       auth: {clientFactory: mockClientFactory, storageArea: mockStorage},
     })
-    const initialState = authStore.getInitialState(instance)
+    const initialState = authStore.getInitialState(instance, null)
     initialState.authState = {
       type: AuthStateType.LOGGED_IN,
       token: 'sk-nonstamped-token',
@@ -412,7 +412,7 @@ describe('refreshStampedToken', () => {
     }
     const state = createStoreState(initialState)
 
-    const subscription = refreshStampedToken({state, instance})
+    const subscription = refreshStampedToken({state, instance, key: null})
     subscriptions.push(subscription)
 
     await vi.advanceTimersByTimeAsync(0)

package/src/auth/subscribeToStateAndFetchCurrentUser.test.ts

@@ -20,8 +20,8 @@ describe('subscribeToStateAndFetchCurrentUser', () => {
     const clientFactory = vi.fn().mockReturnValue(mockClient)
     const instance = createSanityInstance({projectId: 'p', dataset: 'd', auth: {clientFactory}})
 
-    const state = createStoreState(authStore.getInitialState(instance))
-    const subscription = subscribeToStateAndFetchCurrentUser({state, instance})
+    const state = createStoreState(authStore.getInitialState(instance, null))
+    const subscription = subscribeToStateAndFetchCurrentUser({state, instance, key: null})
 
     expect(state.get()).toMatchObject({authState: {type: AuthStateType.LOGGED_OUT}})
 
@@ -52,8 +52,8 @@ describe('subscribeToStateAndFetchCurrentUser', () => {
     const clientFactory = vi.fn().mockReturnValue(mockClient)
     const instance = createSanityInstance({projectId: 'p', dataset: 'd', auth: {clientFactory}})
 
-    const state = createStoreState(authStore.getInitialState(instance))
-    const subscription = subscribeToStateAndFetchCurrentUser({state, instance})
+    const state = createStoreState(authStore.getInitialState(instance, null))
+    const subscription = subscribeToStateAndFetchCurrentUser({state, instance, key: null})
 
     expect(state.get()).toMatchObject({authState: {type: AuthStateType.LOGGED_OUT}})
 
@@ -88,8 +88,8 @@ describe('subscribeToStateAndFetchCurrentUser', () => {
     const clientFactory = vi.fn().mockReturnValue(mockClient)
     const instance = createSanityInstance({projectId: 'p', dataset: 'd', auth: {clientFactory}})
 
-    const state = createStoreState(authStore.getInitialState(instance))
-    const subscription = subscribeToStateAndFetchCurrentUser({state, instance})
+    const state = createStoreState(authStore.getInitialState(instance, null))
+    const subscription = subscribeToStateAndFetchCurrentUser({state, instance, key: null})
 
     expect(state.get()).toMatchObject({authState: {type: AuthStateType.LOGGED_OUT}})
 

package/src/auth/subscribeToStorageEventsAndSetToken.test.ts

@@ -34,9 +34,9 @@ describe('subscribeToStorageEventsAndSetToken', () => {
   })
 
   it('sets the state to logged in when a matching storage event returns a token', () => {
-    const state = createStoreState(authStore.getInitialState(instance))
+    const state = createStoreState(authStore.getInitialState(instance, null))
     const {storageKey} = state.get().options
-    const subscription = subscribeToStorageEventsAndSetToken({state, instance})
+    const subscription = subscribeToStorageEventsAndSetToken({state, instance, key: null})
 
     expect(state.get()).toMatchObject({
       authState: {type: AuthStateType.LOGGED_OUT, isDestroyingSession: false},
@@ -54,10 +54,10 @@ describe('subscribeToStorageEventsAndSetToken', () => {
 
   it('sets the state to logged in when a matching storage event returns null', () => {
     vi.mocked(getTokenFromStorage).mockReturnValue('existing-token')
-    const state = createStoreState(authStore.getInitialState(instance))
+    const state = createStoreState(authStore.getInitialState(instance, null))
     const {storageKey} = state.get().options
 
-    const subscription = subscribeToStorageEventsAndSetToken({state, instance})
+    const subscription = subscribeToStorageEventsAndSetToken({state, instance, key: null})
 
     expect(state.get()).toMatchObject({
       authState: {type: AuthStateType.LOGGED_IN, token: 'existing-token', currentUser: null},