From 6ba0eaedfab28ffdeffca9c3eaf09db72441fc14 Mon Sep 17 00:00:00 2001
From: haguro <1076+haguro@users.noreply.github.com>
Date: Fri, 12 Jan 2024 21:39:07 +1100
Subject: [PATCH] Update VoiceSettings model with newly added fields

---
 elevenlabs_test.go     |  2 +-
 models.go              |  2 ++
 request_bodies_test.go | 12 +++++++++---
 3 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/elevenlabs_test.go b/elevenlabs_test.go
index 1d02b5d..3569857 100644
--- a/elevenlabs_test.go
+++ b/elevenlabs_test.go
@@ -465,7 +465,7 @@ func TestEditVoiceSettings(t *testing.T) {
 	})
 	defer server.Close()
 	client := elevenlabs.NewMockClient(context.Background(), server.URL, mockAPIKey, mockTimeout)
-	err := client.EditVoiceSettings("TestVoiceID", elevenlabs.VoiceSettings{Stability: 0.2, SimilarityBoost: 0.7})
+	err := client.EditVoiceSettings("TestVoiceID", elevenlabs.VoiceSettings{Stability: 0.2, SimilarityBoost: 0.7, Style: 0.3, SpeakerBoost: false})
 	if err != nil {
 		t.Errorf("Expected no errors, got error: %q", err)
 	}
diff --git a/models.go b/models.go
index c95f062..6d2efdb 100644
--- a/models.go
+++ b/models.go
@@ -57,6 +57,8 @@ type Voice struct {
 type VoiceSettings struct {
 	SimilarityBoost float32 `json:"similarity_boost"`
 	Stability       float32 `json:"stability"`
+	Style           float32 `json:"style,omitempty"`
+	SpeakerBoost    bool    `json:"use_speaker_boost,omitempty"`
 }
 
 type VoiceSharing struct {
diff --git a/request_bodies_test.go b/request_bodies_test.go
index 0291f14..a48db9f 100644
--- a/request_bodies_test.go
+++ b/request_bodies_test.go
@@ -110,12 +110,16 @@ var testRespBodies = map[string][]byte{
 
 	"TestGetDefaultVoiceSettings": []byte(`{
   "stability": 0.1,
-  "similarity_boost": 0.2
+  "similarity_boost": 0.2,
+  "style": 0,
+  "use_speaker_boost": true
 }`),
 
 	"TestGetVoiceSettings": []byte(`{
   "stability": 0.7,
-  "similarity_boost": 0.9
+  "similarity_boost": 0.9,
+  "style": 0.1,
+  "use_speaker_boost": false
 }`),
 
 	"TestGetVoice": []byte(`{
@@ -173,7 +177,9 @@ var testRespBodies = map[string][]byte{
   ],
   "settings": {
     "stability": 0.3,
-    "similarity_boost": 0.7
+    "similarity_boost": 0.7,
+    "style": 0.4,
+    "use_speaker_boost": false
   },
   "sharing": {
     "status": "string",