From a5ab1868305bcfdb5e932055a2c0dd30837e7dfb Mon Sep 17 00:00:00 2001 From: northes Date: Thu, 6 Jun 2024 21:14:41 +0800 Subject: [PATCH] test(enum/chat): add test methods --- enum_chat_completions_test.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/enum_chat_completions_test.go b/enum_chat_completions_test.go index 5404385..b31232f 100644 --- a/enum_chat_completions_test.go +++ b/enum_chat_completions_test.go @@ -13,6 +13,7 @@ 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()) @@ -20,4 +21,8 @@ func TestEnumChatCompletions(t *testing.T) { 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()) }