diff --git a/.mock/definition/__package__.yml b/.mock/definition/__package__.yml
index b3c7268..dfe652e 100644
--- a/.mock/definition/__package__.yml
+++ b/.mock/definition/__package__.yml
@@ -2398,7 +2398,7 @@ types:
         docs: Status of the refinement job
         default: Pending
       total_cost:
-        type: optional<float>
+        type: optional<string>
         docs: Total cost of the refinement job (in USD)
       previous_version:
         type: optional<PromptVersion>
diff --git a/.mock/definition/prompts/versions.yml b/.mock/definition/prompts/versions.yml
index 3846665..f3e2d86 100644
--- a/.mock/definition/prompts/versions.yml
+++ b/.mock/definition/prompts/versions.yml
@@ -248,7 +248,7 @@ service:
               prompt: prompt
               refinement_job_id: refinement_job_id
               refinement_status: Pending
-              total_cost: 1.1
+              total_cost: total_cost
               previous_version:
                 title: title
                 parent_model: 1
@@ -309,7 +309,7 @@ service:
               prompt: prompt
               refinement_job_id: refinement_job_id
               refinement_status: Pending
-              total_cost: 1.1
+              total_cost: total_cost
               previous_version:
                 title: title
                 parent_model: 1
diff --git a/poetry.lock b/poetry.lock
index 516e8e4..09368ae 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -278,13 +278,13 @@ files = [
 
 [[package]]
 name = "datamodel-code-generator"
-version = "0.26.1"
+version = "0.26.2"
 description = "Datamodel Code Generator"
 optional = false
 python-versions = "<4.0,>=3.8"
 files = [
-    {file = "datamodel_code_generator-0.26.1-py3-none-any.whl", hash = "sha256:bbe8a6cc0b9cfdbfd294e336e02b4c50b481ffc3b3c608b5578b6d7aa02cc8ae"},
-    {file = "datamodel_code_generator-0.26.1.tar.gz", hash = "sha256:3b7b49c4230fa197ca28847e1e8996cd664638a7e91796c826a61c60d4ccd8a2"},
+    {file = "datamodel_code_generator-0.26.2-py3-none-any.whl", hash = "sha256:f62576a27c9083f2b22cf8c97ed79a394155f131db3e3bf55cd72893f48c5d80"},
+    {file = "datamodel_code_generator-0.26.2.tar.gz", hash = "sha256:03c153434d5a308e31fb4528c0199015054570642ccda8cd2f2cb3cc2c497622"},
 ]
 
 [package.dependencies]
@@ -1808,13 +1808,13 @@ files = [
 
 [[package]]
 name = "tqdm"
-version = "4.66.5"
+version = "4.66.6"
 description = "Fast, Extensible Progress Meter"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "tqdm-4.66.5-py3-none-any.whl", hash = "sha256:90279a3770753eafc9194a0364852159802111925aa30eb3f9d85b0e805ac7cd"},
-    {file = "tqdm-4.66.5.tar.gz", hash = "sha256:e1020aef2e5096702d8a025ac7d16b1577279c9d63f8375b63083e9a5f0fcbad"},
+    {file = "tqdm-4.66.6-py3-none-any.whl", hash = "sha256:223e8b5359c2efc4b30555531f09e9f2f3589bcd7fdd389271191031b49b7a63"},
+    {file = "tqdm-4.66.6.tar.gz", hash = "sha256:4bdd694238bef1485ce839d67967ab50af8f9272aab687c0d7702a01da0be090"},
 ]
 
 [package.dependencies]
diff --git a/src/label_studio_sdk/types/refined_prompt_response.py b/src/label_studio_sdk/types/refined_prompt_response.py
index 54c28ab..5c53a94 100644
--- a/src/label_studio_sdk/types/refined_prompt_response.py
+++ b/src/label_studio_sdk/types/refined_prompt_response.py
@@ -35,7 +35,7 @@ class RefinedPromptResponse(pydantic_v1.BaseModel):
     Status of the refinement job
     """
 
-    total_cost: typing.Optional[float] = pydantic_v1.Field(default=None)
+    total_cost: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     Total cost of the refinement job (in USD)
     """
diff --git a/tests/prompts/test_versions.py b/tests/prompts/test_versions.py
index 021feb4..1af9e4d 100644
--- a/tests/prompts/test_versions.py
+++ b/tests/prompts/test_versions.py
@@ -182,7 +182,7 @@ async def test_get_refined_prompt(client: LabelStudio, async_client: AsyncLabelS
         "prompt": "prompt",
         "refinement_job_id": "refinement_job_id",
         "refinement_status": "Pending",
-        "total_cost": 1.1,
+        "total_cost": "total_cost",
         "previous_version": {
             "title": "title",
             "parent_model": 1,
@@ -234,7 +234,7 @@ async def test_refine_prompt(client: LabelStudio, async_client: AsyncLabelStudio
         "prompt": "prompt",
         "refinement_job_id": "refinement_job_id",
         "refinement_status": "Pending",
-        "total_cost": 1.1,
+        "total_cost": "total_cost",
         "previous_version": {
             "title": "title",
             "parent_model": 1,