model-service/model_service/schemas/aimodel_schema.py

#!/usr/bin/env python3
from marshmallow import fields
from marshmallow_sqlalchemy import ModelSchema
from model import AIModel


class AIModelSchema(ModelSchema):
    class Meta:
        model = AIModel