chore: import upstream snapshot with attribution
TypeScript SDK Compatibility V1.x E2E Tests / Select Node version matrix (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / TypeScript SDK Compatibility V1.x E2E Tests Node ${{matrix.node_version}} (push) Has been cancelled
TypeScript SDK E2E Tests / TypeScript SDK E2E Tests Node ${{matrix.node_version}} (push) Has been cancelled
Opik Optimizer - E2E Tests / build-opik (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / build-opik (push) Has been cancelled
Python SDK E2E Tests / Select Python version matrix (push) Has been cancelled
Python SDK E2E Tests / Python SDK E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Python SDK E2E Tests / build-opik (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / Select Python version matrix (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / Python SDK Compatibility V1.x E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / build-opik (push) Has been cancelled
TypeScript SDK E2E Tests / Select Node version matrix (push) Has been cancelled
TypeScript SDK E2E Tests / build-opik (push) Has been cancelled
Opik Optimizer - E2E Tests / Opik Optimizer E2E Tests Python ${{matrix.python_version}} (push) Has been cancelled
Opik Optimizer - E2E Tests / Opik Optimizer Integration Smoke Tests (push) Has been cancelled
🐙 Code Quality / detect (push) Has been cancelled
🐙 Code Quality / lint (${{ matrix.leg.name }}) (push) Has been cancelled
🐙 Code Quality / summary (push) Has been cancelled
TypeScript SDK Library Integration Tests / Check Secrets (push) Has been cancelled
TypeScript SDK Library Integration Tests / opik-vercel (Vercel AI SDK / eve) (push) Has been cancelled
SDK Library Integration Tests Runner / Check Secrets (push) Has been cancelled
SDK Library Integration Tests Runner / Missed OpenAI API Key Warning (push) Has been cancelled
SDK Library Integration Tests Runner / Build (push) Has been cancelled
SDK Library Integration Tests Runner / openai_tests (push) Has been cancelled
SDK Library Integration Tests Runner / langchain_tests (push) Has been cancelled
SDK Library Integration Tests Runner / langchain_legacy_tests (push) Has been cancelled
SDK Library Integration Tests Runner / llama_index_tests (push) Has been cancelled
SDK Library Integration Tests Runner / anthropic_tests (push) Has been cancelled
SDK Library Integration Tests Runner / mistral_tests (push) Has been cancelled
SDK Library Integration Tests Runner / groq_tests (push) Has been cancelled
SDK Library Integration Tests Runner / aisuite_tests (push) Has been cancelled
SDK Library Integration Tests Runner / haystack_tests (push) Has been cancelled
SDK Library Integration Tests Runner / dspy_tests (push) Has been cancelled
SDK Library Integration Tests Runner / crewai_v0_tests (push) Has been cancelled
SDK Library Integration Tests Runner / crewai_v1_tests (push) Has been cancelled
SDK Library Integration Tests Runner / genai_tests (push) Has been cancelled
SDK Library Integration Tests Runner / adk_tests (push) Has been cancelled
SDK Library Integration Tests Runner / adk_legacy_1_3_0_tests (push) Has been cancelled
SDK Library Integration Tests Runner / evaluation_metrics_tests (push) Has been cancelled
SDK Library Integration Tests Runner / bedrock_tests (push) Has been cancelled
SDK Library Integration Tests Runner / litellm_tests (push) Has been cancelled
SDK Library Integration Tests Runner / harbor_tests (push) Has been cancelled
SDK Library Integration Tests Runner / Slack Notification (push) Has been cancelled
Lint Opik Helm Chart / render-equality (push) Has been cancelled
Opik Optimizer - Unit Tests / Opik Optimizer Unit Tests Python ${{matrix.python_version}} (push) Has been cancelled
Python BE E2E Tests / Python BE E2E (push) Has been cancelled
Python Backend Tests / run-python-backend-tests (push) Has been cancelled
Python SDK Unit Tests / Python SDK Unit Tests ${{matrix.python_version}} (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
SDK E2E Libraries Integration Tests / Check Secrets (push) Has been cancelled
SDK E2E Libraries Integration Tests / Missed OpenAI API Key Warning (push) Has been cancelled
SDK E2E Libraries Integration Tests / build-opik (push) Has been cancelled
SDK E2E Libraries Integration Tests / E2E Lib Integration Python ${{matrix.python_version}} (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-gemini) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-langchain) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-openai) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-otel) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-vercel) (push) Has been cancelled
TypeScript SDK Build & Publish / build-and-publish (push) Has been cancelled
TypeScript SDK Unit Tests / Test on Node ${{ matrix.node-version }} (push) Has been cancelled
Backend Tests / discover-tests (push) Has been cancelled
Backend Tests / ${{ matrix.name }} (push) Has been cancelled
Build and Publish SDK / build-and-publish (push) Has been cancelled
Build Opik Docker Images / set-version (push) Has been cancelled
Build Opik Docker Images / build-backend (push) Has been cancelled
Build Opik Docker Images / build-sandbox-executor-python (push) Has been cancelled
Build Opik Docker Images / build-python-backend (push) Has been cancelled
Build Opik Docker Images / build-frontend (push) Has been cancelled
Build Opik Docker Images / create-git-tag (push) Has been cancelled
ClickHouse Migration Cluster Check / validate-clickhouse-migrations (push) Has been cancelled
Docs - Publish / run (push) Has been cancelled
E2E Tests - Post Merge (v2) / 🧪 E2E v2 Tests (${{ github.event.inputs.tier || 't1' }}) (push) Has been cancelled
E2E Tests - Post Merge (v2) / 📢 Slack Notification (push) Has been cancelled
Frontend Unit Tests / Test on Node 20 (push) Has been cancelled
Guardrails E2E Tests / Select Python version matrix (push) Has been cancelled
Guardrails E2E Tests / Guardrails E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Guardrails E2E Tests / 📢 Slack Notification (push) Has been cancelled
Guardrails Backend Unit Tests / Guardrails Backend Unit Tests (push) Has been cancelled
Guardrails Backend Unit Tests / 📢 Slack Notification (push) Has been cancelled
Lint Opik Helm Chart / lint-helm-chart (Helm v3.21.0) (push) Has been cancelled
Lint Opik Helm Chart / lint-helm-chart (Helm v4.2.0) (push) Has been cancelled
Lint Opik Helm Chart / unittest-helm-chart (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / Select Node version matrix (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / TypeScript SDK Compatibility V1.x E2E Tests Node ${{matrix.node_version}} (push) Has been cancelled
TypeScript SDK E2E Tests / TypeScript SDK E2E Tests Node ${{matrix.node_version}} (push) Has been cancelled
Opik Optimizer - E2E Tests / build-opik (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / build-opik (push) Has been cancelled
Python SDK E2E Tests / Select Python version matrix (push) Has been cancelled
Python SDK E2E Tests / Python SDK E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Python SDK E2E Tests / build-opik (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / Select Python version matrix (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / Python SDK Compatibility V1.x E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / build-opik (push) Has been cancelled
TypeScript SDK E2E Tests / Select Node version matrix (push) Has been cancelled
TypeScript SDK E2E Tests / build-opik (push) Has been cancelled
Opik Optimizer - E2E Tests / Opik Optimizer E2E Tests Python ${{matrix.python_version}} (push) Has been cancelled
Opik Optimizer - E2E Tests / Opik Optimizer Integration Smoke Tests (push) Has been cancelled
🐙 Code Quality / detect (push) Has been cancelled
🐙 Code Quality / lint (${{ matrix.leg.name }}) (push) Has been cancelled
🐙 Code Quality / summary (push) Has been cancelled
TypeScript SDK Library Integration Tests / Check Secrets (push) Has been cancelled
TypeScript SDK Library Integration Tests / opik-vercel (Vercel AI SDK / eve) (push) Has been cancelled
SDK Library Integration Tests Runner / Check Secrets (push) Has been cancelled
SDK Library Integration Tests Runner / Missed OpenAI API Key Warning (push) Has been cancelled
SDK Library Integration Tests Runner / Build (push) Has been cancelled
SDK Library Integration Tests Runner / openai_tests (push) Has been cancelled
SDK Library Integration Tests Runner / langchain_tests (push) Has been cancelled
SDK Library Integration Tests Runner / langchain_legacy_tests (push) Has been cancelled
SDK Library Integration Tests Runner / llama_index_tests (push) Has been cancelled
SDK Library Integration Tests Runner / anthropic_tests (push) Has been cancelled
SDK Library Integration Tests Runner / mistral_tests (push) Has been cancelled
SDK Library Integration Tests Runner / groq_tests (push) Has been cancelled
SDK Library Integration Tests Runner / aisuite_tests (push) Has been cancelled
SDK Library Integration Tests Runner / haystack_tests (push) Has been cancelled
SDK Library Integration Tests Runner / dspy_tests (push) Has been cancelled
SDK Library Integration Tests Runner / crewai_v0_tests (push) Has been cancelled
SDK Library Integration Tests Runner / crewai_v1_tests (push) Has been cancelled
SDK Library Integration Tests Runner / genai_tests (push) Has been cancelled
SDK Library Integration Tests Runner / adk_tests (push) Has been cancelled
SDK Library Integration Tests Runner / adk_legacy_1_3_0_tests (push) Has been cancelled
SDK Library Integration Tests Runner / evaluation_metrics_tests (push) Has been cancelled
SDK Library Integration Tests Runner / bedrock_tests (push) Has been cancelled
SDK Library Integration Tests Runner / litellm_tests (push) Has been cancelled
SDK Library Integration Tests Runner / harbor_tests (push) Has been cancelled
SDK Library Integration Tests Runner / Slack Notification (push) Has been cancelled
Lint Opik Helm Chart / render-equality (push) Has been cancelled
Opik Optimizer - Unit Tests / Opik Optimizer Unit Tests Python ${{matrix.python_version}} (push) Has been cancelled
Python BE E2E Tests / Python BE E2E (push) Has been cancelled
Python Backend Tests / run-python-backend-tests (push) Has been cancelled
Python SDK Unit Tests / Python SDK Unit Tests ${{matrix.python_version}} (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
SDK E2E Libraries Integration Tests / Check Secrets (push) Has been cancelled
SDK E2E Libraries Integration Tests / Missed OpenAI API Key Warning (push) Has been cancelled
SDK E2E Libraries Integration Tests / build-opik (push) Has been cancelled
SDK E2E Libraries Integration Tests / E2E Lib Integration Python ${{matrix.python_version}} (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-gemini) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-langchain) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-openai) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-otel) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-vercel) (push) Has been cancelled
TypeScript SDK Build & Publish / build-and-publish (push) Has been cancelled
TypeScript SDK Unit Tests / Test on Node ${{ matrix.node-version }} (push) Has been cancelled
Backend Tests / discover-tests (push) Has been cancelled
Backend Tests / ${{ matrix.name }} (push) Has been cancelled
Build and Publish SDK / build-and-publish (push) Has been cancelled
Build Opik Docker Images / set-version (push) Has been cancelled
Build Opik Docker Images / build-backend (push) Has been cancelled
Build Opik Docker Images / build-sandbox-executor-python (push) Has been cancelled
Build Opik Docker Images / build-python-backend (push) Has been cancelled
Build Opik Docker Images / build-frontend (push) Has been cancelled
Build Opik Docker Images / create-git-tag (push) Has been cancelled
ClickHouse Migration Cluster Check / validate-clickhouse-migrations (push) Has been cancelled
Docs - Publish / run (push) Has been cancelled
E2E Tests - Post Merge (v2) / 🧪 E2E v2 Tests (${{ github.event.inputs.tier || 't1' }}) (push) Has been cancelled
E2E Tests - Post Merge (v2) / 📢 Slack Notification (push) Has been cancelled
Frontend Unit Tests / Test on Node 20 (push) Has been cancelled
Guardrails E2E Tests / Select Python version matrix (push) Has been cancelled
Guardrails E2E Tests / Guardrails E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Guardrails E2E Tests / 📢 Slack Notification (push) Has been cancelled
Guardrails Backend Unit Tests / Guardrails Backend Unit Tests (push) Has been cancelled
Guardrails Backend Unit Tests / 📢 Slack Notification (push) Has been cancelled
Lint Opik Helm Chart / lint-helm-chart (Helm v3.21.0) (push) Has been cancelled
Lint Opik Helm Chart / lint-helm-chart (Helm v4.2.0) (push) Has been cancelled
Lint Opik Helm Chart / unittest-helm-chart (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,187 @@
|
||||
SimulatedUser
|
||||
=============
|
||||
|
||||
.. currentmodule:: opik.simulation
|
||||
|
||||
.. autoclass:: SimulatedUser
|
||||
:members:
|
||||
:special-members: __init__
|
||||
|
||||
Description
|
||||
-----------
|
||||
|
||||
The ``SimulatedUser`` class generates realistic user responses for multi-turn conversation simulations. It can use either LLM-generated responses or predefined fixed responses, making it flexible for different testing scenarios.
|
||||
|
||||
Key Features
|
||||
------------
|
||||
|
||||
- **LLM-powered responses**: Uses any supported LLM model to generate context-aware user responses
|
||||
- **Fixed responses**: Option to use predefined responses for deterministic testing
|
||||
- **Persona-based behavior**: Simulates different user personalities and behaviors
|
||||
- **Conversation context**: Generates responses based on full conversation history
|
||||
|
||||
Constructor
|
||||
-----------
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
SimulatedUser(
|
||||
persona: str,
|
||||
model: Optional[str] = None,
|
||||
fixed_responses: Optional[List[str]] = None
|
||||
)
|
||||
|
||||
Parameters
|
||||
----------
|
||||
|
||||
**persona** (str)
|
||||
Description of the user's personality and behavior. This is used as a system prompt to guide the LLM's response generation.
|
||||
|
||||
**model** (str, optional)
|
||||
LLM model to use for generating responses. If omitted, defaults to the value of ``OPIK_DEFAULT_LLM`` (or ``openai/gpt-5-nano`` when unset). Supports any model available through Opik's model factory.
|
||||
|
||||
**fixed_responses** (List[str], optional)
|
||||
List of predefined responses to cycle through. If provided, these responses will be used instead of LLM generation.
|
||||
|
||||
Methods
|
||||
-------
|
||||
|
||||
generate_response
|
||||
~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
generate_response(conversation_history: List[Dict[str, str]]) -> str
|
||||
|
||||
Generates a response based on the conversation history.
|
||||
|
||||
**Parameters:**
|
||||
|
||||
- **conversation_history** (List[Dict[str, str]]): List of message dictionaries with 'role' and 'content' keys
|
||||
|
||||
**Returns:**
|
||||
|
||||
- **str**: String response from the simulated user
|
||||
|
||||
**Behavior:**
|
||||
|
||||
- If ``fixed_responses`` are provided, cycles through them in order
|
||||
- Otherwise, uses the LLM to generate context-aware responses based on the persona and conversation history
|
||||
- Automatically limits conversation history to last 10 messages to avoid token limits
|
||||
|
||||
Examples
|
||||
--------
|
||||
|
||||
Basic Usage
|
||||
~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from opik.simulation import SimulatedUser
|
||||
|
||||
# Create a simulated user with a specific persona
|
||||
user_simulator = SimulatedUser(
|
||||
persona="You are a frustrated customer who wants a refund for a broken product",
|
||||
model="openai/gpt-5-nano"
|
||||
)
|
||||
|
||||
# Generate a response based on conversation history
|
||||
conversation = [
|
||||
{"role": "assistant", "content": "Hello, how can I help you today?"},
|
||||
{"role": "user", "content": "My product broke after 2 days, I want a refund."},
|
||||
{"role": "assistant", "content": "I'm sorry to hear that. What happened?"}
|
||||
]
|
||||
|
||||
response = user_simulator.generate_response(conversation)
|
||||
print(response) # Output: "It just stopped working! I've barely used it..."
|
||||
|
||||
Fixed Responses
|
||||
~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
# Use predefined responses for deterministic testing
|
||||
user_simulator = SimulatedUser(
|
||||
persona="Test user",
|
||||
fixed_responses=[
|
||||
"I want a refund",
|
||||
"This is taking too long",
|
||||
"Can I speak to a manager?",
|
||||
"I'm not satisfied with this service"
|
||||
]
|
||||
)
|
||||
|
||||
# Responses will cycle through the fixed list
|
||||
response1 = user_simulator.generate_response([]) # "I want a refund"
|
||||
response2 = user_simulator.generate_response([]) # "This is taking too long"
|
||||
response3 = user_simulator.generate_response([]) # "Can I speak to a manager?"
|
||||
|
||||
Different Personas
|
||||
~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
# Happy customer persona
|
||||
happy_customer = SimulatedUser(
|
||||
persona="You are a satisfied customer who loves the product and wants to buy more",
|
||||
model="openai/gpt-5-nano"
|
||||
)
|
||||
|
||||
# Confused user persona
|
||||
confused_user = SimulatedUser(
|
||||
persona="You are a confused user who needs help understanding how to use the product",
|
||||
model="openai/gpt-5-nano"
|
||||
)
|
||||
|
||||
# Technical user persona
|
||||
technical_user = SimulatedUser(
|
||||
persona="You are a technical user who asks detailed questions about implementation and integration",
|
||||
model="openai/gpt-5-nano"
|
||||
)
|
||||
|
||||
Integration with run_simulation
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from opik.simulation import SimulatedUser, run_simulation
|
||||
from opik import track
|
||||
|
||||
@track
|
||||
def customer_service_agent(user_message: str, *, thread_id: str, **kwargs):
|
||||
# Your agent logic here
|
||||
return {"role": "assistant", "content": "I understand your concern..."}
|
||||
|
||||
# Create multiple user personas for testing
|
||||
personas = [
|
||||
"You are a frustrated customer who wants a refund",
|
||||
"You are a happy customer who wants to buy more products",
|
||||
"You are a confused user who needs help with setup"
|
||||
]
|
||||
|
||||
for i, persona in enumerate(personas):
|
||||
simulator = SimulatedUser(persona=persona)
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=simulator,
|
||||
max_turns=5,
|
||||
project_name="customer_service_evaluation"
|
||||
)
|
||||
print(f"Simulation {i+1} completed: {simulation['thread_id']}")
|
||||
|
||||
Best Practices
|
||||
--------------
|
||||
|
||||
1. **Clear Personas**: Write detailed, specific personas to get consistent behavior
|
||||
2. **Model Selection**: Choose appropriate models based on your needs (faster models for testing, more capable models for realistic simulation)
|
||||
3. **Fixed Responses**: Use fixed responses for deterministic testing scenarios
|
||||
4. **Context Management**: The class automatically handles conversation context, but be aware of token limits
|
||||
5. **Error Handling**: The class includes fallback responses if LLM generation fails
|
||||
|
||||
Notes
|
||||
-----
|
||||
|
||||
- The class uses Opik's model factory for LLM integration, ensuring consistency with other Opik features
|
||||
- Responses are generated as strings, not message dictionaries
|
||||
- The persona is used as a system prompt to guide response generation
|
||||
- Fixed responses cycle through the list in order, starting over when exhausted
|
||||
@@ -0,0 +1,89 @@
|
||||
Simulation
|
||||
==========
|
||||
|
||||
The Opik simulation module provides tools for creating multi-turn conversation simulations between simulated users and your applications. This is particularly useful for evaluating agent behavior over multiple conversation turns.
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
|
||||
SimulatedUser
|
||||
run_simulation
|
||||
|
||||
Overview
|
||||
--------
|
||||
|
||||
Multi-turn simulation allows you to:
|
||||
|
||||
- **Simulate realistic user interactions** with your agent over multiple conversation turns
|
||||
- **Generate context-aware user responses** based on conversation history
|
||||
- **Evaluate agent behavior** across extended conversations
|
||||
- **Test different user personas** and scenarios systematically
|
||||
|
||||
Key Components
|
||||
---------------
|
||||
|
||||
**SimulatedUser**: A class that generates realistic user responses using LLMs or predefined responses.
|
||||
|
||||
**run_simulation**: A function that orchestrates multi-turn conversations between a simulated user and your application.
|
||||
|
||||
Basic Usage
|
||||
-----------
|
||||
|
||||
Here's a simple example of how to use the simulation module:
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from opik.simulation import SimulatedUser, run_simulation
|
||||
from opik import track
|
||||
|
||||
# Create a simulated user
|
||||
user_simulator = SimulatedUser(
|
||||
persona="You are a frustrated customer who wants a refund",
|
||||
model="openai/gpt-5-nano"
|
||||
)
|
||||
|
||||
# Define your agent
|
||||
@track
|
||||
def my_agent(user_message: str, *, thread_id: str, **kwargs):
|
||||
# Your agent logic here
|
||||
return {"role": "assistant", "content": "I can help you with that..."}
|
||||
|
||||
# Run the simulation
|
||||
simulation = run_simulation(
|
||||
app=my_agent,
|
||||
user_simulator=user_simulator,
|
||||
max_turns=5
|
||||
)
|
||||
|
||||
print(f"Thread ID: {simulation['thread_id']}")
|
||||
print(f"Conversation: {simulation['conversation_history']}")
|
||||
|
||||
Integration with Evaluation
|
||||
---------------------------
|
||||
|
||||
Simulations work seamlessly with Opik's evaluation framework:
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from opik.evaluation import evaluate_threads
|
||||
from opik.evaluation.metrics import ConversationThreadMetric
|
||||
|
||||
# Run multiple simulations
|
||||
simulations = []
|
||||
for persona in ["frustrated_user", "happy_customer", "confused_user"]:
|
||||
simulator = SimulatedUser(persona=f"You are a {persona}")
|
||||
simulation = run_simulation(
|
||||
app=my_agent,
|
||||
user_simulator=simulator,
|
||||
max_turns=5
|
||||
)
|
||||
simulations.append(simulation)
|
||||
|
||||
# Evaluate the threads
|
||||
results = evaluate_threads(
|
||||
project_name="my_project",
|
||||
filter_string='tags contains "simulation"',
|
||||
metrics=[ConversationThreadMetric()]
|
||||
)
|
||||
|
||||
For more detailed examples and advanced usage patterns, see the individual component documentation.
|
||||
@@ -0,0 +1,285 @@
|
||||
run_simulation
|
||||
==============
|
||||
|
||||
.. currentmodule:: opik.simulation
|
||||
|
||||
.. autofunction:: run_simulation
|
||||
|
||||
Description
|
||||
-----------
|
||||
|
||||
The ``run_simulation`` function orchestrates multi-turn conversation simulations between a simulated user and your application. It manages the conversation flow, tracks traces, and returns comprehensive results for evaluation.
|
||||
|
||||
Key Features
|
||||
------------
|
||||
|
||||
- **Multi-turn conversations**: Runs conversations for a specified number of turns
|
||||
- **Automatic tracing**: Automatically decorates your app with ``@track`` if not already decorated
|
||||
- **Thread management**: Groups all traces from a simulation under a single thread ID
|
||||
- **Error handling**: Gracefully handles errors and continues simulation
|
||||
- **Flexible configuration**: Supports custom parameters and metadata
|
||||
|
||||
Function Signature
|
||||
------------------
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
run_simulation(
|
||||
app: Callable,
|
||||
user_simulator: SimulatedUser,
|
||||
initial_message: Optional[str] = None,
|
||||
max_turns: int = 5,
|
||||
thread_id: Optional[str] = None,
|
||||
project_name: Optional[str] = None,
|
||||
**app_kwargs: Any
|
||||
) -> Dict[str, Any]
|
||||
|
||||
Parameters
|
||||
----------
|
||||
|
||||
**app** (Callable)
|
||||
Your application function that processes messages. Must have signature:
|
||||
``app(message: str, *, thread_id: str, **kwargs) -> Dict[str, str]``
|
||||
|
||||
The function will be automatically decorated with ``@track`` if not already decorated.
|
||||
|
||||
**user_simulator** (SimulatedUser)
|
||||
Instance of ``SimulatedUser`` that generates user responses.
|
||||
|
||||
**initial_message** (str, optional)
|
||||
Optional initial message from the user. If ``None``, the simulator will generate one.
|
||||
|
||||
**max_turns** (int, optional)
|
||||
Maximum number of conversation turns. Defaults to 5.
|
||||
|
||||
**thread_id** (str, optional)
|
||||
Thread ID for grouping traces. If ``None``, a new ID will be generated.
|
||||
|
||||
**project_name** (str, optional)
|
||||
Project name for trace logging. Included in trace metadata.
|
||||
|
||||
**app_kwargs** (Any)
|
||||
Additional keyword arguments passed to the app function.
|
||||
|
||||
Returns
|
||||
-------
|
||||
|
||||
**Dict[str, Any]**
|
||||
Dictionary containing:
|
||||
|
||||
- **thread_id** (str): The thread ID used for this simulation
|
||||
- **conversation_history** (List[Dict[str, str]]): Complete conversation as message dictionaries
|
||||
- **project_name** (str, optional): Project name if provided
|
||||
|
||||
App Function Requirements
|
||||
-------------------------
|
||||
|
||||
Your app function must follow this signature:
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
def my_app(user_message: str, *, thread_id: str, **kwargs) -> Dict[str, str]:
|
||||
# Process the user message
|
||||
# Manage conversation history internally using thread_id
|
||||
# Return assistant response as message dict
|
||||
return {"role": "assistant", "content": "Your response"}
|
||||
|
||||
**Key Requirements:**
|
||||
|
||||
1. **First parameter**: Must accept the user message as a string
|
||||
2. **thread_id parameter**: Must accept thread_id as a keyword-only argument
|
||||
3. **Return format**: Must return a dictionary with 'role' and 'content' keys
|
||||
4. **History management**: Your app is responsible for managing conversation history internally
|
||||
|
||||
Examples
|
||||
--------
|
||||
|
||||
Basic Usage
|
||||
~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from opik.simulation import SimulatedUser, run_simulation
|
||||
from opik import track
|
||||
|
||||
# Create a simulated user
|
||||
user_simulator = SimulatedUser(
|
||||
persona="You are a customer who wants help with a product",
|
||||
model="openai/gpt-5-nano"
|
||||
)
|
||||
|
||||
# Define your agent with conversation history management
|
||||
agent_history = {}
|
||||
|
||||
@track
|
||||
def customer_service_agent(user_message: str, *, thread_id: str, **kwargs):
|
||||
if thread_id not in agent_history:
|
||||
agent_history[thread_id] = []
|
||||
|
||||
# Add user message to history
|
||||
agent_history[thread_id].append({"role": "user", "content": user_message})
|
||||
|
||||
# Process with full conversation context
|
||||
messages = agent_history[thread_id]
|
||||
|
||||
# Your agent logic here (e.g., call LLM)
|
||||
response = "I can help you with that. What specific issue are you experiencing?"
|
||||
|
||||
# Add assistant response to history
|
||||
agent_history[thread_id].append({"role": "assistant", "content": response})
|
||||
|
||||
return {"role": "assistant", "content": response}
|
||||
|
||||
# Run the simulation
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=user_simulator,
|
||||
max_turns=5,
|
||||
project_name="customer_service_evaluation"
|
||||
)
|
||||
|
||||
print(f"Thread ID: {simulation['thread_id']}")
|
||||
print(f"Conversation length: {len(simulation['conversation_history'])}")
|
||||
|
||||
Custom Initial Message
|
||||
~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
# Start with a specific initial message
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=user_simulator,
|
||||
initial_message="I'm having trouble with my order",
|
||||
max_turns=3
|
||||
)
|
||||
|
||||
Custom Thread ID
|
||||
~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
# Use a custom thread ID for easier tracking
|
||||
custom_thread_id = "simulation_test_001"
|
||||
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=user_simulator,
|
||||
thread_id=custom_thread_id,
|
||||
max_turns=5
|
||||
)
|
||||
|
||||
Multiple Simulations
|
||||
~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
# Run multiple simulations with different personas
|
||||
personas = [
|
||||
"You are a frustrated customer who wants a refund",
|
||||
"You are a happy customer who wants to buy more",
|
||||
"You are a confused user who needs help with setup"
|
||||
]
|
||||
|
||||
simulations = []
|
||||
for i, persona in enumerate(personas):
|
||||
simulator = SimulatedUser(persona=persona)
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=simulator,
|
||||
max_turns=5,
|
||||
project_name="multi_persona_evaluation"
|
||||
)
|
||||
simulations.append(simulation)
|
||||
print(f"Simulation {i+1} completed: {simulation['thread_id']}")
|
||||
|
||||
Integration with Evaluation
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from opik.evaluation import evaluate_threads
|
||||
from opik.evaluation.metrics import ConversationThreadMetric
|
||||
|
||||
# Run simulations
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=user_simulator,
|
||||
max_turns=5,
|
||||
project_name="evaluation_test"
|
||||
)
|
||||
|
||||
# Evaluate the simulation thread
|
||||
results = evaluate_threads(
|
||||
project_name="evaluation_test",
|
||||
filter_string=f'thread_id = "{simulation["thread_id"]}"',
|
||||
metrics=[ConversationThreadMetric()]
|
||||
)
|
||||
|
||||
Advanced Usage with Tags
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
# Add custom tags and metadata to traces
|
||||
simulation = run_simulation(
|
||||
app=customer_service_agent,
|
||||
user_simulator=user_simulator,
|
||||
max_turns=5,
|
||||
project_name="tagged_simulation",
|
||||
simulation_id="test_001", # Custom parameter
|
||||
tags=["simulation", "customer_service"] # Custom parameter
|
||||
)
|
||||
|
||||
# Your app can access these parameters
|
||||
@track
|
||||
def tagged_agent(user_message: str, *, thread_id: str, simulation_id: str = None, tags: List[str] = None, **kwargs):
|
||||
# Use simulation_id and tags for custom logic
|
||||
if simulation_id:
|
||||
print(f"Running simulation: {simulation_id}")
|
||||
|
||||
return {"role": "assistant", "content": "Response"}
|
||||
|
||||
Error Handling
|
||||
~~~~~~~~~~~~~~
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
@track
|
||||
def error_prone_agent(user_message: str, *, thread_id: str, **kwargs):
|
||||
# This might raise an exception
|
||||
if "error" in user_message.lower():
|
||||
raise ValueError("Simulated error")
|
||||
|
||||
return {"role": "assistant", "content": "Normal response"}
|
||||
|
||||
# run_simulation handles errors gracefully
|
||||
simulation = run_simulation(
|
||||
app=error_prone_agent,
|
||||
user_simulator=user_simulator,
|
||||
max_turns=3
|
||||
)
|
||||
|
||||
# Errors are captured in the conversation history
|
||||
for message in simulation['conversation_history']:
|
||||
if "Error processing message" in message.get('content', ''):
|
||||
print(f"Error occurred: {message['content']}")
|
||||
|
||||
Best Practices
|
||||
--------------
|
||||
|
||||
1. **Thread Management**: Always use the provided ``thread_id`` to manage conversation history
|
||||
2. **Error Handling**: Implement proper error handling in your app function
|
||||
3. **Return Format**: Always return message dictionaries with 'role' and 'content' keys
|
||||
4. **History Management**: Keep conversation history in a thread-safe way if running concurrent simulations
|
||||
5. **Resource Management**: Be mindful of token usage with long conversations
|
||||
6. **Testing**: Use fixed responses in SimulatedUser for deterministic testing
|
||||
|
||||
Notes
|
||||
-----
|
||||
|
||||
- The function automatically decorates your app with ``@track`` if not already decorated
|
||||
- All traces from a simulation are grouped under the same thread ID
|
||||
- The function handles errors gracefully and continues the simulation
|
||||
- Conversation history is returned as a list of message dictionaries
|
||||
- Custom parameters passed via ``**app_kwargs`` are forwarded to your app function
|
||||
Reference in New Issue
Block a user