37 lines
777 B
Python
37 lines
777 B
Python
"""
|
|
Pydantic models for API request/response schemas
|
|
"""
|
|
from .training import (
|
|
TrainingStartRequest,
|
|
TrainingStartResponse,
|
|
TrainingStatusResponse,
|
|
TrainingMetricsResponse,
|
|
TrainingProgressResponse,
|
|
)
|
|
from .models import (
|
|
ModelSearchRequest,
|
|
ModelSearchResponse,
|
|
ModelListResponse,
|
|
ModelConfigResponse,
|
|
LoRAScanResponse,
|
|
LoRAInfo,
|
|
ModelInfo,
|
|
)
|
|
|
|
__all__ = [
|
|
# Training schemas
|
|
"TrainingStartRequest",
|
|
"TrainingStartResponse",
|
|
"TrainingStatusResponse",
|
|
"TrainingMetricsResponse",
|
|
"TrainingProgressResponse",
|
|
# Model management schemas
|
|
"ModelSearchRequest",
|
|
"ModelSearchResponse",
|
|
"ModelListResponse",
|
|
"ModelConfigResponse",
|
|
"LoRAScanResponse",
|
|
"LoRAInfo",
|
|
"ModelInfo",
|
|
]
|
|
|