fix: DIA-1562: false precision in cost reports (#347)

Co-authored-by: fern-api <115122769+fern-api[bot]@users.noreply.github.com> Co-authored-by: Matt Bernstein <[email protected]>
HumanSignal · Oct 30, 2024 · 9163715 · 9163715
1 parent f68b723
commit 9163715
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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,