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/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,