16 lines
316 B
Python
16 lines
316 B
Python
from typing import Literal
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class Argument(BaseModel):
|
|
type: Literal['str', 'int', 'float']
|
|
description: str | None = None
|
|
|
|
|
|
class ModelDescription(BaseModel):
|
|
engine: str
|
|
name: str
|
|
arguments: dict[str, Argument]
|
|
description: None | str = None
|