together 2.0.0a17__py3-none-any.whl → 2.0.0a19__py3-none-any.whl

together/resources/endpoints.py

@@ -7,7 +7,12 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..types import endpoint_list_params, endpoint_create_params, endpoint_update_params
+from ..types import (
+    endpoint_list_params,
+    endpoint_create_params,
+    endpoint_update_params,
+    endpoint_list_hardware_params,
+)
 from .._types import Body, Omit, Query, Headers, NoneType, NotGiven, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -23,6 +28,7 @@ from ..types.autoscaling_param import AutoscalingParam
 from ..types.dedicated_endpoint import DedicatedEndpoint
 from ..types.endpoint_list_response import EndpointListResponse
 from ..types.endpoint_list_avzones_response import EndpointListAvzonesResponse
+from ..types.endpoint_list_hardware_response import EndpointListHardwareResponse
 
 __all__ = ["EndpointsResource", "AsyncEndpointsResource"]
 
@@ -320,6 +326,47 @@ class EndpointsResource(SyncAPIResource):
             cast_to=EndpointListAvzonesResponse,
         )
 
+    def list_hardware(
+        self,
+        *,
+        model: 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,
+    ) -> EndpointListHardwareResponse:
+        """Returns a list of available hardware configurations for deploying models.
+
+        When a
+        model parameter is provided, it returns only hardware configurations compatible
+        with that model, including their current availability status.
+
+        Args:
+          model: Filter hardware configurations by model compatibility. When provided, the
+              response includes availability status for each compatible configuration.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._get(
+            "/hardware",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform({"model": model}, endpoint_list_hardware_params.EndpointListHardwareParams),
+            ),
+            cast_to=EndpointListHardwareResponse,
+        )
+
 
 class AsyncEndpointsResource(AsyncAPIResource):
     @cached_property
@@ -614,6 +661,49 @@ class AsyncEndpointsResource(AsyncAPIResource):
             cast_to=EndpointListAvzonesResponse,
         )
 
+    async def list_hardware(
+        self,
+        *,
+        model: 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,
+    ) -> EndpointListHardwareResponse:
+        """Returns a list of available hardware configurations for deploying models.
+
+        When a
+        model parameter is provided, it returns only hardware configurations compatible
+        with that model, including their current availability status.
+
+        Args:
+          model: Filter hardware configurations by model compatibility. When provided, the
+              response includes availability status for each compatible configuration.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._get(
+            "/hardware",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {"model": model}, endpoint_list_hardware_params.EndpointListHardwareParams
+                ),
+            ),
+            cast_to=EndpointListHardwareResponse,
+        )
+
 
 class EndpointsResourceWithRawResponse:
     def __init__(self, endpoints: EndpointsResource) -> None:
@@ -637,6 +727,9 @@ class EndpointsResourceWithRawResponse:
         self.list_avzones = to_raw_response_wrapper(
             endpoints.list_avzones,
         )
+        self.list_hardware = to_raw_response_wrapper(
+            endpoints.list_hardware,
+        )
 
 
 class AsyncEndpointsResourceWithRawResponse:
@@ -661,6 +754,9 @@ class AsyncEndpointsResourceWithRawResponse:
         self.list_avzones = async_to_raw_response_wrapper(
             endpoints.list_avzones,
         )
+        self.list_hardware = async_to_raw_response_wrapper(
+            endpoints.list_hardware,
+        )
 
 
 class EndpointsResourceWithStreamingResponse:
@@ -685,6 +781,9 @@ class EndpointsResourceWithStreamingResponse:
         self.list_avzones = to_streamed_response_wrapper(
             endpoints.list_avzones,
         )
+        self.list_hardware = to_streamed_response_wrapper(
+            endpoints.list_hardware,
+        )
 
 
 class AsyncEndpointsResourceWithStreamingResponse:
@@ -709,3 +808,6 @@ class AsyncEndpointsResourceWithStreamingResponse:
         self.list_avzones = async_to_streamed_response_wrapper(
             endpoints.list_avzones,
         )
+        self.list_hardware = async_to_streamed_response_wrapper(
+            endpoints.list_hardware,
+        )

together/resources/models/__init__.py

@@ -0,0 +1,33 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .models import (
+    ModelsResource,
+    AsyncModelsResource,
+    ModelsResourceWithRawResponse,
+    AsyncModelsResourceWithRawResponse,
+    ModelsResourceWithStreamingResponse,
+    AsyncModelsResourceWithStreamingResponse,
+)
+from .uploads import (
+    UploadsResource,
+    AsyncUploadsResource,
+    UploadsResourceWithRawResponse,
+    AsyncUploadsResourceWithRawResponse,
+    UploadsResourceWithStreamingResponse,
+    AsyncUploadsResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "UploadsResource",
+    "AsyncUploadsResource",
+    "UploadsResourceWithRawResponse",
+    "AsyncUploadsResourceWithRawResponse",
+    "UploadsResourceWithStreamingResponse",
+    "AsyncUploadsResourceWithStreamingResponse",
+    "ModelsResource",
+    "AsyncModelsResource",
+    "ModelsResourceWithRawResponse",
+    "AsyncModelsResourceWithRawResponse",
+    "ModelsResourceWithStreamingResponse",
+    "AsyncModelsResourceWithStreamingResponse",
+]

together/resources/{models.py → models/models.py}

@@ -6,25 +6,37 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..types import model_list_params, model_upload_params
-from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from .._utils import maybe_transform, async_maybe_transform
-from .._compat import cached_property
-from .._resource import SyncAPIResource, AsyncAPIResource
-from .._response import (
+from ...types import model_list_params, model_upload_params
+from .uploads import (
+    UploadsResource,
+    AsyncUploadsResource,
+    UploadsResourceWithRawResponse,
+    AsyncUploadsResourceWithRawResponse,
+    UploadsResourceWithStreamingResponse,
+    AsyncUploadsResourceWithStreamingResponse,
+)
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ..._utils import maybe_transform, async_maybe_transform
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
     to_raw_response_wrapper,
     to_streamed_response_wrapper,
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from .._base_client import make_request_options
-from ..types.model_list_response import ModelListResponse
-from ..types.model_upload_response import ModelUploadResponse
+from ..._base_client import make_request_options
+from ...types.model_list_response import ModelListResponse
+from ...types.model_upload_response import ModelUploadResponse
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
 
 
 class ModelsResource(SyncAPIResource):
+    @cached_property
+    def uploads(self) -> UploadsResource:
+        return UploadsResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> ModelsResourceWithRawResponse:
         """
@@ -148,6 +160,10 @@ class ModelsResource(SyncAPIResource):
 
 
 class AsyncModelsResource(AsyncAPIResource):
+    @cached_property
+    def uploads(self) -> AsyncUploadsResource:
+        return AsyncUploadsResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> AsyncModelsResourceWithRawResponse:
         """
@@ -281,6 +297,10 @@ class ModelsResourceWithRawResponse:
             models.upload,
         )
 
+    @cached_property
+    def uploads(self) -> UploadsResourceWithRawResponse:
+        return UploadsResourceWithRawResponse(self._models.uploads)
+
 
 class AsyncModelsResourceWithRawResponse:
     def __init__(self, models: AsyncModelsResource) -> None:
@@ -293,6 +313,10 @@ class AsyncModelsResourceWithRawResponse:
             models.upload,
         )
 
+    @cached_property
+    def uploads(self) -> AsyncUploadsResourceWithRawResponse:
+        return AsyncUploadsResourceWithRawResponse(self._models.uploads)
+
 
 class ModelsResourceWithStreamingResponse:
     def __init__(self, models: ModelsResource) -> None:
@@ -305,6 +329,10 @@ class ModelsResourceWithStreamingResponse:
             models.upload,
         )
 
+    @cached_property
+    def uploads(self) -> UploadsResourceWithStreamingResponse:
+        return UploadsResourceWithStreamingResponse(self._models.uploads)
+
 
 class AsyncModelsResourceWithStreamingResponse:
     def __init__(self, models: AsyncModelsResource) -> None:
@@ -316,3 +344,7 @@ class AsyncModelsResourceWithStreamingResponse:
         self.upload = async_to_streamed_response_wrapper(
             models.upload,
         )
+
+    @cached_property
+    def uploads(self) -> AsyncUploadsResourceWithStreamingResponse:
+        return AsyncUploadsResourceWithStreamingResponse(self._models.uploads)

together/resources/models/uploads.py

@@ -0,0 +1,163 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..._types import Body, Query, Headers, NotGiven, not_given
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ..._base_client import make_request_options
+from ...types.models.upload_status_response import UploadStatusResponse
+
+__all__ = ["UploadsResource", "AsyncUploadsResource"]
+
+
+class UploadsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> UploadsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/togethercomputer/together-py#accessing-raw-response-data-eg-headers
+        """
+        return UploadsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> UploadsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/togethercomputer/together-py#with_streaming_response
+        """
+        return UploadsResourceWithStreamingResponse(self)
+
+    def status(
+        self,
+        job_id: str,
+        *,
+        # 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,
+    ) -> UploadStatusResponse:
+        """
+        Get the status of a specific job
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not job_id:
+            raise ValueError(f"Expected a non-empty value for `job_id` but received {job_id!r}")
+        return self._get(
+            f"/jobs/{job_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=UploadStatusResponse,
+        )
+
+
+class AsyncUploadsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncUploadsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/togethercomputer/together-py#accessing-raw-response-data-eg-headers
+        """
+        return AsyncUploadsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncUploadsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/togethercomputer/together-py#with_streaming_response
+        """
+        return AsyncUploadsResourceWithStreamingResponse(self)
+
+    async def status(
+        self,
+        job_id: str,
+        *,
+        # 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,
+    ) -> UploadStatusResponse:
+        """
+        Get the status of a specific job
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not job_id:
+            raise ValueError(f"Expected a non-empty value for `job_id` but received {job_id!r}")
+        return await self._get(
+            f"/jobs/{job_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=UploadStatusResponse,
+        )
+
+
+class UploadsResourceWithRawResponse:
+    def __init__(self, uploads: UploadsResource) -> None:
+        self._uploads = uploads
+
+        self.status = to_raw_response_wrapper(
+            uploads.status,
+        )
+
+
+class AsyncUploadsResourceWithRawResponse:
+    def __init__(self, uploads: AsyncUploadsResource) -> None:
+        self._uploads = uploads
+
+        self.status = async_to_raw_response_wrapper(
+            uploads.status,
+        )
+
+
+class UploadsResourceWithStreamingResponse:
+    def __init__(self, uploads: UploadsResource) -> None:
+        self._uploads = uploads
+
+        self.status = to_streamed_response_wrapper(
+            uploads.status,
+        )
+
+
+class AsyncUploadsResourceWithStreamingResponse:
+    def __init__(self, uploads: AsyncUploadsResource) -> None:
+        self._uploads = uploads
+
+        self.status = async_to_streamed_response_wrapper(
+            uploads.status,
+        )

together/types/__init__.py

@@ -25,7 +25,6 @@ from .eval_list_params import EvalListParams as EvalListParams
 from .execute_response import ExecuteResponse as ExecuteResponse
 from .autoscaling_param import AutoscalingParam as AutoscalingParam
 from .finetune_response import FinetuneResponse as FinetuneResponse
-from .job_list_response import JobListResponse as JobListResponse
 from .model_list_params import ModelListParams as ModelListParams
 from .tool_choice_param import ToolChoiceParam as ToolChoiceParam
 from .dedicated_endpoint import DedicatedEndpoint as DedicatedEndpoint
@@ -41,16 +40,13 @@ from .endpoint_list_params import EndpointListParams as EndpointListParams
 from .eval_create_response import EvalCreateResponse as EvalCreateResponse
 from .eval_status_response import EvalStatusResponse as EvalStatusResponse
 from .file_delete_response import FileDeleteResponse as FileDeleteResponse
-from .hardware_list_params import HardwareListParams as HardwareListParams
 from .rerank_create_params import RerankCreateParams as RerankCreateParams
 from .batch_create_response import BatchCreateResponse as BatchCreateResponse
 from .image_generate_params import ImageGenerateParams as ImageGenerateParams
-from .job_retrieve_response import JobRetrieveResponse as JobRetrieveResponse
 from .model_upload_response import ModelUploadResponse as ModelUploadResponse
 from .endpoint_create_params import EndpointCreateParams as EndpointCreateParams
 from .endpoint_list_response import EndpointListResponse as EndpointListResponse
 from .endpoint_update_params import EndpointUpdateParams as EndpointUpdateParams
-from .hardware_list_response import HardwareListResponse as HardwareListResponse
 from .rerank_create_response import RerankCreateResponse as RerankCreateResponse
 from .embedding_create_params import EmbeddingCreateParams as EmbeddingCreateParams
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams
@@ -60,8 +56,10 @@ from .fine_tuning_list_response import FineTuningListResponse as FineTuningListR
 from .fine_tuning_content_params import FineTuningContentParams as FineTuningContentParams
 from .fine_tuning_cancel_response import FineTuningCancelResponse as FineTuningCancelResponse
 from .fine_tuning_delete_response import FineTuningDeleteResponse as FineTuningDeleteResponse
+from .endpoint_list_hardware_params import EndpointListHardwareParams as EndpointListHardwareParams
 from .endpoint_list_avzones_response import EndpointListAvzonesResponse as EndpointListAvzonesResponse
 from .code_interpreter_execute_params import CodeInterpreterExecuteParams as CodeInterpreterExecuteParams
+from .endpoint_list_hardware_response import EndpointListHardwareResponse as EndpointListHardwareResponse
 from .fine_tuning_list_events_response import FineTuningListEventsResponse as FineTuningListEventsResponse
 from .fine_tuning_estimate_price_params import FineTuningEstimatePriceParams as FineTuningEstimatePriceParams
 from .fine_tuning_estimate_price_response import FineTuningEstimatePriceResponse as FineTuningEstimatePriceResponse

together/types/beta/__init__.py

@@ -3,8 +3,14 @@
 from __future__ import annotations
 
 from .cluster import Cluster as Cluster
+from .deployment import Deployment as Deployment
+from .deployment_logs import DeploymentLogs as DeploymentLogs
+from .jig_deploy_params import JigDeployParams as JigDeployParams
+from .jig_list_response import JigListResponse as JigListResponse
+from .jig_update_params import JigUpdateParams as JigUpdateParams
 from .cluster_create_params import ClusterCreateParams as ClusterCreateParams
 from .cluster_list_response import ClusterListResponse as ClusterListResponse
 from .cluster_update_params import ClusterUpdateParams as ClusterUpdateParams
 from .cluster_delete_response import ClusterDeleteResponse as ClusterDeleteResponse
+from .jig_retrieve_logs_params import JigRetrieveLogsParams as JigRetrieveLogsParams
 from .cluster_list_regions_response import ClusterListRegionsResponse as ClusterListRegionsResponse