Integrations

The SDK sits between your agent and your trainer. Everything below is a string or one call; keys stay in your environment and never reach While.

Models the agent runs on

The agent, the situation writer and the simulated person take the same spec or backend object. Connect your agent shows each one in a run.

Provider	How	Key
OpenAI	`wai.OpenAI("gpt-4.1-mini")` or `"openai:<model>"`	`OPENAI_API_KEY`; `OPENAI_BASE_URL` for a compatible server
Anthropic	`wai.Anthropic("claude-haiku-4-5")` or `"anthropic:<model>"`	`ANTHROPIC_API_KEY`
Fireworks	`wai.Fireworks("accounts/fireworks/models/qwen3-4b")` or `"fireworks:<model>"`	`FIREWORKS_API_KEY`
Amazon Bedrock	`wai.models.Bedrock(model_id, region=)` or `"bedrock:<model-id>[@<region>]"`	`AWS_BEARER_TOKEN_BEDROCK`, else AWS credentials
vLLM, SGLang, TGI, LM Studio, any OpenAI-compatible server	`wai.Endpoint(model, url="http://localhost:8000/v1")` or `"vllm:<model>@<url>"`	`VLLM_API_KEY` when the server wants one
Ollama	`wai.Ollama("llama3")` or `"ollama:<model>"`	none
The model While hosts	`wai.Hosted()`, the default when nothing is configured	`whileai login`
Your own code	`my_agent(message) -> {"steps": [...], "final_text": "..."}`	yours

Judge	How	Page
Any model above	`wai.Judge(rubric=..., model=wai.OpenAI("gpt-4.1"))`	Evals
TypeSafe Jev	`"typesafe:<model>"` as the judge spec: one probability per verdict	Overview
A verifier	`wai.verify.MathEqual()`, `wai.verify.CodeExec(tests=...)`, or your function	Verifiers

Rows leave the SDK in the shape each trainer reads, with the tool-call round trip checked for that shape.

Trainer	How	Page
TRL (`SFTTrainer`, `DPOTrainer`, `GRPOTrainer`)	`select(mode="sft").export("train.jsonl", format="trl")`; `wai.HackMonitor` inside the loop	SFT, DPO, GRPO
Fireworks managed training and serving	`export(..., format="fireworks")`, then `firectl sftj create` and a deployment	Train on Fireworks
prime-rl and verifiers on Prime Intellect	`wai.prime_rl_config(...)` writes the run; `export_environment` writes the environment	prime-rl, Prime Intellect RL
Modal	the training recipes run there on your account, one A10G	SFT, Text-to-SQL
While hosted GPUs	`wai.train(...)`, then `wai.serve(...)` gives back an OpenAI-compatible endpoint	Hosted loop

Source or destination	How	Page
OpenTelemetry traces	`wai.simulations.rows_from_otel(batch)`, then `simulate(..., traces=rows)`	Connect your agent
Your own traces as JSONL	`simulate(..., traces="traces.jsonl")`	Connect your agent
Hugging Face Hub	`export(..., push_to="me/my-set")` for datasets, `wai.hub.push("out/adapter", "me/my-lora")` for adapters	Hugging Face
The While platform	`push`, `train`, `serve`, and the experiments page that tracks every run	Platform

Something you use is not here? Open an issue with the provider and the call you expected.