test(client): fix the priority of reading env variables

northes · Apr 2, 2024 · a947fd9 · a947fd9
1 parent 6aaa40a
commit a947fd9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/api_chat_completions_test.go b/api_chat_completions_test.go
@@ -117,6 +117,14 @@ func TestChatStream(t *testing.T) {
  t.Error(err)
  continue
  }
- t.Logf("Role: %s,Content: %s", msg.Role, msg.Content)
+
+ switch msg.Role {
+ case moonshot.RoleSystem:
+ case moonshot.RoleUser:
+ case moonshot.RoleAssistant:
+ default:
+ t.Logf("Role: %s,Content: %s", msg.Role, msg.Content)
+ }
+
  }
 }
diff --git a/client_test.go b/client_test.go
@@ -12,11 +12,14 @@ import (
 )
 
 func NewTestClient() (*moonshot.Client, error) {
- err := godotenv.Load()
- if err != nil {
- panic(err)
- }
  key, ok := os.LookupEnv("MOONSHOT_KEY")
+ if !ok {
+ err := godotenv.Load()
+ if err != nil {
+ return nil, err
+ }
+ key, ok = os.LookupEnv("MOONSHOT_KEY")
+ }
  if !ok {
  return nil, errors.New("missing environment variable: MOONSHOT_KEY")
  }