zenml-codebase-analyzer

An automated review of a Python codebase that looks for ZenML pipelines, steps, and other ZenML features. It reports which features are already used and where improvements may be useful.

In plain words
What is it for?
Use it to inspect pipeline and step definitions, metadata logging, tags, scheduling, secrets, lifecycle hooks, reports, and other ZenML usage.
Why use it?
It removes the need to search the project manually for scattered ZenML patterns. It turns those findings into a short list of possible improvements.

Agent

Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

agentmods
npx agentmods add agents/zenml-io/skills/zenml-codebase-analyzer
Clone the repo
git clone --depth 1 https://github.com/zenml-io/skills
Per session 46 Only the description is in the session, so the agent can decide to use it. The body loads when it is invoked.
When invoked 970 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. Scan, not verified.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5 $0.00046 $0.00970
Opus 5 $0.00023 $0.00485
Sonnet 5 $0.00009 $0.00194
Haiku 4.5 $0.00005 $0.00097

Measured 2d ago against content hash 81ae1e6a1c61, method: parsed. Prices are Anthropic first-party input rates as of 2026-08-30, from the pricing page.

Security

Grade A, and why

zenml-codebase-analyzer scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 2d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

skills/zenml-quick-wins/agents/zenml-codebase-analyzer.md · 126 lines

How it starts

The opening of the file, as written. The whole thing — 126 lines — stays where its author put it; the contents beside it link to each section on GitHub.

ZenML Codebase Analyzer

Specialized agent for analyzing Python codebases to identify ZenML usage patterns and quick win opportunities. Searches for specific patterns and returns a concise, structured summary of findings.

Mission

Scan the codebase for ZenML-related patterns, identify what features are already in use, and flag opportunities for improvement. The main conversation doesn't need all the search results—just the actionable insights.

Patterns to Search For

1. Pipeline and Step Definitions

@pipeline - Find all pipeline definitions
@step - Find all step definitions

2. Current Feature Usage (Already Adopted)

log_metadata - Metadata logging
tags= - Pipeline/step tags
Model( - Model Control Plane usage
HTMLString - HTML report visualizations
Schedule( - Scheduling configuration
get_secret or Secret( - Secrets management
on_failure= or on_success= or on_start= or on_end= - Lifecycle hooks

3. Quick Win Opportunities (Missing Patterns)

Look for these anti-patterns or missing features:

Search For If Found Without... Quick Win
@pipeline tags= parameter #9 Tags
@step log_metadata calls #1 Metadata
Hardcoded API keys/passwords get_secret() #7 Secrets
@pipeline Model( usage #12 Model Control Plane
Training metrics (accuracy, loss) log_metadata #1 Metadata
.fit( or .train( Experiment tracker #3 Autologging
try/except notify in step on_failure=/on_success= hooks #18 Lifecycle hooks

4. Security Concerns

# Look for potential hardcoded secrets
password=
api_key=
secret_key=
token=
AWS_ACCESS_KEY

Search Strategy

  1. Find Python files with ZenML imports first
  2. Identify pipeline files (files with @pipeline decorator)
  3. Check each pipeline for feature adoption
  4. Scan for security issues (hardcoded credentials)
  5. Count and categorize findings

Output Format

Read the full file on GitHub · 126 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 2d ago First seen · 126 lines · 46 tokens per session scan A 81ae1e6a1c61

Subscribe to this mod's changes

zenml-codebase-analyzer is an agent published in the GitHub repository zenml-io/skills (6 stars, last pushed 2mo ago), licensed MIT. It adds 46 tokens to every session and 970 once invoked, about $0.0002 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-31.

Related

Other agents, from other repositories

agent-sdk-verifier-py

Use this agent to verify that a Python Agent SDK application is properly configured, follows SDK best practices and documentation recommendations, and is ready for deployment or testing. This agent should be invoked after a Python Agent SDK app has been created or modified.

anthropics/claude-plugins-official · 55 tokens

mlops-engineer

ML operations agent for experiment tracking, model registry, feature stores, ML pipelines, model serving, drift monitoring, and AIOps.

pjt222/agent-almanac · 31 tokens

mainframe-python-backend-engineer

Use for server-side Python work in FastAPI, Django, Flask, and other established services: HTTP APIs, business logic, authentication, data access, PostgreSQL, migrations, workers, realtime communication, caching, object storage, external integrations, generated documents, observability, and backend tests. Not for data…

CATWILLgh/MAINFRAME · 88 tokens

polyglot-engineer

Multi-language expert specializing in Rust, Python, Zig, Go, TypeScript, and Swift. Use for cross-language projects, porting code, or choosing the right language for a task.

rand/cc-polymath · 44 tokens

django-reviewer

Reviews and refines Django/Python code for clarity, consistency, and maintainability while preserving all functionality. Focuses on recently modified code unless instructed otherwise. Use proactively after Django code changes.

vintasoftware/django-ai-plugins · 42 tokens

test-engineer

QA engineer specialized in test strategy, test writing, and coverage analysis. Use for designing test suites, writing tests for existing code, or evaluating test quality.

addyosmani/agent-skills · 35 tokens