Skip to content

Add bank freshness datasets (48 queries, three DB layouts) - #129

Closed
tjworks wants to merge 2 commits into
ucbepic:mainfrom
tjworks:add-bank-fresh-dataset
Closed

tjworks wants to merge 2 commits into
ucbepic:mainfrom
tjworks:add-bank-fresh-dataset

Conversation

@tjworks

@tjworks tjworks commented Oct 28, 2026

Copy link
Copy Markdown

Summary

Adds the bank freshness test: 48 time-sensitive banking questions, each shipped over three database layouts so the same questions can be run against different backends.

  • query_bank_fresh_direct: six source systems (Postgres core banking, SQLite cards/payments, DuckDB lending/insurance, Mongo CRM)
  • query_bank_fresh_live: single Mongo collection (bank_live)
  • query_bank_fresh_warehouse: single Mongo collection plus _meta (bank_warehouse)

Each query directory has query.json, ground_truth.csv and validate.py, matching the existing dataset layout. Binary datasets (.db, .sql, .bson) go through LFS per .gitattributes. Agent run logs are excluded, as for the other datasets.

Scaffold changes

The Mongo-only variants rely on aggregation queries, so this also includes:

  • query_db accepts {"collection": ..., "pipeline": [...]} for MongoDB and runs it via aggregate(); documented in the tool schema and prompt examples.
  • Mongo find() no longer silently caps results at 5 documents, matching SQL behaviour.
  • DAB_PERSISTENT_DBS env var skips reload/cleanup of Mongo and Postgres databases that already exist, so repeated runs reuse loaded data.

🤖 Generated with Claude Code

https://claude.ai/code/session_01SQhxfVKEX2459gdzzSfZEw

mewmonsicha and others added 2 commits October 28, 2026 09:59
- query_db now accepts {"collection": ..., "pipeline": [...]} for MongoDB
  and runs it via aggregate(); documented in the tool schema and prompt.
- Mongo find() no longer silently caps results at 5 documents, matching
  SQL behaviour of returning every row.
- DAB_PERSISTENT_DBS env var skips reload/cleanup of Mongo and Postgres
  databases that already exist, so repeated runs reuse loaded data.

Needed by the bank freshness datasets, whose Mongo-only variants rely on
aggregation queries.

Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01SQhxfVKEX2459gdzzSfZEw
Three variants of the same 48 time-sensitive banking questions, each
with its own db_config.yaml, db_description.txt, ground_truth.csv and
validate.py per query:

- query_bank_fresh_direct:    six source systems (Postgres core banking,
  SQLite cards/payments, DuckDB lending/insurance, Mongo CRM)
- query_bank_fresh_live:      single Mongo collection (bank_live)
- query_bank_fresh_warehouse: single Mongo collection plus _meta (bank_warehouse)

Agent run logs are intentionally left out, as for the other datasets.

Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01SQhxfVKEX2459gdzzSfZEw
@tjworks tjworks closed this Oct 28, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants