Skip to content

Commit e2571c7

Browse files
committed
fix!: model dimension defaults
1 parent 31c8e96 commit e2571c7

8 files changed

Lines changed: 296 additions & 283 deletions

File tree

package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
"name": "retriv",
33
"type": "module",
44
"version": "0.1.0",
5-
"packageManager": "pnpm@10.28.2+sha512.41872f037ad22f7348e3b1debbaf7e867cfd448f2726d9cf74c08f19507c31d2c8e7a11525b983febc2df640b5438dee6023ebb1f84ed43cc2d654d2bc326264",
5+
"packageManager": "pnpm@10.28.2",
66
"description": "Unified document retrieval for search - semantic, full-text, and fuzzy.",
77
"author": {
88
"name": "Harlan Wilton",

pnpm-lock.yaml

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

pnpm-workspace.yaml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,17 +12,17 @@ catalog:
1212
'@huggingface/transformers': ^3.8.1
1313
'@libsql/client': ^0.17.0
1414
'@types/node': ^25.2.0
15-
'@types/pg': ^8.11.0
16-
'@upstash/vector': ^1.1.0
15+
'@types/pg': ^8.16.0
16+
'@upstash/vector': ^1.2.2
1717
'@vitest/coverage-v8': ^4.0.18
1818
ai: ^6.0.67
19-
bumpp: ^10.3.2
19+
bumpp: ^10.4.0
2020
eslint: ^9.39.2
21-
obuild: ^0.4.14
21+
obuild: ^0.4.22
2222
pg: ^8.18.0
2323
sqlite-vec: ^0.1.7-alpha.2
2424
typescript: ^5.9.3
25-
vitest: ^4.0.16
25+
vitest: ^4.0.18
2626
zod: ^4.3.6
2727
ignoredBuiltDependencies:
2828
- better-sqlite3

src/embeddings/model-info.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ export const DEFAULT_MODELS: Record<EmbeddingPreset, { model: string, dimensions
1818
'mistral': { model: 'mistral-embed', dimensions: 1024 },
1919
'cohere': { model: 'embed-english-v3.0', dimensions: 1024 },
2020
'ollama': { model: 'nomic-embed-text', dimensions: 768 },
21-
'transformers.js': { model: 'Xenova/bge-base-en-v1.5', dimensions: 768 },
21+
'transformers.js': { model: 'Xenova/bge-small-en-v1.5', dimensions: 384 },
2222
}
2323

2424
/**

src/embeddings/transformers-js.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ async function clearCorruptedCache(error: unknown, model: string): Promise<boole
4242
* ```
4343
*/
4444
export function transformersJs(options: TransformersEmbeddingOptions = {}): EmbeddingConfig {
45-
const baseModel = options.model ?? 'bge-base-en-v1.5'
45+
const baseModel = options.model ?? 'bge-small-en-v1.5'
4646
const model = resolveModelForPreset(baseModel, 'transformers.js')
4747
let cached: ResolvedEmbedding | null = null
4848

0 commit comments

Comments
 (0)