Generate realistic, high-fidelity, multi-turn user interactions conditioned on user intents and user personas (traits and attributes). Simulations can be used to evaluate agents, to create training data for long horizon RL training, or to prototype product flows.

TraitBasis is a method for highly-controllable generations that does not suffer from the limitations of prompt-based approaches like the user persona fading with number of turns or the user forgetting the intent in a long context. We will be releasing our paper shortly.

Demo notebooks can be found in the examples/ directory. Contact [email protected] for an API key to generate simulations using TraitBasis.

• Generate multi-turn customer interactions directly from the example notebooks.
• Tune personas, intents, and sampling parameters through the JSON configs in examples/**/configs/.
• Call Together or any OpenAI-compatible endpoint by supplying your API keys.
• (Together example) Upload simulation outputs to Together and poll evaluation jobs entirely within the notebook flow.
• (Together example) Review persona summaries and scored transcripts inline to decide next changes.

• examples/quick_start/: Minimal notebook to generate TraitBasis-driven rollouts.
• examples/rl/: Rollouts tailored for long-horizon RL training workflows using examples similar to Tau Bench
• examples/simulations_together_evals/: Together model as the agent + Together Evals judging end-to-end.

Install uv if you don't already have it, then prepare the environment:

curl -LsSf https://astral.sh/uv/install.sh | sh
source .venv/bin/activate
uv venv --python 3.12 --seed
uv pip install --upgrade pip
uv pip install jupyterlab ipykernel together collinear nest_asyncio jinja2 --no-cache
uv pip install "openai>=1.13.3" "mistralai>=0.4.0" "anthropic>=0.26.1" "google-generativeai>=0.5.4" "tenacity>=8.3.0" "termcolor>=2.4.0" "numpy>=1.26.4" "litellm==1.41.0"
uv pip install tau-trait

If you plan to call Together endpoints, export your API key in the active shell:

export TOGETHER_API_KEY="YOUR_TOGETHER_API_KEY"

Start Jupyter from the project root:

uv run --with jupyter jupyter lab

Select the trait-basis kernel in each notebook and run the grouped import cell before generating simulations.