PyroMind SDK Example

Python example project for the PyroMind SDK.

Provides an automated testing framework to verify SDK node behavior against documentation.

Directory Structure

├── docs/              # Node definitions, I/O specs, and command examples
├── workflow/          # Workflow definition files
├── imgs/              # Flow charts, architecture diagrams, etc.
├── tests/             # Automated test suite
│   ├── test_node_doc_consistency.py  # 全量节点校验
│   ├── test_single_node.py           # 单节点测试
│   └── test_output/                  # 测试结果输出

Workflow Test

Submit a workflow JSON file via the SDK CLI, wait for completion, and export each node's input and output.

export PYROMIND_API_KEY=<your-api-key>
python -m pyromind_sdk.test_run_workflow_cli <workflow.json> [options]

Options

Option	Default	Description
--name	from JSON	Task name
--output	-	Output results file
--poll-interval	5	Poll interval in seconds
--timeout	600	Max wait time in seconds
--pretty	false	Pretty-print JSON output
--max-retries	0	Max retries for API requests
-h, --help	-	Show help message

Examples

python -m pyromind_sdk.test_run_workflow_cli workflow/LoadDataset/LoadDataset.json --pretty
python -m pyromind_sdk.test_run_workflow_cli workflow/LoadImage/LoadImage.json --pretty

Test Modes (tests/)

The test framework is document-driven and supports two modes:

1. Full Node Validation

Read all system nodes via the SDK, then automatically compare each node's input/output against the definitions in docs/, and run their workflows.

2. Single / Specific Node Test

Use test_single_node.py with --node filter or PYROMIND_NODES env var to test specific nodes in isolation for quick troubleshooting.

SDK

This project depends on the PyroMind SDK.

Setup

pip install pyromind-sdk

Set the required environment variables:

export PYROMIND_API_KEY=<your-api-key>
export PYROMIND_BASE_URL=https://api-portal.pyromind.ai/api/v1
export PYROMIND_CLUSTER=us-west-2

Running Tests

# Full node validation — compare all system nodes against docs
python tests/test_node_doc_consistency.py

# Single / specific node tests
python tests/test_single_node.py --node LoadImage
python tests/test_single_node.py --node "LoadImage or LoadDataset"

# Or via environment variable
PYROMIND_NODES="LoadImage" python tests/test_single_node.py