Skip to content

fix: prompt download when maia3 cache is missing #274

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ashtonanderson merged 1 commit into from
Mar 28, 2026
Merged

fix: prompt download when maia3 cache is missing #274

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion public/maia-worker.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,10 @@ const DB_NAME = 'MaiaModels'
const STORE_NAME = 'models'
const MODEL_KEY = 'maia-rapid-model'

function isCompatibleModelCache(data, expectedUrl, expectedVersion) {
return data.url === expectedUrl && data.version === expectedVersion
}

function openDB() {
return new Promise((resolve, reject) => {
const request = indexedDB.open(DB_NAME, 1)
Expand Down Expand Up @@ -51,7 +55,7 @@ async function getCachedModel(modelUrl, modelVersion) {

if (!data) return null

if (data.version && data.version !== modelVersion) {
if (!isCompatibleModelCache(data, modelUrl, modelVersion)) {
const rwTx = db.transaction([STORE_NAME], 'readwrite')
rwTx.objectStore(STORE_NAME).delete(MODEL_KEY)
return null
Expand Down
57 changes: 28 additions & 29 deletions src/lib/engine/storage.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,16 @@ interface ModelStorage {
size: number
}

const MODEL_STORAGE_KEY = 'maia-rapid-model'

function isCompatibleModelCache(
modelData: ModelStorage,
modelUrl: string,
modelVersion: string,
): boolean {
return modelData.version === modelVersion && modelData.url === modelUrl
}

export class MaiaModelStorage {
private dbName = 'MaiaModels'
private storeName = 'models'
Expand Down Expand Up @@ -46,7 +56,7 @@ export class MaiaModelStorage {
const store = transaction.objectStore(this.storeName)

const modelData: ModelStorage = {
id: 'maia-rapid-model',
id: MODEL_STORAGE_KEY,
url: modelUrl,
version: modelVersion,
data: new Blob([buffer]),
Expand Down Expand Up @@ -82,7 +92,7 @@ export class MaiaModelStorage {
console.log('Storage: Requesting model data...')
const modelData = await new Promise<ModelStorage | null>(
(resolve, reject) => {
const request = store.get('maia-rapid-model')
const request = store.get(MODEL_STORAGE_KEY)
request.onsuccess = () => {
console.log(
'Storage: IndexedDB request successful, result:',
Expand All @@ -104,36 +114,25 @@ export class MaiaModelStorage {
return null
}

// Version mismatch = genuinely new model, must re-download.
if (modelData.version && modelData.version !== modelVersion) {
console.log('Storage: Model version changed, clearing old cache')
console.log('Storage: Cached version:', modelData.version)
// Maia cache records must match both the expected model version and URL.
// Legacy Maia-2 downloads may not include a version field; treating those
// as incompatible forces the download modal to appear instead of leaving
// analysis in a broken "loaded old model" state.
if (!isCompatibleModelCache(modelData, modelUrl, modelVersion)) {
console.log(
'Storage: Cached Maia model is incompatible, clearing old cache',
)
console.log('Storage: Cached URL:', modelData.url || '(missing)')
console.log(
'Storage: Cached version:',
modelData.version || '(missing)',
)
console.log('Storage: Required URL:', modelUrl)
console.log('Storage: Required version:', modelVersion)
await this.deleteModel()
return null
}

// URL changed but same version (e.g. moved hosts) — keep the cache,
// just update the stored URL so future loads match immediately.
if (modelData.url !== modelUrl || modelData.version !== modelVersion) {
console.log('Storage: Updating stored URL/version')
try {
const rwTx = db.transaction([this.storeName], 'readwrite')
const rwStore = rwTx.objectStore(this.storeName)
await new Promise<void>((resolve, reject) => {
const req = rwStore.put({
...modelData,
url: modelUrl,
version: modelVersion,
})
req.onsuccess = () => resolve()
req.onerror = () => reject(req.error)
})
} catch (e) {
console.warn('Storage: Failed to update stored URL/version:', e)
}
}

console.log('Storage: Converting Blob to ArrayBuffer...')
// Convert Blob back to ArrayBuffer
const buffer = await modelData.data.arrayBuffer()
Expand All @@ -156,7 +155,7 @@ export class MaiaModelStorage {
const store = transaction.objectStore(this.storeName)

await new Promise<void>((resolve, reject) => {
const request = store.delete('maia-rapid-model')
const request = store.delete(MODEL_STORAGE_KEY)
request.onsuccess = () => resolve()
request.onerror = () => reject(request.error)
})
Expand Down Expand Up @@ -193,7 +192,7 @@ export class MaiaModelStorage {

const modelData = await new Promise<ModelStorage | null>(
(resolve, reject) => {
const request = store.get('maia-rapid-model')
const request = store.get(MODEL_STORAGE_KEY)
request.onsuccess = () => resolve(request.result || null)
request.onerror = () => reject(request.error)
},
Expand Down
Loading