Skip to main content

PDF extraction pipeline for scientific papers

Project description

acatome-extract

PDF extraction and enrichment pipeline for scientific papers. Converts PDFs into structured, searchable bundles with block-level summaries and embeddings.

Features

  • Marker PDF extraction — structured block extraction with headings, tables, figures
  • Fitz fallback — recursive character chunking when Marker is unavailable
  • LLM enrichment — block and paper summaries via Ollama or litellm
  • Embeddings — sentence-transformer embeddings for semantic search
  • File watcheracatome-extract watch monitors an inbox folder
  • Bundle format.acatome companion files for sharing pre-built extractions
  • CLIacatome-extract command for extract, enrich, and watch workflows

Installation

uv pip install -e .

With GPU acceleration:

uv pip install -e ".[gpu]"

Usage

from acatome_extract.pipeline import extract

bundle = extract("/path/to/paper.pdf")

CLI

# Extract (RAKE summaries included automatically, no LLM needed)
acatome-extract extract paper.pdf
acatome-extract extract --type datasheet TI_LM317.pdf   # non-article types

# Enrich — embeddings only by default; add --summarize for LLM summaries
acatome-extract enrich /path/to/bundle
acatome-extract enrich --summarize /path/to/bundle       # enable LLM summaries
acatome-extract enrich --summarize --skip-existing dir/   # incremental LLM pass

# Watch — extract + embed + ingest; LLM summaries off by default
acatome-extract watch ~/papers/inbox
acatome-extract watch ~/papers/inbox --summarize          # enable LLM summaries

# Migrate old bundles to new summaries dict format + add RAKE
acatome-extract migrate ~/.acatome/papers
acatome-extract migrate ~/.acatome/papers --dry-run       # preview changes

# Supplements
acatome-extract attach parent-slug supplement.pdf --name s1

Summaries

Extraction always generates RAKE (extractive keyword) summaries — instant, no LLM required. LLM-based summaries are opt-in via --summarize and require an Ollama or litellm-compatible model.

RAKE summaries are used as the default for search and display. To add LLM summaries later:

acatome-extract enrich --summarize --skip-existing ~/.acatome/papers

Sidecar metadata

Place a <stem>.meta.json alongside any PDF to override metadata:

{"type": "datasheet", "title": "LM317 Regulator", "author": "Texas Instruments", "year": 2022}

Supported fields: type, title, author (string or list), year, doi, abstract, journal.

Dependencies

  • acatome-meta — metadata lookup and verification
  • marker-pdf — structured PDF extraction
  • litellm / Ollama — LLM-based enrichment

Testing

uv run python -m pytest tests/ -v

License

GPL-3.0-or-later — see LICENSE.

Project details


Download files

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

Source Distribution

acatome_extract-0.2.1.tar.gz (38.8 kB view details)

Uploaded Source

Built Distribution

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

acatome_extract-0.2.1-py3-none-any.whl (32.4 kB view details)

Uploaded Python 3

File details

Details for the file acatome_extract-0.2.1.tar.gz.

File metadata

  • Download URL: acatome_extract-0.2.1.tar.gz
  • Upload date:
  • Size: 38.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for acatome_extract-0.2.1.tar.gz
Algorithm Hash digest
SHA256 428ccb056253411c85e7aca4696c94b70500b0ecc5c3c3ef610c2b5e28208479
MD5 6639fc600f8b0a0c9a8d5f9d9bb4a43c
BLAKE2b-256 9743f64ab720bf6fbf7a9b040d21d5edce9e11f5257642a58c374e3a7f724856

See more details on using hashes here.

Provenance

The following attestation bundles were made for acatome_extract-0.2.1.tar.gz:

Publisher: publish.yml on retospect/acatome-extract

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

File details

Details for the file acatome_extract-0.2.1-py3-none-any.whl.

File metadata

File hashes

Hashes for acatome_extract-0.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 f96ea3c22eccf8ace5ea483907d1097941edbda8869ed48d0b3530a542a556d5
MD5 06cca307041cc13beafdf0a348712f15
BLAKE2b-256 32fdde5454e33df213260a3391a074b478e22b7967c7f7af79419cc259934849

See more details on using hashes here.

Provenance

The following attestation bundles were made for acatome_extract-0.2.1-py3-none-any.whl:

Publisher: publish.yml on retospect/acatome-extract

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

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page