together 1.5.25__py3-none-any.whl → 1.5.27__py3-none-any.whl

together/resources/finetune.py

@@ -13,6 +13,7 @@ from together.types import (
     CosineLRScheduler,
     CosineLRSchedulerArgs,
     FinetuneCheckpoint,
+    FinetuneDeleteResponse,
     FinetuneDownloadResult,
     FinetuneList,
     FinetuneListEvents,
@@ -570,6 +571,37 @@ class FineTuning:
 
         return FinetuneResponse(**response.data)
 
+    def delete(self, id: str, force: bool = False) -> FinetuneDeleteResponse:
+        """
+        Method to delete a fine-tuning job
+
+        Args:
+            id (str): Fine-tune ID to delete. A string that starts with `ft-`.
+            force (bool, optional): Force deletion. Defaults to False.
+
+        Returns:
+            FinetuneDeleteResponse: Object containing deletion confirmation message.
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        params = {"force": str(force).lower()}
+
+        response, _, _ = requestor.request(
+            options=TogetherRequest(
+                method="DELETE",
+                url=f"fine-tunes/{id}",
+                params=params,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return FinetuneDeleteResponse(**response.data)
+
     def list_events(self, id: str) -> FinetuneListEvents:
         """
         Lists events of a fine-tune job
@@ -1007,6 +1039,37 @@ class AsyncFineTuning:
 
         return FinetuneResponse(**response.data)
 
+    async def delete(self, id: str, force: bool = False) -> FinetuneDeleteResponse:
+        """
+        Async method to delete a fine-tuning job
+
+        Args:
+            id (str): Fine-tune ID to delete. A string that starts with `ft-`.
+            force (bool, optional): Force deletion. Defaults to False.
+
+        Returns:
+            FinetuneDeleteResponse: Object containing deletion confirmation message.
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        params = {"force": str(force).lower()}
+
+        response, _, _ = await requestor.arequest(
+            options=TogetherRequest(
+                method="DELETE",
+                url=f"fine-tunes/{id}",
+                params=params,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return FinetuneDeleteResponse(**response.data)
+
     async def list_events(self, id: str) -> FinetuneListEvents:
         """
         List fine-tuning events

together/resources/models.py

@@ -6,6 +6,8 @@ from together.abstract import api_requestor
 from together.together_response import TogetherResponse
 from together.types import (
     ModelObject,
+    ModelUploadRequest,
+    ModelUploadResponse,
     TogetherClient,
     TogetherRequest,
 )
@@ -85,6 +87,64 @@ class Models(ModelsBase):
 
         return models
 
+    def upload(
+        self,
+        *,
+        model_name: str,
+        model_source: str,
+        model_type: str = "model",
+        hf_token: str | None = None,
+        description: str | None = None,
+        base_model: str | None = None,
+        lora_model: str | None = None,
+    ) -> ModelUploadResponse:
+        """
+        Upload a custom model or adapter from Hugging Face or S3.
+
+        Args:
+            model_name (str): The name to give to your uploaded model
+            model_source (str): The source location of the model (Hugging Face repo or S3 path)
+            model_type (str, optional): Whether the model is a full model or an adapter. Defaults to "model".
+            hf_token (str, optional): Hugging Face token (if uploading from Hugging Face)
+            description (str, optional): A description of your model
+            base_model (str, optional): The base model to use for an adapter if setting it to run against a serverless pool. Only used for model_type "adapter".
+            lora_model (str, optional): The lora pool to use for an adapter if setting it to run against, say, a dedicated pool. Only used for model_type "adapter".
+
+        Returns:
+            ModelUploadResponse: Object containing upload job information
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        data = {
+            "model_name": model_name,
+            "model_source": model_source,
+            "model_type": model_type,
+        }
+
+        if hf_token is not None:
+            data["hf_token"] = hf_token
+        if description is not None:
+            data["description"] = description
+        if base_model is not None:
+            data["base_model"] = base_model
+        if lora_model is not None:
+            data["lora_model"] = lora_model
+
+        response, _, _ = requestor.request(
+            options=TogetherRequest(
+                method="POST",
+                url="models",
+                params=data,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return ModelUploadResponse.from_api_response(response.data)
+
 
 class AsyncModels(ModelsBase):
     async def list(
@@ -132,3 +192,61 @@ class AsyncModels(ModelsBase):
         models.sort(key=lambda x: x.id.lower())
 
         return models
+
+    async def upload(
+        self,
+        *,
+        model_name: str,
+        model_source: str,
+        model_type: str = "model",
+        hf_token: str | None = None,
+        description: str | None = None,
+        base_model: str | None = None,
+        lora_model: str | None = None,
+    ) -> ModelUploadResponse:
+        """
+        Upload a custom model or adapter from Hugging Face or S3.
+
+        Args:
+            model_name (str): The name to give to your uploaded model
+            model_source (str): The source location of the model (Hugging Face repo or S3 path)
+            model_type (str, optional): Whether the model is a full model or an adapter. Defaults to "model".
+            hf_token (str, optional): Hugging Face token (if uploading from Hugging Face)
+            description (str, optional): A description of your model
+            base_model (str, optional): The base model to use for an adapter if setting it to run against a serverless pool. Only used for model_type "adapter".
+            lora_model (str, optional): The lora pool to use for an adapter if setting it to run against, say, a dedicated pool. Only used for model_type "adapter".
+
+        Returns:
+            ModelUploadResponse: Object containing upload job information
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        data = {
+            "model_name": model_name,
+            "model_source": model_source,
+            "model_type": model_type,
+        }
+
+        if hf_token is not None:
+            data["hf_token"] = hf_token
+        if description is not None:
+            data["description"] = description
+        if base_model is not None:
+            data["base_model"] = base_model
+        if lora_model is not None:
+            data["lora_model"] = lora_model
+
+        response, _, _ = await requestor.arequest(
+            options=TogetherRequest(
+                method="POST",
+                url="models",
+                params=data,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return ModelUploadResponse.from_api_response(response.data)

together/resources/videos.py

@@ -0,0 +1,303 @@
+from __future__ import annotations
+
+from typing import Any, Dict, List
+import sys
+
+from together.abstract import api_requestor
+from together.together_response import TogetherResponse
+from together.types import (
+    TogetherClient,
+    TogetherRequest,
+)
+from together.types.videos import (
+    CreateVideoResponse,
+    CreateVideoBody,
+    VideoJob,
+)
+
+if sys.version_info >= (3, 8):
+    from typing import Literal
+else:
+    from typing_extensions import Literal
+
+
+class Videos:
+    def __init__(self, client: TogetherClient) -> None:
+        self._client = client
+
+    def create(
+        self,
+        *,
+        model: str,
+        prompt: str | None = None,
+        height: int | None = None,
+        width: int | None = None,
+        seconds: str | None = None,
+        fps: int | None = None,
+        steps: int | None = None,
+        seed: int | None = None,
+        guidance_scale: float | None = None,
+        output_format: Literal["MP4", "WEBM"] | None = None,
+        output_quality: int | None = None,
+        negative_prompt: str | None = None,
+        frame_images: List[Dict[str, Any]] | None = None,
+        reference_images: List[str] | None = None,
+        **kwargs: Any,
+    ) -> CreateVideoResponse:
+        """
+        Method to generate videos based on a given prompt using a specified model.
+
+        Args:
+            model (str): The model to use for video generation.
+
+            prompt (str): A description of the desired video. Positive prompt for the generation.
+
+            height (int, optional): Height of the video to generate in pixels.
+
+            width (int, optional): Width of the video to generate in pixels.
+
+            seconds (str, optional): Length of generated video in seconds. Min 1 max 10.
+
+            fps (int, optional): Frames per second, min 15 max 60. Defaults to 24.
+
+            steps (int, optional): The number of denoising steps the model performs during video
+                generation. More steps typically result in higher quality output but require longer
+                processing time. Min 10 max 50. Defaults to 20.
+
+            seed (int, optional): Seed to use in initializing the video generation. Using the same
+                seed allows deterministic video generation. If not provided, a random seed is
+                generated for each request. Note: When requesting multiple videos with the same
+                seed, the seed will be incremented by 1 (+1) for each video generated.
+
+            guidance_scale (float, optional): Controls how closely the video generation follows your
+                prompt. Higher values make the model adhere more strictly to your text description,
+                while lower values allow more creative freedom. Recommended range is 6.0-10.0 for
+                most video models. Values above 12 may cause over-guidance artifacts or unnatural
+                motion patterns. Defaults to 8.
+
+            output_format (str, optional): Specifies the format of the output video. Either "MP4"
+                or "WEBM". Defaults to "MP4".
+
+            output_quality (int, optional): Compression quality. Defaults to 20.
+
+            negative_prompt (str, optional): Similar to prompt, but specifies what to avoid instead
+                of what to include. Defaults to None.
+
+            frame_images (List[Dict[str, Any]], optional): Array of images to guide video generation,
+                like keyframes. If size 1, starting frame; if size 2, starting and ending frame;
+                if more than 2 then frame must be specified. Defaults to None.
+
+            reference_images (List[str], optional): An array containing reference images
+                used to condition the generation process. These images provide visual guidance to
+                help the model generate content that aligns with the style, composition, or
+                characteristics of the reference materials. Defaults to None.
+
+        Returns:
+            CreateVideoResponse: Object containing video generation job id
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = CreateVideoBody(
+            prompt=prompt,
+            model=model,
+            height=height,
+            width=width,
+            seconds=seconds,
+            fps=fps,
+            steps=steps,
+            seed=seed,
+            guidance_scale=guidance_scale,
+            output_format=output_format,
+            output_quality=output_quality,
+            negative_prompt=negative_prompt,
+            frame_images=frame_images,
+            reference_images=reference_images,
+            **kwargs,
+        ).model_dump(exclude_none=True)
+
+        response, _, _ = requestor.request(
+            options=TogetherRequest(
+                method="POST",
+                url="../v2/videos",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return CreateVideoResponse(**response.data)
+
+    def retrieve(
+        self,
+        id: str,
+    ) -> VideoJob:
+        """
+        Method to retrieve a video creation job.
+
+        Args:
+            id (str): The ID of the video creation job to retrieve.
+
+        Returns:
+            VideoJob: Object containing the current status and details of the video creation job
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=TogetherRequest(
+                method="GET",
+                url=f"../v2/videos/{id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return VideoJob(**response.data)
+
+
+class AsyncVideos:
+    def __init__(self, client: TogetherClient) -> None:
+        self._client = client
+
+    async def create(
+        self,
+        *,
+        prompt: str,
+        model: str,
+        height: int | None = None,
+        width: int | None = None,
+        seconds: float | None = None,
+        fps: int | None = None,
+        steps: int | None = None,
+        seed: int | None = None,
+        guidance_scale: float | None = None,
+        output_format: Literal["MP4", "WEBM"] | None = None,
+        output_quality: int | None = None,
+        negative_prompt: str | None = None,
+        frame_images: List[Dict[str, Any]] | None = None,
+        reference_images: List[str] | None = None,
+        **kwargs: Any,
+    ) -> CreateVideoResponse:
+        """
+        Async method to create videos based on a given prompt using a specified model.
+
+        Args:
+            prompt (str): A description of the desired video. Positive prompt for the generation.
+
+            model (str): The model to use for video generation.
+
+            height (int, optional): Height of the video to generate in pixels.
+
+            width (int, optional): Width of the video to generate in pixels.
+
+            seconds (float, optional): Length of generated video in seconds. Min 1 max 10.
+
+            fps (int, optional): Frames per second, min 15 max 60. Defaults to 24.
+
+            steps (int, optional): The number of denoising steps the model performs during video
+                generation. More steps typically result in higher quality output but require longer
+                processing time. Min 10 max 50. Defaults to 20.
+
+            seed (int, optional): Seed to use in initializing the video generation. Using the same
+                seed allows deterministic video generation. If not provided, a random seed is
+                generated for each request. Note: When requesting multiple videos with the same
+                seed, the seed will be incremented by 1 (+1) for each video generated.
+
+            guidance_scale (float, optional): Controls how closely the video generation follows your
+                prompt. Higher values make the model adhere more strictly to your text description,
+                while lower values allow more creative freedom. Recommended range is 6.0-10.0 for
+                most video models. Values above 12 may cause over-guidance artifacts or unnatural
+                motion patterns. Defaults to 8.
+
+            output_format (Literal["MP4", "WEBM"], optional): Specifies the format of the output video. Either "MP4"
+                or "WEBM". Defaults to "MP4".
+
+            output_quality (int, optional): Compression quality. Defaults to 20.
+
+            negative_prompt (str, optional): Similar to prompt, but specifies what to avoid instead
+                of what to include. Defaults to None.
+
+            frame_images (List[Dict[str, Any]], optional): Array of images to guide video generation,
+                like keyframes. If size 1, starting frame; if size 2, starting and ending frame;
+                if more than 2 then frame must be specified. Defaults to None.
+
+            reference_images (List[str], optional): An array containing reference images
+                used to condition the generation process. These images provide visual guidance to
+                help the model generate content that aligns with the style, composition, or
+                characteristics of the reference materials. Defaults to None.
+
+        Returns:
+            CreateVideoResponse: Object containing video creation job id
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = CreateVideoBody(
+            prompt=prompt,
+            model=model,
+            height=height,
+            width=width,
+            seconds=seconds,
+            fps=fps,
+            steps=steps,
+            seed=seed,
+            guidance_scale=guidance_scale,
+            output_format=output_format,
+            output_quality=output_quality,
+            negative_prompt=negative_prompt,
+            frame_images=frame_images,
+            reference_images=reference_images,
+            **kwargs,
+        ).model_dump(exclude_none=True)
+
+        response, _, _ = await requestor.arequest(
+            options=TogetherRequest(
+                method="POST",
+                url="../v2/videos",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return CreateVideoResponse(**response.data)
+
+    async def retrieve(
+        self,
+        id: str,
+    ) -> VideoJob:
+        """
+        Async method to retrieve a video creation job.
+
+        Args:
+            id (str): The ID of the video creation job to retrieve.
+
+        Returns:
+            VideoJob: Object containing the current status and details of the video creation job
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=TogetherRequest(
+                method="GET",
+                url=f"../v2/videos/{id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, TogetherResponse)
+
+        return VideoJob(**response.data)

together/types/__init__.py

@@ -52,13 +52,14 @@ from together.types.finetune import (
     FinetuneListEvents,
     FinetuneRequest,
     FinetuneResponse,
+    FinetuneDeleteResponse,
     FinetuneTrainingLimits,
     FullTrainingType,
     LoRATrainingType,
     TrainingType,
 )
 from together.types.images import ImageRequest, ImageResponse
-from together.types.models import ModelObject
+from together.types.models import ModelObject, ModelUploadRequest, ModelUploadResponse
 from together.types.rerank import RerankRequest, RerankResponse
 from together.types.batch import BatchJob, BatchJobStatus, BatchEndpoint
 from together.types.evaluation import (
@@ -74,6 +75,11 @@ from together.types.evaluation import (
     EvaluationJob,
     EvaluationStatusResponse,
 )
+from together.types.videos import (
+    CreateVideoBody,
+    CreateVideoResponse,
+    VideoJob,
+)
 
 
 __all__ = [
@@ -92,6 +98,7 @@ __all__ = [
     "FinetuneResponse",
     "FinetuneList",
     "FinetuneListEvents",
+    "FinetuneDeleteResponse",
     "FinetuneDownloadResult",
     "FinetuneLRScheduler",
     "LinearLRScheduler",
@@ -108,6 +115,8 @@ __all__ = [
     "ImageRequest",
     "ImageResponse",
     "ModelObject",
+    "ModelUploadRequest",
+    "ModelUploadResponse",
     "TrainingType",
     "FullTrainingType",
     "LoRATrainingType",
@@ -148,4 +157,7 @@ __all__ = [
     "EvaluationCreateResponse",
     "EvaluationJob",
     "EvaluationStatusResponse",
+    "CreateVideoBody",
+    "CreateVideoResponse",
+    "VideoJob",
 ]

together/types/audio_speech.py

@@ -158,6 +158,17 @@ class AudioTranscriptionWord(BaseModel):
     word: str
     start: float
     end: float
+    id: Optional[int] = None
+    speaker_id: Optional[str] = None
+
+
+class AudioSpeakerSegment(BaseModel):
+    id: int
+    speaker_id: str
+    start: float
+    end: float
+    text: str
+    words: List[AudioTranscriptionWord]
 
 
 class AudioTranscriptionResponse(BaseModel):
@@ -165,11 +176,13 @@ class AudioTranscriptionResponse(BaseModel):
 
 
 class AudioTranscriptionVerboseResponse(BaseModel):
+    id: Optional[str] = None
     language: Optional[str] = None
     duration: Optional[float] = None
     text: str
     segments: Optional[List[AudioTranscriptionSegment]] = None
     words: Optional[List[AudioTranscriptionWord]] = None
+    speaker_segments: Optional[List[AudioSpeakerSegment]] = None
 
 
 class AudioTranslationResponse(BaseModel):

together/types/evaluation.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from datetime import datetime
 from enum import Enum
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Literal, Optional, Union
 
 from pydantic import BaseModel, Field
 
@@ -23,16 +23,22 @@ class EvaluationStatus(str, Enum):
 
 
 class JudgeModelConfig(BaseModel):
-    model_name: str
+    model: str
+    model_source: Literal["serverless", "dedicated", "external"]
     system_template: str
+    external_api_token: Optional[str] = None
+    external_base_url: Optional[str] = None
 
 
 class ModelRequest(BaseModel):
-    model_name: str
+    model: str
+    model_source: Literal["serverless", "dedicated", "external"]
     max_tokens: int
     temperature: float
     system_template: str
     input_template: str
+    external_api_token: Optional[str] = None
+    external_base_url: Optional[str] = None
 
 
 class ClassifyParameters(BaseModel):

together/types/files.py

@@ -52,7 +52,7 @@ class FileResponse(BaseModel):
     """
 
     id: str
-    object: Literal[ObjectType.File]
+    object: str
     # created timestamp
     created_at: int | None = None
     type: FileType | None = None

together/types/finetune.py

@@ -322,6 +322,11 @@ class FinetuneListEvents(BaseModel):
     data: List[FinetuneEvent] | None = None
 
 
+class FinetuneDeleteResponse(BaseModel):
+    # delete message
+    message: str
+
+
 class FinetuneDownloadResult(BaseModel):
     # object type
     object: Literal["local"] | None = None