36 lines
799 B
JavaScript
36 lines
799 B
JavaScript
const { ChromaClient } = require('chromadb')
|
|
|
|
const chromaPath = 'http://localhost:8000'
|
|
|
|
const chroma = new ChromaClient({
|
|
path: chromaPath,
|
|
// auth: {
|
|
// provider: 'basic',
|
|
// credentials: 'username:password',
|
|
// // provider: "token",
|
|
// // credentials: "test-token",
|
|
// },
|
|
})
|
|
|
|
async function main() {
|
|
const collection = await chroma.createCollection({ name: 'test-collection' })
|
|
for (let i = 0; i < 2000; i++) {
|
|
await collection.add({
|
|
ids: ['test-id-' + i.toString()],
|
|
embeddings: [[1, 2, 3, 4, 5]],
|
|
documents: ['test'],
|
|
})
|
|
process.stdout.write('.')
|
|
}
|
|
console.log('')
|
|
console.log('Done')
|
|
|
|
const queryData = await collection.query({
|
|
queryEmbeddings: [1, 2, 3, 4, 5],
|
|
})
|
|
|
|
console.log(queryData)
|
|
}
|
|
|
|
main().catch(console.error)
|