together 1.1.5__py3-none-any.whl → 1.2.1__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/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/types/completions.py

@@ -61,7 +61,7 @@ class CompletionRequest(BaseModel):
 class CompletionChoicesData(BaseModel):
     index: int
     logprobs: LogprobsPart | None = None
-    finish_reason: FinishReason
+    finish_reason: FinishReason | None = None
     text: str
 
 

together/types/finetune.py

@@ -24,6 +24,7 @@ class FinetuneJobStatus(str, Enum):
     STATUS_CANCEL_REQUESTED = "cancel_requested"
     STATUS_CANCELLED = "cancelled"
     STATUS_ERROR = "error"
+    STATUS_USER_ERROR = "user_error"
     STATUS_COMPLETED = "completed"
 
 
@@ -67,6 +68,7 @@ class FinetuneEventType(str, Enum):
     MODEL_UPLOAD_COMPLETE = "MODEL_UPLOAD_COMPLETE"
     JOB_COMPLETE = "JOB_COMPLETE"
     JOB_ERROR = "JOB_ERROR"
+    JOB_USER_ERROR = "JOB_USER_ERROR"
     CANCEL_REQUESTED = "CANCEL_REQUESTED"
     JOB_RESTARTED = "JOB_RESTARTED"
     REFUND = "REFUND"

{together-1.1.5.dist-info → together-1.2.1.dist-info}/METADATA

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

{together-1.1.5.dist-info → together-1.2.1.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/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
@@ -36,11 +36,11 @@ together/types/__init__.py,sha256=K7Gv6hLmobIfqfmijZbZwFrwxK_YKIDR_v94_ElFmVA,14
 together/types/abstract.py,sha256=1lFQI_3WjsR_t1128AeKW0aTk6EiM6Gh1J3ZuyLLPao,642
 together/types/chat_completions.py,sha256=VxuvcHM9IQVWfD1qKu6gDWdrV5UainjkV3pGm_4Oiyo,4335
 together/types/common.py,sha256=0evjGduXV_tStd0TkGNwvU4fDjfLGUOuUJz6Vl5KYbs,1491
-together/types/completions.py,sha256=qOvTqoxc5ZLGFgukVm9-Vx9UKNnBQZGuFIz7HCGHAoQ,2887
+together/types/completions.py,sha256=yydloTQePGaY97Lx-kbkvgCqBFhHFl7jU5s7uf9Ncg0,2901
 together/types/embeddings.py,sha256=J7grkYYn7xhqeKaBO2T-8XQRtHhkzYzymovtGdIUK5A,751
 together/types/error.py,sha256=OVlCs3cx_2WhZK4JzHT8SQyRIIqKOP1AZQ4y1PydjAE,370
 together/types/files.py,sha256=-rEUfsV6f2vZB9NrFxT4_933ubsDIUNkPB-3OlOFk4A,1954
-together/types/finetune.py,sha256=C5z5iaH3E_wBiW1USLfUzv3trJxCf36hYh3l55pUaME,5775
+together/types/finetune.py,sha256=eK2Zv47ccvbenGv87PmNCBsqPQleiA3KxoK7KjsuYTM,5850
 together/types/images.py,sha256=zX4Vt38tFDKU6yGb_hBY_N5eSTn3KPdpP5Ce_qnRHXQ,915
 together/types/models.py,sha256=3zag9x2fus2McNmLkr3fKzQL-2RNIT1-tiabI-z21p8,1013
 together/utils/__init__.py,sha256=VpjeRTya1m5eEE-Qe1zYTFsNAvuEA-dy7M2eG9Xu4fc,662
@@ -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.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-together-1.1.5.dist-info/METADATA,sha256=cByMGtmLlDdbVgw5Pl_xEvi3cSyMDzE1YC3ZMqj0Qu0,11812
-together-1.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-together-1.1.5.dist-info/entry_points.txt,sha256=G-b5NKW6lUUf1V1fH8IPTBb7jXnK7lhbX9H1zTEJXPs,50
-together-1.1.5.dist-info/RECORD,,
+together-1.2.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+together-1.2.1.dist-info/METADATA,sha256=61lkQQva68_obMLTd0sLYhgHi7hWJnCxBkEJA9_eCS4,11812
+together-1.2.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+together-1.2.1.dist-info/entry_points.txt,sha256=G-b5NKW6lUUf1V1fH8IPTBb7jXnK7lhbX9H1zTEJXPs,50
+together-1.2.1.dist-info/RECORD,,