Skip to content

Commit 55796b0

Browse files
committed
chore: update readme for agents per model
1 parent 1c371b8 commit 55796b0

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

README.md

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,11 @@ generator_config = ConversationsGeneratorConfig(agent1="You're a shop assistant
142142
lengths=[4, 5],
143143
temperatures=[0.1, 0.2],
144144
options=[("n", "2"), ("n", "3")],
145-
model="gpt-3.5-turbo" # or model="gpt-4"
145+
model="gpt-3.5-turbo", # or model="gpt-4"
146+
# Specific models per agent can be set. If one of the parameters is set, the other must also be provided, otherwise both agents use the "model" value.
147+
model_agent_one="gpt-3.5-turbo",
148+
model_agent_two="gpt-4"
149+
146150
)
147151

148152
conversations_generator = ConversationsGenerator(generator_config)
@@ -238,6 +242,19 @@ Options:
238242
OpenAI Chat model to use. GPT-4 is only
239243
supported if provided API key has access to
240244
GPT4. Defaults to GPT-3.5-Turbo.
245+
-m1, --model-agent1 [gpt-3.5-turbo|gpt-4]
246+
OpenAI Chat model to use for agent1. GPT-4
247+
is only supported if given API key has
248+
access to GPT4. Defaults to GPT-3.5-Turbo.
249+
If set, --model-agent2 must also be
250+
provided, otherwise --model value will be
251+
used.
252+
-m2, --model-agent2 [gpt-3.5-turbo|gpt-4]
253+
OpenAI Chat model to use for agent2. GPT-4
254+
is only supported if given API key has
255+
access to GPT4. Defaults to GPT-3.5-Turbo If
256+
set, --model-agent1 must also be provided,
257+
otherwise --model value will be used.
241258
-n, --num-samples INTEGER Number of conversations for each
242259
configuration.
243260
-o, --option <TEXT TEXT>... Values for additional options denoted in

0 commit comments

Comments
 (0)