Skip to content

Commit

Permalink
gen model
Browse files Browse the repository at this point in the history
  • Loading branch information
goldenxinxing committed Nov 13, 2023
1 parent f786bb3 commit d28f942
Showing 1 changed file with 31 additions and 12 deletions.
43 changes: 31 additions & 12 deletions client/starwhale/base/client/models/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -391,6 +391,31 @@ class FineTuneCreateRequest(SwBaseModel):
)


class FineTuneEvalCreateRequest(SwBaseModel):
model_version_id: Optional[int] = Field(None, alias='modelVersionId')
dataset_version_ids: Optional[List[int]] = Field(None, alias='datasetVersionIds')
runtime_version_id: Optional[int] = Field(None, alias='runtimeVersionId')
time_to_live_in_sec: Optional[int] = Field(None, alias='timeToLiveInSec')
model_version_url: Optional[str] = Field(None, alias='modelVersionUrl')
dataset_version_urls: Optional[str] = Field(None, alias='datasetVersionUrls')
runtime_version_url: Optional[str] = Field(None, alias='runtimeVersionUrl')
comment: Optional[str] = None
resource_pool: str = Field(..., alias='resourcePool')
handler: Optional[str] = None
step_spec_over_writes: Optional[str] = Field(None, alias='stepSpecOverWrites')
type: Optional[Type3] = None
dev_mode: Optional[bool] = Field(None, alias='devMode')
dev_password: Optional[str] = Field(None, alias='devPassword')
dev_way: Optional[DevWay] = Field(None, alias='devWay')
space_id: Optional[int] = Field(None, alias='spaceId')


class ResponseMessageLong(SwBaseModel):
code: str
message: str
data: int


class RecordValueDesc(SwBaseModel):
key: str
value: Optional[Dict[str, Any]] = None
Expand Down Expand Up @@ -553,7 +578,7 @@ class Flag(Enum):
unchanged = 'unchanged'


class Type4(Enum):
class Type5(Enum):
directory = 'directory'
file = 'file'

Expand All @@ -563,7 +588,7 @@ class FileNode(SwBaseModel):
signature: Optional[str] = None
flag: Optional[Flag] = None
mime: Optional[str] = None
type: Optional[Type4] = None
type: Optional[Type5] = None
desc: Optional[str] = None
size: Optional[str] = None

Expand Down Expand Up @@ -683,12 +708,6 @@ class ResponseMessageListString(SwBaseModel):
data: List[str]


class ResponseMessageLong(SwBaseModel):
code: str
message: str
data: int


class RuntimeVersionViewVo(SwBaseModel):
id: str
version_name: str = Field(..., alias='versionName')
Expand Down Expand Up @@ -920,13 +939,13 @@ class DatasetVo(SwBaseModel):
version: DatasetVersionVo


class Type5(Enum):
class Type6(Enum):
dev_mode = 'DEV_MODE'
web_handler = 'WEB_HANDLER'


class ExposedLinkVo(SwBaseModel):
type: Type5
type: Type6
name: str
link: str

Expand Down Expand Up @@ -1227,7 +1246,7 @@ class Status2(Enum):
unknown = 'UNKNOWN'


class Type6(Enum):
class Type7(Enum):
image = 'IMAGE'
video = 'VIDEO'
audio = 'AUDIO'
Expand All @@ -1242,7 +1261,7 @@ class BuildRecordVo(SwBaseModel):
task_id: str = Field(..., alias='taskId')
dataset_name: str = Field(..., alias='datasetName')
status: Status2
type: Type6
type: Type7
create_time: int = Field(..., alias='createTime')


Expand Down

0 comments on commit d28f942

Please sign in to comment.