(msg, gpt_version)
| 214 | }] |
| 215 | |
| 216 | def api_call(msg, gpt_version): |
| 217 | if "o3-mini" in gpt_version: |
| 218 | completion = client.chat.completions.create( |
| 219 | model=gpt_version, |
| 220 | reasoning_effort="high", |
| 221 | messages=msg |
| 222 | ) |
| 223 | else: |
| 224 | completion = client.chat.completions.create( |
| 225 | model=gpt_version, |
| 226 | messages=msg |
| 227 | ) |
| 228 | |
| 229 | return completion |
| 230 | |
| 231 | responses = [] |
| 232 | trajectories = [] |