Skip to content

Commit 97d67b1

Browse files
committed
chore(search): use retriv device support
Upgrade retriv to 0.15.0 and remove the temporary local Transformers.js provider now that device forwarding is available upstream.
1 parent 473d135 commit 97d67b1

5 files changed

Lines changed: 23 additions & 175 deletions

File tree

pnpm-lock.yaml

Lines changed: 17 additions & 74 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pnpm-workspace.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ catalogMode: prefer
22
minimumReleaseAgeExclude:
33
- verkit@0.2.0
44
- '@mdream/rust-wasm32-wasi@1.5.12'
5+
- retriv@0.15.0
56
shellEmulator: true
67

78
trustPolicy: no-downgrade
@@ -23,7 +24,7 @@ catalog:
2324
ofetch: ^1.5.1
2425
pathe: ^2.0.3
2526
publint: ^0.3.23
26-
retriv: ^0.14.7
27+
retriv: ^0.15.0
2728
std-env: ^4.2.0
2829
tsx: ^4.23.8
2930
typebox: ^1.3.10

src/retriv/index.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ import { readConfig } from '../core/config.ts'
33
import { stripFrontmatter } from '../core/markdown.ts'
44
import { checkIndexEmbeddingIdentity, recordIndexEmbeddingIdentity } from './index-embedding-identity.ts'
55
import { getEmbeddingIdentity, resolveEmbedDevice, resolveEmbedModel } from './models.ts'
6-
import { transformersEmbeddings } from './transformers-embeddings.ts'
76

87
export type { ChunkEntity, Document, IndexConfig, IndexPhase, IndexProgress, SearchFilter, SearchOptions, SearchResult, SearchSnippet }
98

@@ -67,19 +66,21 @@ export async function getDb(config: Pick<IndexConfig, 'dbPath'>) {
6766
if (identityState._tag === 'Mismatch')
6867
throw new EmbeddingIndexMismatchError(config.dbPath, identityState.stored, identityState.current)
6968

70-
let createRetriv, autoChunker, sqliteMod, sqliteVec, cachedEmbeddings
69+
let createRetriv, autoChunker, sqliteMod, sqliteVec, transformersJs, cachedEmbeddings
7170
try {
7271
;([
7372
{ createRetriv },
7473
{ autoChunker },
7574
sqliteMod,
7675
sqliteVec,
76+
{ transformersJs },
7777
{ cachedEmbeddings },
7878
] = await Promise.all([
7979
import('retriv'),
8080
import('retriv/chunkers/auto'),
8181
import('retriv/db/sqlite'),
8282
import('sqlite-vec'),
83+
import('retriv/embeddings/transformers-js'),
8384
import('./embedding-cache.ts'),
8485
]))
8586
}
@@ -89,7 +90,7 @@ export async function getDb(config: Pick<IndexConfig, 'dbPath'>) {
8990
throw err
9091
}
9192
const embeddings = await cachedEmbeddings(
92-
transformersEmbeddings({
93+
transformersJs({
9394
model: embedModel,
9495
...(device ? { device } : {}),
9596
}),

src/retriv/transformers-embeddings.ts

Lines changed: 0 additions & 63 deletions
This file was deleted.

test/unit/transformers-embeddings.test.ts

Lines changed: 0 additions & 34 deletions
This file was deleted.

0 commit comments

Comments
 (0)