This is the aitaem library, pronounced "i-tame".
Full documentation: https://chaturv3di.github.io/aitaem/
TL;DR: Point this Python library toward your OLAP database or a local CSV file and start generating insights, i.e. metrics, slices, segments, and time series, in no time. This library is LLM friendly (more on that later).
Business leaders, PMs, EMs, and even individual contributors constantly require deep understanding of their businesses and products. Another term for this understanding is "data insights". The most common way to obtain these insights is to rely on a data scientist or a business analyst to dive into the data and compute the insights. Why is this a bad idea?
- Practically: Dashboards are limited in how much they can hold apriori. There's always a new question which existing dashboards cannot answer. There are always either too many dashboards or too few
- Operationally: It is a waste of time if a DS or a BA has to dive into the source tables and (re)write SQL queries to compute customized metrics, slices, or segments repeatedly
- Scientifically: The accuracy of ad-hoc analysis depends upon the individual; the same analysis done by different individuals can yield different results
- Organisationally: Inter-org trust should be built upon processes/toolings rather than on individuals
This library provides powerful functionality in a compact API. The core consists of two componenents.
- Specifications: A simple declarative structure to modularly define metric specs, slice/breakdown specs, and segment specs
- Computation: A small collection of Python classes with compact signatures which compute the metrics
Additionally, there are utilities to connect to various data backends (simultaneously) and helpers to visualize/render charts and trends.
pip install aitaemfrom aitaem import SpecCache, ConnectionManager, MetricCompute
# 1. Load metric, slice, and segment specs from YAML files or directories
cache = SpecCache.from_yaml(
metric_paths="examples/metrics/",
slice_paths="examples/slices/",
segment_paths="examples/segments/",
)
# 2. Configure the data backend (DuckDB, BigQuery, etc.)
conn = ConnectionManager()
conn.add_connection("duckdb", path=":memory:") # or path to .duckdb file
# 3. Compute metrics
mc = MetricCompute(cache, conn)
table = mc.compute(
metrics="ctr",
slices="campaign_type",
segments="platform",
time_window=("2024-01-01", "2024-04-01"),
)
print(table.to_pandas())Every compute() call returns a lazy ibis.Table with exactly these 11 columns.
Call .to_pandas() to materialise into a DataFrame.
| Column | Description |
|---|---|
period_type |
"all_time" or a named period |
period_start_date |
ISO date string or None |
period_end_date |
ISO date string or None |
entity_id |
Entity column value when by_entity is set; None otherwise |
metric_name |
Name of the metric (e.g. "ctr") |
metric_format |
Format hint from the spec (e.g. "percentage"), or None |
slice_type |
Slice name or "none" for the all-data baseline |
slice_value |
Slice value (e.g. "Search") or "all" |
segment_name |
Segment name or "none" for the all-data baseline |
segment_value |
Segment value (e.g. "Google Ads") or "all" |
metric_value |
Computed numeric result |
The examples/ directory contains sample YAML specs and a CSV dataset for an ad campaigns use case. You can run the following end-to-end:
from aitaem import SpecCache, ConnectionManager, MetricCompute
from aitaem.helpers import load_csvs_to_duckdb
# Load the sample CSV into a DuckDB file and get back a connector
connector = load_csvs_to_duckdb("examples/data/ad_campaigns.csv", "examples/data/ad_campaigns.duckdb")
conn = ConnectionManager()
conn.add_connection("duckdb", connector=connector)
cache = SpecCache.from_yaml(
metric_paths="examples/metrics/",
slice_paths="examples/slices/",
segment_paths="examples/segments/",
)
mc = MetricCompute(cache, conn)
# CTR by campaign type for Q1 2024
table = mc.compute(
metrics="ctr",
slices="campaign_type",
time_window=("2024-01-01", "2024-04-01"),
)
print(table.to_pandas()[["slice_value", "metric_value"]])