chore: initial commit

script/generate_mock_data.js

@@ -0,0 +1,35 @@
+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)