File tree Expand file tree Collapse file tree 3 files changed +15
-4
lines changed
apps/twig/src/renderer/features Expand file tree Collapse file tree 3 files changed +15
-4
lines changed Original file line number Diff line number Diff line change 11import { PostHogAPIClient } from "@api/posthogClient" ;
2- import { resetSessionService } from "@features/sessions/service/service" ;
32import { identifyUser , resetUser , track } from "@renderer/lib/analytics" ;
43import { electronStorage } from "@renderer/lib/electronStorage" ;
54import { logger } from "@renderer/lib/logger" ;
@@ -21,6 +20,12 @@ const log = logger.scope("auth-store");
2120let refreshPromise : Promise < void > | null = null ;
2221let initializePromise : Promise < boolean > | null = null ;
2322
23+ let sessionResetCallback : ( ( ) => void ) | null = null ;
24+
25+ export function setSessionResetCallback ( callback : ( ) => void ) {
26+ sessionResetCallback = callback ;
27+ }
28+
2429const REFRESH_MAX_RETRIES = 3 ;
2530const REFRESH_INITIAL_DELAY_MS = 1000 ;
2631
@@ -644,7 +649,7 @@ export const useAuthStore = create<AuthState>()(
644649 }
645650
646651 // Clean up all existing sessions before switching projects
647- resetSessionService ( ) ;
652+ sessionResetCallback ?. ( ) ;
648653
649654 const apiHost = getCloudUrlFromRegion ( cloudRegion ) ;
650655
@@ -706,7 +711,7 @@ export const useAuthStore = create<AuthState>()(
706711 resetUser ( ) ;
707712
708713 // Clean up session service subscriptions before clearing auth state
709- resetSessionService ( ) ;
714+ sessionResetCallback ?. ( ) ;
710715
711716 trpcVanilla . analytics . resetUser . mutate ( ) ;
712717
Original file line number Diff line number Diff line change @@ -64,6 +64,7 @@ const mockAuthStore = vi.hoisted(() => ({
6464 } ,
6565 } ) ) ,
6666 } ,
67+ setSessionResetCallback : vi . fn ( ) ,
6768} ) ) ;
6869
6970vi . mock ( "@features/auth/stores/authStore" , ( ) => mockAuthStore ) ;
Original file line number Diff line number Diff line change @@ -3,7 +3,10 @@ import type {
33 RequestPermissionRequest ,
44 SessionConfigOption ,
55} from "@agentclientprotocol/sdk" ;
6- import { useAuthStore } from "@features/auth/stores/authStore" ;
6+ import {
7+ setSessionResetCallback ,
8+ useAuthStore ,
9+ } from "@features/auth/stores/authStore" ;
710import { useModelsStore } from "@features/sessions/stores/modelsStore" ;
811import { useSessionAdapterStore } from "@features/sessions/stores/sessionAdapterStore" ;
912import {
@@ -89,6 +92,8 @@ export function resetSessionService(): void {
8992 } ) ;
9093}
9194
95+ setSessionResetCallback ( resetSessionService ) ;
96+
9297export class SessionService {
9398 private connectingTasks = new Map < string , Promise < void > > ( ) ;
9499 private subscriptions = new Map <
You can’t perform that action at this time.
0 commit comments