19 lines
325 B
Python
19 lines
325 B
Python
from dataclasses import dataclass
|
|
from typing import Callable
|
|
|
|
@dataclass
|
|
class Test:
|
|
name: str
|
|
runnable: Callable
|
|
runnable_input: dict
|
|
validator: Callable
|
|
validation_input: dict
|
|
|
|
@dataclass
|
|
class Model:
|
|
display_name: str
|
|
identifier: str
|
|
supports_tools: bool
|
|
parameter_count_in_b: float
|
|
|