soulcraft/brainy
2
0
Fork 0
Code Issues 1 Pull requests Projects Releases 335 Packages Wiki Activity Actions

feat: add persistent storage support for OPFS

Browse source 
Introduced an optional `requestPersistentStorage` flag to enable persistent storage permission requests in the browser. Updated `OPFSStorage` to include methods for requesting and checking persistent storage status, and modified `createStorage` to handle the new flag.
This commit is contained in:
David Snelling 2025-05-28 11:37:28 -07:00
parent 5f4e6ccc70
commit ac5955a922
2 changed files with 74 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/brainyData.ts
View file

@ -29,6 +29,12 @@ export interface BrainyDataConfig {
* Embedding function to convert data to vectors
*/
embeddingFunction?: EmbeddingFunction
/**
* Request persistent storage when running in a browser
* This will prompt the user for permission to use persistent storage
*/
requestPersistentStorage?: boolean
}
export class BrainyData<T = any> {
@ -36,6 +42,7 @@ export class BrainyData<T = any> {
private storage: StorageAdapter | null = null
private isInitialized = false
private embeddingFunction: EmbeddingFunction
private requestPersistentStorage: boolean
/**
* Create a new vector database
@ -52,6 +59,9 @@ export class BrainyData<T = any> {
// Set embedding function if provided, otherwise use default
this.embeddingFunction = config.embeddingFunction || defaultEmbeddingFunction
// Set persistent storage request flag
this.requestPersistentStorage = config.requestPersistentStorage || false
}
/**
@ -66,7 +76,9 @@ export class BrainyData<T = any> {
try {
// Initialize storage if not provided in constructor
if (!this.storage) {
this.storage = await createStorage()
this.storage = await createStorage({
requestPersistentStorage: this.requestPersistentStorage
})
}
// Initialize storage