from rest_framework.serializers import ModelSerializer
from .models import AgentModel, Agent, AgentRun, AgentEvent


class AgentModelSerializer(ModelSerializer):
    class Meta:
        model = AgentModel
        fields = ['id', 'uuid', 'name', 'version', 'path']
        read_only_fields = ['id', 'uuid']


class AgentSerializer(ModelSerializer):
    model = AgentModelSerializer(read_only=True)

    class Meta:
        model = Agent
        fields = [
            'id',
            'uuid',
            'model',
            'organization',
            'status',
            'description',
            'started_at',
            'completed_at',
        ]
        read_only_fields = ['id', 'uuid', 'started_at', 'completed_at']


class AgentRunSerializer(ModelSerializer):
    class Meta:
        model = AgentRun
        fields = [
            'id',
            'uuid',
            'agent',
            'user',
            'status',
            'input_data',
            'output_data',
            'error_message',
            'started_at',
            'completed_at',
        ]
        read_only_fields = ['id', 'uuid', 'started_at', 'completed_at']


class AgentEventSerializer(ModelSerializer):
    class Meta:
        model = AgentEvent
        fields = ['id', 'uuid', 'execution', 'event_type', 'content', 'timestamp']
        read_only_fields = ['id', 'uuid', 'timestamp']