supported models wip

This commit is contained in:
Xi Yan 2024-09-21 18:37:22 -07:00
parent 20a4302877
commit c0199029e5
10 changed files with 215 additions and 34 deletions

View file

@ -29,17 +29,12 @@ class ModelServingSpec(BaseModel):
@json_schema_type
class ModelsListResponse(BaseModel):
models_list: List[ModelSpec]
models_list: List[ModelServingSpec]
@json_schema_type
class ModelsGetResponse(BaseModel):
core_model_spec: Optional[ModelSpec] = None
@json_schema_type
class ModelsRegisterResponse(BaseModel):
core_model_spec: Optional[ModelSpec] = None
core_model_spec: Optional[ModelServingSpec] = None
class Models(Protocol):