river-client
Python client for the River ML training API — sampling, LoRA fine-tuning, and reinforcement learning against River-hosted models.
Installation
pip install river-client
Requires Python 3.12+.
Quick start
import river_client as river
client = river.Client(api_key="your-key", endpoint="api.river.ai")
# Stateless sampling from a base model
samples = client.sample(
"What is 2+2?",
base_model="Qwen/Qwen3.6-35B-A3B-FP8",
max_tokens=50,
)
print(samples[0].text)
# Training with a session
with client.session() as session:
model = session.create_model(
base_model="Qwen/Qwen3.6-35B-A3B-FP8",
lora=river.LoraConfig(rank=16),
)
# Forward + backward, then an optimizer step
result = model.forward_backward(data, loss_fn="cross_entropy")
model.optim_step(lr=1e-4)
# Sample from the current weights
sample_groups = model.sample("Continue:", max_tokens=100)
AI agent skill
The package bundles an agent skill — a SKILL.md that teaches AI coding
agents (Claude Code and compatible tools) the current training API:
train_step semantics, data formats, RL/SFT/distillation loop patterns,
and fault-tolerant auto-recovery. Because it ships inside the wheel, the
skill always matches the installed client version.
Install it into your agent's skills directory:
python -m river_client.skill --install
This copies the skill into ~/.claude/skills/; pass --dest for a
different location (e.g. a project's .claude/skills/). Run without
--install to print the bundled skill's path instead.
License
Apache-2.0. See LICENSE.
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file river_client-0.6.1.tar.gz.
File metadata
- Download URL: river_client-0.6.1.tar.gz
- Upload date:
- Size: 99.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
uv/0.12.0 {"installer":{"name":"uv","version":"0.12.0","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
98d2a77b097dfaa201d7cc7e8058b060b72e8fe577ff237c251dd3a89e886449
|
|
| MD5 |
6300f3aa84602efe9301b7b965c732a1
|
|
| BLAKE2b-256 |
3c84ed74d07870deaad8d4db100ac64b25432bf7e687c8e2e3134950792fe778
|
File details
Details for the file river_client-0.6.1-py3-none-any.whl.
File metadata
- Download URL: river_client-0.6.1-py3-none-any.whl
- Upload date:
- Size: 113.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
uv/0.12.0 {"installer":{"name":"uv","version":"0.12.0","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
32b4e2057954bc160dabd095f0191c231a427ac74a9d95fa959a3172ae2b8ab6
|
|
| MD5 |
fa86dbfae851805e1c5bd26c60594e1e
|
|
| BLAKE2b-256 |
eab1ef476c439b3363680cd66eb32dc578a7723278c0622508628f17f7c28515
|
