Skip to content

Commit c9c411c

Browse files
authored
upgrade default model to handle bad-foratted outputs to gpt-4o-mini as gpt-3.5-turbo is deprecated (#183)
1 parent 88d043d commit c9c411c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sotopia/generation_utils/generate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ def format_bad_output_for_script(
391391
ill_formed_output: str,
392392
format_instructions: str,
393393
agents: list[str],
394-
model_name: str = "gpt-3.5-turbo",
394+
model_name: str = "gpt-4o-mini",
395395
) -> BaseMessage:
396396
template = """
397397
Given the string that can not be parsed by a parser, reformat it to a string that can be parsed by the parser which uses the following format instructions. Do not add or delete any information.
@@ -425,7 +425,7 @@ def format_bad_output_for_script(
425425
def format_bad_output(
426426
ill_formed_output: BaseMessage,
427427
format_instructions: str,
428-
model_name: str = "gpt-3.5-turbo",
428+
model_name: str = "gpt-4o-mini",
429429
) -> BaseMessage:
430430
template = """
431431
Given the string that can not be parsed by json parser, reformat it to a string that can be parsed by json parser.

0 commit comments

Comments
 (0)