"""Module to upsert data into AstraDB"""
import os
import logging

from langchain_astradb import AstraDBVectorStore

logging.basicConfig(
    format='%(asctime)s - %(levelname)s - %(funcName)s - %(message)s',
    datefmt="%Y-%m-%d %H:%M:%S",
    level=logging.INFO)

logging.info("Initializing AstraDB client...")

ASTRA_DB_APPLICATION_TOKEN = os.environ['ASTRA_DB_APPLICATION_TOKEN']
ASTRA_DB_API_ENDPOINT = os.environ['ASTRA_DB_API_ENDPOINT']
COLLECTION_NAME = "article"

vstore = AstraDBVectorStore(
    namespace="default_keyspace",
    collection_name="article",
    token=os.environ["ASTRA_DB_APPLICATION_TOKEN"],
    api_endpoint=os.environ["ASTRA_DB_API_ENDPOINT"],
    autodetect_collection=True)

results = vstore.similarity_search(query="thud",k=1)
for doc in results:
    print(f"* {doc.page_content} [{doc.metadata}]")