This Developer Guide is designed to help you contribute to the OpenLLM project. Follow these steps to set up your development environment and learn the process of contributing to our open-source project.
Join our Discord Channel and reach out to us if you have any question!
Before you can start developing, you'll need to set up your environment:
Important
We recommend using the Python version from .python-version-default
file within the project root
to avoid any version mismatch. You can use pyenv to manage your python version.
Note that hatch run setup
will symlink the python version from .python-version-default
to .python-version
in the project root.
Therefore any tools that understand .python-version
will use the correct Python version.
-
Ensure you have Git, and Python3.8+ installed.
-
Fork the OpenLLM repository from GitHub.
-
Clone the forked repository from GitHub:
git clone [email protected]:username/OpenLLM.git && cd openllm
-
Add the OpenLLM upstream remote to your local OpenLLM clone:
git remote add upstream [email protected]:bentoml/OpenLLM.git
-
Configure git to pull from the upstream remote:
git switch main # ensure you're on the main branch git fetch upstream --tags git branch --set-upstream-to=upstream/main
-
Install hatch:
pip install hatch pre-commit
-
Run the following to setup all pre-commit hooks:
hatch run setup
-
Enter a project's environment with.
hatch shell
This will automatically enter a virtual environment and update the relevant dependencies.
Note
If you don't want to work with hatch, you can use the editable workflow with running bash local.sh
Here's a high-level overview of our project structure:
openllm/
├── ADDING_NEW_MODEL.md # How to add a new model
├── CHANGELOG.md # Generated changelog
├── CITATION.cff # Citation File Format
├── DEVELOPMENT.md # The project's Developer Guide
├── Formula # Homebrew Formula
├── LICENSE.md # Use terms and conditions
├── README.md # The project's README file
├── STYLE.md # The project's Style Guide
├── cz.py # code-golf commitizen
├── examples # Usage demonstration scripts
├── openllm-node # openll node library
├── openllm-python # openllm python library
│ └── src
│ └── openllm # openllm core implementation
├── pyproject.toml # Python Project Specification File (PEP 518)
└── tools # Utilities Script
After setting up your environment, here's how you can start contributing:
-
Create a new branch for your feature or fix:
git checkout -b feature/my-feature
-
Make your changes to the codebase.
-
Run all formatter and linter with
hatch
:hatch run quality
-
Write tests that verify your feature or fix (see Writing Tests below).
-
Run all tests to ensure your changes haven't broken anything:
hatch run tests:python
-
Commit your changes:
git commit -m "Add my feature"
-
Push your changes to your fork:
git push origin feature/my-feature
-
Submit a Pull Request on GitHub.
If you wish to use a modified version of OpenLLM, install your fork from source
with pip install -e
and set OPENLLM_DEV_BUILD=True
, so that Bentos built
will include the generated wheels for OpenLLM in the bundle.
Good tests are crucial for the stability of our codebase. Always write tests for your features and fixes.
We use pytest
for our tests. Make sure your tests are in the tests/
directory and their filenames start with test_
.
Run all tests with:
hatch run tests:python
Run snapshot testing for model outputs:
hatch run tests:models
To update the snapshot, do the following:
hatch run tests:snapshot-models
To filter out most of the generated commits for infrastructure, use
--invert-grep
in conjunction with --grep
to filter out all commits with
regex "[generated]"
You can run the following to test the behaviour of the compiled module:
hatch run compile
Important
This will compiled some performance sensitive modules with mypyc.
The compiled .so
or .pyd
can be found
under /openllm-python/src/openllm
. If you run into any issue, run hatch run recompile
See STYLE.md for our style guide.
After you change or update any CI related under .github
, run bash tools/lock-actions.sh
to lock the action version.
See this docs for more information on OpenLLM's CI/CD workflow.
See ClojureScript UI's README.md for more information.
See Documentation's README.md for more information on running locally.
pip install 'https://github.com/bentoml/OpenLLM/archive/main.tar.gz#subdirectory=openllm-python'
To release a new version, use ./tools/run-release-action
. It requires gh
,
jq
and hatch
:
./tools/run-release-action --release <major|minor|patch>
Once the tag is release, run the release for base container to the latest release tag.
Note that currently this workflow can only be run by the BentoML team.
modeled after the attrs workflow
If the change is noteworthy, there needs to be a changelog entry so users can learn about it!
To avoid merge conflicts, we use the
Towncrier package to manage our
changelog. towncrier uses independent Markdown files for each pull request –
so called news fragments – instead of one monolithic changelog file. On
release, those news fragments are compiled into
CHANGELOG.md
.
You don't need to install Towncrier yourself, you just have to abide by a few simple rules:
-
For each pull request, add a new file into
changelog.d
with a filename adhering to the<pr#>.(change|deprecation|breaking|feature).md
schema: For example,changelog.d/42.change.md
for a non-breaking change that is proposed in pull request #42. -
As with other docs, please use semantic newlines within news fragments.
-
Wrap symbols like modules, functions, or classes into backticks so they are rendered in a
monospace font
. -
Wrap arguments into asterisks like in docstrings:
Added new argument *an_argument*.
-
If you mention functions or other callables, add parentheses at the end of their names:
openllm.func()
oropenllm.LLMClass.method()
. This makes the changelog a lot more readable. -
Prefer simple past tense or constructions with "now". For example:
- Added
LLM.func()
. LLM.func()
now doesn't do X.Y.Z anymore when passed the foobar argument.
- Added
-
If you want to reference multiple issues, copy the news fragment to another filename. Towncrier will merge all news fragments with identical contents into one entry with multiple links to the respective pull requests.
Example entries:
Added `LLM.func()`. The feature really _is_ awesome.
or:
`openllm.utils.func()` now doesn't X.Y.Z anymore when passed the _foobar_
argument. The bug really _was_ nasty.
hatch run changelog
will render the current changelog to the terminal if you
have any doubts.