Skip to main content

AKMS Node Generation

Tooling for turning NotebookLM-curated paper batches into validated AKMS knowledge nodes.

The package has two complementary flows:

  • Batch Picker: a local UI for assigning papers to one NotebookLM notebook per AKMS batch.
  • NLM batch generator: a serial CLI that asks a batch's NotebookLM notebook for structured YAML/JSON node output, then validates and writes the results locally.

Install

From the AKMS repo root:

uv sync --project packages/akms_nodes_gen

If you also want the documentation toolchain:

uv sync --project packages/akms_nodes_gen --group docs

Batch picker

uv --project packages/akms_nodes_gen run akms-pick

The picker runs a local FastAPI UI for selecting papers, staging PDFs, creating NotebookLM notebooks, and writing batch plan JSON files under Sources_Evals/NLM/Inputs/.

Serial NotebookLM batch generation

nlm_batch.py runs one selected batch/cluster at a time. It does not hardcode the NotebookLM prompt, source IDs, timeout, or output format; pass those at the CLI boundary.

uv run python -m akms_nodes_gen.nlm_batch \
  --plan Sources_Evals/NLM/Inputs/R7_B2_pf_energy_solvers_plan.json \
  --batch-id R7_B2 \
  --out-dir Sources_Evals/NLM/Outputs/R7_B2_pf_energy_solvers \
  --prompt-file path/to/notebooklm_prompt.md \
  --template-file path/to/output_template.yaml \
  --source-ids source_a,source_b \
  --timeout 180 \
  --output-format yaml

Useful safety controls:

  • --require-source-refs is on by default and requires citations/source refs on equations, algorithms, pitfalls, and references.
  • --allow-invented-edges is off by default; edge targets must match known plan IDs or Tier-1 node IDs.
  • --max-retries controls local repair attempts when NotebookLM returns invalid structure.
  • --converter and --validator can run the existing YAML-to-Markdown and node validation scripts after each YAML write.

The generator writes:

  • {node_id}.yaml
  • _raw_responses/{node_id}.txt
  • _nlm_cache/*.json
  • _nlm_batch_state.json

Docs

Build the docs locally:

uv --project packages/akms_nodes_gen run --group docs \
  mkdocs build -f packages/akms_nodes_gen/mkdocs.yml

Main docs live in packages/akms_nodes_gen/docs/.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

akms_nodes_gen-0.3.1.tar.gz (58.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

akms_nodes_gen-0.3.1-py3-none-any.whl (68.3 kB view details)

Uploaded Python 3

File details

Details for the file akms_nodes_gen-0.3.1.tar.gz.

File metadata

  • Download URL: akms_nodes_gen-0.3.1.tar.gz
  • Upload date:
  • Size: 58.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for akms_nodes_gen-0.3.1.tar.gz
Algorithm Hash digest
SHA256 44754c9bb91435e6566aeb36bf3beafb41d17f7ed0450df1325ec0bda573854e
MD5 7a9cb7e7b07b1f4166b8df1a16da8727
BLAKE2b-256 670cb14b5f4b5498e41d501e2d7af953e37747e535bdc28cace0c836cded99e1

See more details on using hashes here.

Provenance

The following attestation bundles were made for akms_nodes_gen-0.3.1.tar.gz:

Publisher: release.yml on CEmM2/AKMS

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file akms_nodes_gen-0.3.1-py3-none-any.whl.

File metadata

  • Download URL: akms_nodes_gen-0.3.1-py3-none-any.whl
  • Upload date:
  • Size: 68.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for akms_nodes_gen-0.3.1-py3-none-any.whl
Algorithm Hash digest
SHA256 63951e71b747c5fa7353fbbdad836aa6c0b31bb34289aa91794ecd23e5c94934
MD5 4bbd35c56912459fdb570011cfba5081
BLAKE2b-256 39734c73d95ebddf127d44bd68f28e5066e3021e639a76e25e41af633b692bdf

See more details on using hashes here.

Provenance

The following attestation bundles were made for akms_nodes_gen-0.3.1-py3-none-any.whl:

Publisher: release.yml on CEmM2/AKMS

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

This release

0.3.1 This release

2 files

0.3.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page