@myrialabs/clopen 0.2.1 → 0.2.3

package/backend/ws/README.md

@@ -466,7 +466,7 @@ try {
 
 **Backend:**
 ```typescript
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 .on('terminal:input', {
   data: t.Object({
@@ -589,7 +589,7 @@ ws.emit.global(event, payload)                  // → all connected clients
 Emit events from external sources (PTY, file watchers, etc.):
 
 ```typescript
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 // PTY output
 pty.onData((output) => {
@@ -632,7 +632,7 @@ browser.on('screencast-frame', (frame) => {
 
 **Backend - Connection Setup:**
 ```typescript
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 app.ws('/ws', {
   open(wsRaw) {
@@ -751,10 +751,10 @@ Merge another router.
 
 ### WebSocket Singleton API
 
-**File: `backend/lib/utils/ws.ts`**
+**File: `backend/utils/ws.ts`**
 
 ```typescript
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 // Emit events
 ws.emit.user(userId, 'event', payload)
@@ -781,7 +781,7 @@ ws.getConnectionCount(projectId?)
 
 ### Frontend Client API
 
-**File: `frontend/lib/utils/ws.ts`**
+**File: `frontend/shared/ws.ts`**
 
 ```typescript
 import ws from '$lib/utils/ws';
@@ -865,7 +865,7 @@ renderTree(tree);
 ```typescript
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 export const streamHandler = createRouter()
   .on('terminal:input', {
@@ -942,7 +942,7 @@ onDestroy(() => {
 ```typescript
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 export const interactHandler = createRouter()
   .on('browser:interact', {
@@ -1485,7 +1485,7 @@ cleanup();
 ### Event Emission
 
 ```typescript
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 // User-specific (all connections of a user)
 ws.emit.user(userId, 'event', payload);

package/backend/ws/auth/invites.ts

@@ -1,7 +1,7 @@
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { createInvite, listInvites, revokeInvite } from '$backend/lib/auth/auth-service';
-import { ws } from '$backend/lib/utils/ws';
+import { createInvite, listInvites, revokeInvite } from '$backend/auth/auth-service';
+import { ws } from '$backend/utils/ws';
 
 export const inviteHandler = createRouter()
 	// Create invite token (admin only — enforced by auth gate)

package/backend/ws/auth/login.ts

@@ -12,11 +12,11 @@ import {
 	updateUserName,
 	createOrGetNoAuthAdmin,
 	needsSetup
-} from '$backend/lib/auth/auth-service';
-import { settingsQueries } from '$backend/lib/database/queries';
-import { getTokenType } from '$backend/lib/auth/tokens';
-import { authRateLimiter } from '$backend/lib/auth/rate-limiter';
-import { ws } from '$backend/lib/utils/ws';
+} from '$backend/auth/auth-service';
+import { settingsQueries } from '$backend/database/queries';
+import { getTokenType } from '$backend/auth/tokens';
+import { authRateLimiter } from '$backend/auth/rate-limiter';
+import { ws } from '$backend/utils/ws';
 
 const authUserSchema = t.Object({
 	id: t.String(),
@@ -51,7 +51,7 @@ export const loginHandler = createRouter()
 		settingsQueries.set('system:settings', JSON.stringify(parsed));
 
 		// Set auth on connection
-		const tokenHash = (await import('$backend/lib/auth/tokens')).hashToken(result.sessionToken);
+		const tokenHash = (await import('$backend/auth/tokens')).hashToken(result.sessionToken);
 		ws.setAuth(conn, result.user.id, result.user.role, tokenHash);
 
 		return result;
@@ -82,7 +82,7 @@ export const loginHandler = createRouter()
 		const result = createOrGetNoAuthAdmin();
 
 		// Set auth on connection
-		const tokenHash = (await import('$backend/lib/auth/tokens')).hashToken(result.sessionToken);
+		const tokenHash = (await import('$backend/auth/tokens')).hashToken(result.sessionToken);
 		ws.setAuth(conn, result.user.id, result.user.role, tokenHash);
 
 		return result;
@@ -104,7 +104,7 @@ export const loginHandler = createRouter()
 		const result = createOrGetNoAuthAdmin();
 
 		// Set auth on connection
-		const tokenHash = (await import('$backend/lib/auth/tokens')).hashToken(result.sessionToken);
+		const tokenHash = (await import('$backend/auth/tokens')).hashToken(result.sessionToken);
 		ws.setAuth(conn, result.user.id, result.user.role, tokenHash);
 
 		return result;
@@ -187,7 +187,7 @@ export const loginHandler = createRouter()
 			authRateLimiter.recordSuccess(ip);
 
 			// Set auth on connection
-			const tokenHash = (await import('$backend/lib/auth/tokens')).hashToken(result.sessionToken);
+			const tokenHash = (await import('$backend/auth/tokens')).hashToken(result.sessionToken);
 			ws.setAuth(conn, result.user.id, result.user.role, tokenHash);
 
 			return result;

package/backend/ws/auth/status.ts

@@ -1,7 +1,7 @@
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { needsSetup, getUserById, getAuthMode, isOnboardingComplete } from '$backend/lib/auth/auth-service';
-import { ws } from '$backend/lib/utils/ws';
+import { needsSetup, getUserById, getAuthMode, isOnboardingComplete } from '$backend/auth/auth-service';
+import { ws } from '$backend/utils/ws';
 
 export const statusHandler = createRouter()
 	.http('auth:status', {

package/backend/ws/auth/users.ts

@@ -1,6 +1,6 @@
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { listUsers, removeUser } from '$backend/lib/auth/auth-service';
+import { listUsers, removeUser } from '$backend/auth/auth-service';
 
 export const usersHandler = createRouter()
 	// List all users (admin only — enforced by auth gate)

package/backend/ws/chat/background.ts

@@ -7,9 +7,9 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { streamManager } from '../../lib/chat/stream-manager';
+import { streamManager } from '../../chat/stream-manager';
 import { debug } from '$shared/utils/logger';
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 export const backgroundHandler = createRouter()
 	// Start background stream (fire-and-forget)

package/backend/ws/chat/stream.ts

@@ -9,11 +9,11 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { streamManager, type StreamEvent } from '../../lib/chat/stream-manager';
+import { streamManager, type StreamEvent } from '../../chat/stream-manager';
 import { debug } from '$shared/utils/logger';
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 import { broadcastPresence } from '../projects/status';
-import { sessionQueries, messageQueries } from '../../lib/database/queries';
+import { sessionQueries, messageQueries } from '../../database/queries';
 
 // ============================================================================
 // Global stream lifecycle handler (module-level, not per-connection)

package/backend/ws/engine/claude/accounts.ts

@@ -16,11 +16,11 @@
 import { t } from 'elysia';
 import { spawn } from 'bun-pty';
 import { createRouter } from '$shared/utils/ws-server';
-import { ws } from '$backend/lib/utils/ws';
-import { engineQueries } from '../../../lib/database/queries';
-import { resetEnvironment, getClaudeUserConfigDir } from '../../../lib/engine/adapters/claude/environment';
+import { ws } from '$backend/utils/ws';
+import { engineQueries } from '../../../database/queries';
+import { resetEnvironment, getClaudeUserConfigDir } from '../../../engine/adapters/claude/environment';
 import { debug } from '$shared/utils/logger';
-import { getCleanSpawnEnv } from '../../../lib/shared/env';
+import { getCleanSpawnEnv } from '../../../utils/env';
 import { resolveCommand } from '../utils';
 
 // ── Helpers ──

package/backend/ws/engine/claude/status.ts

@@ -6,7 +6,7 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { engineQueries } from '../../../lib/database/queries';
+import { engineQueries } from '../../../database/queries';
 import { debug } from '$shared/utils/logger';
 import { getBackendOS, detectCLI } from '../utils';
 

package/backend/ws/files/read.ts

@@ -16,8 +16,8 @@ import {
 	buildFileTree,
 	listDirectoryContents,
 	readFileContents
-} from '../../lib/files/file-reading';
-import { handlePathBrowsing } from '../../lib/files/path-browsing';
+} from '../../files/file-reading';
+import { handlePathBrowsing } from '../../files/path-browsing';
 
 // Bun-compatible existsSync implementation
 async function existsSync(path: string): Promise<boolean> {

package/backend/ws/files/watch.ts

@@ -9,9 +9,9 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { fileWatcher } from '$backend/lib/files/file-watcher';
+import { fileWatcher } from '$backend/files/file-watcher';
 import { debug } from '$shared/utils/logger';
-import { ws } from '$backend/lib/utils/ws';
+import { ws } from '$backend/utils/ws';
 
 export const watchHandler = createRouter()
 	// Start watching a project directory

package/backend/ws/files/write.ts

@@ -22,7 +22,7 @@ import {
 	duplicateOperation,
 	uploadFileOperation,
 	deleteOperation
-} from '../../lib/files/file-operations';
+} from '../../files/file-operations';
 
 export const writeHandler = createRouter()
 	// Write file operation

package/backend/ws/git/branch.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 const BranchSchema = t.Object({
 	name: t.String(),

package/backend/ws/git/commit.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 export const commitHandler = createRouter()
 	.http('git:commit', {

package/backend/ws/git/conflict.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 const ConflictMarkerSchema = t.Object({
 	ourStart: t.Number(),

package/backend/ws/git/diff.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 const DiffHunkLineSchema = t.Object({
 	type: t.Union([t.Literal('add'), t.Literal('delete'), t.Literal('context'), t.Literal('header')]),

package/backend/ws/git/log.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 export const logHandler = createRouter()
 	.http('git:log', {

package/backend/ws/git/remote.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 export const remoteHandler = createRouter()
 	.http('git:remotes', {

package/backend/ws/git/staging.ts

@@ -4,8 +4,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
 
 export const stagingHandler = createRouter()
 	.http('git:stage', {

package/backend/ws/git/status.ts

@@ -4,9 +4,9 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { gitService } from '../../lib/git/git-service';
-import { projectQueries } from '../../lib/database/queries/project-queries';
-import { ws as wsServer } from '../../lib/utils/ws';
+import { gitService } from '../../git/git-service';
+import { projectQueries } from '../../database/queries/project-queries';
+import { ws as wsServer } from '../../utils/ws';
 
 export const statusHandler = createRouter()
 	.http('git:status', {

package/backend/ws/messages/crud.ts

@@ -9,9 +9,18 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { messageQueries } from '../../lib/database/queries';
+import { messageQueries } from '../../database/queries';
 import { formatDatabaseMessage } from '$shared/utils/message-formatter';
 
+function extractTextContent(content: unknown): string {
+	if (typeof content === 'string') return content;
+	if (!Array.isArray(content)) return '';
+	return content
+		.filter((c: any) => c.type === 'text')
+		.map((b: any) => b.text || '')
+		.join(' ');
+}
+
 export const crudHandler = createRouter()
 	// List messages
 	.http('messages:list', {
@@ -33,6 +42,49 @@ export const crudHandler = createRouter()
 		}
 	})
 
+	// Bulk session preview — returns title, summary, and message counts for multiple sessions
+	// without loading all messages. Used by the Sessions/History modal.
+	.http('sessions:preview', {
+		data: t.Object({
+			session_ids: t.Array(t.String())
+		}),
+		response: t.Array(t.Any())
+	}, ({ data }) => {
+		return data.session_ids.map(sessionId => {
+			const preview = messageQueries.getSessionPreview(sessionId);
+
+			// Title from first user message
+			let title = 'New Conversation';
+			if (preview.firstUserMessage) {
+				const sdk = JSON.parse(preview.firstUserMessage.sdk_message);
+				const textContent = extractTextContent(sdk.message?.content).trim();
+				if (textContent) {
+					title = textContent.slice(0, 60) + (textContent.length > 60 ? '...' : '');
+				}
+			}
+
+			// Summary from last assistant message
+			let summary = 'No messages yet';
+			if (preview.lastAssistantMessage) {
+				const sdk = JSON.parse(preview.lastAssistantMessage.sdk_message);
+				const rawText = extractTextContent(sdk.message?.content);
+				const cleanText = rawText.replace(/```[\s\S]*?```/g, '').trim();
+				if (cleanText) {
+					summary = cleanText.slice(0, 100) + (cleanText.length > 100 ? '...' : '');
+				}
+			}
+
+			return {
+				session_id: sessionId,
+				title,
+				summary,
+				userCount: preview.userCount,
+				assistantCount: preview.assistantCount,
+				count: preview.userCount + preview.assistantCount
+			};
+		});
+	})
+
 	// Get message by ID
 	.http('messages:get', {
 		data: t.Object({

package/backend/ws/preview/browser/cleanup.ts

@@ -6,8 +6,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 export const cleanupPreviewHandler = createRouter()

package/backend/ws/preview/browser/console.ts

@@ -6,8 +6,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 export const consolePreviewHandler = createRouter()

package/backend/ws/preview/browser/interact.ts

@@ -6,8 +6,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 import type { KeyInput } from 'puppeteer';
 import { debug } from '$shared/utils/logger';
 import { sleep } from '$shared/utils/async';

package/backend/ws/preview/browser/mcp.ts

@@ -6,7 +6,7 @@
 
 import { createRouter } from '$shared/utils/ws-server';
 import { t } from 'elysia';
-import { browserMcpControl } from '$backend/lib/preview';
+import { browserMcpControl } from '$backend/preview';
 import { debug } from '$shared/utils/logger';
 
 // Tab response types

package/backend/ws/preview/browser/native-ui.ts

@@ -10,8 +10,8 @@
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
 import { debug } from '$shared/utils/logger';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 
 // Event forwarding is now handled automatically by BrowserPreviewServiceManager
 // when service instances are created, ensuring proper project isolation.

package/backend/ws/preview/browser/stats.ts

@@ -6,8 +6,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 
 export const statsPreviewHandler = createRouter()
 	.http('preview:browser-tab-stats', {

package/backend/ws/preview/browser/tab-info.ts

@@ -6,8 +6,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 export const tabInfoPreviewHandler = createRouter()

package/backend/ws/preview/browser/tab.ts

@@ -6,8 +6,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { browserPreviewServiceManager } from '../../../lib/preview/index';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '../../../preview/index';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 // Timeout for browser tab open (60 seconds)

package/backend/ws/preview/browser/webcodecs.ts

@@ -8,8 +8,8 @@
 
 import { createRouter } from '$shared/utils/ws-server';
 import { t } from 'elysia';
-import { browserPreviewServiceManager } from '$backend/lib/preview';
-import { ws } from '$backend/lib/utils/ws';
+import { browserPreviewServiceManager } from '$backend/preview';
+import { ws } from '$backend/utils/ws';
 
 export const streamPreviewHandler = createRouter()
 	// Start streaming

package/backend/ws/projects/crud.ts

@@ -11,9 +11,9 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { initializeDatabase } from '../../lib/database';
-import { projectQueries } from '../../lib/database/queries';
-import { ws } from '$backend/lib/utils/ws';
+import { initializeDatabase } from '../../database';
+import { projectQueries } from '../../database/queries';
+import { ws } from '$backend/utils/ws';
 
 export const crudHandler = createRouter()
 	// List all projects for the current user

package/backend/ws/projects/presence.ts

@@ -9,9 +9,9 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { updateUserPresence, getProjectStatusData } from '../../lib/project/status-manager';
-import { streamManager } from '../../lib/chat/stream-manager';
-import { ws } from '$backend/lib/utils/ws';
+import { updateUserPresence, getProjectStatusData } from '../../project/status-manager';
+import { streamManager } from '../../chat/stream-manager';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 export const presenceHandler = createRouter()

package/backend/ws/projects/status.ts

@@ -12,9 +12,9 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { getProjectStatusData, updateUserPresence } from '../../lib/project/status-manager';
-import { streamManager } from '../../lib/chat/stream-manager';
-import { ws } from '$backend/lib/utils/ws';
+import { getProjectStatusData, updateUserPresence } from '../../project/status-manager';
+import { streamManager } from '../../chat/stream-manager';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 /**

package/backend/ws/sessions/crud.ts

@@ -13,8 +13,8 @@
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
 import type { EngineType } from '$shared/types/engine';
-import { sessionQueries, messageQueries, projectQueries } from '../../lib/database/queries';
-import { ws } from '$backend/lib/utils/ws';
+import { sessionQueries, messageQueries, projectQueries } from '../../database/queries';
+import { ws } from '$backend/utils/ws';
 import { debug } from '$shared/utils/logger';
 
 export const crudHandler = createRouter()

package/backend/ws/settings/crud.ts

@@ -9,8 +9,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { settingsQueries } from '../../lib/database/queries';
-import { initializeEngine } from '../../lib/engine';
+import { settingsQueries } from '../../database/queries';
+import { initializeEngine } from '../../engine';
 import { CLAUDE_CODE_MODELS, registerModels } from '$shared/constants/engines';
 import type { EngineType } from '$shared/types/engine';
 

package/backend/ws/snapshot/restore.ts

@@ -8,8 +8,8 @@
 
 import { t } from 'elysia';
 import { createRouter } from '$shared/utils/ws-server';
-import { messageQueries, sessionQueries, projectQueries, checkpointQueries } from '../../lib/database/queries';
-import { snapshotService } from '../../lib/snapshot/snapshot-service';
+import { messageQueries, sessionQueries, projectQueries, checkpointQueries } from '../../database/queries';
+import { snapshotService } from '../../snapshot/snapshot-service';
 import { debug } from '$shared/utils/logger';
 import {
 	buildCheckpointTree,
@@ -17,8 +17,8 @@ import {
 	findCheckpointForHead,
 	findSessionEnd,
 	INITIAL_NODE_ID
-} from '../../lib/snapshot/helpers';
-import { ws } from '$backend/lib/utils/ws';
+} from '../../snapshot/helpers';
+import { ws } from '$backend/utils/ws';
 
 export const restoreHandler = createRouter()
 	/**
@@ -121,7 +121,7 @@ export const restoreHandler = createRouter()
 			debug.log('snapshot', 'HEAD cleared (initial state)');
 
 			// Clear latest_sdk_session_id so next chat starts fresh
-			const db = (await import('../../lib/database')).getDatabase();
+			const db = (await import('../../database')).getDatabase();
 			db.prepare(`UPDATE chat_sessions SET latest_sdk_session_id = NULL WHERE id = ?`).run(sessionId);
 
 			// Clear checkpoint_tree_state