MCPcopy Index your code
hub / github.com/dbt-labs/dbt-extractor

github.com/dbt-labs/dbt-extractor @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
78 symbols 171 edges 7 files 19 documented · 24%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

dbt extractor

Understanding dbt-extractor

This repository contains a tool that processes the most common jinja value templates in dbt model files. The tool depends on tree-sitter and the tree-sitter-jinja2 library.

demo app

Getting started

Strategy

The current strategy is for this processor to be 100% certain when it can accurately extract values from a given model file. Anything less than 100% certainty returns an exception so that the model can be rendered with python Jinja instead.

There are two cases we want to avoid because they would risk correctness to user's projects: 1. Confidently extracting values that would not be extracted by python jinja (false positives) 2. Confidently extracting a set of values that are missing values that python jinja would have extracted. (misses)

If we instead error when we could have confidently extracted values, there is no correctness risk to the user. Only an opportunity to expand the rules to encompass this class of cases as well.

Even though jinja in dbt is not a typed language, the type checker statically determines whether or not the current implementation can confidently extract values without relying on python jinja rendering, which is when these errors would otherwise surface. This type checker will become more permissive over time as this tool expands to include more dbt and jinja features.

Architecture

This architecture is optimized for value extraction and for future flexibility. This architecture is expected to change, and is coded in fp-style stages to make those changes easier for the future.

This processor is composed of several stages: 1. parser 2. type checker 3. extractor

Additionally, the following tools utilize the above processor: 1. browser-based demo of dbt extraction as you type

The tree-sitter parser is located in the tree-sitter-jinja2 library. The rust bindings are used to traverse the concrete syntax tree that tree-sitter creates in order to create a typed abstract syntax tree in the type checking stage. The errors in the type checking stage are not raised to the user, and are instead used by developers to debug tests.

The parser is solely responsible for turning text into recognized values, while the type checker does arity checking, and enforces argument list types (e.g. nested function calls like {{ config(my_ref=ref('table')) }} will parse but not type check even though it is valid dbt syntax. The tool at this time doesn't have an agreed serialization to communicate refs as config values, but could in the future.)

The extractor uses the typed abstract syntax tree to easily identify all the refs, sources, and configs present and extract them.

Join the dbt Community

Reporting bugs and contributing code

Code of Conduct

Everyone interacting in the dbt project's codebases, issue trackers, chat rooms, and mailing lists is expected to follow the dbt Code of Conduct.

Core symbols most depended-on inside this repo

assert_none_type_check
called by 12
src/extractor.rs
to_string
called by 11
src/extractor.rs
mappend
called by 7
src/extractor.rs
text_from_node
called by 7
src/extractor.rs
child_by_field_name
called by 6
src/extractor.rs
map_err
called by 5
src/extractor.rs
assert_all_type_check
called by 5
src/extractor.rs
named_children
called by 4
src/extractor.rs

Shape

Function 61
Enum 8
Method 7
Class 2

Languages

Rust100%

Modules by API surface

src/extractor.rs57 symbols
tests/integration_tests.rs9 symbols
src/exceptions.rs6 symbols
src/python.rs5 symbols
src/main.rs1 symbols

For agents

$ claude mcp add dbt-extractor \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact