feat(api): manual updates (#21)

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 27
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/replicate-mpatankar%2Freplicate-client-b45f922f6a041550870a96f5acec02aa6d8830046fc98b95a275c6486f7586fc.yml
 openapi_spec_hash: ef7fddfb49b4d9c440b0635d2c86f341
-config_hash: 5691e71e792eb8d38c4c91fa7efff22c
+config_hash: 21782974ab54ada5de4d9d9165f60fd0

api.md

@@ -81,7 +81,7 @@ Methods:
 Types:
 
 ```python
-from replicate.types import Prediction, PredictionRequest
+from replicate.types import Prediction, PredictionOutput, PredictionRequest
 ```
 
 Methods:

src/replicate/types/__init__.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from .prediction import Prediction as Prediction
+from .prediction_output import PredictionOutput as PredictionOutput
 from .model_create_params import ModelCreateParams as ModelCreateParams
 from .account_list_response import AccountListResponse as AccountListResponse
 from .hardware_list_response import HardwareListResponse as HardwareListResponse

src/replicate/types/prediction.py

@@ -1,10 +1,11 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Dict, List, Union, Optional
+from typing import Dict, Union, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .prediction_output import PredictionOutput
 
 __all__ = ["Prediction", "URLs"]
 
@@ -38,7 +39,7 @@ class Prediction(BaseModel):
     model: str
     """The name of the model that created the prediction"""
 
-    output: Union[Optional[Dict[str, object]], Optional[List[object]], Optional[str], Optional[float], Optional[bool]]
+    output: PredictionOutput
 
     status: Literal["starting", "processing", "succeeded", "failed", "canceled"]
 

src/replicate/types/prediction_output.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List, Union, Optional
+from typing_extensions import TypeAlias
+
+__all__ = ["PredictionOutput"]
+
+PredictionOutput: TypeAlias = Union[
+    Optional[Dict[str, object]], Optional[List[object]], Optional[str], Optional[float], Optional[bool]
+]