Spaces:
Runtime error
Runtime error
| import { ChromaClient } from '../../src/ChromaClient'; | |
| // import env.ts | |
| window.onload = async () => { | |
| const chroma = new ChromaClient({ path: "http://localhost:8000" }); | |
| await chroma.reset(); | |
| const collection = await chroma.createCollection({ name: "test-from-js" }); | |
| console.log("collection", collection); | |
| // first generate some data | |
| var ids: string[] = []; | |
| var embeddings: Array<any> = []; | |
| var metadatas: Array<any> = []; | |
| for (let i = 0; i < 100; i++) { | |
| ids.push("test-id-" + i.toString()); | |
| embeddings.push([1, 2, 3, 4, 5]); | |
| metadatas.push({ test: "test" }); | |
| } | |
| let add = await collection.add({ ids, embeddings, metadatas }); | |
| console.log("add", add); | |
| let count = await collection.count(); | |
| console.log("count", count); | |
| const queryData = await collection.query({ | |
| queryEmbeddings: [1, 2, 3, 4, 5], | |
| nResults: 5, | |
| where: { test: "test" } | |
| }); | |
| console.log("queryData", queryData); | |
| await collection.delete(); | |
| let count2 = await collection.count(); | |
| console.log("count2", count2); | |
| const collections = await chroma.listCollections(); | |
| console.log("collections", collections); | |
| // this code is commented out so that it is easy to see the output on the page if desired | |
| // let node; | |
| // node = document.querySelector("#list-collections-result"); | |
| // node!.innerHTML = `<pre>${JSON.stringify(collections.data, null, 4)}</pre>`; | |
| // node = document.querySelector("#collection-count"); | |
| // node!.innerHTML = `<pre>${count}</pre>`; | |
| // node = document.querySelector("#collection-get"); | |
| // node!.innerHTML = `<pre>${JSON.stringify(getData, null, 4)}</pre>`; | |
| // node = document.querySelector("#collection-query"); | |
| // node!.innerHTML = `<pre>${JSON.stringify(queryData, null, 4)}</pre>`; | |
| }; | |