def create_db()

in evaluation_pipeline/retrieval.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

def create_db():
    # vector database
    #db = sqlite3.connect(":memory:")
    db = sqlite3.connect(":memory:")
    db.enable_load_extension(True)
    sqlite_vec.load(db)
    db.enable_load_extension(False)

    sqlite_version, vec_version = db.execute(
        "select sqlite_version(), vec_version()"
    ).fetchone()
    print(f"sqlite_version={sqlite_version}, vec_version={vec_version}")
    return db