label-studio-sdk 1.0.20__py3-none-any.whl → 2.0.1__py3-none-any.whl

label_studio_sdk/prompts/versions/client.py

@@ -3,16 +3,12 @@
 import typing
 from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.prompt_version import PromptVersion
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
-from ...types.prompt_version_provider import PromptVersionProvider
-from ...types.prompt_version_created_by import PromptVersionCreatedBy
-import datetime as dt
-from ...types.prompt_version_organization import PromptVersionOrganization
-from ...core.serialization import convert_and_respect_annotation_metadata
+from ...types.third_party_model_version import ThirdPartyModelVersion
+from ...core.unchecked_base_model import construct_type
+from ...types.provider_enum import ProviderEnum
 from ...types.inference_run_cost_estimate import InferenceRunCostEstimate
 from ...types.refined_prompt_response import RefinedPromptResponse
 from ...core.client_wrapper import AsyncClientWrapper
@@ -25,21 +21,72 @@ class VersionsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[PromptVersion]:
+    def get_default_version_name(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
-        Get a list of prompt versions.
+        Get default prompt version name
 
         Parameters
         ----------
         id : int
-            Prompt ID
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[PromptVersion]
+        None
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.prompts.versions.get_default_version_name(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/prompts/{jsonable_encoder(id)}/get-default-version-name",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def list(
+        self,
+        prompt_id_: int,
+        *,
+        prompt_id: int,
+        ordering: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[ThirdPartyModelVersion]:
+        """
+        List all versions of a prompt.
+
+        Parameters
+        ----------
+        prompt_id_ : int
+
+        prompt_id : int
+            A unique integer value identifying the model ID to list versions for.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[ThirdPartyModelVersion]
 
 
         Examples
@@ -50,20 +97,25 @@ class VersionsClient:
             api_key="YOUR_API_KEY",
         )
         client.prompts.versions.list(
-            id=1,
+            prompt_id_=1,
+            prompt_id=1,
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions",
+            f"api/prompts/{jsonable_encoder(prompt_id_)}/versions",
             method="GET",
+            params={
+                "ordering": ordering,
+                "prompt_id": prompt_id,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.List[PromptVersion],
-                    parse_obj_as(
-                        type_=typing.List[PromptVersion],  # type: ignore
+                    typing.List[ThirdPartyModelVersion],
+                    construct_type(
+                        type_=typing.List[ThirdPartyModelVersion],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -74,54 +126,57 @@ class VersionsClient:
 
     def create(
         self,
-        id: int,
+        prompt_id: int,
         *,
-        title: typing.Optional[str] = OMIT,
+        title: str,
+        prompt: str,
+        provider_model_id: str,
         parent_model: typing.Optional[int] = OMIT,
+        provider: typing.Optional[ProviderEnum] = OMIT,
         model_provider_connection: typing.Optional[int] = OMIT,
-        prompt: typing.Optional[str] = OMIT,
-        provider: typing.Optional[PromptVersionProvider] = OMIT,
-        provider_model_id: typing.Optional[str] = OMIT,
-        created_by: typing.Optional[PromptVersionCreatedBy] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        updated_at: typing.Optional[dt.datetime] = OMIT,
-        organization: typing.Optional[PromptVersionOrganization] = OMIT,
+        organization: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> PromptVersion:
+    ) -> ThirdPartyModelVersion:
         """
         Create a new version of a prompt.
 
         Parameters
         ----------
-        id : int
-            Prompt ID
-
-        title : typing.Optional[str]
-
-        parent_model : typing.Optional[int]
+        prompt_id : int
 
-        model_provider_connection : typing.Optional[int]
+        title : str
+            Model name
 
-        prompt : typing.Optional[str]
+        prompt : str
+            Prompt to execute
 
-        provider : typing.Optional[PromptVersionProvider]
+        provider_model_id : str
+            The model ID to use within the given provider, e.g. gpt-3.5
 
-        provider_model_id : typing.Optional[str]
+        parent_model : typing.Optional[int]
+            Parent model interface ID
 
-        created_by : typing.Optional[PromptVersionCreatedBy]
+        provider : typing.Optional[ProviderEnum]
+            The model provider to use e.g. OpenAI
 
-        created_at : typing.Optional[dt.datetime]
+            * `OpenAI` - OpenAI
+            * `AzureOpenAI` - AzureOpenAI
+            * `AzureAIFoundry` - AzureAIFoundry
+            * `VertexAI` - VertexAI
+            * `Gemini` - Gemini
+            * `Anthropic` - Anthropic
+            * `Custom` - Custom
 
-        updated_at : typing.Optional[dt.datetime]
+        model_provider_connection : typing.Optional[int]
 
-        organization : typing.Optional[PromptVersionOrganization]
+        organization : typing.Optional[int]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        PromptVersion
+        ThirdPartyModelVersion
 
 
         Examples
@@ -132,27 +187,23 @@ class VersionsClient:
             api_key="YOUR_API_KEY",
         )
         client.prompts.versions.create(
-            id=1,
+            prompt_id=1,
+            title="title",
+            prompt="prompt",
+            provider_model_id="provider_model_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions",
             method="POST",
             json={
-                "title": title,
                 "parent_model": parent_model,
-                "model_provider_connection": model_provider_connection,
+                "title": title,
                 "prompt": prompt,
                 "provider": provider,
                 "provider_model_id": provider_model_id,
-                "created_by": convert_and_respect_annotation_metadata(
-                    object_=created_by, annotation=PromptVersionCreatedBy, direction="write"
-                ),
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "organization": convert_and_respect_annotation_metadata(
-                    object_=organization, annotation=PromptVersionOrganization, direction="write"
-                ),
+                "model_provider_connection": model_provider_connection,
+                "organization": organization,
             },
             request_options=request_options,
             omit=OMIT,
@@ -160,9 +211,9 @@ class VersionsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    PromptVersion,
-                    parse_obj_as(
-                        type_=PromptVersion,  # type: ignore
+                    ThirdPartyModelVersion,
+                    construct_type(
+                        type_=ThirdPartyModelVersion,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -172,25 +223,23 @@ class VersionsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def get(
-        self, id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> PromptVersion:
+        self, prompt_id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> ThirdPartyModelVersion:
         """
-        Get a prompt version by ID.
+        Retrieve a specific prompt of a model.
 
         Parameters
         ----------
-        id : int
-            Prompt ID
+        prompt_id : int
 
         version_id : int
-            Prompt Version ID
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        PromptVersion
+        ThirdPartyModelVersion
 
 
         Examples
@@ -201,21 +250,21 @@ class VersionsClient:
             api_key="YOUR_API_KEY",
         )
         client.prompts.versions.get(
-            id=1,
+            prompt_id=1,
             version_id=1,
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    PromptVersion,
-                    parse_obj_as(
-                        type_=PromptVersion,  # type: ignore
+                    ThirdPartyModelVersion,
+                    construct_type(
+                        type_=ThirdPartyModelVersion,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -224,17 +273,17 @@ class VersionsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def delete(self, id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+    def delete(
+        self, prompt_id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
-        Delete a prompt version by ID.
+        Delete a prompt version by ID
 
         Parameters
         ----------
-        id : int
-            Prompt ID
+        prompt_id : int
 
         version_id : int
-            Prompt Version ID
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -251,12 +300,12 @@ class VersionsClient:
             api_key="YOUR_API_KEY",
         )
         client.prompts.versions.delete(
-            id=1,
+            prompt_id=1,
             version_id=1,
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}",
             method="DELETE",
             request_options=request_options,
         )
@@ -270,58 +319,60 @@ class VersionsClient:
 
     def update(
         self,
-        id: int,
+        prompt_id: int,
         version_id: int,
         *,
-        title: typing.Optional[str] = OMIT,
         parent_model: typing.Optional[int] = OMIT,
-        model_provider_connection: typing.Optional[int] = OMIT,
+        title: typing.Optional[str] = OMIT,
         prompt: typing.Optional[str] = OMIT,
-        provider: typing.Optional[PromptVersionProvider] = OMIT,
+        provider: typing.Optional[ProviderEnum] = OMIT,
         provider_model_id: typing.Optional[str] = OMIT,
-        created_by: typing.Optional[PromptVersionCreatedBy] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        updated_at: typing.Optional[dt.datetime] = OMIT,
-        organization: typing.Optional[PromptVersionOrganization] = OMIT,
+        model_provider_connection: typing.Optional[int] = OMIT,
+        organization: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> PromptVersion:
+    ) -> ThirdPartyModelVersion:
         """
-        Update a prompt version by ID.
+        Update a specific prompt version by ID.
 
         Parameters
         ----------
-        id : int
-            Prompt ID
+        prompt_id : int
 
         version_id : int
-            Prompt Version ID
-
-        title : typing.Optional[str]
 
         parent_model : typing.Optional[int]
+            Parent model interface ID
 
-        model_provider_connection : typing.Optional[int]
+        title : typing.Optional[str]
+            Model name
 
         prompt : typing.Optional[str]
+            Prompt to execute
 
-        provider : typing.Optional[PromptVersionProvider]
-
-        provider_model_id : typing.Optional[str]
+        provider : typing.Optional[ProviderEnum]
+            The model provider to use e.g. OpenAI
 
-        created_by : typing.Optional[PromptVersionCreatedBy]
+            * `OpenAI` - OpenAI
+            * `AzureOpenAI` - AzureOpenAI
+            * `AzureAIFoundry` - AzureAIFoundry
+            * `VertexAI` - VertexAI
+            * `Gemini` - Gemini
+            * `Anthropic` - Anthropic
+            * `Custom` - Custom
 
-        created_at : typing.Optional[dt.datetime]
+        provider_model_id : typing.Optional[str]
+            The model ID to use within the given provider, e.g. gpt-3.5
 
-        updated_at : typing.Optional[dt.datetime]
+        model_provider_connection : typing.Optional[int]
 
-        organization : typing.Optional[PromptVersionOrganization]
+        organization : typing.Optional[int]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        PromptVersion
+        ThirdPartyModelVersion
 
 
         Examples
@@ -332,28 +383,24 @@ class VersionsClient:
             api_key="YOUR_API_KEY",
         )
         client.prompts.versions.update(
-            id=1,
+            prompt_id=1,
             version_id=1,
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}",
             method="PATCH",
             json={
-                "title": title,
                 "parent_model": parent_model,
-                "model_provider_connection": model_provider_connection,
+                "title": title,
                 "prompt": prompt,
                 "provider": provider,
                 "provider_model_id": provider_model_id,
-                "created_by": convert_and_respect_annotation_metadata(
-                    object_=created_by, annotation=PromptVersionCreatedBy, direction="write"
-                ),
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "organization": convert_and_respect_annotation_metadata(
-                    object_=organization, annotation=PromptVersionOrganization, direction="write"
-                ),
+                "model_provider_connection": model_provider_connection,
+                "organization": organization,
+            },
+            headers={
+                "content-type": "application/json",
             },
             request_options=request_options,
             omit=OMIT,
@@ -361,9 +408,9 @@ class VersionsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    PromptVersion,
-                    parse_obj_as(
-                        type_=PromptVersion,  # type: ignore
+                    ThirdPartyModelVersion,
+                    construct_type(
+                        type_=ThirdPartyModelVersion,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -373,30 +420,16 @@ class VersionsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def cost_estimate(
-        self,
-        prompt_id: int,
-        version_id: int,
-        *,
-        project_id: int,
-        project_subset: int,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, prompt_id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
     ) -> InferenceRunCostEstimate:
         """
-        Get cost estimate for running a prompt version on a particular project/subset
+        Get an estimate of the cost for making an inference run on the selected Prompt Version and Project/ProjectSubset
 
         Parameters
         ----------
         prompt_id : int
-            Prompt ID
 
         version_id : int
-            Prompt Version ID
-
-        project_id : int
-            ID of the project to get an estimate for running on
-
-        project_subset : int
-            Subset of the project to get an estimate for running on (e.g. 'All', 'Sample', or 'HasGT')
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -404,7 +437,7 @@ class VersionsClient:
         Returns
         -------
         InferenceRunCostEstimate
-
+            Cost estimate response
 
         Examples
         --------
@@ -416,24 +449,18 @@ class VersionsClient:
         client.prompts.versions.cost_estimate(
             prompt_id=1,
             version_id=1,
-            project_id=1,
-            project_subset=1,
         )
         """
         _response = self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}/cost-estimate",
-            method="POST",
-            params={
-                "project_id": project_id,
-                "project_subset": project_subset,
-            },
+            method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     InferenceRunCostEstimate,
-                    parse_obj_as(
+                    construct_type(
                         type_=InferenceRunCostEstimate,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -448,7 +475,7 @@ class VersionsClient:
         prompt_id: int,
         version_id: int,
         *,
-        refinement_job_id: str,
+        refinement_job_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> RefinedPromptResponse:
         """
@@ -457,12 +484,10 @@ class VersionsClient:
         Parameters
         ----------
         prompt_id : int
-            Prompt ID
 
         version_id : int
-            Prompt Version ID
 
-        refinement_job_id : str
+        refinement_job_id : typing.Optional[str]
             Refinement Job ID acquired from the `POST /api/prompts/{prompt_id}/versions/{version_id}/refine` endpoint
 
         request_options : typing.Optional[RequestOptions]
@@ -471,7 +496,7 @@ class VersionsClient:
         Returns
         -------
         RefinedPromptResponse
-
+            Refined prompt response
 
         Examples
         --------
@@ -483,7 +508,6 @@ class VersionsClient:
         client.prompts.versions.get_refined_prompt(
             prompt_id=1,
             version_id=1,
-            refinement_job_id="refinement_job_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -498,7 +522,7 @@ class VersionsClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     RefinedPromptResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=RefinedPromptResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -513,10 +537,10 @@ class VersionsClient:
         prompt_id: int,
         version_id: int,
         *,
+        teacher_model_provider_connection_id: int,
+        teacher_model_name: str,
+        project_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,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> RefinedPromptResponse:
         """
@@ -525,30 +549,28 @@ class VersionsClient:
         Parameters
         ----------
         prompt_id : int
-            Prompt ID
 
         version_id : int
-            Base Prompt Version ID
 
-        async_ : typing.Optional[bool]
-            Run the refinement job asynchronously
-
-        teacher_model_provider_connection_id : typing.Optional[int]
+        teacher_model_provider_connection_id : int
             Model Provider Connection ID to use to refine the prompt
 
-        teacher_model_name : typing.Optional[str]
+        teacher_model_name : str
             Name of the model to use to refine the prompt
 
-        project_id : typing.Optional[int]
+        project_id : int
             Project ID to target the refined prompt for
 
+        async_ : typing.Optional[bool]
+            Whether to run the refinement asynchronously
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
         RefinedPromptResponse
-
+            Refined prompt response
 
         Examples
         --------
@@ -560,6 +582,9 @@ class VersionsClient:
         client.prompts.versions.refine_prompt(
             prompt_id=1,
             version_id=1,
+            teacher_model_provider_connection_id=1,
+            teacher_model_name="teacher_model_name",
+            project_id=1,
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -583,7 +608,7 @@ class VersionsClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     RefinedPromptResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=RefinedPromptResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -598,23 +623,82 @@ class AsyncVersionsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list(
+    async def get_default_version_name(
         self, id: int, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[PromptVersion]:
+    ) -> None:
         """
-        Get a list of prompt versions.
+        Get default prompt version name
 
         Parameters
         ----------
         id : int
-            Prompt ID
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[PromptVersion]
+        None
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.prompts.versions.get_default_version_name(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/prompts/{jsonable_encoder(id)}/get-default-version-name",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def list(
+        self,
+        prompt_id_: int,
+        *,
+        prompt_id: int,
+        ordering: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[ThirdPartyModelVersion]:
+        """
+        List all versions of a prompt.
+
+        Parameters
+        ----------
+        prompt_id_ : int
+
+        prompt_id : int
+            A unique integer value identifying the model ID to list versions for.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[ThirdPartyModelVersion]
 
 
         Examples
@@ -630,23 +714,28 @@ class AsyncVersionsClient:
 
         async def main() -> None:
             await client.prompts.versions.list(
-                id=1,
+                prompt_id_=1,
+                prompt_id=1,
             )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions",
+            f"api/prompts/{jsonable_encoder(prompt_id_)}/versions",
             method="GET",
+            params={
+                "ordering": ordering,
+                "prompt_id": prompt_id,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.List[PromptVersion],
-                    parse_obj_as(
-                        type_=typing.List[PromptVersion],  # type: ignore
+                    typing.List[ThirdPartyModelVersion],
+                    construct_type(
+                        type_=typing.List[ThirdPartyModelVersion],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -657,54 +746,57 @@ class AsyncVersionsClient:
 
     async def create(
         self,
-        id: int,
+        prompt_id: int,
         *,
-        title: typing.Optional[str] = OMIT,
+        title: str,
+        prompt: str,
+        provider_model_id: str,
         parent_model: typing.Optional[int] = OMIT,
+        provider: typing.Optional[ProviderEnum] = OMIT,
         model_provider_connection: typing.Optional[int] = OMIT,
-        prompt: typing.Optional[str] = OMIT,
-        provider: typing.Optional[PromptVersionProvider] = OMIT,
-        provider_model_id: typing.Optional[str] = OMIT,
-        created_by: typing.Optional[PromptVersionCreatedBy] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        updated_at: typing.Optional[dt.datetime] = OMIT,
-        organization: typing.Optional[PromptVersionOrganization] = OMIT,
+        organization: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> PromptVersion:
+    ) -> ThirdPartyModelVersion:
         """
         Create a new version of a prompt.
 
         Parameters
         ----------
-        id : int
-            Prompt ID
-
-        title : typing.Optional[str]
-
-        parent_model : typing.Optional[int]
+        prompt_id : int
 
-        model_provider_connection : typing.Optional[int]
+        title : str
+            Model name
 
-        prompt : typing.Optional[str]
+        prompt : str
+            Prompt to execute
 
-        provider : typing.Optional[PromptVersionProvider]
+        provider_model_id : str
+            The model ID to use within the given provider, e.g. gpt-3.5
 
-        provider_model_id : typing.Optional[str]
+        parent_model : typing.Optional[int]
+            Parent model interface ID
 
-        created_by : typing.Optional[PromptVersionCreatedBy]
+        provider : typing.Optional[ProviderEnum]
+            The model provider to use e.g. OpenAI
 
-        created_at : typing.Optional[dt.datetime]
+            * `OpenAI` - OpenAI
+            * `AzureOpenAI` - AzureOpenAI
+            * `AzureAIFoundry` - AzureAIFoundry
+            * `VertexAI` - VertexAI
+            * `Gemini` - Gemini
+            * `Anthropic` - Anthropic
+            * `Custom` - Custom
 
-        updated_at : typing.Optional[dt.datetime]
+        model_provider_connection : typing.Optional[int]
 
-        organization : typing.Optional[PromptVersionOrganization]
+        organization : typing.Optional[int]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        PromptVersion
+        ThirdPartyModelVersion
 
 
         Examples
@@ -720,30 +812,26 @@ class AsyncVersionsClient:
 
         async def main() -> None:
             await client.prompts.versions.create(
-                id=1,
+                prompt_id=1,
+                title="title",
+                prompt="prompt",
+                provider_model_id="provider_model_id",
             )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions",
             method="POST",
             json={
-                "title": title,
                 "parent_model": parent_model,
-                "model_provider_connection": model_provider_connection,
+                "title": title,
                 "prompt": prompt,
                 "provider": provider,
                 "provider_model_id": provider_model_id,
-                "created_by": convert_and_respect_annotation_metadata(
-                    object_=created_by, annotation=PromptVersionCreatedBy, direction="write"
-                ),
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "organization": convert_and_respect_annotation_metadata(
-                    object_=organization, annotation=PromptVersionOrganization, direction="write"
-                ),
+                "model_provider_connection": model_provider_connection,
+                "organization": organization,
             },
             request_options=request_options,
             omit=OMIT,
@@ -751,9 +839,9 @@ class AsyncVersionsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    PromptVersion,
-                    parse_obj_as(
-                        type_=PromptVersion,  # type: ignore
+                    ThirdPartyModelVersion,
+                    construct_type(
+                        type_=ThirdPartyModelVersion,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -763,25 +851,23 @@ class AsyncVersionsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def get(
-        self, id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> PromptVersion:
+        self, prompt_id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> ThirdPartyModelVersion:
         """
-        Get a prompt version by ID.
+        Retrieve a specific prompt of a model.
 
         Parameters
         ----------
-        id : int
-            Prompt ID
+        prompt_id : int
 
         version_id : int
-            Prompt Version ID
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        PromptVersion
+        ThirdPartyModelVersion
 
 
         Examples
@@ -797,7 +883,7 @@ class AsyncVersionsClient:
 
         async def main() -> None:
             await client.prompts.versions.get(
-                id=1,
+                prompt_id=1,
                 version_id=1,
             )
 
@@ -805,16 +891,16 @@ class AsyncVersionsClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    PromptVersion,
-                    parse_obj_as(
-                        type_=PromptVersion,  # type: ignore
+                    ThirdPartyModelVersion,
+                    construct_type(
+                        type_=ThirdPartyModelVersion,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -824,18 +910,16 @@ class AsyncVersionsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def delete(
-        self, id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
+        self, prompt_id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
-        Delete a prompt version by ID.
+        Delete a prompt version by ID
 
         Parameters
         ----------
-        id : int
-            Prompt ID
+        prompt_id : int
 
         version_id : int
-            Prompt Version ID
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -857,7 +941,7 @@ class AsyncVersionsClient:
 
         async def main() -> None:
             await client.prompts.versions.delete(
-                id=1,
+                prompt_id=1,
                 version_id=1,
             )
 
@@ -865,7 +949,7 @@ class AsyncVersionsClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}",
             method="DELETE",
             request_options=request_options,
         )
@@ -879,58 +963,60 @@ class AsyncVersionsClient:
 
     async def update(
         self,
-        id: int,
+        prompt_id: int,
         version_id: int,
         *,
-        title: typing.Optional[str] = OMIT,
         parent_model: typing.Optional[int] = OMIT,
-        model_provider_connection: typing.Optional[int] = OMIT,
+        title: typing.Optional[str] = OMIT,
         prompt: typing.Optional[str] = OMIT,
-        provider: typing.Optional[PromptVersionProvider] = OMIT,
+        provider: typing.Optional[ProviderEnum] = OMIT,
         provider_model_id: typing.Optional[str] = OMIT,
-        created_by: typing.Optional[PromptVersionCreatedBy] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        updated_at: typing.Optional[dt.datetime] = OMIT,
-        organization: typing.Optional[PromptVersionOrganization] = OMIT,
+        model_provider_connection: typing.Optional[int] = OMIT,
+        organization: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> PromptVersion:
+    ) -> ThirdPartyModelVersion:
         """
-        Update a prompt version by ID.
+        Update a specific prompt version by ID.
 
         Parameters
         ----------
-        id : int
-            Prompt ID
+        prompt_id : int
 
         version_id : int
-            Prompt Version ID
-
-        title : typing.Optional[str]
 
         parent_model : typing.Optional[int]
+            Parent model interface ID
 
-        model_provider_connection : typing.Optional[int]
+        title : typing.Optional[str]
+            Model name
 
         prompt : typing.Optional[str]
+            Prompt to execute
 
-        provider : typing.Optional[PromptVersionProvider]
-
-        provider_model_id : typing.Optional[str]
+        provider : typing.Optional[ProviderEnum]
+            The model provider to use e.g. OpenAI
 
-        created_by : typing.Optional[PromptVersionCreatedBy]
+            * `OpenAI` - OpenAI
+            * `AzureOpenAI` - AzureOpenAI
+            * `AzureAIFoundry` - AzureAIFoundry
+            * `VertexAI` - VertexAI
+            * `Gemini` - Gemini
+            * `Anthropic` - Anthropic
+            * `Custom` - Custom
 
-        created_at : typing.Optional[dt.datetime]
+        provider_model_id : typing.Optional[str]
+            The model ID to use within the given provider, e.g. gpt-3.5
 
-        updated_at : typing.Optional[dt.datetime]
+        model_provider_connection : typing.Optional[int]
 
-        organization : typing.Optional[PromptVersionOrganization]
+        organization : typing.Optional[int]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        PromptVersion
+        ThirdPartyModelVersion
 
 
         Examples
@@ -946,7 +1032,7 @@ class AsyncVersionsClient:
 
         async def main() -> None:
             await client.prompts.versions.update(
-                id=1,
+                prompt_id=1,
                 version_id=1,
             )
 
@@ -954,23 +1040,19 @@ class AsyncVersionsClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}",
+            f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}",
             method="PATCH",
             json={
-                "title": title,
                 "parent_model": parent_model,
-                "model_provider_connection": model_provider_connection,
+                "title": title,
                 "prompt": prompt,
                 "provider": provider,
                 "provider_model_id": provider_model_id,
-                "created_by": convert_and_respect_annotation_metadata(
-                    object_=created_by, annotation=PromptVersionCreatedBy, direction="write"
-                ),
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "organization": convert_and_respect_annotation_metadata(
-                    object_=organization, annotation=PromptVersionOrganization, direction="write"
-                ),
+                "model_provider_connection": model_provider_connection,
+                "organization": organization,
+            },
+            headers={
+                "content-type": "application/json",
             },
             request_options=request_options,
             omit=OMIT,
@@ -978,9 +1060,9 @@ class AsyncVersionsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    PromptVersion,
-                    parse_obj_as(
-                        type_=PromptVersion,  # type: ignore
+                    ThirdPartyModelVersion,
+                    construct_type(
+                        type_=ThirdPartyModelVersion,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -990,30 +1072,16 @@ class AsyncVersionsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def cost_estimate(
-        self,
-        prompt_id: int,
-        version_id: int,
-        *,
-        project_id: int,
-        project_subset: int,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, prompt_id: int, version_id: int, *, request_options: typing.Optional[RequestOptions] = None
     ) -> InferenceRunCostEstimate:
         """
-        Get cost estimate for running a prompt version on a particular project/subset
+        Get an estimate of the cost for making an inference run on the selected Prompt Version and Project/ProjectSubset
 
         Parameters
         ----------
         prompt_id : int
-            Prompt ID
 
         version_id : int
-            Prompt Version ID
-
-        project_id : int
-            ID of the project to get an estimate for running on
-
-        project_subset : int
-            Subset of the project to get an estimate for running on (e.g. 'All', 'Sample', or 'HasGT')
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1021,7 +1089,7 @@ class AsyncVersionsClient:
         Returns
         -------
         InferenceRunCostEstimate
-
+            Cost estimate response
 
         Examples
         --------
@@ -1038,8 +1106,6 @@ class AsyncVersionsClient:
             await client.prompts.versions.cost_estimate(
                 prompt_id=1,
                 version_id=1,
-                project_id=1,
-                project_subset=1,
             )
 
 
@@ -1047,18 +1113,14 @@ class AsyncVersionsClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(prompt_id)}/versions/{jsonable_encoder(version_id)}/cost-estimate",
-            method="POST",
-            params={
-                "project_id": project_id,
-                "project_subset": project_subset,
-            },
+            method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     InferenceRunCostEstimate,
-                    parse_obj_as(
+                    construct_type(
                         type_=InferenceRunCostEstimate,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -1073,7 +1135,7 @@ class AsyncVersionsClient:
         prompt_id: int,
         version_id: int,
         *,
-        refinement_job_id: str,
+        refinement_job_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> RefinedPromptResponse:
         """
@@ -1082,12 +1144,10 @@ class AsyncVersionsClient:
         Parameters
         ----------
         prompt_id : int
-            Prompt ID
 
         version_id : int
-            Prompt Version ID
 
-        refinement_job_id : str
+        refinement_job_id : typing.Optional[str]
             Refinement Job ID acquired from the `POST /api/prompts/{prompt_id}/versions/{version_id}/refine` endpoint
 
         request_options : typing.Optional[RequestOptions]
@@ -1096,7 +1156,7 @@ class AsyncVersionsClient:
         Returns
         -------
         RefinedPromptResponse
-
+            Refined prompt response
 
         Examples
         --------
@@ -1113,7 +1173,6 @@ class AsyncVersionsClient:
             await client.prompts.versions.get_refined_prompt(
                 prompt_id=1,
                 version_id=1,
-                refinement_job_id="refinement_job_id",
             )
 
 
@@ -1131,7 +1190,7 @@ class AsyncVersionsClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     RefinedPromptResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=RefinedPromptResponse,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -1146,10 +1205,10 @@ class AsyncVersionsClient:
         prompt_id: int,
         version_id: int,
         *,
+        teacher_model_provider_connection_id: int,
+        teacher_model_name: str,
+        project_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,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> RefinedPromptResponse:
         """
@@ -1158,30 +1217,28 @@ class AsyncVersionsClient:
         Parameters
         ----------
         prompt_id : int
-            Prompt ID
 
         version_id : int
-            Base Prompt Version ID
-
-        async_ : typing.Optional[bool]
-            Run the refinement job asynchronously
 
-        teacher_model_provider_connection_id : typing.Optional[int]
+        teacher_model_provider_connection_id : int
             Model Provider Connection ID to use to refine the prompt
 
-        teacher_model_name : typing.Optional[str]
+        teacher_model_name : str
             Name of the model to use to refine the prompt
 
-        project_id : typing.Optional[int]
+        project_id : int
             Project ID to target the refined prompt for
 
+        async_ : typing.Optional[bool]
+            Whether to run the refinement asynchronously
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
         RefinedPromptResponse
-
+            Refined prompt response
 
         Examples
         --------
@@ -1198,6 +1255,9 @@ class AsyncVersionsClient:
             await client.prompts.versions.refine_prompt(
                 prompt_id=1,
                 version_id=1,
+                teacher_model_provider_connection_id=1,
+                teacher_model_name="teacher_model_name",
+                project_id=1,
             )
 
 
@@ -1224,7 +1284,7 @@ class AsyncVersionsClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     RefinedPromptResponse,
-                    parse_obj_as(
+                    construct_type(
                         type_=RefinedPromptResponse,  # type: ignore
                         object_=_response.json(),
                     ),