Skip to content

Commit 75b69c3

Browse files
committed
docs: add cookbook for ab test
1 parent 36b4fd6 commit 75b69c3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cookbook/ab_testing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@ def generate_email(user: str) -> Tuple[str, str]:
1919
"""Randomly chooses a prompt to perform an A/B test for generating email. Returns the email and the trace ID.
2020
The latter is used to tie-back the collected feedback from the user."""
2121
if random.random() < 0.5:
22-
trace_insert({'metadata': {'ab_test': 'test_0 variant_0'}})
22+
trace_insert({'metadata': {'ab_test_0': 'variant_0'}})
2323
prompt = f'Generate a long email for {user}'
2424
else:
25-
trace_insert({'metadata': {'ab_test': 'test_0 variant_1'}})
25+
trace_insert({'metadata': {'ab_test_0': 'variant_1'}})
2626
prompt = f'Generate a short email for {user}'
2727

2828
email = client.chat.completions.create(

0 commit comments

Comments
 (0)