generate-csv.ts 656 B

123456789101112131415161718192021222324252627282930
  1. import { createObjectCsvWriter as createCsvWriter } from 'csv-writer';
  2. // use to test vector insert
  3. const csvWriter = createCsvWriter({
  4. path: './vectors.csv',
  5. header: [{ id: 'vector', title: 'vector' }],
  6. });
  7. const records = [];
  8. const generateVector = (dimension: number) => {
  9. let index = 0;
  10. const vectors = [];
  11. while (index < dimension) {
  12. vectors.push(1 + Math.random());
  13. index++;
  14. }
  15. return JSON.stringify(vectors);
  16. };
  17. while (records.length < 50000) {
  18. const value = generateVector(8);
  19. records.push({ vector: value });
  20. }
  21. csvWriter
  22. .writeRecords(records) // returns a promise
  23. .then(() => {
  24. console.log('...Done');
  25. });