together 1.5.17__py3-none-any.whl → 2.0.0a8__py3-none-any.whl

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 | None = 32,
-        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 if isinstance(batch_size, int) else "max",
-            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,136 +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]
-
-
-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]

together/resources/code_interpreter.py

@@ -1,82 +0,0 @@
-from __future__ import annotations
-
-from typing import Any, Dict, List, Literal, Optional
-from pydantic import ValidationError
-
-from together.abstract import api_requestor
-from together.together_response import TogetherResponse
-from together.types import TogetherClient, TogetherRequest
-from together.types.code_interpreter import ExecuteResponse, FileInput
-
-
-class CodeInterpreter:
-    """Code Interpreter resource for executing code snippets."""
-
-    def __init__(self, client: TogetherClient) -> None:
-        self._client = client
-
-    def run(
-        self,
-        code: str,
-        language: Literal["python"],
-        session_id: Optional[str] = None,
-        files: Optional[List[Dict[str, Any]]] = None,
-    ) -> ExecuteResponse:
-        """Execute a code snippet, optionally with files.
-
-        Args:
-            code (str): Code snippet to execute
-            language (str): Programming language for the code to execute. Currently only supports Python.
-            session_id (str, optional): Identifier of the current session. Used to make follow-up calls.
-            files (List[Dict], optional): Files to upload to the session before executing the code.
-
-        Returns:
-            ExecuteResponse: Object containing execution results and outputs
-
-        Raises:
-            ValidationError: If any dictionary in the `files` list does not conform to the
-                required structure or types.
-        """
-        requestor = api_requestor.APIRequestor(
-            client=self._client,
-        )
-
-        data: Dict[str, Any] = {
-            "code": code,
-            "language": language,
-        }
-
-        if session_id is not None:
-            data["session_id"] = session_id
-
-        if files is not None:
-            serialized_files = []
-            try:
-                for file_dict in files:
-                    # Validate the dictionary by creating a FileInput instance
-                    validated_file = FileInput(**file_dict)
-                    # Serialize the validated model back to a dict for the API call
-                    serialized_files.append(validated_file.model_dump())
-            except ValidationError as e:
-                raise ValueError(f"Invalid file input format: {e}") from e
-            except TypeError as e:
-                raise ValueError(
-                    f"Invalid file input: Each item in 'files' must be a dictionary. Error: {e}"
-                ) from e
-
-            data["files"] = serialized_files
-
-        # Use absolute URL to bypass the /v1 prefix
-        response, _, _ = requestor.request(
-            options=TogetherRequest(
-                method="POST",
-                url="/tci/execute",
-                params=data,
-            ),
-            stream=False,
-        )
-
-        assert isinstance(response, TogetherResponse)
-
-        # Return the response data directly since our types match the API structure
-        return ExecuteResponse(**response.data)