diff --git a/aidial_client/types/application.py b/aidial_client/types/application.py index 2f4afb7..eb6a8af 100644 --- a/aidial_client/types/application.py +++ b/aidial_client/types/application.py @@ -1,37 +1,15 @@ from typing import Dict, List, Literal, Optional from aidial_client._internal_types._model import ExtraAllowModel +from aidial_client.types.deployment import DeploymentBase -class Features(ExtraAllowModel): - rate: Optional[bool] = None - tokenize: Optional[bool] = None - truncate_prompt: Optional[bool] = None - configuration: Optional[bool] = None - system_prompt: Optional[bool] = None - tools: Optional[bool] = None - seed: Optional[bool] = None - url_attachments: Optional[bool] = None - folder_attachments: Optional[bool] = None - allow_resume: Optional[bool] = None - - -class Application(ExtraAllowModel): +class Application(DeploymentBase): object: Literal["application"] - id: str - description: Optional[str] = None application: str - display_name: Optional[str] = None - display_version: Optional[str] = None - icon_url: Optional[str] = None - reference: Optional[str] = None - owner: Optional[str] = None - status: Optional[str] = None - created_at: int - updated_at: Optional[int] = None - features: Features - input_attachment_types: Optional[List[str]] = None - defaults: Dict = {} + application_type_schema_id: Optional[str] = None + application_properties: Optional[Dict] = None + invalid: Optional[bool] = None class ApplicationsResponse(ExtraAllowModel): diff --git a/aidial_client/types/deployment.py b/aidial_client/types/deployment.py index 85acc97..590de9d 100644 --- a/aidial_client/types/deployment.py +++ b/aidial_client/types/deployment.py @@ -1,4 +1,4 @@ -from typing import List, Literal, Optional +from typing import Dict, List, Literal, Optional from aidial_client._internal_types._model import ExtraAllowModel @@ -7,15 +7,41 @@ class ScaleSettings(ExtraAllowModel): scale_type: Literal["standard"] -class Deployment(ExtraAllowModel): +class Features(ExtraAllowModel): + rate: Optional[bool] = None + tokenize: Optional[bool] = None + temperature: Optional[bool] = None + truncate_prompt: Optional[bool] = None + configuration: Optional[bool] = None + system_prompt: Optional[bool] = None + tools: Optional[bool] = None + seed: Optional[bool] = None + url_attachments: Optional[bool] = None + folder_attachments: Optional[bool] = None + allow_resume: Optional[bool] = None + + +class DeploymentBase(ExtraAllowModel): id: str - model: str - owner: str - object: Literal["deployment", "model"] - status: Literal["succeeded"] + object: str + owner: Optional[str] = None + reference: Optional[str] = None + icon_url: Optional[str] = None + description: Optional[str] = None + display_name: Optional[str] = None + display_version: Optional[str] = None + status: Optional[str] = None created_at: int - updated_at: int + updated_at: Optional[int] = None scale_settings: Optional[ScaleSettings] = None + defaults: Dict = {} + input_attachment_types: Optional[List[str]] = None + features: Optional[Features] = None + + +class Deployment(DeploymentBase): + object: Literal["deployment", "model"] + model: str class DeploymentsResponse(ExtraAllowModel):