Merge pull request #25 from northes/feature/auto_model

feature: add auto model enum
northes · Sep 4, 2024 · 938f70a · 938f70a
2 parents 98d6c6a + 9afd7ba
commit 938f70a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/chat_completions_builder.go b/chat_completions_builder.go
@@ -50,7 +50,7 @@ func (c *chatCompletionsBuilder) preCheck() {
 		c.req.Messages = make([]*ChatCompletionsMessage, 0)
 	}
 	if c.req.Model == "" {
-		c.req.Model = ModelMoonshotV18K
+		c.req.Model = ModelMoonshotV1Auto
 	}
 }
 

diff --git a/enum_chat_completions.go b/enum_chat_completions.go
@@ -17,6 +17,7 @@ func (c ChatCompletionsMessageRole) String() string {
 type ChatCompletionsModelID string
 
 const (
+	ModelMoonshotV1Auto ChatCompletionsModelID = "moonshot-v1-auto"
 	ModelMoonshotV18K   ChatCompletionsModelID = "moonshot-v1-8k"
 	ModelMoonshotV132K  ChatCompletionsModelID = "moonshot-v1-32k"
 	ModelMoonshotV1128K ChatCompletionsModelID = "moonshot-v1-128k"

diff --git a/enum_chat_completions_test.go b/enum_chat_completions_test.go
@@ -19,6 +19,7 @@ func TestEnumChatCompletions(t *testing.T) {
 	tt.EqualValues(moonshot.ModelMoonshotV18K, moonshot.ModelMoonshotV18K.String())
 	tt.EqualValues(moonshot.ModelMoonshotV132K, moonshot.ModelMoonshotV132K.String())
 	tt.EqualValues(moonshot.ModelMoonshotV1128K, moonshot.ModelMoonshotV1128K.String())
+	tt.EqualValues(moonshot.ModelMoonshotV1Auto, moonshot.ModelMoonshotV1Auto.String())
 
 	tt.EqualValues(moonshot.FinishReasonStop, moonshot.FinishReasonStop.String())
 	tt.EqualValues(moonshot.FinishReasonLength, moonshot.FinishReasonLength.String())