change to the openai compatible api

Author: blainekasten

src/together/resources/videos.py

@@ -12,7 +12,7 @@
 from together.types.videos import (
     CreateVideoResponse,
     CreateVideoBody,
-    VideoResource,
+    VideoJob,
 )
 
 if sys.version_info >= (3, 8):
@@ -134,15 +134,15 @@ def create(
     def retrieve(
         self,
         id: str,
-    ) -> VideoResource:
+    ) -> VideoJob:
         """
         Method to retrieve a video creation job.
 
         Args:
             id (str): The ID of the video creation job to retrieve.
 
         Returns:
-            VideoResource: Object containing the current status and details of the video creation job
+            VideoJob: Object containing the current status and details of the video creation job
         """
 
         requestor = api_requestor.APIRequestor(
@@ -152,14 +152,14 @@ def retrieve(
         response, _, _ = requestor.request(
             options=TogetherRequest(
                 method="GET",
-                url=f"../v2/videos/status?id={id}",
+                url=f"../v2/videos/{id}",
             ),
             stream=False,
         )
 
         assert isinstance(response, TogetherResponse)
 
-        return VideoResource(**response.data)
+        return VideoJob(**response.data)
 
 
 class AsyncVideos:
@@ -272,19 +272,18 @@ async def create(
 
         return CreateVideoResponse(**response.data)
 
-    async def status(
+    async def retrieve(
         self,
-        *,
         id: str,
-    ) -> VideoResource:
+    ) -> VideoJob:
         """
-        Async method to check the status of a video generation job.
+        Async method to retrieve a video creation job.
 
         Args:
-            id (str): The ID of the video generation job to check.
+            id (str): The ID of the video creation job to retrieve.
 
         Returns:
-            VideoResource: Object containing the current status and details of the video generation job
+            VideoJob: Object containing the current status and details of the video creation job
         """
 
         requestor = api_requestor.APIRequestor(
@@ -294,11 +293,11 @@ async def status(
         response, _, _ = await requestor.arequest(
             options=TogetherRequest(
                 method="GET",
-                url=f"../v2/videos/status?id={id}",
+                url=f"../v2/videos/{id}",
             ),
             stream=False,
         )
 
         assert isinstance(response, TogetherResponse)
 
-        return VideoResource(**response.data)
+        return VideoJob(**response.data)

src/together/types/__init__.py

@@ -78,7 +78,7 @@
 from together.types.videos import (
     CreateVideoBody,
     CreateVideoResponse,
-    VideoResource,
+    VideoJob,
 )
 
 
@@ -159,5 +159,5 @@
     "EvaluationStatusResponse",
     "CreateVideoBody",
     "CreateVideoResponse",
-    "VideoResource",
+    "VideoJob",
 ]

src/together/types/videos.py

@@ -34,41 +34,36 @@ class CreateVideoBody(BaseModel):
 
 
 class VideoOutputs(BaseModel):
-    """Response from video creation request"""
+    """Artifacts generated from video creation job"""
 
     cost: float
     video_url: str
 
 
-class VideoInfoError(BaseModel):
-    """Error about the video creation request"""
+class Error(BaseModel):
+    """Error information about the video job"""
 
-    code: str
+    code: str | None = None
     message: str
 
 
-class VideoInfo(BaseModel):
-    """Info about the video creation request"""
-
-    user_id: str
-    errors: List[VideoInfoError] | None = None
-
-
 class CreateVideoResponse(BaseModel):
     """Response from video generation request"""
 
     id: str
 
 
-class VideoResource(BaseModel):
-    """Response from video status check"""
+class VideoJob(BaseModel):
+    """Structured information describing a generated video job."""
 
     id: str
     model: str
+    object: Literal["video"]
     status: Literal["queued", "in_progress", "completed", "failed", "cancelled"]
-    info: VideoInfo | None = None
-    inputs: Dict[str, Any]
+    seconds: str
+    size: str
+    created_at: int
+
+    error: Error | None = None
     outputs: VideoOutputs | None = None
-    created_at: str
-    claimed_at: str
-    done_at: str
+    completed_at: int | None = None