@@ -11,30 +11,44 @@ class LlmModel(str, enum.Enum):
11
11
An enumeration.
12
12
"""
13
13
14
+ GPT_3_5_TURBO = "GPT_3_5_TURBO"
14
15
GPT_3_5_TURBO_16_K_0613 = "GPT_3_5_TURBO_16K_0613"
15
16
GPT_3_5_TURBO_0613 = "GPT_3_5_TURBO_0613"
16
17
GPT_3_5_TURBO_1106 = "GPT_3_5_TURBO_1106"
17
18
GPT_3_5_TURBO_0125 = "GPT_3_5_TURBO_0125"
19
+ GPT_4 = "GPT_4"
18
20
GPT_4_0613 = "GPT_4_0613"
21
+ GPT_4_32_K = "GPT_4_32K"
19
22
GPT_4_32_K_0613 = "GPT_4_32K_0613"
20
23
GPT_4_1106_PREVIEW = "GPT_4_1106_PREVIEW"
24
+ GPT_4_0125_PREVIEW = "GPT_4_0125_PREVIEW"
25
+ GPT_4_TURBO = "GPT_4_TURBO"
21
26
GPT_4_TURBO_PREVIEW = "GPT_4_TURBO_PREVIEW"
27
+ GPT_4_TURBO_2024_04_09 = "GPT_4_TURBO_2024_04_09"
22
28
MISTRAL_7_B_INSTRUCT_V_01 = "MISTRAL_7B_INSTRUCT_V01"
23
29
MIXTRAL_8_X_7_B_INSTRUCT_V_01 = "MIXTRAL_8X7B_INSTRUCT_V01"
24
30
25
31
def visit (
26
32
self ,
33
+ gpt_3_5_turbo : typing .Callable [[], T_Result ],
27
34
gpt_3_5_turbo_16_k_0613 : typing .Callable [[], T_Result ],
28
35
gpt_3_5_turbo_0613 : typing .Callable [[], T_Result ],
29
36
gpt_3_5_turbo_1106 : typing .Callable [[], T_Result ],
30
37
gpt_3_5_turbo_0125 : typing .Callable [[], T_Result ],
38
+ gpt_4 : typing .Callable [[], T_Result ],
31
39
gpt_4_0613 : typing .Callable [[], T_Result ],
40
+ gpt_4_32_k : typing .Callable [[], T_Result ],
32
41
gpt_4_32_k_0613 : typing .Callable [[], T_Result ],
33
42
gpt_4_1106_preview : typing .Callable [[], T_Result ],
43
+ gpt_4_0125_preview : typing .Callable [[], T_Result ],
44
+ gpt_4_turbo : typing .Callable [[], T_Result ],
34
45
gpt_4_turbo_preview : typing .Callable [[], T_Result ],
46
+ gpt_4_turbo_2024_04_09 : typing .Callable [[], T_Result ],
35
47
mistral_7_b_instruct_v_01 : typing .Callable [[], T_Result ],
36
48
mixtral_8_x_7_b_instruct_v_01 : typing .Callable [[], T_Result ],
37
49
) -> T_Result :
50
+ if self is LlmModel .GPT_3_5_TURBO :
51
+ return gpt_3_5_turbo ()
38
52
if self is LlmModel .GPT_3_5_TURBO_16_K_0613 :
39
53
return gpt_3_5_turbo_16_k_0613 ()
40
54
if self is LlmModel .GPT_3_5_TURBO_0613 :
@@ -43,14 +57,24 @@ def visit(
43
57
return gpt_3_5_turbo_1106 ()
44
58
if self is LlmModel .GPT_3_5_TURBO_0125 :
45
59
return gpt_3_5_turbo_0125 ()
60
+ if self is LlmModel .GPT_4 :
61
+ return gpt_4 ()
46
62
if self is LlmModel .GPT_4_0613 :
47
63
return gpt_4_0613 ()
64
+ if self is LlmModel .GPT_4_32_K :
65
+ return gpt_4_32_k ()
48
66
if self is LlmModel .GPT_4_32_K_0613 :
49
67
return gpt_4_32_k_0613 ()
50
68
if self is LlmModel .GPT_4_1106_PREVIEW :
51
69
return gpt_4_1106_preview ()
70
+ if self is LlmModel .GPT_4_0125_PREVIEW :
71
+ return gpt_4_0125_preview ()
72
+ if self is LlmModel .GPT_4_TURBO :
73
+ return gpt_4_turbo ()
52
74
if self is LlmModel .GPT_4_TURBO_PREVIEW :
53
75
return gpt_4_turbo_preview ()
76
+ if self is LlmModel .GPT_4_TURBO_2024_04_09 :
77
+ return gpt_4_turbo_2024_04_09 ()
54
78
if self is LlmModel .MISTRAL_7_B_INSTRUCT_V_01 :
55
79
return mistral_7_b_instruct_v_01 ()
56
80
if self is LlmModel .MIXTRAL_8_X_7_B_INSTRUCT_V_01 :
0 commit comments