test(enum/chat): add test methods

northes · Jun 6, 2024 · a5ab186 · a5ab186
1 parent 61dc4fb
commit a5ab186
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/enum_chat_completions_test.go b/enum_chat_completions_test.go
@@ -13,11 +13,16 @@ func TestEnumChatCompletions(t *testing.T) {
 	tt.EqualValues(moonshot.RoleSystem, moonshot.RoleSystem.String())
 	tt.EqualValues(moonshot.RoleUser, moonshot.RoleUser.String())
 	tt.EqualValues(moonshot.RoleAssistant, moonshot.RoleAssistant.String())
+	tt.EqualValues(moonshot.RoleTool, moonshot.RoleTool.String())
 
 	tt.EqualValues(moonshot.ModelMoonshotV18K, moonshot.ModelMoonshotV18K.String())
 	tt.EqualValues(moonshot.ModelMoonshotV132K, moonshot.ModelMoonshotV132K.String())
 	tt.EqualValues(moonshot.ModelMoonshotV1128K, moonshot.ModelMoonshotV1128K.String())
 
 	tt.EqualValues(moonshot.FinishReasonStop, moonshot.FinishReasonStop.String())
 	tt.EqualValues(moonshot.FinishReasonLength, moonshot.FinishReasonLength.String())
+
+	tt.EqualValues(moonshot.ChatCompletionsToolTypeFunction, moonshot.ChatCompletionsToolTypeFunction.String())
+
+	tt.EqualValues(moonshot.ChatCompletionsParametersTypeObject, moonshot.ChatCompletionsParametersTypeObject.String())
 }