fix: DIA-1402: Add ?async=true/false parameter to /refine (#335)

Co-authored-by: fern-api <115122769+fern-api[bot]@users.noreply.github.com>

.mock/definition/prompts/versions.yml

@@ -219,6 +219,10 @@ service:
       display-name: Refine a prompt version
       request:
         name: VersionsRefinePromptRequest
+        query-parameters:
+          async:
+            type: optional<boolean>
+            docs: Run the refinement job asynchronously
         body:
           properties:
             teacher_model_provider_connection_id:

reference.md

@@ -15320,6 +15320,14 @@ client.prompts.versions.refine_prompt(
 <dl>
 <dd>
 
+**async_:** `typing.Optional[bool]` — Run the refinement job asynchronously
+
+</dd>
+</dl>
+
+<dl>
+<dd>
+
 **teacher_model_provider_connection_id:** `typing.Optional[int]` — Model Provider Connection ID to use to refine the prompt
 
 </dd>

src/label_studio_sdk/prompts/versions/client.py

@@ -398,6 +398,7 @@ def refine_prompt(
         prompt_id: int,
         version_id: int,
         *,
+        async_: typing.Optional[bool] = None,
         teacher_model_provider_connection_id: typing.Optional[int] = OMIT,
         teacher_model_name: typing.Optional[str] = OMIT,
         project_id: typing.Optional[int] = OMIT,
@@ -414,6 +415,9 @@ def refine_prompt(
         version_id : int
             Base Prompt Version ID
 
+        async_ : typing.Optional[bool]
+            Run the refinement job asynchronously
+
         teacher_model_provider_connection_id : typing.Optional[int]
             Model Provider Connection ID to use to refine the prompt
 
@@ -446,6 +450,7 @@ def refine_prompt(
         _response = self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}/refine",
             method="POST",
+            params={"async": async_},
             json={
                 "teacher_model_provider_connection_id": teacher_model_provider_connection_id,
                 "teacher_model_name": teacher_model_name,
@@ -846,6 +851,7 @@ async def refine_prompt(
         prompt_id: int,
         version_id: int,
         *,
+        async_: typing.Optional[bool] = None,
         teacher_model_provider_connection_id: typing.Optional[int] = OMIT,
         teacher_model_name: typing.Optional[str] = OMIT,
         project_id: typing.Optional[int] = OMIT,
@@ -862,6 +868,9 @@ async def refine_prompt(
         version_id : int
             Base Prompt Version ID
 
+        async_ : typing.Optional[bool]
+            Run the refinement job asynchronously
+
         teacher_model_provider_connection_id : typing.Optional[int]
             Model Provider Connection ID to use to refine the prompt
 
@@ -894,6 +903,7 @@ async def refine_prompt(
         _response = await self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}/refine",
             method="POST",
+            params={"async": async_},
             json={
                 "teacher_model_provider_connection_id": teacher_model_provider_connection_id,
                 "teacher_model_name": teacher_model_name,