@@ -142,7 +142,11 @@ generator_config = ConversationsGeneratorConfig(agent1="You're a shop assistant
142
142
lengths = [4 , 5 ],
143
143
temperatures = [0.1 , 0.2 ],
144
144
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
+
146
150
)
147
151
148
152
conversations_generator = ConversationsGenerator(generator_config)
@@ -238,6 +242,19 @@ Options:
238
242
OpenAI Chat model to use. GPT-4 is only
239
243
supported if provided API key has access to
240
244
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.
241
258
-n, --num-samples INTEGER Number of conversations for each
242
259
configuration.
243
260
-o, --option <TEXT TEXT>... Values for additional options denoted in
0 commit comments