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())