Metadata-Version: 2.4
Name: vi-sdk
Version: 0.1.0b3
Summary: Datature Vi SDK
Project-URL: Changelog, https://vi.developers.datature.com/docs/vi-sdk-changelog
Project-URL: Documentation, https://vi.developers.datature.com/docs/vi-sdk
Project-URL: Homepage, https://vi.datature.com
Author-email: Wei Loon Cheng <weiloon@datature.io>
License: Apache License 2.0
License-File: LICENSE
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Requires-Python: <3.15,>=3.10
Requires-Dist: filetype>=1.2.0
Requires-Dist: google-crc32c>=1.8.0
Requires-Dist: httpx[http2]>=0.28.1
Requires-Dist: msgspec>=0.20.0
Requires-Dist: pillow>=12.0.0
Requires-Dist: rich>=13.0.0
Provides-Extra: all
Requires-Dist: accelerate>=1.12.0; extra == 'all'
Requires-Dist: bitsandbytes>=0.49.0; (sys_platform != 'darwin') and extra == 'all'
Requires-Dist: coverage[toml]>=7.13.0; extra == 'all'
Requires-Dist: docker>=7.1.0; extra == 'all'
Requires-Dist: ipykernel>=7.1.0; extra == 'all'
Requires-Dist: ipywidgets>=8.1.8; extra == 'all'
Requires-Dist: mkdocs-autorefs>=0.5.0; extra == 'all'
Requires-Dist: mkdocs-git-revision-date-localized-plugin>=1.5.0; extra == 'all'
Requires-Dist: mkdocs-material>=9.7.1; extra == 'all'
Requires-Dist: mkdocs-minify-plugin>=0.7.0; extra == 'all'
Requires-Dist: mkdocs>=1.5.0; extra == 'all'
Requires-Dist: mkdocstrings[python]>=1.0.0; extra == 'all'
Requires-Dist: openai>=2.14.0; extra == 'all'
Requires-Dist: peft>=0.18.0; extra == 'all'
Requires-Dist: pre-commit>=4.5.1; extra == 'all'
Requires-Dist: pymdown-extensions>=10.19.1; extra == 'all'
Requires-Dist: pytest-cov>=7.0.0; extra == 'all'
Requires-Dist: pytest-xdist>=3.5.0; extra == 'all'
Requires-Dist: pytest>=9.0.2; extra == 'all'
Requires-Dist: ruff>=0.14.10; extra == 'all'
Requires-Dist: torch>=2.9.1; extra == 'all'
Requires-Dist: torchvision>=0.24.1; extra == 'all'
Requires-Dist: transformers>=4.57.3; extra == 'all'
Requires-Dist: xgrammar>=0.1.29; extra == 'all'
Provides-Extra: deployment
Requires-Dist: accelerate>=1.12.0; extra == 'deployment'
Requires-Dist: bitsandbytes>=0.49.0; (sys_platform != 'darwin') and extra == 'deployment'
Requires-Dist: docker>=7.1.0; extra == 'deployment'
Requires-Dist: openai>=2.14.0; extra == 'deployment'
Requires-Dist: peft>=0.18.0; extra == 'deployment'
Requires-Dist: torch>=2.9.1; extra == 'deployment'
Requires-Dist: torchvision>=0.24.1; extra == 'deployment'
Requires-Dist: transformers>=4.57.3; extra == 'deployment'
Requires-Dist: xgrammar>=0.1.29; extra == 'deployment'
Provides-Extra: dev
Requires-Dist: coverage[toml]>=7.13.0; extra == 'dev'
Requires-Dist: mkdocs-autorefs>=0.5.0; extra == 'dev'
Requires-Dist: mkdocs-git-revision-date-localized-plugin>=1.5.0; extra == 'dev'
Requires-Dist: mkdocs-material>=9.7.1; extra == 'dev'
Requires-Dist: mkdocs-minify-plugin>=0.7.0; extra == 'dev'
Requires-Dist: mkdocs>=1.5.0; extra == 'dev'
Requires-Dist: mkdocstrings[python]>=1.0.0; extra == 'dev'
Requires-Dist: pre-commit>=4.5.1; extra == 'dev'
Requires-Dist: pymdown-extensions>=10.19.1; extra == 'dev'
Requires-Dist: pytest-cov>=7.0.0; extra == 'dev'
Requires-Dist: pytest-xdist>=3.5.0; extra == 'dev'
Requires-Dist: pytest>=9.0.2; extra == 'dev'
Requires-Dist: ruff>=0.14.10; extra == 'dev'
Provides-Extra: inference
Requires-Dist: accelerate>=1.12.0; extra == 'inference'
Requires-Dist: bitsandbytes>=0.49.0; (sys_platform != 'darwin') and extra == 'inference'
Requires-Dist: peft>=0.18.0; extra == 'inference'
Requires-Dist: torch>=2.9.1; extra == 'inference'
Requires-Dist: torchvision>=0.24.1; extra == 'inference'
Requires-Dist: transformers>=4.57.3; extra == 'inference'
Requires-Dist: xgrammar>=0.1.29; extra == 'inference'
Provides-Extra: jupyter
Requires-Dist: ipykernel>=7.1.0; extra == 'jupyter'
Requires-Dist: ipywidgets>=8.1.8; extra == 'jupyter'
Description-Content-Type: text/markdown

# Python SDK for Datature Vi

[![PyPI version](https://badge.fury.io/py/vi-sdk.svg)](https://pypi.org/project/vi-sdk/)
[![Python Versions](https://img.shields.io/pypi/pyversions/vi-sdk.svg)](https://pypi.org/project/vi-sdk/)
[![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)

The official Python SDK for [Datature Vi](https://vi.datature.com) - a powerful platform for vision-language model training and inference.

## 🚀 Installation

```bash
pip install vi-sdk
```

For inference capabilities:
```bash
pip install vi-sdk[inference]
```

## 📖 Documentation

Visit our comprehensive documentation at [vi.developers.datature.com](https://vi.developers.datature.com/docs/vi-sdk)

## 🎯 Quick Start

```python
import vi

# Initialize client
client = vi.Client()

# List datasets
datasets = client.datasets.list()
for dataset in datasets:
    print(f"Dataset: {dataset.name}")
```

## ✨ Features

- **Dataset Management**: Create, upload, and manage datasets
- **Asset Operations**: Batch upload and download with progress tracking
- **Annotations**: JSONL format support for captions and grounding
- **Model Training**: Track runs and manage trained models
- **Inference**: Local VLM inference (Qwen2.5-VL, NVILA)
- **Dataset Loaders**: PyTorch-compatible data loading utilities

## 📝 License

Apache License 2.0 - see [LICENSE](LICENSE) file for details.

## 🤝 Contributing

See [CONTRIBUTING.md](CONTRIBUTING.md) for contribution guidelines.

### Development Setup

We use pre-commit hooks to ensure code quality:

```bash
# Install dependencies
uv sync --all-extras

# Install pre-commit hooks
uv run pre-commit install

# Verify setup
uv run pre-commit run --all-files
```

📖 **Documentation:**
- [Development Guide](../../.github/DEVELOPMENT.md) - Complete development workflow
- [Pre-commit Setup](../../.github/PRE_COMMIT_SETUP.md) - Quick reference for hooks
- [Setup Summary](../../.github/SETUP_SUMMARY.md) - Overview of CI/CD improvements

## 🐛 Issues

Report issues on [GitHub Issues](https://github.com/datature/Vi-SDK/issues)
