feat(api): add translation endpoint

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 30
+configured_endpoints: 31
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/writerai%2Fwriter-a8e20e751972f7b8c948a833764b02d0835b380189eaf7393d63eb2250dbd598.yml
 openapi_spec_hash: 6f27335468bc5bf01fbcd1454330fb07
-config_hash: ee2e5a914fd298a6f4c740f5ce187f87
+config_hash: 1f6d0bf7309d0007e28ab85b89a0de85

api.md

@@ -182,6 +182,18 @@ Methods:
 
 - <code title="post /v1/tools/comprehend/medical">client.tools.comprehend.<a href="./src/writerai/resources/tools/comprehend.py">medical</a>(\*\*<a href="src/writerai/types/tools/comprehend_medical_params.py">params</a>) -> <a href="./src/writerai/types/tools/comprehend_medical_response.py">ComprehendMedicalResponse</a></code>
 
+# Translation
+
+Types:
+
+```python
+from writerai.types import TranslationRequest, TranslationResponse
+```
+
+Methods:
+
+- <code title="post /v1/translation">client.translation.<a href="./src/writerai/resources/translation.py">translate</a>(\*\*<a href="src/writerai/types/translation_translate_params.py">params</a>) -> <a href="./src/writerai/types/translation_response.py">TranslationResponse</a></code>
+
 # Vision
 
 Types:

src/writerai/_client.py

@@ -25,7 +25,7 @@
     get_async_library,
 )
 from ._version import __version__
-from .resources import chat, files, graphs, models, vision, completions
+from .resources import chat, files, graphs, models, vision, completions, translation
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import WriterError, APIStatusError
 from ._base_client import (
@@ -47,6 +47,7 @@ class Writer(SyncAPIClient):
     graphs: graphs.GraphsResource
     files: files.FilesResource
     tools: tools.ToolsResource
+    translation: translation.TranslationResource
     vision: vision.VisionResource
     with_raw_response: WriterWithRawResponse
     with_streaming_response: WriterWithStreamedResponse
@@ -114,6 +115,7 @@ def __init__(
         self.graphs = graphs.GraphsResource(self)
         self.files = files.FilesResource(self)
         self.tools = tools.ToolsResource(self)
+        self.translation = translation.TranslationResource(self)
         self.vision = vision.VisionResource(self)
         self.with_raw_response = WriterWithRawResponse(self)
         self.with_streaming_response = WriterWithStreamedResponse(self)
@@ -231,6 +233,7 @@ class AsyncWriter(AsyncAPIClient):
     graphs: graphs.AsyncGraphsResource
     files: files.AsyncFilesResource
     tools: tools.AsyncToolsResource
+    translation: translation.AsyncTranslationResource
     vision: vision.AsyncVisionResource
     with_raw_response: AsyncWriterWithRawResponse
     with_streaming_response: AsyncWriterWithStreamedResponse
@@ -298,6 +301,7 @@ def __init__(
         self.graphs = graphs.AsyncGraphsResource(self)
         self.files = files.AsyncFilesResource(self)
         self.tools = tools.AsyncToolsResource(self)
+        self.translation = translation.AsyncTranslationResource(self)
         self.vision = vision.AsyncVisionResource(self)
         self.with_raw_response = AsyncWriterWithRawResponse(self)
         self.with_streaming_response = AsyncWriterWithStreamedResponse(self)
@@ -416,6 +420,7 @@ def __init__(self, client: Writer) -> None:
         self.graphs = graphs.GraphsResourceWithRawResponse(client.graphs)
         self.files = files.FilesResourceWithRawResponse(client.files)
         self.tools = tools.ToolsResourceWithRawResponse(client.tools)
+        self.translation = translation.TranslationResourceWithRawResponse(client.translation)
         self.vision = vision.VisionResourceWithRawResponse(client.vision)
 
 
@@ -428,6 +433,7 @@ def __init__(self, client: AsyncWriter) -> None:
         self.graphs = graphs.AsyncGraphsResourceWithRawResponse(client.graphs)
         self.files = files.AsyncFilesResourceWithRawResponse(client.files)
         self.tools = tools.AsyncToolsResourceWithRawResponse(client.tools)
+        self.translation = translation.AsyncTranslationResourceWithRawResponse(client.translation)
         self.vision = vision.AsyncVisionResourceWithRawResponse(client.vision)
 
 
@@ -440,6 +446,7 @@ def __init__(self, client: Writer) -> None:
         self.graphs = graphs.GraphsResourceWithStreamingResponse(client.graphs)
         self.files = files.FilesResourceWithStreamingResponse(client.files)
         self.tools = tools.ToolsResourceWithStreamingResponse(client.tools)
+        self.translation = translation.TranslationResourceWithStreamingResponse(client.translation)
         self.vision = vision.VisionResourceWithStreamingResponse(client.vision)
 
 
@@ -452,6 +459,7 @@ def __init__(self, client: AsyncWriter) -> None:
         self.graphs = graphs.AsyncGraphsResourceWithStreamingResponse(client.graphs)
         self.files = files.AsyncFilesResourceWithStreamingResponse(client.files)
         self.tools = tools.AsyncToolsResourceWithStreamingResponse(client.tools)
+        self.translation = translation.AsyncTranslationResourceWithStreamingResponse(client.translation)
         self.vision = vision.AsyncVisionResourceWithStreamingResponse(client.vision)
 
 

src/writerai/resources/__init__.py

@@ -56,6 +56,14 @@
     CompletionsResourceWithStreamingResponse,
     AsyncCompletionsResourceWithStreamingResponse,
 )
+from .translation import (
+    TranslationResource,
+    AsyncTranslationResource,
+    TranslationResourceWithRawResponse,
+    AsyncTranslationResourceWithRawResponse,
+    TranslationResourceWithStreamingResponse,
+    AsyncTranslationResourceWithStreamingResponse,
+)
 from .applications import (
     ApplicationsResource,
     AsyncApplicationsResource,
@@ -108,6 +116,12 @@
     "AsyncToolsResourceWithRawResponse",
     "ToolsResourceWithStreamingResponse",
     "AsyncToolsResourceWithStreamingResponse",
+    "TranslationResource",
+    "AsyncTranslationResource",
+    "TranslationResourceWithRawResponse",
+    "AsyncTranslationResourceWithRawResponse",
+    "TranslationResourceWithStreamingResponse",
+    "AsyncTranslationResourceWithStreamingResponse",
     "VisionResource",
     "AsyncVisionResource",
     "VisionResourceWithRawResponse",

src/writerai/resources/translation.py

@@ -0,0 +1,266 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal
+
+import httpx
+
+from ..types import translation_translate_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+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.translation_response import TranslationResponse
+
+__all__ = ["TranslationResource", "AsyncTranslationResource"]
+
+
+class TranslationResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> TranslationResourceWithRawResponse:
+        """
+        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/writer/writer-python#accessing-raw-response-data-eg-headers
+        """
+        return TranslationResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> TranslationResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/writer/writer-python#with_streaming_response
+        """
+        return TranslationResourceWithStreamingResponse(self)
+
+    def translate(
+        self,
+        *,
+        formality: bool,
+        length_control: bool,
+        mask_profanity: bool,
+        model: Literal["palmyra-translate"],
+        source_language_code: str,
+        target_language_code: str,
+        text: 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,
+    ) -> TranslationResponse:
+        """
+        Translate text from one language to another.
+
+        Args:
+          formality: Whether to use formal or informal language in the translation. See the
+              [list of languages that support formality](https://dev.writer.com/api-guides/api-reference/translation-api/language-support#formality).
+              If the language does not support formality, this parameter is ignored.
+
+          length_control: Whether to control the length of the translated text. See the
+              [list of languages that support length control](https://dev.writer.com/api-guides/api-reference/translation-api/language-support#length-control).
+              If the language does not support length control, this parameter is ignored.
+
+          mask_profanity: Whether to mask profane words in the translated text. See the
+              [list of languages that do not support profanity masking](https://dev.writer.com/api-guides/api-reference/translation-api/language-support#profanity-masking).
+              If the language does not support profanity masking, this parameter is ignored.
+
+          model: The model to use for translation.
+
+          source_language_code: The [ISO-639-1](https://en.wikipedia.org/wiki/List_of_ISO_639_language_codes)
+              language code of the original text to translate. For example, `en` for English,
+              `zh` for Chinese, `fr` for French, `es` for Spanish. If the language has a
+              variant, the code appends the two-digit
+              [ISO-3166 country code](https://en.wikipedia.org/wiki/List_of_ISO_3166_country_codes).
+              For example, Mexican Spanish is `es-MX`. See the
+              [list of supported languages and language codes](https://dev.writer.com/api-guides/api-reference/translation-api/language-support).
+
+          target_language_code: The [ISO-639-1](https://en.wikipedia.org/wiki/List_of_ISO_639_language_codes)
+              language code of the target language for the translation. For example, `en` for
+              English, `zh` for Chinese, `fr` for French, `es` for Spanish. If the language
+              has a variant, the code appends the two-digit
+              [ISO-3166 country code](https://en.wikipedia.org/wiki/List_of_ISO_3166_country_codes).
+              For example, Mexican Spanish is `es-MX`. See the
+              [list of supported languages and language codes](https://dev.writer.com/api-guides/api-reference/translation-api/language-support).
+
+          text: The text to translate. Maximum of 100,000 words.
+
+          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(
+            "/v1/translation",
+            body=maybe_transform(
+                {
+                    "formality": formality,
+                    "length_control": length_control,
+                    "mask_profanity": mask_profanity,
+                    "model": model,
+                    "source_language_code": source_language_code,
+                    "target_language_code": target_language_code,
+                    "text": text,
+                },
+                translation_translate_params.TranslationTranslateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=TranslationResponse,
+        )
+
+
+class AsyncTranslationResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncTranslationResourceWithRawResponse:
+        """
+        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/writer/writer-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncTranslationResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncTranslationResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/writer/writer-python#with_streaming_response
+        """
+        return AsyncTranslationResourceWithStreamingResponse(self)
+
+    async def translate(
+        self,
+        *,
+        formality: bool,
+        length_control: bool,
+        mask_profanity: bool,
+        model: Literal["palmyra-translate"],
+        source_language_code: str,
+        target_language_code: str,
+        text: 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,
+    ) -> TranslationResponse:
+        """
+        Translate text from one language to another.
+
+        Args:
+          formality: Whether to use formal or informal language in the translation. See the
+              [list of languages that support formality](https://dev.writer.com/api-guides/api-reference/translation-api/language-support#formality).
+              If the language does not support formality, this parameter is ignored.
+
+          length_control: Whether to control the length of the translated text. See the
+              [list of languages that support length control](https://dev.writer.com/api-guides/api-reference/translation-api/language-support#length-control).
+              If the language does not support length control, this parameter is ignored.
+
+          mask_profanity: Whether to mask profane words in the translated text. See the
+              [list of languages that do not support profanity masking](https://dev.writer.com/api-guides/api-reference/translation-api/language-support#profanity-masking).
+              If the language does not support profanity masking, this parameter is ignored.
+
+          model: The model to use for translation.
+
+          source_language_code: The [ISO-639-1](https://en.wikipedia.org/wiki/List_of_ISO_639_language_codes)
+              language code of the original text to translate. For example, `en` for English,
+              `zh` for Chinese, `fr` for French, `es` for Spanish. If the language has a
+              variant, the code appends the two-digit
+              [ISO-3166 country code](https://en.wikipedia.org/wiki/List_of_ISO_3166_country_codes).
+              For example, Mexican Spanish is `es-MX`. See the
+              [list of supported languages and language codes](https://dev.writer.com/api-guides/api-reference/translation-api/language-support).
+
+          target_language_code: The [ISO-639-1](https://en.wikipedia.org/wiki/List_of_ISO_639_language_codes)
+              language code of the target language for the translation. For example, `en` for
+              English, `zh` for Chinese, `fr` for French, `es` for Spanish. If the language
+              has a variant, the code appends the two-digit
+              [ISO-3166 country code](https://en.wikipedia.org/wiki/List_of_ISO_3166_country_codes).
+              For example, Mexican Spanish is `es-MX`. See the
+              [list of supported languages and language codes](https://dev.writer.com/api-guides/api-reference/translation-api/language-support).
+
+          text: The text to translate. Maximum of 100,000 words.
+
+          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(
+            "/v1/translation",
+            body=await async_maybe_transform(
+                {
+                    "formality": formality,
+                    "length_control": length_control,
+                    "mask_profanity": mask_profanity,
+                    "model": model,
+                    "source_language_code": source_language_code,
+                    "target_language_code": target_language_code,
+                    "text": text,
+                },
+                translation_translate_params.TranslationTranslateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=TranslationResponse,
+        )
+
+
+class TranslationResourceWithRawResponse:
+    def __init__(self, translation: TranslationResource) -> None:
+        self._translation = translation
+
+        self.translate = to_raw_response_wrapper(
+            translation.translate,
+        )
+
+
+class AsyncTranslationResourceWithRawResponse:
+    def __init__(self, translation: AsyncTranslationResource) -> None:
+        self._translation = translation
+
+        self.translate = async_to_raw_response_wrapper(
+            translation.translate,
+        )
+
+
+class TranslationResourceWithStreamingResponse:
+    def __init__(self, translation: TranslationResource) -> None:
+        self._translation = translation
+
+        self.translate = to_streamed_response_wrapper(
+            translation.translate,
+        )
+
+
+class AsyncTranslationResourceWithStreamingResponse:
+    def __init__(self, translation: AsyncTranslationResource) -> None:
+        self._translation = translation
+
+        self.translate = async_to_streamed_response_wrapper(
+            translation.translate,
+        )