together 1.1.4__py3-none-any.whl → 1.2.0__py3-none-any.whl

together/cli/api/files.py

@@ -32,12 +32,17 @@ def files(ctx: click.Context) -> None:
     default=FilePurpose.FineTune.value,
     help="Purpose of file upload. Acceptable values in enum `together.types.FilePurpose`. Defaults to `fine-tunes`.",
 )
-def upload(ctx: click.Context, file: pathlib.Path, purpose: str) -> None:
+@click.option(
+    "--check/--no-check",
+    default=True,
+    help="Whether to check the file before uploading.",
+)
+def upload(ctx: click.Context, file: pathlib.Path, purpose: str, check: bool) -> None:
     """Upload file"""
 
     client: Together = ctx.obj
 
-    response = client.files.upload(file=file, purpose=purpose)
+    response = client.files.upload(file=file, purpose=purpose, check=check)
 
     click.echo(json.dumps(response.model_dump(), indent=4))
 

together/error.py

@@ -23,9 +23,12 @@ class TogetherException(Exception):
             if isinstance(message, TogetherErrorResponse)
             else message
         )
-        self._message = f"Error code: {http_status} - {_message}"
+        if http_status is not None:
+            self._message = f"Error code: {http_status} - {_message}"
+        else:
+            self._message = str(_message)
 
-        super(TogetherException, self).__init__(self._message)
+        super().__init__(self._message)
 
         self.http_status = http_status
         self.headers = headers or {}

together/filemanager.py

@@ -103,13 +103,8 @@ def _prepare_output(
     if "x-tar" in content_type.lower():
         remote_name += ".tar.gz"
 
-    elif "zstd" in content_type.lower() or step != -1:
-        remote_name += ".tar.zst"
-
     else:
-        raise FileTypeError(
-            f"Unknown file type {content_type} found. Aborting download."
-        )
+        remote_name += ".tar.zst"
 
     return Path(remote_name)
 

together/legacy/files.py

@@ -53,9 +53,11 @@ class Files:
 
         client = together.Together(api_key=api_key)
 
-        response = client.files.upload(file=file).model_dump()
+        # disabling the check, because it was run previously
+        response = client.files.upload(file=file, check=False).model_dump()
 
-        response["report_dict"] = report_dict
+        if check:
+            response["report_dict"] = report_dict
 
         return response
 

together/resources/files.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
 from pathlib import Path
+from pprint import pformat
 
 from together.abstract import api_requestor
+from together.error import FileTypeError
 from together.filemanager import DownloadManager, UploadManager
 from together.together_response import TogetherResponse
 from together.types import (
@@ -14,7 +16,7 @@ from together.types import (
     TogetherClient,
     TogetherRequest,
 )
-from together.utils import normalize_key
+from together.utils import check_file, normalize_key
 
 
 class Files:
@@ -22,10 +24,21 @@ class Files:
         self._client = client
 
     def upload(
-        self, file: Path | str, *, purpose: FilePurpose | str = FilePurpose.FineTune
+        self,
+        file: Path | str,
+        *,
+        purpose: FilePurpose | str = FilePurpose.FineTune,
+        check: bool = True,
     ) -> FileResponse:
         upload_manager = UploadManager(self._client)
 
+        if check:
+            report_dict = check_file(file)
+            if not report_dict["is_check_passed"]:
+                raise FileTypeError(
+                    f"Invalid file supplied, failed to upload. Report:\n{pformat(report_dict)}"
+                )
+
         if isinstance(file, str):
             file = Path(file)
 

{together-1.1.4.dist-info → together-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: together
-Version: 1.1.4
+Version: 1.2.0
 Summary: Python client for Together's Cloud Platform!
 Home-page: https://github.com/togethercomputer/together-python
 License: Apache-2.0

{together-1.1.4.dist-info → together-1.2.0.dist-info}/RECORD

@@ -5,20 +5,20 @@ together/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 together/cli/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 together/cli/api/chat.py,sha256=qmDcYohGe1ODfvYEDN6Z2eZO3qODgmeUTll8P5qoKYQ,9170
 together/cli/api/completions.py,sha256=v6KvGsXnPCdSM7WIu_QmKXOA4iVjW93k3dLwfIBkxPQ,4199
-together/cli/api/files.py,sha256=1iOZpCDC-R2s7sOxWe539DU9KKJcrZGayeeksQLutRk,3186
+together/cli/api/files.py,sha256=hOiN1ftwbSAs3gcVqwmPymhmzDYiIUrFWXFHVcWWom8,3329
 together/cli/api/finetune.py,sha256=mSQSasmVQLtNzjJLeoEEwvewfEcpSkjKyozWqwb-7p0,5417
 together/cli/api/images.py,sha256=01dFYa2sK1HqUwVCD9FlwcjqkYWLoNxFZkzok13EriE,2363
 together/cli/api/models.py,sha256=xWEzu8ZpxM_Pz9KEjRPRVuv_v22RayYZ4QcgiezT5tE,1126
 together/cli/cli.py,sha256=RC0tgapkSOFjsRPg8p-8dx9D2LDzm8YmVCHUjk_aVyQ,1977
 together/client.py,sha256=7QT5lwn7-QGf0vtgbhslQujm4986CgfE2spyMYP4JyU,4774
 together/constants.py,sha256=WHe6JA9TliwgErkCnovWPS9w9xXfA3X5PtKJv_y2JxQ,908
-together/error.py,sha256=cILzwDde18INNFYgbYdgvVfOhEjRZM9sg9I8Rl6nc_Y,5329
-together/filemanager.py,sha256=tBr9LOqiv6G63rH7HYePKeKhyylJXfZFHW5Aybk8UUw,11438
+together/error.py,sha256=IWomy3KpDyuWS7uK4US2a4CkG4HC3oQEGQvVtul4nM8,5402
+together/filemanager.py,sha256=QHhBn73oVFdgUpSYXYLmJzHJ9c5wYEMJC0ur6ZgDeYo,11269
 together/legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 together/legacy/base.py,sha256=ehrX1SCfRbK5OA83wL1q7-tfF-yuZOUxzjxYfFtdvvQ,727
 together/legacy/complete.py,sha256=8itEMlyuora2EIO-b-tvjvpHfKhxckDEiqaN-z6ccQs,2343
 together/legacy/embeddings.py,sha256=Sm4XNVBm00JUNQT65hlZE5tIwsKar2PdmK0_lZ4dw_4,591
-together/legacy/files.py,sha256=Xi5hUnfiZX2IqHkoKLjG3PUWrVuVYDcNCvH1FmBbrmo,3863
+together/legacy/files.py,sha256=TFIJZvHlAY-SxQ4PJuCr-A4uRPL2znDhsbSTaTklCD0,3959
 together/legacy/finetune.py,sha256=Uujbmo_wdCko84mZbDTXMkNBySgMUvVXtULb0w96rDI,4917
 together/legacy/images.py,sha256=An5dmwU6Mo8mU-CJmI51J9EhOnYwcG9YXF8UgDcIm8E,582
 together/legacy/models.py,sha256=EEyRNw7AuMoKNXm1nWfT-k2wsAFq213WoyDVGRjwJWs,1053
@@ -27,7 +27,7 @@ together/resources/chat/__init__.py,sha256=RsTptdP8MeGjcdIjze896-J27cRvCbUoMft0X
 together/resources/chat/completions.py,sha256=QGrg0fGsn16k6jpxbQGoIBRivHWgEimPWAEVdKSAVGk,14084
 together/resources/completions.py,sha256=5s-jly2vbpI1rWhFwVijCmCOyhEZuDuwJblrm3HD0Gs,11353
 together/resources/embeddings.py,sha256=yRG0JaF4u_MEynQ4TBgCOVf9bKSvGI1fRaQx3CHUiB0,2546
-together/resources/files.py,sha256=_9mM5e49mMs4BCwW07-dnoxagvPPFhq1oUnsMsLMIYA,4593
+together/resources/files.py,sha256=bnPbaF25e4InBRPvHwXHXT-oSX1Z1sZRsnQW5wq82U4,4990
 together/resources/finetune.py,sha256=DcA0A3I9khS8nMv2KpVNjDVI92knnI2o-O_93Z8gP-U,12416
 together/resources/images.py,sha256=gFzXy7gLzr20KsXJXHEsOZtJJpuR6pH1d8XHCd6Dl9E,4775
 together/resources/models.py,sha256=2dtHhXAqTDOOpwSbYLzWcKTC0-m2Szlb7LDYvp7Jr4w,1786
@@ -49,8 +49,8 @@ together/utils/api_helpers.py,sha256=RSF7SRhbjHzroMOSWAXscflByM1r1ta_1SpxkAT22iE
 together/utils/files.py,sha256=gMLthqfP5hKxVAerHMdy7gLXzdfY6lyOXdpW24Y4X3I,7165
 together/utils/tools.py,sha256=3-lXWP3cBCzOVSZg9tr5zOT1jaVeKAKVWxO2fcXZTh8,1788
 together/version.py,sha256=p03ivHyE0SyWU4jAnRTBi_sOwywVWoZPU4g2gzRgG-Y,126
-together-1.1.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-together-1.1.4.dist-info/METADATA,sha256=t74AKUbStH9MAjck5ZIr55jsZOiPr7rQOyjwEzXKvIU,11812
-together-1.1.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-together-1.1.4.dist-info/entry_points.txt,sha256=G-b5NKW6lUUf1V1fH8IPTBb7jXnK7lhbX9H1zTEJXPs,50
-together-1.1.4.dist-info/RECORD,,
+together-1.2.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+together-1.2.0.dist-info/METADATA,sha256=stIBXOH6i0IqAnq1dB8CNdO40Edf4WVNSEUqiaMfVYM,11812
+together-1.2.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+together-1.2.0.dist-info/entry_points.txt,sha256=G-b5NKW6lUUf1V1fH8IPTBb7jXnK7lhbX9H1zTEJXPs,50
+together-1.2.0.dist-info/RECORD,,