feat: add generation API endpoints

2026-07-02 22:41:37 +08:00
parent 47e6aed7a8
commit 0f7798ec2f
17 changed files with 851 additions and 19 deletions
@@ -0,0 +1,45 @@
+from rest_framework import serializers
+
+
+class GenerateTitleRequestSerializer(serializers.Serializer):
+    prompt = serializers.CharField(trim_whitespace=True, allow_blank=False)
+    model = serializers.CharField(
+        required=False,
+        allow_blank=True,
+        trim_whitespace=True,
+        max_length=64,
+    )
+    image_url = serializers.URLField(required=False, allow_blank=True)
+    image_base64 = serializers.CharField(required=False, allow_blank=True)
+    resolution = serializers.CharField(
+        required=False,
+        allow_blank=True,
+        trim_whitespace=True,
+        default="1K",
+    )
+    parameters = serializers.DictField(required=False, default=dict)
+
+
+class GenerateImageRequestSerializer(serializers.Serializer):
+    prompt = serializers.CharField(trim_whitespace=True, allow_blank=False)
+    model = serializers.CharField(
+        required=False,
+        allow_blank=True,
+        trim_whitespace=True,
+        max_length=64,
+    )
+    image_url = serializers.URLField(required=False, allow_blank=True)
+    image_base64 = serializers.CharField(required=False, allow_blank=True)
+    resolution = serializers.CharField(
+        required=False,
+        allow_blank=True,
+        trim_whitespace=True,
+        default="1K",
+    )
+    aspect_ratio = serializers.CharField(
+        required=False,
+        allow_blank=True,
+        trim_whitespace=True,
+        default="1:1",
+    )
+    parameters = serializers.DictField(required=False, default=dict)