If you want to extend our Python library or if you find a bug, please open a PR!
Also be sure to test your code with the unittest command at the /root level directory.
Before starting your work on the project, set up your development environment:
- Clone the repository:
git clone https://github.com/roboflow-ai/roboflow-python.git
cd roboflow-python- Create and activate a virtual environment:
python3 -m venv env
source env/bin/activate3.Install the package in editable mode with development dependencies:
pip install -e ".[dev]"This project comes with a convenient devcontainer that makes it easier to run tests and has lint configured to run on save.
On rare occasions a full rebuild is needed, you can do it in VSCode by pressing Ctrl+Shift+P and running Dev Containers: Rebuild Container.
python -m unittestWhen creating new functions, please follow the Google style for Python docstrings. See example below:
def example_function(param1: int, param2: str) -> bool:
"""Example function that does something.
Args:
param1: The first parameter.
param2: The second parameter.
Returns:
The return value. True for success, False otherwise.
"""We provide a Makefile to format and ensure code quality. Be sure to run them before creating a PR.
# format code with `ruff`
make style
# check code with `ruff`
make check_code_qualityNote These tests will be run automatically when you commit thanks to git hooks.
The docs can be built with mkdocs serve.
Before that, install the dependencies:
python -m pip install mkdocs mkdocs-material mkdocstrings mkdocstrings[python]The CLI lives in roboflow/cli/ with auto-discovered handler modules. To add a new command:
- Create
roboflow/cli/handlers/mycommand.py:
"""My command description."""
from __future__ import annotations
from typing import TYPE_CHECKING
if TYPE_CHECKING:
import argparse
def register(subparsers: argparse._SubParsersAction) -> None:
parser = subparsers.add_parser("mycommand", help="Do something")
sub = parser.add_subparsers(title="mycommand commands")
p = sub.add_parser("list", help="List things")
p.add_argument("-p", "--project", required=True, help="Project ID")
p.set_defaults(func=_list)
parser.set_defaults(func=lambda args: parser.print_help())
def _list(args: argparse.Namespace) -> None:
from roboflow.cli._output import output, output_error, suppress_sdk_output
with suppress_sdk_output():
try:
# ... your logic here ...
data = [{"id": "example"}]
except Exception as exc:
output_error(args, str(exc), hint="Check your project ID.", exit_code=3)
return
output(args, data, text="Found 1 result.")- Add tests in
tests/cli/test_mycommand_handler.py - Run
make check_code_qualityandpython -m unittest
Agent experience checklist (every command must satisfy):
- Supports
--jsonviaoutput()helper - No interactive prompts when all required flags are provided
- Errors use
output_error(args, message, hint=..., exit_code=N) - SDK calls wrapped in
with suppress_sdk_output(): - Exit codes: 0=success, 1=error, 2=auth, 3=not found
Documentation policy: CLI-COMMANDS.md in this repo is a quickstart only. The comprehensive command reference lives in roboflow-product-docs and is published to docs.roboflow.com. When adding a new command, update both: add a quick example to CLI-COMMANDS.md and the full reference to the product docs CLI page.
To ensure code quality and consistency, we use pre-commit hooks. Follow these steps to set up pre-commit in your development environment:
- Install pre-commit:
pip install pre-commit- Install the git hook scripts:
pre-commit installAfter installation, pre-commit will automatically run on git commit. The hooks perform checks and corrections related to code formatting, linting, and other rules as defined in the .pre-commit-config.yaml file.
Note: If you need to bypass pre-commit hooks temporarily, you can use the --no-verify flag:
git commit --no-verify -m "Your commit message"