together 1.5.35__py3-none-any.whl → 2.0.0a7__py3-none-any.whl

together/legacy/complete.py

@@ -1,93 +0,0 @@
-from __future__ import annotations
-
-import warnings
-from typing import Any, AsyncGenerator, Dict, Iterator
-
-import together
-from together.legacy.base import API_KEY_WARNING, deprecated
-from together.types import CompletionChunk, CompletionResponse
-
-
-class Complete:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        prompt: str,
-        **kwargs: Any,
-    ) -> Dict[str, Any]:
-        """Legacy completion function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        result = client.completions.create(prompt=prompt, stream=False, **kwargs)
-
-        assert isinstance(result, CompletionResponse)
-
-        return result.model_dump(exclude_none=True)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def create_streaming(
-        cls,
-        prompt: str,
-        **kwargs: Any,
-    ) -> Iterator[Dict[str, Any]]:
-        """Legacy streaming completion function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return (
-            token.model_dump(exclude_none=True)  # type: ignore
-            for token in client.completions.create(prompt=prompt, stream=True, **kwargs)
-        )
-
-
-class Completion:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        prompt: str,
-        **kwargs: Any,
-    ) -> CompletionResponse | Iterator[CompletionChunk]:
-        """Completion function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.completions.create(prompt=prompt, **kwargs)
-
-
-class AsyncComplete:
-    @classmethod
-    @deprecated  # type: ignore
-    async def create(
-        cls,
-        prompt: str,
-        **kwargs: Any,
-    ) -> CompletionResponse | AsyncGenerator[CompletionChunk, None]:
-        """Async completion function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.AsyncTogether(api_key=api_key)
-
-        return await client.completions.create(prompt=prompt, **kwargs)

together/legacy/embeddings.py

@@ -1,27 +0,0 @@
-import warnings
-from typing import Any, Dict
-
-import together
-from together.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Embeddings:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        input: str,
-        **kwargs: Any,
-    ) -> Dict[str, Any]:
-        """Legacy embeddings function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.embeddings.create(input=input, **kwargs).model_dump(
-            exclude_none=True
-        )

together/legacy/files.py

@@ -1,146 +0,0 @@
-from __future__ import annotations
-
-import json
-import warnings
-from typing import Any, Dict, List
-
-import together
-from together.legacy.base import API_KEY_WARNING, deprecated
-from together.utils.files import check_file as check_json
-
-
-class Files:
-    @classmethod
-    @deprecated  # type: ignore
-    def list(
-        cls,
-    ) -> Dict[str, Any]:
-        """Legacy file list function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.files.list().model_dump(exclude_none=True)
-
-    @classmethod
-    def check(self, file: str) -> Dict[str, object]:
-        return check_json(file)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def upload(
-        cls,
-        file: str,
-        check: bool = True,
-    ) -> Dict[str, Any]:
-        """Legacy file upload function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        if check:
-            report_dict = check_json(file)
-            if not report_dict["is_check_passed"]:
-                raise together.error.FileTypeError(
-                    f"Invalid file supplied. Failed to upload.\nReport:\n {report_dict}"
-                )
-
-        client = together.Together(api_key=api_key)
-
-        # disabling the check, because it was run previously
-        response = client.files.upload(file=file, check=False).model_dump(
-            exclude_none=True
-        )
-
-        if check:
-            response["report_dict"] = report_dict
-
-        return response
-
-    @classmethod
-    @deprecated  # type: ignore
-    def delete(
-        cls,
-        file_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy file delete function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.files.delete(id=file_id).model_dump(exclude_none=True)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def retrieve(
-        cls,
-        file_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy file retrieve function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.files.retrieve(id=file_id).model_dump(exclude_none=True)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def retrieve_content(
-        cls,
-        file_id: str,
-        output: str | None = None,
-    ) -> Dict[str, Any]:
-        """Legacy file retrieve content function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.files.retrieve_content(id=file_id, output=output).dict(
-            exclude_none=True
-        )
-
-    @classmethod
-    @deprecated  # type: ignore
-    def save_jsonl(
-        self, data: Dict[str, str], output_path: str, append: bool = False
-    ) -> None:
-        """
-        Write list of objects to a JSON lines file.
-        """
-        mode = "a+" if append else "w"
-        with open(output_path, mode, encoding="utf-8") as f:
-            for line in data:
-                json_record = json.dumps(line, ensure_ascii=False)
-                f.write(json_record + "\n")
-        print("Wrote {} records to {}".format(len(data), output_path))
-
-    @classmethod
-    @deprecated  # type: ignore
-    def load_jsonl(self, input_path: str) -> List[Dict[str, str]]:
-        """
-        Read list of objects from a JSON lines file.
-        """
-        data = []
-        with open(input_path, "r", encoding="utf-8") as f:
-            for line in f:
-                data.append(json.loads(line.rstrip("\n|\r")))
-        print("Loaded {} records from {}".format(len(data), input_path))
-        return data

together/legacy/finetune.py

@@ -1,177 +0,0 @@
-from __future__ import annotations
-
-import warnings
-from typing import Any, Dict, List, Literal
-
-import together
-from together.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Finetune:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        training_file: str,  # training file_id
-        model: str,
-        n_epochs: int = 1,
-        n_checkpoints: int | None = 1,
-        batch_size: int | Literal["max"] = "max",
-        learning_rate: float = 0.00001,
-        suffix: (
-            str | None
-        ) = None,  # resulting finetuned model name will include the suffix
-        estimate_price: bool = False,
-        wandb_api_key: str | None = None,
-        confirm_inputs: bool = False,
-    ):
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        if estimate_price:
-            raise ValueError("Price estimation is not supported in version >= 1.0.1")
-
-        if confirm_inputs:
-            raise ValueError("Input confirmation is not supported in version >= 1.0.1")
-
-        client = together.Together(api_key=api_key)
-
-        return client.fine_tuning.create(
-            training_file=training_file,
-            model=model,
-            n_epochs=n_epochs,
-            n_checkpoints=n_checkpoints,
-            batch_size=batch_size,
-            learning_rate=learning_rate,
-            suffix=suffix,
-            wandb_api_key=wandb_api_key,
-        ).model_dump(exclude_none=True)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def list(
-        cls,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning list function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.fine_tuning.list().model_dump(exclude_none=True)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def retrieve(
-        cls,
-        fine_tune_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning retrieve function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.fine_tuning.retrieve(id=fine_tune_id).model_dump(
-            exclude_none=True
-        )
-
-    @classmethod
-    @deprecated  # type: ignore
-    def cancel(
-        cls,
-        fine_tune_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning cancel function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.fine_tuning.cancel(id=fine_tune_id).model_dump(exclude_none=True)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def list_events(
-        cls,
-        fine_tune_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning list events function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.fine_tuning.list_events(id=fine_tune_id).model_dump(
-            exclude_none=True
-        )
-
-    @classmethod
-    @deprecated  # type: ignore
-    def get_checkpoints(
-        cls,
-        fine_tune_id: str,
-    ) -> List[Any]:
-        """Legacy finetuning get checkpoints function."""
-
-        finetune_events = list(cls.retrieve(fine_tune_id=fine_tune_id)["events"])
-
-        saved_events = [i for i in finetune_events if i["type"] in ["CHECKPOINT_SAVE"]]
-
-        return saved_events
-
-    @classmethod
-    @deprecated  # type: ignore
-    def get_job_status(cls, fine_tune_id: str) -> str:
-        """Legacy finetuning get job status function."""
-        return str(cls.retrieve(fine_tune_id=fine_tune_id)["status"])
-
-    @classmethod
-    @deprecated  # type: ignore
-    def is_final_model_available(cls, fine_tune_id: str) -> bool:
-        """Legacy finetuning is final model available function."""
-
-        finetune_events = list(cls.retrieve(fine_tune_id=fine_tune_id)["events"])
-
-        for i in finetune_events:
-            if i["type"] in ["JOB_COMPLETE", "JOB_ERROR"]:
-                if i["checkpoint_path"] != "":
-                    return False
-                else:
-                    return True
-        return False
-
-    @classmethod
-    @deprecated  # type: ignore
-    def download(
-        cls,
-        fine_tune_id: str,
-        output: str | None = None,
-        step: int | None = None,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning download function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.fine_tuning.download(
-            id=fine_tune_id, output=output, checkpoint_step=step
-        ).model_dump(exclude_none=True)

together/legacy/images.py

@@ -1,27 +0,0 @@
-import warnings
-from typing import Any, Dict
-
-import together
-from together.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Image:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        prompt: str,
-        **kwargs: Any,
-    ) -> Dict[str, Any]:
-        """Legacy image function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return client.images.generate(prompt=prompt, **kwargs).model_dump(
-            exclude_none=True
-        )

together/legacy/models.py

@@ -1,44 +0,0 @@
-import warnings
-from typing import Any, Dict, List
-
-import together
-from together.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Models:
-    @classmethod
-    @deprecated  # type: ignore
-    def list(
-        cls,
-    ) -> List[Dict[str, Any]]:
-        """Legacy model list function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        return [item.model_dump(exclude_none=True) for item in client.models.list()]
-
-    @classmethod
-    @deprecated  # type: ignore
-    def info(
-        cls,
-        model: str,
-    ) -> Dict[str, Any]:
-        """Legacy model info function."""
-
-        api_key = None
-        if together.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = together.api_key
-
-        client = together.Together(api_key=api_key)
-
-        model_list = client.models.list()
-
-        for item in model_list:
-            if item.id == model:
-                return item.model_dump(exclude_none=True)

together/resources/batch.py

@@ -1,165 +0,0 @@
-from __future__ import annotations
-
-from typing import List
-
-from together.abstract import api_requestor
-from together.together_response import TogetherResponse
-from together.types import (
-    TogetherClient,
-    TogetherRequest,
-    BatchJob,
-)
-
-
-class Batches:
-    def __init__(self, client: TogetherClient) -> None:
-        self._client = client
-
-    def create_batch(self, file_id: str, endpoint: str) -> BatchJob:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        parameter_payload = {
-            "input_file_id": file_id,
-            "endpoint": endpoint,
-            "completion_window": "24h",
-        }
-
-        response, _, _ = requestor.request(
-            options=TogetherRequest(
-                method="POST",
-                url=f"batches",
-                params=parameter_payload,
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-        response_body = response.data.get("job", {})
-        return BatchJob(**response_body)
-
-    def get_batch(self, batch_job_id: str) -> BatchJob:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        response, _, _ = requestor.request(
-            options=TogetherRequest(
-                method="GET",
-                url=f"batches/{batch_job_id}",
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-        return BatchJob(**response.data)
-
-    def list_batches(self) -> List[BatchJob]:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        response, _, _ = requestor.request(
-            options=TogetherRequest(
-                method="GET",
-                url="batches",
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-        jobs = response.data or []
-        return [BatchJob(**job) for job in jobs]
-
-    def cancel_batch(self, batch_job_id: str) -> BatchJob:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        response, _, _ = requestor.request(
-            options=TogetherRequest(
-                method="POST",
-                url=f"batches/{batch_job_id}/cancel",
-            ),
-            stream=False,
-        )
-
-        return BatchJob(**response.data)
-
-
-class AsyncBatches:
-    def __init__(self, client: TogetherClient) -> None:
-        self._client = client
-
-    async def create_batch(self, file_id: str, endpoint: str) -> BatchJob:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        parameter_payload = {
-            "input_file_id": file_id,
-            "endpoint": endpoint,
-            "completion_window": "24h",
-        }
-
-        response, _, _ = await requestor.arequest(
-            options=TogetherRequest(
-                method="POST",
-                url=f"batches",
-                params=parameter_payload,
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-        response_body = response.data.get("job", {})
-        return BatchJob(**response_body)
-
-    async def get_batch(self, batch_job_id: str) -> BatchJob:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        response, _, _ = await requestor.arequest(
-            options=TogetherRequest(
-                method="GET",
-                url=f"batches/{batch_job_id}",
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-        return BatchJob(**response.data)
-
-    async def list_batches(self) -> List[BatchJob]:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        response, _, _ = await requestor.arequest(
-            options=TogetherRequest(
-                method="GET",
-                url="batches",
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-        jobs = response.data or []
-        return [BatchJob(**job) for job in jobs]
-
-    async def cancel_batch(self, batch_job_id: str) -> BatchJob:
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        response, _, _ = await requestor.arequest(
-            options=TogetherRequest(
-                method="POST",
-                url=f"batches/{batch_job_id}/cancel",
-            ),
-            stream=False,
-        )
-
-        return BatchJob(**response.data)