scale-gp-beta 0.1.0a30__py3-none-any.whl → 0.1.0a32__py3-none-any.whl

scale_gp_beta/resources/models.py

@@ -8,7 +8,7 @@ from typing_extensions import Literal, overload
 import httpx
 
 from ..types import model_list_params, model_create_params, model_update_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -52,16 +52,16 @@ class ModelsResource(SyncAPIResource):
         *,
         name: str,
         vendor_configuration: model_create_params.LaunchModelCreateRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        model_type: Literal["generic"] | NotGiven = NOT_GIVEN,
-        model_vendor: Literal["launch"] | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "update"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        model_type: Literal["generic"] | Omit = omit,
+        model_vendor: Literal["launch"] | Omit = omit,
+        on_conflict: Literal["error", "update"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Create Model
@@ -85,16 +85,16 @@ class ModelsResource(SyncAPIResource):
         *,
         name: str,
         vendor_configuration: model_create_params.LlmEngineModelCreateRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        model_type: Literal["chat_completion"] | NotGiven = NOT_GIVEN,
-        model_vendor: Literal["llmengine"] | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "update"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        model_type: Literal["chat_completion"] | Omit = omit,
+        model_vendor: Literal["llmengine"] | Omit = omit,
+        on_conflict: Literal["error", "update"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Create Model
@@ -119,16 +119,16 @@ class ModelsResource(SyncAPIResource):
         name: str,
         vendor_configuration: model_create_params.LaunchModelCreateRequestVendorConfiguration
         | model_create_params.LlmEngineModelCreateRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        model_type: Literal["generic"] | Literal["chat_completion"] | NotGiven = NOT_GIVEN,
-        model_vendor: Literal["launch"] | Literal["llmengine"] | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "update"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        model_type: Literal["generic"] | Literal["chat_completion"] | Omit = omit,
+        model_vendor: Literal["launch"] | Literal["llmengine"] | Omit = omit,
+        on_conflict: Literal["error", "update"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         return self._post(
             "/v5/models",
@@ -158,7 +158,7 @@ class ModelsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Get Model
@@ -187,13 +187,13 @@ class ModelsResource(SyncAPIResource):
         self,
         model_id: str,
         *,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Update Model
@@ -215,13 +215,13 @@ class ModelsResource(SyncAPIResource):
         model_id: str,
         *,
         vendor_configuration: model_update_params.ModelConfigurationPatchRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Update Model
@@ -243,13 +243,13 @@ class ModelsResource(SyncAPIResource):
         model_id: str,
         *,
         name: str,
-        on_conflict: Literal["error", "swap"] | NotGiven = NOT_GIVEN,
+        on_conflict: Literal["error", "swap"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Update Model
@@ -269,17 +269,16 @@ class ModelsResource(SyncAPIResource):
         self,
         model_id: str,
         *,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        vendor_configuration: model_update_params.ModelConfigurationPatchRequestVendorConfiguration
-        | NotGiven = NOT_GIVEN,
-        name: str | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "swap"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        vendor_configuration: model_update_params.ModelConfigurationPatchRequestVendorConfiguration | Omit = omit,
+        name: str | Omit = omit,
+        on_conflict: Literal["error", "swap"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         if not model_id:
             raise ValueError(f"Expected a non-empty value for `model_id` but received {model_id!r}")
@@ -303,8 +302,8 @@ class ModelsResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: str | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
+        ending_before: str | Omit = omit,
+        limit: int | Omit = omit,
         model_vendor: Literal[
             "openai",
             "cohere",
@@ -319,16 +318,16 @@ class ModelsResource(SyncAPIResource):
             "xai",
             "fireworks_ai",
         ]
-        | NotGiven = NOT_GIVEN,
-        name: str | NotGiven = NOT_GIVEN,
-        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
-        starting_after: str | NotGiven = NOT_GIVEN,
+        | Omit = omit,
+        name: str | Omit = omit,
+        sort_order: Literal["asc", "desc"] | Omit = omit,
+        starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SyncCursorPage[InferenceModel]:
         """
         List Models
@@ -374,7 +373,7 @@ class ModelsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> ModelDeleteResponse:
         """
         Delete Model
@@ -425,16 +424,16 @@ class AsyncModelsResource(AsyncAPIResource):
         *,
         name: str,
         vendor_configuration: model_create_params.LaunchModelCreateRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        model_type: Literal["generic"] | NotGiven = NOT_GIVEN,
-        model_vendor: Literal["launch"] | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "update"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        model_type: Literal["generic"] | Omit = omit,
+        model_vendor: Literal["launch"] | Omit = omit,
+        on_conflict: Literal["error", "update"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Create Model
@@ -458,16 +457,16 @@ class AsyncModelsResource(AsyncAPIResource):
         *,
         name: str,
         vendor_configuration: model_create_params.LlmEngineModelCreateRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        model_type: Literal["chat_completion"] | NotGiven = NOT_GIVEN,
-        model_vendor: Literal["llmengine"] | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "update"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        model_type: Literal["chat_completion"] | Omit = omit,
+        model_vendor: Literal["llmengine"] | Omit = omit,
+        on_conflict: Literal["error", "update"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Create Model
@@ -492,16 +491,16 @@ class AsyncModelsResource(AsyncAPIResource):
         name: str,
         vendor_configuration: model_create_params.LaunchModelCreateRequestVendorConfiguration
         | model_create_params.LlmEngineModelCreateRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        model_type: Literal["generic"] | Literal["chat_completion"] | NotGiven = NOT_GIVEN,
-        model_vendor: Literal["launch"] | Literal["llmengine"] | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "update"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        model_type: Literal["generic"] | Literal["chat_completion"] | Omit = omit,
+        model_vendor: Literal["launch"] | Literal["llmengine"] | Omit = omit,
+        on_conflict: Literal["error", "update"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         return await self._post(
             "/v5/models",
@@ -531,7 +530,7 @@ class AsyncModelsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Get Model
@@ -560,13 +559,13 @@ class AsyncModelsResource(AsyncAPIResource):
         self,
         model_id: str,
         *,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Update Model
@@ -588,13 +587,13 @@ class AsyncModelsResource(AsyncAPIResource):
         model_id: str,
         *,
         vendor_configuration: model_update_params.ModelConfigurationPatchRequestVendorConfiguration,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Update Model
@@ -616,13 +615,13 @@ class AsyncModelsResource(AsyncAPIResource):
         model_id: str,
         *,
         name: str,
-        on_conflict: Literal["error", "swap"] | NotGiven = NOT_GIVEN,
+        on_conflict: Literal["error", "swap"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         """
         Update Model
@@ -642,17 +641,16 @@ class AsyncModelsResource(AsyncAPIResource):
         self,
         model_id: str,
         *,
-        model_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        vendor_configuration: model_update_params.ModelConfigurationPatchRequestVendorConfiguration
-        | NotGiven = NOT_GIVEN,
-        name: str | NotGiven = NOT_GIVEN,
-        on_conflict: Literal["error", "swap"] | NotGiven = NOT_GIVEN,
+        model_metadata: Dict[str, object] | Omit = omit,
+        vendor_configuration: model_update_params.ModelConfigurationPatchRequestVendorConfiguration | Omit = omit,
+        name: str | Omit = omit,
+        on_conflict: Literal["error", "swap"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> InferenceModel:
         if not model_id:
             raise ValueError(f"Expected a non-empty value for `model_id` but received {model_id!r}")
@@ -676,8 +674,8 @@ class AsyncModelsResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: str | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
+        ending_before: str | Omit = omit,
+        limit: int | Omit = omit,
         model_vendor: Literal[
             "openai",
             "cohere",
@@ -692,16 +690,16 @@ class AsyncModelsResource(AsyncAPIResource):
             "xai",
             "fireworks_ai",
         ]
-        | NotGiven = NOT_GIVEN,
-        name: str | NotGiven = NOT_GIVEN,
-        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
-        starting_after: str | NotGiven = NOT_GIVEN,
+        | Omit = omit,
+        name: str | Omit = omit,
+        sort_order: Literal["asc", "desc"] | Omit = omit,
+        starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AsyncPaginator[InferenceModel, AsyncCursorPage[InferenceModel]]:
         """
         List Models
@@ -747,7 +745,7 @@ class AsyncModelsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> ModelDeleteResponse:
         """
         Delete Model