c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a6__py3-none-any.whl → 0.1.0a8__py3-none-any.whl

gradientai/types/api_model.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from .._models import BaseModel
+from .api_agreement import APIAgreement
+from .api_model_version import APIModelVersion
+
+__all__ = ["APIModel"]
+
+
+class APIModel(BaseModel):
+    agreement: Optional[APIAgreement] = None
+
+    created_at: Optional[datetime] = None
+
+    is_foundational: Optional[bool] = None
+
+    name: Optional[str] = None
+
+    parent_uuid: Optional[str] = None
+
+    updated_at: Optional[datetime] = None
+
+    upload_complete: Optional[bool] = None
+
+    url: Optional[str] = None
+
+    uuid: Optional[str] = None
+
+    version: Optional[APIModelVersion] = None

gradientai/types/chat/__init__.py

@@ -4,4 +4,3 @@ from __future__ import annotations
 
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams
 from .completion_create_response import CompletionCreateResponse as CompletionCreateResponse
-from .chat_completion_token_logprob import ChatCompletionTokenLogprob as ChatCompletionTokenLogprob

gradientai/types/chat/completion_create_response.py

@@ -4,7 +4,7 @@ from typing import List, Optional
 from typing_extensions import Literal
 
 from ..._models import BaseModel
-from .chat_completion_token_logprob import ChatCompletionTokenLogprob
+from ..shared.chat_completion_token_logprob import ChatCompletionTokenLogprob
 
 __all__ = ["CompletionCreateResponse", "Choice", "ChoiceLogprobs", "ChoiceMessage", "Usage"]
 

gradientai/types/model_list_params.py

@@ -0,0 +1,42 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["ModelListParams"]
+
+
+class ModelListParams(TypedDict, total=False):
+    page: int
+    """page number."""
+
+    per_page: int
+    """items per page."""
+
+    public_only: bool
+    """only include models that are publicly available."""
+
+    usecases: List[
+        Literal[
+            "MODEL_USECASE_UNKNOWN",
+            "MODEL_USECASE_AGENT",
+            "MODEL_USECASE_FINETUNED",
+            "MODEL_USECASE_KNOWLEDGEBASE",
+            "MODEL_USECASE_GUARDRAIL",
+            "MODEL_USECASE_REASONING",
+            "MODEL_USECASE_SERVERLESS",
+        ]
+    ]
+    """include only models defined for the listed usecases.
+
+    - MODEL_USECASE_UNKNOWN: The use case of the model is unknown
+    - MODEL_USECASE_AGENT: The model maybe used in an agent
+    - MODEL_USECASE_FINETUNED: The model maybe used for fine tuning
+    - MODEL_USECASE_KNOWLEDGEBASE: The model maybe used for knowledge bases
+      (embedding models)
+    - MODEL_USECASE_GUARDRAIL: The model maybe used for guardrails
+    - MODEL_USECASE_REASONING: The model usecase for reasoning
+    - MODEL_USECASE_SERVERLESS: The model usecase for serverless inference
+    """

gradientai/types/model_list_response.py

@@ -1,15 +1,18 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import List
-from typing_extensions import Literal
+from typing import List, Optional
 
-from .model import Model
 from .._models import BaseModel
+from .api_model import APIModel
+from .shared.api_meta import APIMeta
+from .shared.api_links import APILinks
 
 __all__ = ["ModelListResponse"]
 
 
 class ModelListResponse(BaseModel):
-    data: List[Model]
+    links: Optional[APILinks] = None
 
-    object: Literal["list"]
+    meta: Optional[APIMeta] = None
+
+    models: Optional[List[APIModel]] = None

gradientai/types/shared/__init__.py

@@ -2,3 +2,4 @@
 
 from .api_meta import APIMeta as APIMeta
 from .api_links import APILinks as APILinks
+from .chat_completion_token_logprob import ChatCompletionTokenLogprob as ChatCompletionTokenLogprob

gradientai/types/model.py

@@ -1,21 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from typing_extensions import Literal
-
-from .._models import BaseModel
-
-__all__ = ["Model"]
-
-
-class Model(BaseModel):
-    id: str
-    """The model identifier, which can be referenced in the API endpoints."""
-
-    created: int
-    """The Unix timestamp (in seconds) when the model was created."""
-
-    object: Literal["model"]
-    """The object type, which is always "model"."""
-
-    owned_by: str
-    """The organization that owns the model."""