Skip to main content

okfgen — turn any repo, database, or open-data portal into an AI-ready knowledge graph

okfgen

Point it at a repo, database, docs site, or open-data portal.
Get a portable, agent-ready knowledge graph in seconds — no LLM, no API key, no lock-in.

PyPI CI Python 3.9+ License Apache-2.0 No LLM required Zero core dependencies Live demo GitHub stars

▶ Explore the live interactive knowledge graphs →


okfgen is a deterministic reference implementation of both sides of the Open Knowledge Format (OKF) — Google's vendor-neutral standard for representing the knowledge around your data and systems as just markdown files with YAML frontmatter (announcement).

  • Producers turn a source system, database, docs site, or live open-data portal into a bundle.
  • Consumers read a bundle back out — a viewer, a search index, a reasoning agent.

It extracts structured facts straight from the source (schemas, file structure, READMEs, dependency manifests, page headings). No LLM and no API key are required; an optional --llm flag adds Claude-powered enrichment where you want it.

              PRODUCERS                              CONSUMERS
   git repo  ─┐                          ┌─  visualize  → interactive HTML graph
   database  ─┤                          ├─  search     → full-text index
  open data  ─┼─►  generate ─► BUNDLE ─► ┼─  ask        → reasoning agent
   local dir ─┤        │       (.md +    └─  validate   → conformance check
   web docs  ─┘        ▼      frontmatter)
                    enrich  (pass 2: join paths, backlinks, citations)

Quickstart (30 seconds)

Zero-install with uv — turn this directory into a knowledge graph and open it:

uvx okfgen generate . -o my-okf
uvx okfgen visualize my-okf -o my-okf/graph.html
# then open my-okf/graph.html

Or with pip:

pip install okfgen
okfgen generate . -o my-okf && okfgen visualize my-okf -o my-okf/graph.html

🎥 See the demo above, or explore the live gallery — every graph is fully interactive.


Why okfgen

  • One command, any source → a knowledge graph. Code, databases, docs, and live open-data portals — the same tool, the same output.
  • Deterministic, offline, no API key. Reproducible facts, not LLM hallucinations. Runs in air-gapped environments. The LLM is strictly opt-in.
  • Open format, zero lock-in. Output is plain markdown + YAML you can read, diff in git, and grep — not a proprietary database.
  • Agent-ready. Search, a citation-backed reasoning agent, and a portable JSON index make bundles first-class context for RAG and AI agents.
  • Provenance built in (OKF v0.2). Every concept records how it was produced (generated) and what it derives from (sources) — with real credibility signals where the source exposes them: publisher, view counts, and freshness dates from live open-data portals.
  • A viewer you can email. The visualizer is a single self-contained HTML file — no backend, no CDN, data never leaves the page.
  • Reference implementation of an open standard. Tracks the OKF v0.2 spec; every bundle it emits passes its own conformance validator.

How it compares

okfgen Data catalogs
(DataHub / Amundsen)
LLM auto-doc tools Hand-written wiki
Runs with no server/DB to deploy ⚠️
No API key / no LLM required
Deterministic & reproducible
Open, plain-markdown output (no lock-in) ⚠️
Code and DB and docs and live open data ⚠️ data only ⚠️ manual
Self-contained interactive graph viewer ⚠️ needs server
Agent-ready (search + reasoning over bundle) ⚠️
Time to first result seconds hours–days minutes

Install

pip install okfgen                 # core: git, local, web, schema (zero deps)
pip install "okfgen[all]"          # + BigQuery, Firebase, MCP, PyYAML

Optional extras: [bigquery], [firebase], [mcp], [yaml], [dev]. For development from a clone: pip install -e '.[dev]'.


Producers — make a bundle from your data

okfgen generate https://github.com/psf/requests.git   # a source system (git)
okfgen generate ./my-project                          # a source system (local)
okfgen generate schema:./warehouse.schema.json        # a database (offline)
okfgen generate schema:./ddl.sql                      # a database (SQL DDL)
okfgen generate bq:my-gcp-project                     # BigQuery datasets/tables
okfgen generate firebase:my-firebase-project          # Firestore collections
okfgen generate https://docs.mytool.dev/              # a documentation site
okfgen generate ckan:https://portal/dataset/some-set  # a live CKAN open-data portal
okfgen generate socrata:https://data.cityofnewyork.us/d/erm2-nwe9  # a live Socrata dataset
Input Detected as What it extracts
git@… / *.git / github URL git shallow-clones, then scans like a local dir
a directory path local README overview, per-directory code modules (functions/classes/types), doc files, dependency inventory
schema:FILE.json / .sql schema dataset + table concepts with full column schemas — no cloud creds
bq:PROJECT bigquery one concept per dataset and per table, with column schemas
firebase:PROJECT firebase one concept per Firestore collection, fields/types inferred from sampled docs
ckan:PORTAL/dataset/SLUG ckan a live CKAN open-data dataset → one concept per resource, with live column schemas + example rows from the DataStore and v0.2 credibility signals (publisher, freshness). No auth; works against data.gov, data.gov.au, the EU portal, city portals, etc.
socrata:DOMAIN/d/4x4-ID socrata a live Socrata dataset (NYC Open Data, Seattle, Chicago, many state portals) → Dataset + Table concepts with live column schema + example rows and credibility signals (publisher, real view counts as usage_count, last-updated). No auth.
http(s)://… web crawls same-host pages (depth/page budget) into one concept per page

Cloud sources use Application Default Credentials (gcloud auth application-default login). Output goes to ./<name>-okf/.

The enrichment agent (pass 2)

Producers draft concepts; the enrichment agent enriches them — exactly the two-pass pattern from the OKF blog. Deterministically, it infers join paths between tables from foreign-key naming (customer_id → customers) and wires backlinks so the graph is navigable both ways:

okfgen enrich ./my-okf                 # in place
okfgen enrich ./my-okf -o ./enriched   # to a new directory
okfgen enrich ./my-okf --llm           # also rewrite descriptions via Claude

Consumers — read a bundle back out

The OKF value proposition is producer/consumer independence: any consumer works on any bundle, regardless of who produced it.

# Viewer: a self-contained interactive graph (no backend, no CDN, data stays local)
okfgen visualize ./my-okf -o graph.html

# Search index: full-text, TF-IDF ranked
okfgen search ./my-okf "weekly active users"
okfgen search ./my-okf --export index.json      # portable JSON index

# Reasoning agent: retrieves concepts, follows join links, answers with citations
okfgen ask ./my-okf "how do orders relate to customers?"
okfgen ask ./my-okf "..." --llm                 # phrase answer via Claude

# Conformance validation
okfgen validate ./my-okf --strict

okfgen ask shows its work — the retrieved concepts, the links it traversed, and the citations behind the answer — so the reasoning is auditable.


Use it inside your AI agent (MCP)

okfgen ships an MCP server, so Claude Desktop, Claude Code, Cursor, and any Model Context Protocol client can produce and reason over OKF bundles without leaving the agent.

pip install "okfgen[mcp]"
okfgen-mcp            # stdio MCP server

Register it (e.g. Claude Desktop claude_desktop_config.json, or Cursor's MCP settings):

{
  "mcpServers": {
    "okfgen": { "command": "okfgen-mcp" }
  }
}

Exposed tools: okfgen_generate, okfgen_search, okfgen_ask, okfgen_validate, okfgen_visualize, okfgen_list_source_types. Now an agent can say "catalog this database and tell me how orders join to customers" and get grounded, cited answers.


Sample bundles

Browse the sample knowledge graphs online: https://bushans.github.io/okfgen/

Ready-to-browse bundles live in samples/bundles/. Open any graph.html in a browser, or point the consumers at them. The same visualizers are published to GitHub Pages from docs/ (regenerate with python samples/build_pages.py).

  • Three offline, reproducible bundles (database, source system, docs site): python samples/build_samples.py
  • One live public-data bundle — Toronto Beaches Water Quality from the Toronto Open Data CKAN portal: python samples/build_live_samples.py

See samples/README.md for details.


Output layout

<name>-okf/
├── index.md            # root listing + okf_version: "0.2"
├── log.md              # generation / enrichment log (ISO-dated)
├── overview.md         # the root "Project" / "Data Project" concept
├── dependencies.md     # parsed manifests (git/local)
├── docs/…              # documentation concepts
├── modules/…           # per-directory code concepts (git/local)
├── datasets/… tables/… # database / BigQuery concepts
├── collections/…       # Firestore concepts
├── pages/…             # web page concepts
└── graph.html          # (after `visualize`) the interactive viewer

Every concept carries the required type frontmatter field plus recommended title/description/resource/tags, and an OKF v0.2 generated: { by, at } provenance stamp. Bodies use the conventional OKF # Schema, # Examples, # Computation, # Joins headings.


Design notes

  • Deterministic by default. git/local/web/schema run on the standard library alone (zero third-party deps). Cloud SDKs and the LLM are optional extras, loaded lazily and off unless you ask.
  • Producer/consumer split. Consumers depend only on markdown + frontmatter (okfgen/consumer.py), never on producer internals.
  • Scriptable. Every command prints its primary output path to stdout and logs to stderr: BUNDLE=$(okfgen generate ./repo).

Development

pip install -e '.[dev]'
pytest

New to the project? TESTING.md is a step-by-step VS Code walkthrough: environment setup, running the test suite, and driving every producer/consumer command locally.

Contributing

Contributions are welcome — new source adapters especially. See CONTRIBUTING.md for the dev setup and a guide to adding a producer in a few lines.

License

Apache-2.0.

Download files

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

Source Distribution

okfgen-0.1.1.tar.gz (66.0 kB view details)

Uploaded Source

Built Distribution

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

okfgen-0.1.1-py3-none-any.whl (64.6 kB view details)

Uploaded Python 3

File details

Details for the file okfgen-0.1.1.tar.gz.

File metadata

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

File hashes

Hashes for okfgen-0.1.1.tar.gz
Algorithm Hash digest
SHA256 f93fb55ad4bde3c96e956756f09c394cb5e44e0a3daabd0ef86cb02430988ba4
MD5 91a4117f2d7885ac8efbc08d74557924
BLAKE2b-256 2af790c4127604ce6ca04f61a563cb6b91b74981e70f68c8f5872347833cb0d6

See more details on using hashes here.

Provenance

The following attestation bundles were made for okfgen-0.1.1.tar.gz:

Publisher: publish.yml on bushans/okfgen

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

File details

Details for the file okfgen-0.1.1-py3-none-any.whl.

File metadata

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

File hashes

Hashes for okfgen-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 927bb83922fb993fd5543c9c9d70dfc2b47e98ab3922418a0a80f950c06517f9
MD5 3a7b1a9c05a4f1ea1da4a6b9c23ce2ef
BLAKE2b-256 080701f9afa2f79172ce491a3d3e0f201de9f6ea3abd799fa47db629d59762e5

See more details on using hashes here.

Provenance

The following attestation bundles were made for okfgen-0.1.1-py3-none-any.whl:

Publisher: publish.yml on bushans/okfgen

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 Sentry Error logging StatusPage Status page