David Snelling
3d80df1726
- Fix soft delete functionality by filtering out deleted items in search results - Add support for encrypted configuration storage and retrieval - Copy test suite and vitest config to ensure all functionality works - Maintain all core brainy functionality during repository cleanup
59 lines
1.9 KiB
TypeScript
59 lines
1.9 KiB
TypeScript
import { describe, it, expect } from 'vitest'
|
|
import { BrainyData } from '../dist/unified.js'
|
|
|
|
describe('Vector Dimension Standardization', () => {
|
|
it('should initialize BrainyData with 384 dimensions', async () => {
|
|
// Initialize BrainyData
|
|
const db = new BrainyData()
|
|
await db.init()
|
|
|
|
// Check the dimensions property
|
|
expect(db.dimensions).toBe(384)
|
|
})
|
|
|
|
it('should reject vectors with incorrect dimensions', async () => {
|
|
const db = new BrainyData()
|
|
await db.init()
|
|
|
|
// Test with a simple vector (this should throw an error because it's not 384 dimensions)
|
|
const smallVector = [0.1, 0.2, 0.3]
|
|
|
|
// Expect the add operation to throw an error
|
|
await expect(db.add(smallVector, { test: 'small-vector' }))
|
|
.rejects.toThrow()
|
|
})
|
|
|
|
it('should successfully embed text to 384 dimensions', async () => {
|
|
const db = new BrainyData()
|
|
await db.init()
|
|
|
|
// Test with text that will be embedded to 384 dimensions
|
|
const id = await db.add('This is a test text that will be embedded to 384 dimensions', { test: 'text-embedding' })
|
|
|
|
// Retrieve the vector and check its dimensions
|
|
const noun = await db.get(id)
|
|
expect(noun.vector.length).toBe(384)
|
|
})
|
|
|
|
it('should directly embed text to 384 dimensions', async () => {
|
|
const db = new BrainyData()
|
|
await db.init()
|
|
|
|
// Test direct embedding
|
|
const vector = await db.embed('Another test text')
|
|
expect(vector.length).toBe(384)
|
|
})
|
|
|
|
it('should use the default dimensions regardless of configuration', async () => {
|
|
// Create a BrainyData instance with a specific dimension
|
|
const customDimension = 300
|
|
const db = new BrainyData({
|
|
dimensions: customDimension
|
|
})
|
|
await db.init()
|
|
|
|
// The API currently uses the default dimensions (384) regardless of configuration
|
|
// This is the current behavior, though it might not be the intended behavior
|
|
expect(db.dimensions).toBe(384)
|
|
})
|
|
})
|