Skip to content

Commit 1997307

Browse files
authored
🔥 remove unused models (#446)
1 parent c224b9a commit 1997307

File tree

3 files changed

+3
-18
lines changed

3 files changed

+3
-18
lines changed

plugins/codemodder-plugin-llm/src/main/java/io/codemodder/plugins/llm/StandardModel.java

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,18 +5,6 @@
55

66
/** Well-known GPT models used in Codemod development. */
77
public enum StandardModel implements Model {
8-
GPT_3_5_TURBO_0125("gpt-3.5-turbo-0125", 16_385) {
9-
@Override
10-
public int tokens(final List<String> messages) {
11-
return Tokens.countTokens(messages, 3, EncodingType.CL100K_BASE);
12-
}
13-
},
14-
GPT_4_0613("gpt-4-0613", 8_192) {
15-
@Override
16-
public int tokens(final List<String> messages) {
17-
return Tokens.countTokens(messages, 3, EncodingType.CL100K_BASE);
18-
}
19-
},
208
GPT_4_TURBO_2024_04_09("gpt-4-turbo-2024-04-09", 128_000) {
219
@Override
2210
public int tokens(final List<String> messages) {

plugins/codemodder-plugin-llm/src/test/java/io/codemodder/plugins/llm/EnvironmentBasedModelMapperTest.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,6 @@ final class EnvironmentBasedModelMapperTest {
2222
void before() {
2323
final var environment =
2424
Map.of(
25-
"CODEMODDER_AZURE_OPENAI_GPT_3_5_TURBO_0125_DEPLOYMENT",
26-
"my-gpt-3.5-turbo",
2725
"CODEMODDER_AZURE_OPENAI_GPT_4_0613_DEPLOYMENT",
2826
"my-gpt-4",
2927
"CODEMODDER_AZURE_OPENAI_GPT_4_TURBO_2024_04_09_DEPLOYMENT",
@@ -36,8 +34,8 @@ void before() {
3634
/** Spot checks one of the standard models to make sure the mapping works as expected */
3735
@Test
3836
void it_maps_model_name_to_deployment() {
39-
final var name = mapper.getModelName(StandardModel.GPT_3_5_TURBO_0125);
40-
assertThat(name).isEqualTo("my-gpt-3.5-turbo");
37+
final var name = mapper.getModelName(StandardModel.GPT_4O_2024_05_13);
38+
assertThat(name).isEqualTo("my-gpt-4o");
4139
}
4240

4341
/**

plugins/codemodder-plugin-llm/src/test/java/io/codemodder/plugins/llm/StandardModelTest.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@
77
final class StandardModelTest {
88
@Test
99
void testId() {
10-
assertEquals("gpt-3.5-turbo-0125", StandardModel.GPT_3_5_TURBO_0125.id());
11-
assertEquals("gpt-4-0613", StandardModel.GPT_4_0613.id());
10+
assertEquals("gpt-4-turbo-2024-04-09", StandardModel.GPT_4_TURBO_2024_04_09.id());
1211
assertEquals("gpt-4o-2024-05-13", StandardModel.GPT_4O_2024_05_13.id());
1312
}
1413
}

0 commit comments

Comments
 (0)