together 1.5.34__py3-none-any.whl → 2.0.0a6__py3-none-any.whl

together/client.py

@@ -1,186 +0,0 @@
-from __future__ import annotations
-
-import os
-import sys
-from typing import Dict, TYPE_CHECKING
-
-from together import resources
-from together.constants import BASE_URL, MAX_RETRIES, TIMEOUT_SECS
-from together.error import AuthenticationError
-from together.resources.code_interpreter import CodeInterpreter
-from together.types import TogetherClient
-from together.utils import enforce_trailing_slash
-from together.utils.api_helpers import get_google_colab_secret
-
-
-class Together:
-    completions: resources.Completions
-    chat: resources.Chat
-    embeddings: resources.Embeddings
-    files: resources.Files
-    images: resources.Images
-    models: resources.Models
-    fine_tuning: resources.FineTuning
-    rerank: resources.Rerank
-    audio: resources.Audio
-    batches: resources.Batches
-    code_interpreter: CodeInterpreter
-    evaluation: resources.Evaluation
-    videos: resources.Videos
-
-    # client options
-    client: TogetherClient
-
-    def __init__(
-        self,
-        *,
-        api_key: str | None = None,
-        base_url: str | None = None,
-        timeout: float | None = None,
-        max_retries: int | None = None,
-        supplied_headers: Dict[str, str] | None = None,
-    ) -> None:
-        """Construct a new synchronous together client instance.
-
-        This automatically infers the following arguments from their corresponding environment variables if they are not provided:
-        - `api_key` from `TOGETHER_API_KEY`
-        - `base_url` from `TOGETHER_BASE_URL`
-        """
-
-        # get api key
-        if not api_key:
-            api_key = os.environ.get("TOGETHER_API_KEY")
-
-        if not api_key and "google.colab" in sys.modules:
-            api_key = get_google_colab_secret("TOGETHER_API_KEY")
-
-        if not api_key:
-            raise AuthenticationError(
-                "The api_key client option must be set either by passing api_key to the client or by setting the "
-                "TOGETHER_API_KEY environment variable"
-            )
-
-        # get base url
-        if not base_url:
-            base_url = os.environ.get("TOGETHER_BASE_URL")
-
-        if not base_url:
-            base_url = BASE_URL
-
-        if timeout is None:
-            timeout = TIMEOUT_SECS
-
-        if max_retries is None:
-            max_retries = MAX_RETRIES
-
-        # TogetherClient object
-        self.client = TogetherClient(
-            api_key=api_key,
-            base_url=enforce_trailing_slash(base_url),
-            timeout=timeout,
-            max_retries=max_retries,
-            supplied_headers=supplied_headers,
-        )
-
-        self.completions = resources.Completions(self.client)
-        self.chat = resources.Chat(self.client)
-        self.embeddings = resources.Embeddings(self.client)
-        self.files = resources.Files(self.client)
-        self.images = resources.Images(self.client)
-        self.models = resources.Models(self.client)
-        self.fine_tuning = resources.FineTuning(self.client)
-        self.rerank = resources.Rerank(self.client)
-        self.audio = resources.Audio(self.client)
-        self.endpoints = resources.Endpoints(self.client)
-        self.code_interpreter = CodeInterpreter(self.client)
-        self.batches = resources.Batches(self.client)
-        self.evaluation = resources.Evaluation(self.client)
-        self.videos = resources.Videos(self.client)
-
-
-class AsyncTogether:
-    completions: resources.AsyncCompletions
-    chat: resources.AsyncChat
-    embeddings: resources.AsyncEmbeddings
-    files: resources.AsyncFiles
-    images: resources.AsyncImages
-    models: resources.AsyncModels
-    fine_tuning: resources.AsyncFineTuning
-    rerank: resources.AsyncRerank
-    audio: resources.AsyncAudio
-    code_interpreter: CodeInterpreter
-    batches: resources.AsyncBatches
-    evaluation: resources.AsyncEvaluation
-    videos: resources.AsyncVideos
-    # client options
-    client: TogetherClient
-
-    def __init__(
-        self,
-        *,
-        api_key: str | None = None,
-        base_url: str | None = None,
-        timeout: float | None = None,
-        max_retries: int | None = None,
-        supplied_headers: Dict[str, str] | None = None,
-    ) -> None:
-        """Construct a new async together client instance.
-
-        This automatically infers the following arguments from their corresponding environment variables if they are not provided:
-        - `api_key` from `TOGETHER_API_KEY`
-        - `base_url` from `TOGETHER_BASE_URL`
-        """
-
-        # get api key
-        if not api_key:
-            api_key = os.environ.get("TOGETHER_API_KEY")
-
-        if not api_key and "google.colab" in sys.modules:
-            api_key = get_google_colab_secret("TOGETHER_API_KEY")
-
-        if not api_key:
-            raise AuthenticationError(
-                "The api_key client option must be set either by passing api_key to the client or by setting the "
-                "TOGETHER_API_KEY environment variable"
-            )
-
-        # get base url
-        if not base_url:
-            base_url = os.environ.get("TOGETHER_BASE_URL")
-
-        if not base_url:
-            base_url = BASE_URL
-
-        if timeout is None:
-            timeout = TIMEOUT_SECS
-
-        if max_retries is None:
-            max_retries = MAX_RETRIES
-
-        # TogetherClient object
-        self.client = TogetherClient(
-            api_key=api_key,
-            base_url=enforce_trailing_slash(base_url),
-            timeout=timeout,
-            max_retries=max_retries,
-            supplied_headers=supplied_headers,
-        )
-
-        self.completions = resources.AsyncCompletions(self.client)
-        self.chat = resources.AsyncChat(self.client)
-        self.embeddings = resources.AsyncEmbeddings(self.client)
-        self.files = resources.AsyncFiles(self.client)
-        self.images = resources.AsyncImages(self.client)
-        self.models = resources.AsyncModels(self.client)
-        self.fine_tuning = resources.AsyncFineTuning(self.client)
-        self.rerank = resources.AsyncRerank(self.client)
-        self.audio = resources.AsyncAudio(self.client)
-        self.code_interpreter = CodeInterpreter(self.client)
-        self.batches = resources.AsyncBatches(self.client)
-        self.evaluation = resources.AsyncEvaluation(self.client)
-        self.videos = resources.AsyncVideos(self.client)
-
-
-Client = Together
-
-AsyncClient = AsyncTogether

together/error.py

@@ -1,194 +0,0 @@
-from __future__ import annotations
-
-import json
-from typing import Any, Dict
-
-from requests import RequestException
-
-from together.types.error import TogetherErrorResponse
-
-
-class TogetherException(Exception):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        headers: str | Dict[Any, Any] | None = None,
-        request_id: str | None = None,
-        http_status: int | None = None,
-    ) -> None:
-        if isinstance(message, TogetherErrorResponse):
-            self.api_response = message
-
-        _message = (
-            json.dumps(message.model_dump(exclude_none=True))
-            if isinstance(message, TogetherErrorResponse)
-            else message
-        )
-        if http_status is not None:
-            self._message = f"Error code: {http_status} - {_message}"
-        else:
-            self._message = str(_message)
-
-        super().__init__(self._message)
-
-        self.http_status = http_status
-        self.headers = headers or {}
-        self.request_id = request_id
-
-    def __repr__(self) -> str:
-        repr_message = json.dumps(
-            {
-                "response": self._message,
-                "status": self.http_status,
-                "request_id": self.request_id,
-                "headers": self.headers,
-            }
-        )
-        return "%s(%r)" % (self.__class__.__name__, repr_message)
-
-
-class AuthenticationError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class ResponseError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class JSONError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class InstanceError(TogetherException):
-    def __init__(self, model: str | None = "model", **kwargs: Any) -> None:
-        super().__init__(**kwargs)
-        self.message = f"""No running instances for {model}.
-                You can start an instance with one of the following methods:
-                  1. navigating to the Together Playground at api.together.ai
-                  2. starting one in python using together.Models.start(model_name)
-                  3. `$ together models start <MODEL_NAME>` at the command line.
-                See `together.Models.list()` in python or `$ together models list` in command line
-                to get an updated list of valid model names.
-                """
-
-
-class RateLimitError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class FileTypeError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class AttributeError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class Timeout(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class APIConnectionError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class InvalidRequestError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class APIError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class ServiceUnavailableError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)
-
-
-class DownloadError(TogetherException):
-    def __init__(
-        self,
-        message: (
-            TogetherErrorResponse | Exception | str | RequestException | None
-        ) = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(message=message, **kwargs)