chore: Minimize breaking changes on common import paths and alias names (#206)

Author: blainekasten

src/together/constants.py

@@ -0,0 +1,34 @@
+# Manually added to minimize breaking changes from V1
+from ._constants import (
+    MAX_RETRY_DELAY as MAX_RETRY_DELAY,
+    DEFAULT_MAX_RETRIES,
+    INITIAL_RETRY_DELAY as INITIAL_RETRY_DELAY,
+)
+from .lib.constants import (
+    MIN_SAMPLES as MIN_SAMPLES,
+    DISABLE_TQDM as DISABLE_TQDM,
+    MAX_IMAGE_BYTES as MAX_IMAGE_BYTES,
+    NUM_BYTES_IN_GB as NUM_BYTES_IN_GB,
+    MAX_FILE_SIZE_GB as MAX_FILE_SIZE_GB,
+    MIN_PART_SIZE_MB as MIN_PART_SIZE_MB,
+    DOWNLOAD_BLOCK_SIZE as DOWNLOAD_BLOCK_SIZE,
+    MAX_MULTIPART_PARTS as MAX_MULTIPART_PARTS,
+    TARGET_PART_SIZE_MB as TARGET_PART_SIZE_MB,
+    MAX_CONCURRENT_PARTS as MAX_CONCURRENT_PARTS,
+    MAX_IMAGES_PER_EXAMPLE as MAX_IMAGES_PER_EXAMPLE,
+    MULTIPART_THRESHOLD_GB as MULTIPART_THRESHOLD_GB,
+    MAX_BASE64_IMAGE_LENGTH as MAX_BASE64_IMAGE_LENGTH,
+    MULTIPART_UPLOAD_TIMEOUT as MULTIPART_UPLOAD_TIMEOUT,
+    PARQUET_EXPECTED_COLUMNS as PARQUET_EXPECTED_COLUMNS,
+    REQUIRED_COLUMNS_MESSAGE as REQUIRED_COLUMNS_MESSAGE,
+    JSONL_REQUIRED_COLUMNS_MAP as JSONL_REQUIRED_COLUMNS_MAP,
+    POSSIBLE_ROLES_CONVERSATION as POSSIBLE_ROLES_CONVERSATION,
+    DatasetFormat as DatasetFormat,
+)
+
+TIMEOUT_SECS = 600
+MAX_SESSION_LIFETIME_SECS = 180
+MAX_CONNECTION_RETRIES = 2
+MAX_RETRIES = DEFAULT_MAX_RETRIES
+
+BASE_URL = "https://api.together.xyz/v1"

src/together/error.py

@@ -0,0 +1,16 @@
+
+# Manually added to minimize breaking changes from V1
+from ._exceptions import (
+    APIError as APIError,
+    RateLimitError as RateLimitError,
+    APITimeoutError,
+    BadRequestError,
+    APIConnectionError as APIConnectionError,
+    AuthenticationError as AuthenticationError,
+    APIResponseValidationError,
+)
+
+Timeout = APITimeoutError
+InvalidRequestError = BadRequestError
+TogetherException = APIError
+ResponseError = APIResponseValidationError

src/together/types/__init__.py

@@ -1,6 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from __future__ import annotations
+from __future__ import annotations # noqa
 
 from .batch_job import BatchJob as BatchJob
 from .embedding import Embedding as Embedding
@@ -75,3 +75,14 @@
 from .fine_tuning_list_checkpoints_response import (
     FineTuningListCheckpointsResponse as FineTuningListCheckpointsResponse,
 )
+
+# Manually added to minimize breaking changes from V1
+from .chat.chat_completion import ChatCompletion 
+from .chat.chat_completion_chunk import ChatCompletionChunk as ChatCompletionChunk
+from .chat.chat_completion_usage import ChatCompletionUsage
+UsageData = ChatCompletionUsage
+ChatCompletionResponse = ChatCompletion
+CompletionResponse = Completion
+ListEndpoint = EndpointListResponse
+ImageRequest = ImageGenerateParams
+ImageResponse = ImageFile

src/together/types/chat_completions.py

@@ -0,0 +1,7 @@
+# Manually added to minimize breaking changes from V1
+from .tool_choice import ToolChoice
+from .chat.chat_completion import ChatCompletion
+from .chat.chat_completion_chunk import ChatCompletionChunk as ChatCompletionChunk
+
+ChatCompletionResponse = ChatCompletion
+ToolCalls = ToolChoice

src/together/types/endpoints.py

@@ -0,0 +1,4 @@
+# Manually added to minimize breaking changes from V1
+from together.types import DedicatedEndpoint as DedicatedEndpoint
+
+ListEndpoint = DedicatedEndpoint

src/together/types/files.py

@@ -0,0 +1,8 @@
+# Manually added to minimize breaking changes from V1
+from . import (
+    FileList as FileList,
+    FileType as FileType,
+    FilePurpose as FilePurpose,
+    FileResponse as FileResponse,
+    FileDeleteResponse as FileDeleteResponse,
+)

src/together/types/finetune.py

@@ -0,0 +1,27 @@
+# Manually added to minimize breaking changes from V1
+from ..lib.types.fine_tuning import (
+    COMPLETED_STATUSES as COMPLETED_STATUSES,
+    TrainingType as TrainingType,
+    FinetuneEvent as FinetuneEvent,
+    TrainingMethod as TrainingMethod,
+    FinetuneRequest as FinetuneRequest,
+    EmptyLRScheduler as EmptyLRScheduler,
+    FinetuneProgress as FinetuneProgress,
+    FinetuneResponse as FinetuneResponse,
+    FullTrainingType as FullTrainingType,
+    LoRATrainingType as LoRATrainingType,
+    CosineLRScheduler as CosineLRScheduler,
+    FinetuneEventType as FinetuneEventType,
+    FinetuneJobStatus as FinetuneJobStatus,
+    LinearLRScheduler as LinearLRScheduler,
+    TrainingMethodDPO as TrainingMethodDPO,
+    TrainingMethodSFT as TrainingMethodSFT,
+    FinetuneEventLevels as FinetuneEventLevels,
+    FinetuneLRScheduler as FinetuneLRScheduler,
+    CosineLRSchedulerArgs as CosineLRSchedulerArgs,
+    LinearLRSchedulerArgs as LinearLRSchedulerArgs,
+    FinetuneTrainingLimits as FinetuneTrainingLimits,
+    FinetuneMultimodalParams as FinetuneMultimodalParams,
+    FinetuneFullTrainingLimits as FinetuneFullTrainingLimits,
+    FinetuneLoraTrainingLimits as FinetuneLoraTrainingLimits,
+)

src/together/types/models.py

@@ -0,0 +1,2 @@
+# Manually added to minimize breaking changes from V1
+from together.types import ModelObject as ModelObject