import { AstraDB } from "@datastax/astra-db-ts";

async function main() {

  const ASTRA_DB_API_ENDPOINT = process.env['ASTRA_DB_API_ENDPOINT'];
  const ASTRA_DB_APPLICATION_TOKEN = process.env['ASTRA_DB_APPLICATION_TOKEN'];

  const db = new AstraDB(ASTRA_DB_APPLICATION_TOKEN, ASTRA_DB_API_ENDPOINT);

  // Create a vector collection
  await db.createCollection("vector_5_collection", { vector: { dimension: 5, metric: "cosine" } });
  const collection = await db.collection("vector_5_collection");

  const documents = [
    {
      "_id": "1",
      "text": "ChatGPT integrated sneakers that talk to you",
      "$vector": [0.1, 0.15, 0.3, 0.12, 0.05],
    },
    {
      "_id": "2",
      "text": "An AI quilt to help you sleep forever",
      "$vector": [0.45, 0.09, 0.01, 0.2, 0.11],
    },
    {
      "_id": "3",
      "text": "Vision Vector Frame - A deep learning display that controls your mood",
      "$vector": [0.1, 0.05, 0.08, 0.3, 0.6],
    }
  ];
  await collection.insertMany(documents);

  // Return the number of documents in the collection
  const results = await collection.countDocuments();
  console.log(results);
}

main().catch(console.error);