Skip to main content

Structured LLM task runner with schema validation, confidence scoring, and subagent orchestration

Project description

Cognitive Modules

CI npm version PyPI version npm downloads Node.js 18+ Python 3.9+ License: MIT

Verifiable Structured AI Task Specification

English | ไธญๆ–‡

Cognitive Modules is an AI task definition specification designed for generation tasks that require strong constraints, verifiability, and auditability.

What's New in v2.2

Feature Description
Control/Data Separation meta control plane + data data plane, middleware can route without parsing business logic
Module Tiers exec / decision / exploration with different strictness levels
Recoverable Overflow extensions.insights preserves LLM's additional insights
Extensible Enums Allow custom types without sacrificing type safety
Repair Pass Auto-fix formatting issues, reduce validation failures

Features

  • Strong Type Contracts - JSON Schema bidirectional validation for input/output
  • Explainable Output - Mandatory confidence + rationale output
  • Control/Data Separation - meta.explain for quick routing + data.rationale for detailed audit
  • Module Tiers - exec / decision / exploration with different constraint levels
  • Subagent Orchestration - @call:module supports inter-module calls
  • Parameter Passing - $ARGUMENTS runtime substitution
  • Multi-LLM Support - OpenAI / Anthropic / MiniMax / Ollama
  • Public Registry - cog install registry:module-name

Version Selection

Version Spec npm PyPI Status
v2.2 v2.2 2.2.0 2.2.1 โœ… Stable (recommended)
v2.5 v2.5 2.5.0-beta.x 2.5.0bx ๐Ÿงช Beta (streaming + multimodal)
# Install stable v2.2
npm install cognitive-modules-cli@2.2.0
pip install cognitive-modules==2.2.1

# Install beta v2.5 (streaming + multimodal)
npm install cognitive-modules-cli@beta
pip install cognitive-modules==2.5.0b1

Installation

Node.js (npm) - Recommended

# Zero-install quick start (recommended)
npx cognitive-modules-cli@2.2.0 run code-reviewer --args "your code"

# Global installation
npm install -g cognitive-modules-cli@2.2.0

Python (pip)

pip install cognitive-modules==2.2.1

# With LLM support
pip install "cognitive-modules[openai]==2.2.1"      # OpenAI
pip install "cognitive-modules[anthropic]==2.2.1"   # Claude
pip install "cognitive-modules[all]==2.2.1"         # All providers
Platform Package Command Features
npm cognitive-modules-cli cog โœ… Recommended, zero-install, full features
pip cognitive-modules cog โœ… Full features

Quick Start

# Configure LLM
export LLM_PROVIDER=openai
export OPENAI_API_KEY=sk-xxx

# Run code review (npm)
npx cognitive-modules-cli run code-reviewer --args "def login(u,p): return db.query(f'SELECT * FROM users WHERE name={u}')" --pretty

# Or use globally installed cog command
cog run code-reviewer --args "..." --pretty

# Run task prioritization
cog run task-prioritizer --args "fix bug(urgent), write docs, optimize performance" --pretty

# Run API design
cog run api-designer --args "user system CRUD API" --pretty

# Start HTTP service (API integration)
cog serve --port 8000

# Start MCP server (Claude Code / Cursor integration)
cog mcp

v2.2 Response Format

All modules now return the unified v2.2 envelope format:

{
  "ok": true,
  "meta": {
    "confidence": 0.92,
    "risk": "low",
    "explain": "Brief summary for quick routing decisions (โ‰ค280 chars)"
  },
  "data": {
    "...business fields...",
    "rationale": "Detailed reasoning for audit and human review",
    "extensions": {
      "insights": [
        {
          "text": "Additional insight",
          "suggested_mapping": "Suggested field to add to schema"
        }
      ]
    }
  }
}

Control vs Data Plane

Layer Field Purpose
Control Plane meta.confidence Routing/fallback decisions
Control Plane meta.risk Human review trigger
Control Plane meta.explain Logs/card UI
Data Plane data.rationale Detailed audit
Data Plane data.extensions Recoverable insights

Core Features

Feature Description
JSON Schema Validation Bidirectional input/output validation
Confidence Every output must include 0-1 confidence
Reasoning meta.explain (brief) + data.rationale (detailed)
Module Tiers tier: exec | decision | exploration
Risk Aggregation meta.risk = max(changes[*].risk)
Parameter Passing $ARGUMENTS runtime substitution
Subagents @call:module for inter-module calls
Validation Tools cog validate / cog validate --v22

Integration Methods

Method Command Use Case
CLI cog run Command line
HTTP API cog serve n8n, Coze, Dify
MCP Server cog mcp Claude, Cursor

CLI Commands

# Module management
cog list                    # List installed modules
cog info <module>           # View module details
cog validate <module>       # Validate module structure
cog validate <module> --v22 # Validate v2.2 format

# Run modules
cog run <module> input.json -o output.json --pretty
cog run <module> --args "requirements" --pretty
cog run <module> --args "requirements" --subagent  # Enable subagent

# Create modules
cog init <name> -d "description"
cog init <name> --format v22  # Create v2.2 format module

# Migrate modules
cog migrate <module>        # Migrate v1/v2.1 module to v2.2

# Install from GitHub (recommended)
cog add ziel-io/cognitive-modules -m code-simplifier
cog add org/repo -m module-name --tag v1.0.0   # Install specific version
cog remove <module>                             # Remove module

# Version management
cog update <module>                 # Update to latest version
cog update <module> --tag v2.0.0    # Update to specific version
cog versions <url>                  # View available versions

# Other installation methods
cog install github:user/repo/path
cog install registry:module-name
cog uninstall <module>

# Registry
cog registry                # View public modules
cog search <query>          # Search modules

# Environment check
cog doctor

Built-in Modules

Module Tier Function Example
code-reviewer decision Code review cog run code-reviewer --args "your code"
code-simplifier decision Code simplification cog run code-simplifier --args "complex code"
task-prioritizer decision Task priority sorting cog run task-prioritizer --args "task1,task2"
api-designer decision REST API design cog run api-designer --args "order system"
ui-spec-generator exploration UI spec generation cog run ui-spec-generator --args "e-commerce homepage"
product-analyzer exploration Product analysis (subagent) cog run product-analyzer --args "health product" -s

Module Format

v2.2 Format (Recommended)

my-module/
โ”œโ”€โ”€ module.yaml     # Machine-readable manifest (with tier/overflow/enums)
โ”œโ”€โ”€ prompt.md       # Human-readable prompt
โ”œโ”€โ”€ schema.json     # meta + input + data + error schemas
โ””โ”€โ”€ tests/          # Golden test cases
    โ”œโ”€โ”€ case1.input.json
    โ””โ”€โ”€ case1.expected.json

module.yaml (v2.2)

name: my-module
version: 2.2.0
responsibility: One-line description

tier: decision           # exec | decision | exploration
schema_strictness: medium # high | medium | low

excludes:
  - things not to do

policies:
  network: deny
  filesystem_write: deny
  side_effects: deny

overflow:
  enabled: true
  recoverable: true
  max_items: 5
  require_suggested_mapping: true

enums:
  strategy: extensible   # strict | extensible

failure:
  contract: error_union
  partial_allowed: true

compat:
  accepts_v21_payload: true
  runtime_auto_wrap: true

v1 Format (Still Supported)

my-module/
โ”œโ”€โ”€ MODULE.md       # Metadata + instructions
โ”œโ”€โ”€ schema.json     # Input/output schema
โ””โ”€โ”€ examples/
    โ”œโ”€โ”€ input.json
    โ””โ”€โ”€ output.json

Tier Explanation

Tier Purpose Schema Strictness Overflow
exec Auto-execution (patch, instruction generation) high Disabled
decision Judgment/evaluation/classification medium Enabled
exploration Exploration/research/inspiration low Enabled

Using with AI Tools

Cursor / Codex CLI

Create AGENTS.md in your project root:

## Code Review

When code review is needed:
1. Read `~/.cognitive/modules/code-reviewer/MODULE.md`
2. Output in schema.json format
3. Include meta.explain + data.rationale

Direct Conversation

Read ~/.cognitive/modules/code-reviewer/MODULE.md,
review this code: def login(u,p): ...

LLM Configuration

# OpenAI
export LLM_PROVIDER=openai
export OPENAI_API_KEY=sk-xxx

# Anthropic Claude
export LLM_PROVIDER=anthropic
export ANTHROPIC_API_KEY=sk-ant-xxx

# MiniMax
export LLM_PROVIDER=minimax
export MINIMAX_API_KEY=sk-xxx

# Ollama (local)
export LLM_PROVIDER=ollama

# Check configuration
cog doctor

Migrating to v2.2

Migrate from v1 or v2.1 modules to v2.2:

# Auto-migrate single module
cog migrate code-reviewer

# Migrate all modules
cog migrate --all

# Verify migration result
cog validate code-reviewer --v22

Manual migration steps:

  1. Create module.yaml (add tier/overflow/enums)
  2. Update schema.json (add meta schema)
  3. Create/update prompt.md (describe v2.2 envelope format)
  4. Keep MODULE.md (backward compatibility)

Development

# Clone
git clone https://github.com/ziel-io/cognitive-modules.git
cd cognitive-modules

# Install dev dependencies
pip install -e ".[dev]"

# Run tests
pytest tests/ -v

# Create new module (v2.2 format)
cog init my-module -d "module description" --format v22
cog validate my-module --v22

Project Structure

cognitive-modules/
โ”œโ”€โ”€ src/cognitive/          # Python CLI source
โ”‚   โ”œโ”€โ”€ cli.py              # Command entry
โ”‚   โ”œโ”€โ”€ loader.py           # Module loader (supports v0/v1/v2.2)
โ”‚   โ”œโ”€โ”€ runner.py           # Module executor (v2.2 envelope)
โ”‚   โ”œโ”€โ”€ validator.py        # Module validator (includes v2.2 validation)
โ”‚   โ”œโ”€โ”€ migrate.py          # v2.2 migration tool
โ”‚   โ”œโ”€โ”€ subagent.py         # Subagent orchestration
โ”‚   โ”œโ”€โ”€ registry.py         # Module installation
โ”‚   โ”œโ”€โ”€ templates.py        # Module templates
โ”‚   โ””โ”€โ”€ providers/          # LLM backends
โ”œโ”€โ”€ packages/
โ”‚   โ””โ”€โ”€ cli-node/           # Node.js CLI (npm: cognitive-modules-cli)
โ”‚       โ”œโ”€โ”€ src/            # TypeScript source
โ”‚       โ””โ”€โ”€ package.json
โ”œโ”€โ”€ cognitive/modules/      # Built-in modules (all v2.2)
โ”œโ”€โ”€ coze-plugin/            # Coze integration plugin
โ”œโ”€โ”€ tests/                  # Unit tests
โ”œโ”€โ”€ SPEC.md                 # v0.1 specification (historical)
โ”œโ”€โ”€ SPEC-v2.2.md            # v2.2 specification (latest)
โ”œโ”€โ”€ INTEGRATION.md          # Integration guide
โ””โ”€โ”€ cognitive-registry.json # Public registry

Multi-Platform Support

Platform Package Command Installation
Python cognitive-modules cog pip install cognitive-modules
Node.js cognitive-modules-cli cog npm install -g cognitive-modules-cli

Both versions share the same module format and v2.2 specification.

Documentation

Specification

Document Description
SPEC-v2.2.md v2.2 full specification (Control/Data separation, Tier, Overflow)
SPEC-v2.2_zh.md v2.2 ่ง„่Œƒไธญๆ–‡็‰ˆ
SPEC.md v0.1 specification (context philosophy)

For Implementers

Document Description
IMPLEMENTERS-GUIDE.md Step-by-step guide for building a runtime
CONFORMANCE.md Conformance levels (Level 1/2/3)
ERROR-CODES.md Standard error code taxonomy (E1xxx-E4xxx)
templates/runtime-starter/ Starter template for new implementations

Advanced Features

Document Description
COMPOSITION.md Module composition and dataflow specification
CONTEXT-PROTOCOL.md Context protocol for stateful workflows

Schemas & Test Vectors

Resource Description
spec/response-envelope.schema.json JSON Schema for v2.2 envelope validation
spec/module.yaml.schema.json JSON Schema for module.yaml
spec/test-vectors/ Official test vectors for compliance

Registry & Distribution

Resource Description
REGISTRY-PROTOCOL.md Registry protocol specification
spec/registry-entry.schema.json Registry entry JSON Schema
cognitive-registry.json Current public registry
CERTIFICATION.md Certification program (badges, verification)

Governance

Document Description
GOVERNANCE.md Project governance structure
CMEP-PROCESS.md Enhancement proposal process

Integration

Document Description
INTEGRATION.md Agent tool integration guide
COGNITIVE-PROTOCOL.md Protocol details

License

MIT

Project details


Download files

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

Source Distribution

cognitive_modules-2.2.2.tar.gz (70.4 kB view details)

Uploaded Source

Built Distribution

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

cognitive_modules-2.2.2-py3-none-any.whl (54.6 kB view details)

Uploaded Python 3

File details

Details for the file cognitive_modules-2.2.2.tar.gz.

File metadata

  • Download URL: cognitive_modules-2.2.2.tar.gz
  • Upload date:
  • Size: 70.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cognitive_modules-2.2.2.tar.gz
Algorithm Hash digest
SHA256 15ad4808d31a2603cf4468c4bf9075faa4949bd14da05c10b6e0ce60e8d73091
MD5 ebe3be21363748c4987499c63b12edff
BLAKE2b-256 c8319b5d06d42b2c60c44e99206f5684d23d3506c519191fa8e39e982cff392a

See more details on using hashes here.

File details

Details for the file cognitive_modules-2.2.2-py3-none-any.whl.

File metadata

File hashes

Hashes for cognitive_modules-2.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 70abcf6028ed6ddc435305f49c0090235299ac8c8a71b98d958d29c8620d4689
MD5 5fc4cb92ecad38cd20ef66b10b8b088b
BLAKE2b-256 20ae90d9f5e319da39d42911a104e5a4c1ee9ffb3f68780da9af678c7d89aefe

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page