| """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}]") | |