Use VersionGetResponse type for latest_version

Author: dgellow

src/replicate/lib/_predictions_use.py

@@ -417,7 +417,13 @@ def create(self, *_: Input.args, **inputs: Input.kwargs) -> Run[Output]:
         version = self._version
 
         if version:
-            prediction = self._client.predictions.create(version=version, input=processed_inputs)
+            if isinstance(version, VersionGetResponse):
+                version_id = version.id
+            elif isinstance(version, dict) and "id" in version:
+                version_id = version["id"]
+            else:
+                version_id = str(version)
+            prediction = self._client.predictions.create(version=version_id, input=processed_inputs)
         else:
             prediction = self._client.models.predictions.create(model=self._model, input=processed_inputs)
 
@@ -629,7 +635,13 @@ async def create(self, *_: Input.args, **inputs: Input.kwargs) -> AsyncRun[Outpu
         version = await self._version()
 
         if version:
-            prediction = await self._client.predictions.create(version=version, input=processed_inputs)
+            if isinstance(version, VersionGetResponse):
+                version_id = version.id
+            elif isinstance(version, dict) and "id" in version:
+                version_id = version["id"]
+            else:
+                version_id = str(version)
+            prediction = await self._client.predictions.create(version=version_id, input=processed_inputs)
         else:
             model = await self._model()
             prediction = await self._client.models.predictions.create(

src/replicate/types/collection_get_response.py

@@ -3,6 +3,8 @@
 from typing import List, Optional
 from typing_extensions import Literal
 
+from replicate.types.models.version_get_response import VersionGetResponse
+
 from .._models import BaseModel
 
 __all__ = ["CollectionGetResponse", "Model"]
@@ -21,7 +23,7 @@ class Model(BaseModel):
     github_url: Optional[str] = None
     """A URL for the model's source code on GitHub"""
 
-    latest_version: Optional[object] = None
+    latest_version: Optional[VersionGetResponse] = None
     """The model's latest version"""
 
     license_url: Optional[str] = None

src/replicate/types/model_create_response.py

@@ -4,6 +4,7 @@
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .models.version_get_response import VersionGetResponse
 
 __all__ = ["ModelCreateResponse"]
 
@@ -21,7 +22,7 @@ class ModelCreateResponse(BaseModel):
     github_url: Optional[str] = None
     """A URL for the model's source code on GitHub"""
 
-    latest_version: Optional[object] = None
+    latest_version: Optional[VersionGetResponse] = None
     """The model's latest version"""
 
     license_url: Optional[str] = None

src/replicate/types/model_get_response.py

@@ -4,6 +4,7 @@
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .models.version_get_response import VersionGetResponse
 
 __all__ = ["ModelGetResponse"]
 
@@ -21,7 +22,7 @@ class ModelGetResponse(BaseModel):
     github_url: Optional[str] = None
     """A URL for the model's source code on GitHub"""
 
-    latest_version: Optional[object] = None
+    latest_version: Optional[VersionGetResponse] = None
     """The model's latest version"""
 
     license_url: Optional[str] = None

src/replicate/types/model_list_response.py

@@ -4,6 +4,7 @@
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .models.version_get_response import VersionGetResponse
 
 __all__ = ["ModelListResponse"]
 
@@ -21,7 +22,7 @@ class ModelListResponse(BaseModel):
     github_url: Optional[str] = None
     """A URL for the model's source code on GitHub"""
 
-    latest_version: Optional[object] = None
+    latest_version: Optional[VersionGetResponse] = None
     """The model's latest version"""
 
     license_url: Optional[str] = None

src/replicate/types/model_search_response.py

@@ -4,6 +4,7 @@
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .models.version_get_response import VersionGetResponse
 
 __all__ = ["ModelSearchResponse"]
 
@@ -21,7 +22,7 @@ class ModelSearchResponse(BaseModel):
     github_url: Optional[str] = None
     """A URL for the model's source code on GitHub"""
 
-    latest_version: Optional[object] = None
+    latest_version: Optional[VersionGetResponse] = None
     """The model's latest version"""
 
     license_url: Optional[str] = None