from chalk import DataFrame, offline
from chalk.sql import scan_parquet

from features.quest import Quest


@offline
def get_quests() -> DataFrame[Quest]:
    return scan_parquet(
        files="s3://q-chalk-test-data/surveyors-guild/quest-rows/",
        columns=["id", "party_id", "title", "difficulty", "reward_gold", "status", "started_date"],
    )