Source code for smarter.apps.prompt.api.v1.serializers

# pylint: disable=C0115
"""Django REST framework serializers for the API admin app."""

from rest_framework import serializers

from smarter.apps.plugin.serializers import PluginMetaSerializer
from smarter.apps.prompt.models import (
    Prompt,
    PromptHistory,
    PromptPluginUsage,
    PromptToolCall,
)



[docs]
class PromptSerializer(serializers.ModelSerializer):
    class Meta:
        model = Prompt
        fields = "__all__"




[docs]
class ChatHistorySerializer(serializers.ModelSerializer):
    """Serializer for the PromptHistory model."""

    prompt = PromptSerializer(read_only=True)

    class Meta:
        model = PromptHistory
        fields = "__all__"




[docs]
class PromptPluginUsageSerializer(serializers.ModelSerializer):
    """Serializer for the PromptPluginUsage model."""

    prompt = PromptSerializer(read_only=True)
    plugin = PluginMetaSerializer()

    class Meta:
        model = PromptPluginUsage
        fields = "__all__"




[docs]
class PromptToolCallSerializer(serializers.ModelSerializer):
    """Serializer for the PromptToolCall model."""

    prompt = PromptSerializer(read_only=True)

    class Meta:
        model = PromptToolCall
        fields = "__all__"