Open Source

Generate an AI Bill of Materials
for your AI apps

Scan your AI codebase and automatically map the models, agents, prompts, tools, datasets, and APIs inside it. Supports Python, Jupyter notebooks, and JavaScript/TypeScript. The Trivy for AI applications.

MIT License

What AIBOM Detects

AIBOM finds AI components in your code and maps them to a structured inventory with OWASP LLM risk analysis.

your_code.py
from langchain import ChatOpenAI

llm = ChatOpenAI(
    model="gpt-4",
    temperature=0.7
)
gpt-4
OpenAI / LangChain
your_app.ts
import OpenAI from "openai";

const client = new OpenAI({
    model: "gpt-4o-mini"
});
gpt-4o-mini
OpenAI / TypeScript
your_code.py
from langchain.agents import initialize_agent

agent = initialize_agent(
    tools,
    llm,
    agent="zero-shot-react-description"
)
ReAct Agent
zero-shot-react-description
your_code.py
from langchain.tools import Tool

search = Tool(
    name="SerpAPI",
    func=search_run
)
SerpAPI
External Search Tool
your_code.py
embeddings = OpenAIEmbeddings(
    model="text-embedding-3-small"
)
vectorstore = FAISS.from_documents(
    docs, embeddings
)
text-embedding-3-small
FAISS Vector Store
your_code.py
prompt = ChatPromptTemplate.from_messages([
    ("system", "You are a helpful assistant..."),
    ("human", "{input}")
])
System Prompt
2 message templates
6 Source Detectors
50+ Model Providers
20+ AI Frameworks

Example Output

AIBOM generates a structured JSON document mapping all AI components in your codebase with risk findings and provenance tracking.

aibom generate .
$ aibom generate . --output AI_BOM.json --audit-mode
Scanning repository...
Analyzing 247 Python files
Analyzing 56 TypeScript files
Parsing runtime manifests
Detecting AI components
Evaluating risk policies
Generated AI_BOM.json
Project: customer-support-agent
AI Components Detected:
Models:
- gpt-4 (OpenAI)
- text-embedding-3-small
Frameworks:
- langchain
- openai
Agent:
- react-agent
Tools:
- serpapi
- wikipedia
Risk Findings:
- 2 third-party provider warnings
- 1 exfil surface detected

Key Capabilities

Beyond simple detection, AIBOM provides enterprise-grade features for AI supply chain security.

OWASP LLM Risk Analysis

Built-in heuristics aligned with OWASP LLM Top 10. Detect third-party providers, exfiltration surfaces, and prompt injection risks.

Multi-Language Support

Detect AI components in Python, Jupyter notebooks, JavaScript, and TypeScript codebases with framework-specific parsers.

Cryptographic Attestation

Sign evidence bundles with X.509 certificates. Verify provenance, certificate chains, and enforce signer allowlists.

Drift Detection

Compare AIBOM versions to detect new models, tools, or external providers. Gate CI/CD pipelines on unauthorized changes.

Custom Risk Policies

Define organization-specific risk rules with allowlists, thresholds, and severity overrides in JSON or YAML format.

Periodic Scanning

Schedule recurring scans with trend analysis. Track novel components over time and maintain historical snapshots.

Why AIBOM Matters

AI systems contain many hidden dependencies that traditional tooling cannot see.

01

Hidden Dependencies

Models, prompts, tools, datasets, and APIs form complex dependency graphs that are invisible to traditional SBOM tools.

02

Security Risks

Prompt injection, model poisoning, and data leakage require visibility into how AI components interact with your systems.

03

Governance

Organizations need to track AI usage for compliance, risk management, and responsible AI practices.

04

Supply Chain

Third-party models and APIs introduce supply chain risks that must be audited and monitored continuously.

Traditional SBOM Dependencies Libraries AI Models Prompts & Agents AIBOM Dependencies Libraries AI Models Prompts & Agents + = Complete visibility Security coverage Compliance ready

How It Works

Three simple steps to complete AI supply chain visibility.

1

Scan Repository

AIBOM analyzes your codebase using AST parsing to identify Python files, notebooks, JavaScript/TypeScript files, and configuration files.

2

Detect AI Components

Multiple specialized detectors identify models, agents, prompts, tools, datasets, and frameworks with precise source locations and provenance tracking.

3

Generate AIBOM

Produces a structured JSON document with full inventory, OWASP LLM risk findings, and export formats like SPDX and CycloneDX.

Get Started

Install AIBOM and scan your first AI project in minutes.

Install

pip install aibom

Generate

aibom generate .

Export

aibom export --format spdx-json

Advanced Usage

aibom generate . --audit-mode --bundle-out evidence.zip Create signed evidence bundle
aibom diff baseline.json new.json --fail-on new-model Detect drift between versions
aibom periodic-scan . --interval daily Schedule recurring scans

Open Source

AIBOM is built by and for the AI security community. We welcome contributions from AI security researchers, AI engineers, and LangChain developers.

Copied to clipboard