deploywizard 0.0.1.23062025

A CLI tool for deploying ML models as production-ready REST APIs

DeployWizard

DeployWizard is a powerful CLI tool that automates the deployment of machine learning models as production-ready REST APIs with Docker support. It generates all the necessary code and configuration files to containerize your ML models with FastAPI, making deployment a breeze.

Features

• Multi-Framework Support: Works with scikit-learn, PyTorch, and TensorFlow models
• Production-Ready: Generates Dockerfiles and optimized FastAPI applications
• Environment-Aware: Handles both development and production environments
• Secure by Default: Uses non-root users in containers and secure defaults
• Easy to Use: Simple CLI interface for generating deployment code
• Customizable: Flexible template system for advanced customization
• Tested: Comprehensive test suite with 100% code coverage

Installation

1. Create and activate a virtual environment (recommended):

   python -m venv venv
   source venv/bin/activate  # On Windows use: venv\Scripts\activate
   

2. Install DeployWizard using pip:

   pip install deploywizard
   

3. Verify installation:

   deploywizard --version
   

Quick Start

1. Register a Model

# Register a scikit-learn model
deploywizard register iris_model.pkl --name iris_classifier --version 1.0.0 --framework sklearn --description "Iris classifier with 95% accuracy"

# Register a PyTorch model
deploywizard register sentiment_model.pt --name sentiment_analyzer --version 2.1.0 --framework pytorch --description "BERT-based sentiment analysis"

# Register a TensorFlow model
deploywizard register image_model.h5 --name image_classifier --version 3.0.0 --framework tensorflow --description "CNN for image classification"

2. List and Inspect Models

# List all registered models
deploywizard list
# Output:
# ┏━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━┳━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━┓
# ┃ Name                ┃ Version ┃ Framework ┃ Description                                      ┃ Registered At ┃
# ┡━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━╇━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━┩
# │ test-model          │ 1.0.0   │ sklearn   │ Test model                                       │ 2025-06-23    │
# │ test-model-db0357e8 │ 1.0.0   │ sklearn   │ Test model                                       │ 2025-06-23    │
# │ sklearn_test_model  │ 1.0.0   │ sklearn   │ Logistic regression model trained on 100         │ 2025-06-23    │
# │                     │         │           │ samples, ...                                     │               │
# └─────────────────────┴─────────┴───────────┴──────────────────────────────────────────────────┴───────────────┘

# Get detailed info about a specific model
deploywizard info --name sklearn_test_model
# Output:
# 
# Model Information
#   • Name: sklearn_test_model
#   • Version: 1.0.0
#   • Framework: sklearn
#   • Path: /path/to/test_models/sklearn_model.pkl
#   • Registered: 2025-06-23T12:35:12.495113+00:00
#
# Description:
#   Logistic regression model trained on 100 samples, 4 features and 2 classes.

3. Deploy a Model

# Deploy a specific version of a model
deploywizard deploy --name sklearn_test_model --version 1.0.0 --output sklearn_test_model_api
# Output:
# Deploying sklearn_test_model (version: 1.0.0)...
# Generating project for sklearn_test_model v 1.0.0
# Output directory: sklearn_test_model_api
# [SUCCESS] Model file copied to /path/to/sklearn_test_model_api/app/sklearn_model.pkl
# [SUCCESS] Project generated successfully in sklearn_test_model_api
# Successfully deployed sklearn_test_model to sklearn_test_model_api
#
# Next steps:
# 1. cd sklearn_test_model_api
# 2. docker-compose up --build
#
# Your API will be available at http://localhost:8000
# API documentation: http://localhost:8000/docs

# Or deploy the latest version (omitting --version)
deploywizard deploy --name sklearn_test_model --output my_model_api

4. Run the Deployed API

After deploying, navigate to the output directory and start the API:

cd sklearn_test_model_api
docker-compose up --build

Once the containers are up and running, you can access:

• API: http://localhost:8000
• Interactive API documentation: http://localhost:8000/docs
• Alternative documentation: http://localhost:8000/redoc

5. Test the API

You can test the API using curl or any HTTP client:

# Health check
curl http://localhost:8000/health
# Expected output: {"status":"healthy"}

# Make predictions (example for a scikit-learn model)
curl -X POST http://localhost:8000/predict \
  -H "Content-Type: application/json" \
  -d '{"data": [[5.1, 3.5, 1.4, 0.2]]}'

Testing

DeployWizard includes a comprehensive test suite. To run the tests:

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

# Run all tests
pytest -v

# Run tests with coverage report
pytest --cov=deploywizard --cov-report=term-missing

# Run a specific test file
pytest tests/test_cli.py -v

Troubleshooting

Common Issues

1. Model not found

   • Ensure the model file exists at the specified path
   • Verify the framework is correctly specified

2. Port already in use

   • Stop any containers using port 8000
   • Or specify a different port: docker run -p 8080:8000

3. Docker build fails

   • Check your internet connection
   • Verify Docker is running
   • Check the Docker logs for specific error messages

4. UnicodeDecodeError on Windows

   • Ensure your terminal supports UTF-8 encoding
   • Set the following environment variable: set PYTHONUTF8=1

Contributing

Contributions are welcome! Please follow these steps:

1. Fork the repository
2. Create a new branch for your feature
3. Commit your changes with descriptive messages
4. Push to the branch
5. Create a new Pull Request

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

• Built with ❤️ using FastAPI and Docker
• Inspired by the need for simple ML model deployment solutions
• Thanks to all contributors who have helped improve this project!