feat: Refactor application and deployment models to reflect same structures in core

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):

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):