src/vectordb/create_db.py

from src.vectordb.search import *
import logging

logger = logging.getLogger(__name__)
logging.basicConfig(encoding='utf-8', level=logging.DEBUG)

from src.vectordb.ingest import create_wikivoyage_docs_db_and_add_data, create_wikivoyage_listings_db_and_add_data


def run():
    logging.info("Creating database for Wikivoyage Documents")
    try:
        create_wikivoyage_docs_db_and_add_data()
    except Exception as e:
        logger.error(f"Error for Wikivoyage Documents: {e}")

    logging.info("Creating database for Wikivoyage Listings")
    try:
        create_wikivoyage_listings_db_and_add_data()
        print("Completed")
    except Exception as e:
        logger.error(f"Error for Wikivoyage Listings: {e}")


if __name__ == "__main__":
    run()