together 1.5.17__py3-none-any.whl → 2.0.0a8__py3-none-any.whl

together/resources/hardware.py

@@ -0,0 +1,181 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..types import hardware_list_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 (
+    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.hardware_list_response import HardwareListResponse
+
+__all__ = ["HardwareResource", "AsyncHardwareResource"]
+
+
+class HardwareResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> HardwareResourceWithRawResponse:
+        """
+        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 HardwareResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> HardwareResourceWithStreamingResponse:
+        """
+        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 HardwareResourceWithStreamingResponse(self)
+
+    def list(
+        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,
+    ) -> HardwareListResponse:
+        """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}, hardware_list_params.HardwareListParams),
+            ),
+            cast_to=HardwareListResponse,
+        )
+
+
+class AsyncHardwareResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncHardwareResourceWithRawResponse:
+        """
+        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 AsyncHardwareResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncHardwareResourceWithStreamingResponse:
+        """
+        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 AsyncHardwareResourceWithStreamingResponse(self)
+
+    async def list(
+        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,
+    ) -> HardwareListResponse:
+        """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}, hardware_list_params.HardwareListParams),
+            ),
+            cast_to=HardwareListResponse,
+        )
+
+
+class HardwareResourceWithRawResponse:
+    def __init__(self, hardware: HardwareResource) -> None:
+        self._hardware = hardware
+
+        self.list = to_raw_response_wrapper(
+            hardware.list,
+        )
+
+
+class AsyncHardwareResourceWithRawResponse:
+    def __init__(self, hardware: AsyncHardwareResource) -> None:
+        self._hardware = hardware
+
+        self.list = async_to_raw_response_wrapper(
+            hardware.list,
+        )
+
+
+class HardwareResourceWithStreamingResponse:
+    def __init__(self, hardware: HardwareResource) -> None:
+        self._hardware = hardware
+
+        self.list = to_streamed_response_wrapper(
+            hardware.list,
+        )
+
+
+class AsyncHardwareResourceWithStreamingResponse:
+    def __init__(self, hardware: AsyncHardwareResource) -> None:
+        self._hardware = hardware
+
+        self.list = async_to_streamed_response_wrapper(
+            hardware.list,
+        )

together/resources/images.py

@@ -1,162 +1,310 @@
-from __future__ import annotations
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Any
+from __future__ import annotations
 
-from together.abstract import api_requestor
-from together.together_response import TogetherResponse
-from together.types import (
-    ImageRequest,
-    ImageResponse,
-    TogetherClient,
-    TogetherRequest,
+from typing import Union, Iterable
+from typing_extensions import Literal
+
+import httpx
+
+from ..types import image_generate_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 (
+    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.image_file import ImageFile
 
+__all__ = ["ImagesResource", "AsyncImagesResource"]
 
-class Images:
-    def __init__(self, client: TogetherClient) -> None:
-        self._client = client
+
+class ImagesResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> ImagesResourceWithRawResponse:
+        """
+        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 ImagesResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> ImagesResourceWithStreamingResponse:
+        """
+        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 ImagesResourceWithStreamingResponse(self)
 
     def generate(
         self,
         *,
+        model: Union[
+            Literal[
+                "black-forest-labs/FLUX.1-schnell-Free",
+                "black-forest-labs/FLUX.1-schnell",
+                "black-forest-labs/FLUX.1.1-pro",
+            ],
+            str,
+        ],
         prompt: str,
-        model: str,
-        steps: int | None = 20,
-        seed: int | None = None,
-        n: int | None = 1,
-        height: int | None = 1024,
-        width: int | None = 1024,
-        negative_prompt: str | None = None,
-        **kwargs: Any,
-    ) -> ImageResponse:
+        disable_safety_checker: bool | Omit = omit,
+        guidance_scale: float | Omit = omit,
+        height: int | Omit = omit,
+        image_loras: Iterable[image_generate_params.ImageLora] | Omit = omit,
+        image_url: str | Omit = omit,
+        n: int | Omit = omit,
+        negative_prompt: str | Omit = omit,
+        output_format: Literal["jpeg", "png"] | Omit = omit,
+        response_format: Literal["base64", "url"] | Omit = omit,
+        seed: int | Omit = omit,
+        steps: int | Omit = omit,
+        width: int | 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,
+    ) -> ImageFile:
         """
-        Method to generate images based on a given prompt using a specified model.
+        Use an image model to generate an image for a given prompt.
 
         Args:
-            prompt (str): A description of the desired images. Maximum length varies by model.
+          model: The model to use for image generation.
 
-            model (str, optional): The model to use for image generation.
+              [See all of Together AI's image models](https://docs.together.ai/docs/serverless-models#image-models)
 
-            steps (int, optional): Number of generation steps. Defaults to 20
+          prompt: A description of the desired images. Maximum length varies by model.
 
-            seed (int, optional): Seed used for generation. Can be used to reproduce image generations.
-                Defaults to None.
+          disable_safety_checker: If true, disables the safety checker for image generation.
 
-            n (int, optional): Number of image results to generate. Defaults to 1.
+          guidance_scale: Adjusts the alignment of the generated image with the input prompt. Higher
+              values (e.g., 8-10) make the output more faithful to the prompt, while lower
+              values (e.g., 1-5) encourage more creative freedom.
 
-            height (int, optional): Height of the image to generate in number of pixels. Defaults to 1024
+          height: Height of the image to generate in number of pixels.
 
-            width (int, optional): Width of the image to generate in number of pixels. Defaults to 1024
+          image_loras: An array of objects that define LoRAs (Low-Rank Adaptations) to influence the
+              generated image.
 
-            negative_prompt (str, optional): The prompt or prompts not to guide the image generation.
-                Defaults to None
+          image_url: URL of an image to use for image models that support it.
 
-            image_base64: (str, optional): Reference image used for generation. Defaults to None.
+          n: Number of image results to generate.
 
-        Returns:
-            ImageResponse: Object containing image data
-        """
+          negative_prompt: The prompt or prompts not to guide the image generation.
 
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
+          output_format: The format of the image response. Can be either be `jpeg` or `png`. Defaults to
+              `jpeg`.
+
+          response_format: Format of the image response. Can be either a base64 string or a URL.
 
-        parameter_payload = ImageRequest(
-            prompt=prompt,
-            model=model,
-            steps=steps,
-            seed=seed,
-            n=n,
-            height=height,
-            width=width,
-            negative_prompt=negative_prompt,
-            **kwargs,
-        ).model_dump(exclude_none=True)
-
-        response, _, _ = requestor.request(
-            options=TogetherRequest(
-                method="POST",
-                url="images/generations",
-                params=parameter_payload,
+          seed: Seed used for generation. Can be used to reproduce image generations.
+
+          steps: Number of generation steps.
+
+          width: Width of the image to generate in number of pixels.
+
+          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._post(
+            "/images/generations",
+            body=maybe_transform(
+                {
+                    "model": model,
+                    "prompt": prompt,
+                    "disable_safety_checker": disable_safety_checker,
+                    "guidance_scale": guidance_scale,
+                    "height": height,
+                    "image_loras": image_loras,
+                    "image_url": image_url,
+                    "n": n,
+                    "negative_prompt": negative_prompt,
+                    "output_format": output_format,
+                    "response_format": response_format,
+                    "seed": seed,
+                    "steps": steps,
+                    "width": width,
+                },
+                image_generate_params.ImageGenerateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            stream=False,
+            cast_to=ImageFile,
         )
 
-        assert isinstance(response, TogetherResponse)
 
-        return ImageResponse(**response.data)
+class AsyncImagesResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncImagesResourceWithRawResponse:
+        """
+        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 AsyncImagesResourceWithRawResponse(self)
 
-class AsyncImages:
-    def __init__(self, client: TogetherClient) -> None:
-        self._client = client
+    @cached_property
+    def with_streaming_response(self) -> AsyncImagesResourceWithStreamingResponse:
+        """
+        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 AsyncImagesResourceWithStreamingResponse(self)
 
     async def generate(
         self,
         *,
+        model: Union[
+            Literal[
+                "black-forest-labs/FLUX.1-schnell-Free",
+                "black-forest-labs/FLUX.1-schnell",
+                "black-forest-labs/FLUX.1.1-pro",
+            ],
+            str,
+        ],
         prompt: str,
-        model: str,
-        steps: int | None = 20,
-        seed: int | None = None,
-        n: int | None = 1,
-        height: int | None = 1024,
-        width: int | None = 1024,
-        negative_prompt: str | None = None,
-        **kwargs: Any,
-    ) -> ImageResponse:
+        disable_safety_checker: bool | Omit = omit,
+        guidance_scale: float | Omit = omit,
+        height: int | Omit = omit,
+        image_loras: Iterable[image_generate_params.ImageLora] | Omit = omit,
+        image_url: str | Omit = omit,
+        n: int | Omit = omit,
+        negative_prompt: str | Omit = omit,
+        output_format: Literal["jpeg", "png"] | Omit = omit,
+        response_format: Literal["base64", "url"] | Omit = omit,
+        seed: int | Omit = omit,
+        steps: int | Omit = omit,
+        width: int | 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,
+    ) -> ImageFile:
         """
-        Async method to generate images based on a given prompt using a specified model.
+        Use an image model to generate an image for a given prompt.
 
         Args:
-            prompt (str): A description of the desired images. Maximum length varies by model.
+          model: The model to use for image generation.
+
+              [See all of Together AI's image models](https://docs.together.ai/docs/serverless-models#image-models)
+
+          prompt: A description of the desired images. Maximum length varies by model.
+
+          disable_safety_checker: If true, disables the safety checker for image generation.
+
+          guidance_scale: Adjusts the alignment of the generated image with the input prompt. Higher
+              values (e.g., 8-10) make the output more faithful to the prompt, while lower
+              values (e.g., 1-5) encourage more creative freedom.
+
+          height: Height of the image to generate in number of pixels.
 
-            model (str, optional): The model to use for image generation.
+          image_loras: An array of objects that define LoRAs (Low-Rank Adaptations) to influence the
+              generated image.
 
-            steps (int, optional): Number of generation steps. Defaults to 20
+          image_url: URL of an image to use for image models that support it.
 
-            seed (int, optional): Seed used for generation. Can be used to reproduce image generations.
-                Defaults to None.
+          n: Number of image results to generate.
 
-            n (int, optional): Number of image results to generate. Defaults to 1.
+          negative_prompt: The prompt or prompts not to guide the image generation.
 
-            height (int, optional): Height of the image to generate in number of pixels. Defaults to 1024
+          output_format: The format of the image response. Can be either be `jpeg` or `png`. Defaults to
+              `jpeg`.
 
-            width (int, optional): Width of the image to generate in number of pixels. Defaults to 1024
+          response_format: Format of the image response. Can be either a base64 string or a URL.
 
-            negative_prompt (str, optional): The prompt or prompts not to guide the image generation.
-                Defaults to None
+          seed: Seed used for generation. Can be used to reproduce image generations.
 
-            image_base64: (str, optional): Reference image used for generation. Defaults to None.
+          steps: Number of generation steps.
 
-        Returns:
-            ImageResponse: Object containing image data
+          width: Width of the image to generate in number of pixels.
+
+          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._post(
+            "/images/generations",
+            body=await async_maybe_transform(
+                {
+                    "model": model,
+                    "prompt": prompt,
+                    "disable_safety_checker": disable_safety_checker,
+                    "guidance_scale": guidance_scale,
+                    "height": height,
+                    "image_loras": image_loras,
+                    "image_url": image_url,
+                    "n": n,
+                    "negative_prompt": negative_prompt,
+                    "output_format": output_format,
+                    "response_format": response_format,
+                    "seed": seed,
+                    "steps": steps,
+                    "width": width,
+                },
+                image_generate_params.ImageGenerateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ImageFile,
+        )
+
+
+class ImagesResourceWithRawResponse:
+    def __init__(self, images: ImagesResource) -> None:
+        self._images = images
 
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
+        self.generate = to_raw_response_wrapper(
+            images.generate,
         )
 
-        parameter_payload = ImageRequest(
-            prompt=prompt,
-            model=model,
-            steps=steps,
-            seed=seed,
-            n=n,
-            height=height,
-            width=width,
-            negative_prompt=negative_prompt,
-            **kwargs,
-        ).model_dump(exclude_none=True)
-
-        response, _, _ = await requestor.arequest(
-            options=TogetherRequest(
-                method="POST",
-                url="images/generations",
-                params=parameter_payload,
-            ),
-            stream=False,
+
+class AsyncImagesResourceWithRawResponse:
+    def __init__(self, images: AsyncImagesResource) -> None:
+        self._images = images
+
+        self.generate = async_to_raw_response_wrapper(
+            images.generate,
         )
 
-        assert isinstance(response, TogetherResponse)
 
-        return ImageResponse(**response.data)
+class ImagesResourceWithStreamingResponse:
+    def __init__(self, images: ImagesResource) -> None:
+        self._images = images
+
+        self.generate = to_streamed_response_wrapper(
+            images.generate,
+        )
+
+
+class AsyncImagesResourceWithStreamingResponse:
+    def __init__(self, images: AsyncImagesResource) -> None:
+        self._images = images
+
+        self.generate = async_to_streamed_response_wrapper(
+            images.generate,
+        )