diff --git a/packages/core/src/BaseIndex.ts b/packages/core/src/BaseIndex.ts index 0a0d00dfae1982a6273ce072a80366a76132b2ce..dd6e7461b20ea86a6133aad94dd0adf71a014748 100644 --- a/packages/core/src/BaseIndex.ts +++ b/packages/core/src/BaseIndex.ts @@ -141,7 +141,12 @@ export class VectorStoreIndex extends BaseIndex<IndexDict> { vectorStore.add(embeddingResults); - throw new Error("not implemented"); + const indexDict = new IndexDict(); + for (const { node } of embeddingResults) { + indexDict.addNode(node); + } + + return indexDict; } static async fromDocuments( diff --git a/packages/core/src/storage/StorageContext.ts b/packages/core/src/storage/StorageContext.ts index d7c5e03bcd3254586aab19c2427b9f3efed528cc..d5c51c8d2abdd30e09baea086bde63d6bbaa3ea0 100644 --- a/packages/core/src/storage/StorageContext.ts +++ b/packages/core/src/storage/StorageContext.ts @@ -32,21 +32,24 @@ export async function storageContextFromDefaults({ persistDir, fs, }: BuilderParams): Promise<StorageContext> { - persistDir = persistDir || DEFAULT_PERSIST_DIR; - - fs = fs || DEFAULT_FS; - - docStore = - docStore || - (await SimpleDocumentStore.fromPersistDir( - persistDir, - DEFAULT_NAMESPACE, - fs - )); - indexStore = - indexStore || (await SimpleIndexStore.fromPersistDir(persistDir, fs)); - vectorStore = - vectorStore || (await SimpleVectorStore.fromPersistDir(persistDir, fs)); + if (!persistDir) { + docStore = docStore || new SimpleDocumentStore(); + indexStore = indexStore || new SimpleIndexStore(); + vectorStore = vectorStore || new SimpleVectorStore(); + } else { + fs = fs || DEFAULT_FS; + docStore = + docStore || + (await SimpleDocumentStore.fromPersistDir( + persistDir, + DEFAULT_NAMESPACE, + fs + )); + indexStore = + indexStore || (await SimpleIndexStore.fromPersistDir(persistDir, fs)); + vectorStore = + vectorStore || (await SimpleVectorStore.fromPersistDir(persistDir, fs)); + } return { docStore,