admin/gbrain
1
0
Fork 0
mirror of https://github.com/garrytan/gbrain.git synced 2026-07-31 04:07:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: harden Voyage schema templating

Browse Source
This commit is contained in:
Eva
2026-05-06 17:55:28 -07:00
committed by Garry Tan
parent 3004a87550
commit d52d4fd617
2 changed files with 34 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/ai/schema-templating.test.ts
+17
View File
@@ -1,5 +1,6 @@
import { describe, test, expect } from 'bun:test';
import { getPGLiteSchema, PGLITE_SCHEMA_SQL } from '../../src/core/pglite-schema.ts';
import { getPostgresSchema } from '../../src/core/postgres-engine.ts';
describe('getPGLiteSchema', () => {
test('default produces v0.13-compatible schema (1536d + text-embedding-3-large)', () => {
@@ -41,3 +42,19 @@ describe('getPGLiteSchema', () => {
expect(PGLITE_SCHEMA_SQL).toBe(getPGLiteSchema());
});
});
describe('getPostgresSchema', () => {
test('Voyage 2048d updates vector column and seeded config but skips HNSW', () => {
const sql = getPostgresSchema(2048, 'voyage-4-large');
expect(sql).toMatch(/vector\(2048\)/);
expect(sql).toMatch(/\('embedding_model', 'voyage-4-large'\)/);
expect(sql).toMatch(/\('embedding_dimensions', '2048'\)/);
expect(sql).not.toContain('idx_chunks_embedding ON content_chunks USING hnsw');
});
test('escapes configured model before inserting into schema SQL literals', () => {
const sql = getPostgresSchema(1024, "voyage-weird'quoted");
expect(sql).toContain("'voyage-weird''quoted'");
expect(sql).not.toContain("'voyage-weird'quoted'");
});
});