@@ -25,6 +25,7 @@ class LlmModel(str, enum.Enum):
25
25
GPT_4_TURBO = "GPT_4_TURBO"
26
26
GPT_4_TURBO_PREVIEW = "GPT_4_TURBO_PREVIEW"
27
27
GPT_4_TURBO_2024_04_09 = "GPT_4_TURBO_2024_04_09"
28
+ GPT_4_0 = "GPT_4_0"
28
29
MISTRAL_7_B_INSTRUCT_V_01 = "MISTRAL_7B_INSTRUCT_V01"
29
30
MIXTRAL_8_X_7_B_INSTRUCT_V_01 = "MIXTRAL_8X7B_INSTRUCT_V01"
30
31
@@ -44,6 +45,7 @@ def visit(
44
45
gpt_4_turbo : typing .Callable [[], T_Result ],
45
46
gpt_4_turbo_preview : typing .Callable [[], T_Result ],
46
47
gpt_4_turbo_2024_04_09 : typing .Callable [[], T_Result ],
48
+ gpt_4_0 : typing .Callable [[], T_Result ],
47
49
mistral_7_b_instruct_v_01 : typing .Callable [[], T_Result ],
48
50
mixtral_8_x_7_b_instruct_v_01 : typing .Callable [[], T_Result ],
49
51
) -> T_Result :
@@ -75,6 +77,8 @@ def visit(
75
77
return gpt_4_turbo_preview ()
76
78
if self is LlmModel .GPT_4_TURBO_2024_04_09 :
77
79
return gpt_4_turbo_2024_04_09 ()
80
+ if self is LlmModel .GPT_4_0 :
81
+ return gpt_4_0 ()
78
82
if self is LlmModel .MISTRAL_7_B_INSTRUCT_V_01 :
79
83
return mistral_7_b_instruct_v_01 ()
80
84
if self is LlmModel .MIXTRAL_8_X_7_B_INSTRUCT_V_01 :
0 commit comments